(
session_id, query, bedrock_agent='solution', enable_trace=True, end_session=False)
| 34 | |
| 35 | |
| 36 | def invoke_bedrock_agent( |
| 37 | session_id, query, bedrock_agent='solution', enable_trace=True, end_session=False): |
| 38 | agent_id = retrieve_environment_variables("BEDROCK_AGENT_ID") |
| 39 | agent_alias_id = retrieve_environment_variables("BEDROCK_AGENT_ALIAS_ID") |
| 40 | |
| 41 | return bedrock_agent_runtime_client.invoke_agent( |
| 42 | inputText=query, |
| 43 | agentId=agent_id, |
| 44 | agentAliasId=agent_alias_id, |
| 45 | enableTrace=enable_trace, |
| 46 | endSession=end_session, |
| 47 | sessionId=session_id |
| 48 | ) |
| 49 | |
| 50 | |
| 51 | @st.fragment |
no test coverage detected