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

Configure max iterations per operator run via environment variable

Browse Source
This commit is contained in:
Debanjum
2025-05-20 01:03:11 -07:00
parent 06a1a22e3b
commit 19b4c18b69
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,6 +1,7 @@
import asyncio import asyncio
import json import json
import logging import logging
import os
from typing import Callable, List, Optional from typing import Callable, List, Optional
import requests import requests
@@ -46,7 +47,7 @@ async def operate_browser(
raise ValueError(f"No vision enabled chat model found. Configure a vision chat model to operate browser.") raise ValueError(f"No vision enabled chat model found. Configure a vision chat model to operate browser.")
# Initialize Agent # Initialize Agent
max_iterations = 40 # TODO: Configurable? max_iterations = int(os.getenv("KHOJ_OPERATOR_ITERATIONS", 40))
operator_agent: OperatorAgent operator_agent: OperatorAgent
if reasoning_model.name.startswith("gpt-4o"): 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)