Merge pull request #94 from brnaba-aws/main

Added support for sessionState in Agents for Bedrock TS
awslabs · Nov 21, 2024 · 0a9f7cc · 0a9f7cc
2 parents da9563f + efd295e
commit 0a9f7cc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/typescript/src/agents/amazonBedrockAgent.ts b/typescript/src/agents/amazonBedrockAgent.ts
@@ -52,14 +52,15 @@ export class AmazonBedrockAgent extends Agent {
     userId: string,
     sessionId: string,
     chatHistory: ConversationMessage[],
-    additionalParams?: Record<string, string>
+    additionalParams?: Record<any, any>
   ): Promise<ConversationMessage> {
     // Construct the command to invoke the Amazon Bedrock agent with user input
     const command = new InvokeAgentCommand({
       agentId: this.agentId,
       agentAliasId: this.agentAliasId,
-      sessionId,
-      inputText
+      sessionId: sessionId,
+      inputText: inputText,
+      sessionState: additionalParams ? additionalParams.sessionState?  additionalParams.sessionState : undefined : undefined,
     });
 
     try {

diff --git a/typescript/src/orchestrator.ts b/typescript/src/orchestrator.ts
@@ -345,7 +345,7 @@ export class MultiAgentOrchestrator {
     userInput: string,
     userId: string,
     sessionId: string,
-    additionalParams: Record<string, string> = {}
+    additionalParams: Record<any, any> = {}
   ): Promise<AgentResponse> {
     this.executionTimes = new Map();
     let classifierResult: ClassifierResult;