sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-17 08:04:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Include attach files in calls to extract questions

Browse source
This commit is contained in:
sabaimran 2024-11-07 15:59:15 -08:00
parent de73cbc610
commit 3b1e8462cd
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/khoj/routers/api.py
View file

@ -351,6 +351,7 @@ async def extract_references_and_questions(
query_images: Optional[List[str]] = None, query_images: Optional[List[str]] = None,
agent: Agent = None, agent: Agent = None,
tracer: dict = {}, tracer: dict = {},
attached_files: str = None,
): ):
user = request.user.object if request.user.is_authenticated else None user = request.user.object if request.user.is_authenticated else None
@ -425,6 +426,7 @@ async def extract_references_and_questions(
max_prompt_size=conversation_config.max_prompt_size, max_prompt_size=conversation_config.max_prompt_size,
personality_context=personality_context, personality_context=personality_context,
tracer=tracer, tracer=tracer,
attached_files=attached_files,
) )
elif conversation_config.model_type == ChatModelOptions.ModelType.OPENAI: elif conversation_config.model_type == ChatModelOptions.ModelType.OPENAI:
openai_chat_config = conversation_config.openai_config openai_chat_config = conversation_config.openai_config
@ -443,6 +445,7 @@ async def extract_references_and_questions(
vision_enabled=vision_enabled, vision_enabled=vision_enabled,
personality_context=personality_context, personality_context=personality_context,
tracer=tracer, tracer=tracer,
attached_files=attached_files,
) )
elif conversation_config.model_type == ChatModelOptions.ModelType.ANTHROPIC: elif conversation_config.model_type == ChatModelOptions.ModelType.ANTHROPIC:
api_key = conversation_config.openai_config.api_key api_key = conversation_config.openai_config.api_key
@ -458,6 +461,7 @@ async def extract_references_and_questions(
vision_enabled=vision_enabled, vision_enabled=vision_enabled,
personality_context=personality_context, personality_context=personality_context,
tracer=tracer, tracer=tracer,
attached_files=attached_files,
) )
elif conversation_config.model_type == ChatModelOptions.ModelType.GOOGLE: elif conversation_config.model_type == ChatModelOptions.ModelType.GOOGLE:
api_key = conversation_config.openai_config.api_key api_key = conversation_config.openai_config.api_key
@ -474,6 +478,7 @@ async def extract_references_and_questions(
vision_enabled=vision_enabled, vision_enabled=vision_enabled,
personality_context=personality_context, personality_context=personality_context,
tracer=tracer, tracer=tracer,
attached_files=attached_files,
) )
# Collate search results as context for GPT # Collate search results as context for GPT