Use json schema to enforce generate online queries format

This commit is contained in:
Debanjum
2025-03-19 14:31:39 +05:30
parent 2c53eb9de1
commit a5627ef787

View File

@@ -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,