klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 05:39:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use subscriber type specific context window to generate response

Browse Source
This commit is contained in:
Debanjum
2025-08-18 23:24:00 -07:00
parent ff73d30106
commit 222cc19b7f
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/khoj/routers/helpers.py
View File

@@ -1625,6 +1625,7 @@ async def agenerate_chat_response(
deepthought = True deepthought = True
chat_model = await ConversationAdapters.aget_valid_chat_model(user, conversation, is_subscribed) chat_model = await ConversationAdapters.aget_valid_chat_model(user, conversation, is_subscribed)
max_prompt_size = await ConversationAdapters.aget_max_context_size(chat_model, user)
vision_available = chat_model.vision_enabled vision_available = chat_model.vision_enabled
if not vision_available and query_images: if not vision_available and query_images:
vision_enabled_config = await ConversationAdapters.aget_vision_enabled_config() vision_enabled_config = await ConversationAdapters.aget_vision_enabled_config()
@@ -1656,7 +1657,7 @@ async def agenerate_chat_response(
model=chat_model_name, model=chat_model_name,
api_key=api_key, api_key=api_key,
api_base_url=openai_chat_config.api_base_url, api_base_url=openai_chat_config.api_base_url,
max_prompt_size=chat_model.max_prompt_size, max_prompt_size=max_prompt_size,
tokenizer_name=chat_model.tokenizer, tokenizer_name=chat_model.tokenizer,
agent=agent, agent=agent,
vision_available=vision_available, vision_available=vision_available,
@@ -1687,7 +1688,7 @@ async def agenerate_chat_response(
model=chat_model.name, model=chat_model.name,
api_key=api_key, api_key=api_key,
api_base_url=api_base_url, api_base_url=api_base_url,
max_prompt_size=chat_model.max_prompt_size, max_prompt_size=max_prompt_size,
tokenizer_name=chat_model.tokenizer, tokenizer_name=chat_model.tokenizer,
agent=agent, agent=agent,
vision_available=vision_available, vision_available=vision_available,
@@ -1717,7 +1718,7 @@ async def agenerate_chat_response(
model=chat_model.name, model=chat_model.name,
api_key=api_key, api_key=api_key,
api_base_url=api_base_url, api_base_url=api_base_url,
max_prompt_size=chat_model.max_prompt_size, max_prompt_size=max_prompt_size,
tokenizer_name=chat_model.tokenizer, tokenizer_name=chat_model.tokenizer,
agent=agent, agent=agent,
vision_available=vision_available, vision_available=vision_available,