Skip to content

Commit 3c70e0d

Browse files
committed
define bedrock client
1 parent 1b1e875 commit 3c70e0d

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

python/devopsbot.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -62,15 +62,9 @@
6262

6363

6464
# Function to retrieve info from RAG with knowledge base
65-
def ask_bedrock_llm_with_knowledge_base(flat_conversation, knowledge_base_id) -> str:
66-
# Create a Bedrock agent runtime client
67-
bedrock_agent_runtime_client = boto3.client(
68-
"bedrock-agent-runtime",
69-
region_name=model_region_name
70-
)
71-
65+
def ask_bedrock_llm_with_knowledge_base(flat_conversation, knowledge_base_id, bedrock_client) -> str:
7266
# uses embedding model to retrieve and generate a response
73-
response = bedrock_agent_runtime_client.retrieve(
67+
response = bedrock_client.retrieve(
7468
retrievalQuery={
7569
'text': flat_conversation
7670
},
@@ -544,7 +538,7 @@ def handle_message_event(client, body, say, bedrock_client, app, token, register
544538
print(f"🚀 Flat conversation: {flat_conversation}")
545539

546540
# Get context data from the knowledge base
547-
knowledge_base_response = ask_bedrock_llm_with_knowledge_base(flat_conversation, ConfluenceKnowledgeBaseId)
541+
knowledge_base_response = ask_bedrock_llm_with_knowledge_base(flat_conversation, ConfluenceKnowledgeBaseId, bedrock_client)
548542

549543
if os.environ.get("VERA_DEBUG", "False") == "True":
550544
print(f"🚀 Knowledge base response: {knowledge_base_response}")

0 commit comments

Comments
 (0)