From 4c0125379d37f24d051d050ac22af9b8661384e7 Mon Sep 17 00:00:00 2001 From: Jan <18519371+jankrepl@users.noreply.github.com> Date: Tue, 19 Nov 2024 10:20:16 +0100 Subject: [PATCH] Add a new config section --- swarm_copy/app/config.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/swarm_copy/app/config.py b/swarm_copy/app/config.py index f0334e9..0204814 100644 --- a/swarm_copy/app/config.py +++ b/swarm_copy/app/config.py @@ -17,6 +17,16 @@ class SettingsAgent(BaseModel): model_config = ConfigDict(frozen=True) +class SettingsHIL(BaseModel): + """Agent setting.""" + + redis_uri: str + poll_interval: int = 5 + ttl_seconds: int = 600 # 10 minutes + + model_config = ConfigDict(frozen=True) + + class SettingsDB(BaseModel): """DB settings for retrieving history.""" @@ -213,6 +223,7 @@ class Settings(BaseSettings): tools: SettingsTools knowledge_graph: SettingsKnowledgeGraph + hil: SettingsHIL agent: SettingsAgent = SettingsAgent() # has no required db: SettingsDB = SettingsDB() # has no required openai: SettingsOpenAI = SettingsOpenAI() # has no required