Prefer agent chat model to extract document search queries

Make chat model preference order for document search consistent with
all other tools.
This commit is contained in:
Debanjum
2025-08-27 13:36:15 -07:00
parent 7a42042488
commit 2091044db5

View File

@@ -1224,6 +1224,7 @@ async def search_documents(
personality_context=personality_context, personality_context=personality_context,
location_data=location_data, location_data=location_data,
chat_history=chat_history, chat_history=chat_history,
agent=agent,
tracer=tracer, tracer=tracer,
) )
@@ -1275,6 +1276,7 @@ async def extract_questions(
location_data: LocationData = None, location_data: LocationData = None,
chat_history: List[ChatMessageModel] = [], chat_history: List[ChatMessageModel] = [],
max_queries: int = 5, max_queries: int = 5,
agent: Agent = None,
tracer: dict = {}, tracer: dict = {},
): ):
""" """
@@ -1319,6 +1321,8 @@ async def extract_questions(
description="List of semantic search queries to run on user documents.", description="List of semantic search queries to run on user documents.",
) )
agent_chat_model = AgentAdapters.get_agent_chat_model(agent, user) if agent else None
raw_response = await send_message_to_model_wrapper( raw_response = await send_message_to_model_wrapper(
query=prompt, query=prompt,
query_files=query_files, query_files=query_files,
@@ -1327,6 +1331,7 @@ async def extract_questions(
response_type="json_object", response_type="json_object",
response_schema=DocumentQueries, response_schema=DocumentQueries,
fast_model=False, fast_model=False,
agent_chat_model=agent_chat_model,
user=user, user=user,
tracer=tracer, tracer=tracer,
) )