mirror of
https://github.com/khoaliber/khoj.git
synced 2026-03-04 13:20:17 +00:00
Use json schema to enforce generate online queries format
This commit is contained in:
@@ -540,11 +540,15 @@ async def generate_online_subqueries(
|
||||
|
||||
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):
|
||||
response = await send_message_to_model_wrapper(
|
||||
online_queries_prompt,
|
||||
query_images=query_images,
|
||||
response_type="json_object",
|
||||
response_schema=OnlineQueries,
|
||||
user=user,
|
||||
query_files=query_files,
|
||||
agent_chat_model=agent_chat_model,
|
||||
|
||||
Reference in New Issue
Block a user