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

Use json schema to enforce generate online queries format

Browse Source
This commit is contained in:
Debanjum
2025-03-19 14:31:39 +05:30
parent 2c53eb9de1
commit a5627ef787
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -540,11 +540,15 @@ async def generate_online_subqueries(
agent_chat_model = agent.chat_model if agent else None agent_chat_model = agent.chat_model if agent else None
class OnlineQueries(BaseModel):
queries: List[str]
with timer("Chat actor: Generate online search subqueries", logger): with timer("Chat actor: Generate online search subqueries", logger):
response = await send_message_to_model_wrapper( response = await send_message_to_model_wrapper(
online_queries_prompt, online_queries_prompt,
query_images=query_images, query_images=query_images,
response_type="json_object", response_type="json_object",
response_schema=OnlineQueries,
user=user, user=user,
query_files=query_files, query_files=query_files,
agent_chat_model=agent_chat_model, agent_chat_model=agent_chat_model,