From 4385bfb3a5d780177e1e096bd022ea779d69e7ea Mon Sep 17 00:00:00 2001 From: selimseker Date: Thu, 6 Jun 2024 19:46:15 +0300 Subject: [PATCH] get agent max_iter config from env-var --- src/dria_searching_agent/config/config.py | 4 ++++ src/dria_searching_agent/main.py | 1 + 2 files changed, 5 insertions(+) diff --git a/src/dria_searching_agent/config/config.py b/src/dria_searching_agent/config/config.py index aa2b68e..9fedc55 100644 --- a/src/dria_searching_agent/config/config.py +++ b/src/dria_searching_agent/config/config.py @@ -7,6 +7,7 @@ def __init__(self): self.agent_model_provider = os.getenv('AGENT_MODEL_PROVIDER') self.agent_model = os.getenv('AGENT_MODEL') + self.agent_max_iter = os.getenv('AGENT_MAX_ITER', 10) self.anthropic_key = os.getenv('ANTHROPIC_KEY') self.openai_api_key = os.getenv('OPENAI_API_KEY') @@ -30,6 +31,9 @@ def AGENT_MODEL_PROVIDER(): def AGENT_MODEL(): return config.agent_model +def AGENT_MAX_ITER(): + return config.agent_max_iter + def ANTHROPIC_KEY(): return config.anthropic_key diff --git a/src/dria_searching_agent/main.py b/src/dria_searching_agent/main.py index 397ff64..c99f36c 100644 --- a/src/dria_searching_agent/main.py +++ b/src/dria_searching_agent/main.py @@ -120,6 +120,7 @@ def __create_agents(self): for k,v in self.agents_config.items(): agent = Agent( **v, + max_iter=config.AGENT_MAX_ITER(), # TODO: maybe we can get this from agents.json verbose=True, llm = self.__get_model(), tools=[