Merge pull request #4 from KyMidd/feature/restore-bedrock-agent-runtime-client

restore bedrock agent runtime client

Author: KyMidd

python/devopsbot.py

@@ -64,10 +64,16 @@
 
 
 # Function to retrieve info from RAG with knowledge base
-def ask_bedrock_llm_with_knowledge_base(flat_conversation, knowledge_base_id, bedrock_client) -> str:
+def ask_bedrock_llm_with_knowledge_base(flat_conversation, knowledge_base_id) -> str:
+    
+    # Create a Bedrock agent runtime client
+    bedrock_agent_runtime_client = boto3.client(
+        "bedrock-agent-runtime", 
+        region_name=model_region_name
+    )
 
     # uses embedding model to retrieve and generate a response
-    response = bedrock_client.retrieve(
+    response = bedrock_agent_runtime_client.retrieve(
         retrievalQuery={
           'text': flat_conversation
         },
@@ -617,7 +623,7 @@ def handle_message_event(client, body, say, bedrock_client, app, token, register
             print(f"🚀 Flat conversation: {flat_conversation}")
 
         # Get context data from the knowledge base
-        knowledge_base_response = ask_bedrock_llm_with_knowledge_base(flat_conversation, ConfluenceKnowledgeBaseId, bedrock_client)
+        knowledge_base_response = ask_bedrock_llm_with_knowledge_base(flat_conversation, ConfluenceKnowledgeBaseId)
 
         if os.environ.get("VERA_DEBUG", "False") == "True":
             print(f"🚀 Knowledge base response: {knowledge_base_response}")