MongoDB With Bedrock Agent: Quick Tutorial

1
{
2
  "fields": [
3
    {
4
      "type": "vector",
5
      "path": "embedding",
6
      "numDimensions": 1024,
7
      "similarity": "cosine"
8
    },
9
    {
10
      "type" : "filter",
11
      "path" : "metadata"
12
    },
13
    {
14
      "type" : "filter",
15
      "path" : "text"
16
    },
17
  ]
18
}

1
!pip install boto3

1
import boto3
2
from botocore.exceptions import ClientError
3
import os
4
import random
5
import getpass
6
7
8
# Get AWS credentials from user
9
aws_access_key = getpass.getpass("Enter your AWS Access Key: ")
10
aws_secret_key = getpass.getpass("Enter your AWS Secret Key: ")

1
bedrock_agent_runtime = boto3.client('bedrock-agent-runtime',
2
                                    aws_access_key_id=aws_access_key,
3
                                    aws_secret_access_key=aws_secret_key,
4
                                    region_name="us-east-1")
5
6
7
# Define agent IDs (replace these with your actual agent IDs)
8
agent_id = getpass.getpass("Enter your agent ID")
9
agent_alias_id = getpass.getpass("Enter your agent Alias ID")

1
def randomise_session_id():
2
   """
3
   Generate a random session ID.
4
5
6
   Returns:
7
       str: A random session ID.
8
   """
9
   return str(random.randint(1000, 9999))
10
11
12
def data_stream_generator(response):
13
   """
14
   Generator to yield data chunks from the response.
15
16
17
   Args:
18
       response (dict): The response dictionary.
19
20
21
   Yields:
22
       str: The next chunk of data.
23
   """
24
   for event in response["completion"]:
25
       chunk = event.get("chunk", {})
26
       if "bytes" in chunk:
27
           yield chunk["bytes"].decode()
28
29
30
def invoke_agent(bedrock_agent_runtime, agent_id, agent_alias_id, session_id, prompt):
31
   """
32
   Sends a prompt for the agent to process and respond to, streaming the response data.
33
34
35
   Args:
36
       bedrock_agent_runtime (boto3 client): The runtime client to invoke the agent.
37
       agent_id (str): The unique identifier of the agent to use.
38
       agent_alias_id (str): The alias of the agent to use.
39
       session_id (str): The unique identifier of the session. Use the same value across requests to continue the same conversation.
40
       prompt (str): The prompt that you want the agent to complete.
41
42
43
   Returns:
44
       str: The response from the agent.
45
   """
46
   try:
47
       response = bedrock_agent_runtime.invoke_agent(
48
           agentId=agent_id,
49
           agentAliasId=agent_alias_id,
50
           sessionId=session_id,
51
           inputText=prompt,
52
       )
53
54
55
       # Use the data stream generator to stream the response
56
       ret_response = ''.join(data_stream_generator(response))
57
58
59
       return ret_response
60
61
62
   except Exception as e:
63
       return f"Error invoking agent: {e}"

1
# Initialize chat history and session ID
2
session_id = randomise_session_id()
3
4
5
while True:
6
   prompt = input("Enter your prompt (or type 'exit' to quit): ")
7
8
9
   if prompt.lower() == 'exit':
10
       break
11
12
13
   response = invoke_agent(bedrock_agent_runtime, agent_id, agent_alias_id, session_id, prompt)
14
15
16
   print("Agent Response:")
17
   print(response)