check agent id

awslabs · Jul 26, 2024 · 0070cf0 · 0070cf0
1 parent ff7fb97
commit 0070cf0
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/src/orchestrator.ts b/src/orchestrator.ts
@@ -251,6 +251,9 @@ export class MultiAgentOrchestrator {
   }
 
   addAgent(agent: Agent): void {
+    if (this.agents[agent.id]) {
+      throw new Error(`An agent with ID '${agent.id}' already exists.`);
+    }
     this.agents[agent.id] = agent;
     this.classifier.setAgents(this.agents);
   }

diff --git a/tests/Classifier.test.ts b/tests/Classifier.test.ts
@@ -58,6 +58,25 @@ describe("Classifier", () => {
         expect(response.selectedAgent?.name).toBe('Math Agent'); 
       }, 15000);
 
+      it('should throw an error when adding an agent with an existing ID', () => {
+        const existingAgent = new BedrockLLMAgent({
+          name: "EXISTING",
+          streaming: true,
+          description: 'Existing agent',
+        });
+        orchestrator.addAgent(existingAgent);
+
+        const duplicateAgent = new BedrockLLMAgent({
+          name: "EXISTING",
+          streaming: true,
+          description: 'Duplicate agent',
+        });
+
+        expect(() => {
+          orchestrator.addAgent(duplicateAgent);
+        }).toThrow(`An agent with ID '${duplicateAgent.id}' already exists.`);
+      });
+
 });