Spaces:
Running
on
T4
Running
on
T4
change ksize in RAG
Browse files- RAG/bedrock_agent.py +6 -6
RAG/bedrock_agent.py
CHANGED
@@ -30,24 +30,24 @@ logger = logging.getLogger(__name__)
|
|
30 |
#bedrock_agent_client = boto3.client('bedrock-agent',region_name=region)
|
31 |
bedrock_agent_runtime_client = boto3.client(
|
32 |
'bedrock-agent-runtime',
|
33 |
-
aws_access_key_id=st.secrets['
|
34 |
-
aws_secret_access_key=st.secrets['
|
35 |
)
|
36 |
enable_trace:bool = True
|
37 |
end_session:bool = False
|
38 |
|
39 |
def delete_memory():
|
40 |
response = bedrock_agent_runtime_client.delete_agent_memory(
|
41 |
-
agentAliasId='
|
42 |
-
agentId='
|
43 |
)
|
44 |
|
45 |
def query_(inputs):
|
46 |
# invoke the agent API
|
47 |
agentResponse = bedrock_agent_runtime_client.invoke_agent(
|
48 |
inputText=inputs['shopping_query'],
|
49 |
-
agentId='
|
50 |
-
agentAliasId='
|
51 |
sessionId=st.session_state.session_id_,
|
52 |
enableTrace=enable_trace,
|
53 |
endSession= end_session
|
|
|
30 |
#bedrock_agent_client = boto3.client('bedrock-agent',region_name=region)
|
31 |
bedrock_agent_runtime_client = boto3.client(
|
32 |
'bedrock-agent-runtime',
|
33 |
+
aws_access_key_id=st.secrets['user_access_key_us_west_2'],
|
34 |
+
aws_secret_access_key=st.secrets['user_secret_key_us_west_2'], region_name = 'us-west-2'
|
35 |
)
|
36 |
enable_trace:bool = True
|
37 |
end_session:bool = False
|
38 |
|
39 |
def delete_memory():
|
40 |
response = bedrock_agent_runtime_client.delete_agent_memory(
|
41 |
+
agentAliasId='DEEEEZM2TM',
|
42 |
+
agentId='EJVGQW1BH7'
|
43 |
)
|
44 |
|
45 |
def query_(inputs):
|
46 |
# invoke the agent API
|
47 |
agentResponse = bedrock_agent_runtime_client.invoke_agent(
|
48 |
inputText=inputs['shopping_query'],
|
49 |
+
agentId='EJVGQW1BH7',
|
50 |
+
agentAliasId='DEEEEZM2TM',
|
51 |
sessionId=st.session_state.session_id_,
|
52 |
enableTrace=enable_trace,
|
53 |
endSession= end_session
|