fixed MultiAgentOrchestrator init

awslabs · Oct 16, 2024 · 0aac968 · 0aac968
1 parent 864dea8
commit 0aac968
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/python/src/multi_agent_orchestrator/orchestrator.py b/python/src/multi_agent_orchestrator/orchestrator.py
@@ -19,10 +19,10 @@
 @dataclass
 class MultiAgentOrchestrator:
     def __init__(self,
-                 options: OrchestratorConfig = DEFAULT_CONFIG,
-                 storage: ChatStorage = InMemoryChatStorage(),
-                 classifier: Classifier = BedrockClassifier(options=BedrockClassifierOptions()),
-                 logger: Logger = None):
+                 options: Optional[OrchestratorConfig] = None,
+                 storage: Optional[ChatStorage] = None,
+                 classifier: Optional[Classifier] = None,
+                 logger: Optional[Logger] = None):
 
         if options is None:
             options = {}
@@ -41,7 +41,8 @@ def __init__(self,
 
         self.logger = Logger(self.config, logger)
         self.agents: Dict[str, Agent] = {}
-        self.classifier: Classifier = classifier
+        self.storage = storage or InMemoryChatStorage()
+        self.classifier: Classifier = classifier or BedrockClassifier(options=BedrockClassifierOptions())
         self.execution_times: Dict[str, float] = {}
         self.default_agent: Agent = BedrockLLMAgent(
             options=BedrockLLMAgentOptions(