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

Only trigger claude, openai monolithic operators with specific models

Browse Source 
To use Anthropic monolithic operator, set chat model to claude-3.7-sonnet
To use Openai monolithic operator, set chat model to gpt-4o
This commit is contained in:
Debanjum
2025-05-09 15:26:54 -06:00
parent 258b5a0372
commit 3a75cd3c3d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/khoj/processor/operator/operate_browser.py
View File

@@ -48,17 +48,17 @@ async def operate_browser(
# Initialize Agent # Initialize Agent
max_iterations = 40 # TODO: Configurable? max_iterations = 40 # TODO: Configurable?
operator_agent: OperatorAgent operator_agent: OperatorAgent
if reasoning_model.name.startswith("gpt-"): if reasoning_model.name.startswith("gpt-4o"):
operator_agent = OpenAIOperatorAgent(query, reasoning_model, max_iterations, tracer) operator_agent = OpenAIOperatorAgent(query, reasoning_model, max_iterations, tracer)
elif reasoning_model.name.startswith("claude-"): elif reasoning_model.name.startswith("claude-3-7-sonnet"):
operator_agent = AnthropicOperatorAgent(query, reasoning_model, max_iterations, tracer) operator_agent = AnthropicOperatorAgent(query, reasoning_model, max_iterations, tracer)
else: else:
grounding_model_name = "ui-tars-1.5-7b" grounding_model_name = "ui-tars-1.5"
grounding_model = await ConversationAdapters.aget_chat_model_by_name(grounding_model_name) grounding_model = await ConversationAdapters.aget_chat_model_by_name(grounding_model_name)
if ( if (
not grounding_model not grounding_model
or not grounding_model.vision_enabled or not grounding_model.vision_enabled
or grounding_model.model_type != ChatModel.ModelType.OPENAI or not grounding_model.model_type == ChatModel.ModelType.OPENAI
): ):
raise ValueError("No supported visual grounding model for binary operator agent found.") raise ValueError("No supported visual grounding model for binary operator agent found.")
operator_agent = BinaryOperatorAgent(query, reasoning_model, grounding_model, max_iterations, tracer) operator_agent = BinaryOperatorAgent(query, reasoning_model, grounding_model, max_iterations, tracer)