Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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
|