prasadnu commited on
Commit
3e0e0e3
·
1 Parent(s): 84a6766

change ksize in RAG

Browse files
Files changed (1) hide show
  1. 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['user_access_key'],
34
- aws_secret_access_key=st.secrets['user_secret_key'], region_name = 'us-east-1'
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='TSTALIASID',
42
- agentId='B4Z7BTURC4'
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='B4Z7BTURC4',
50
- agentAliasId='TSTALIASID',
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