Add a new config section

BlueBrain · Nov 19, 2024 · 4c01253 · 4c01253
1 parent b5fbdd9
commit 4c01253
Showing 1 changed file with 11 additions and 0 deletions.
diff --git 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