File tree Expand file tree Collapse file tree 1 file changed +2
-5
lines changed
instrumentation/opentelemetry-instrumentation-botocore/tests Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Original file line number Diff line number Diff line change @@ -1383,7 +1383,7 @@ def invoke_model_tool_call(
13831383 llm_model_value ,
13841384 expect_content ,
13851385):
1386- # pylint:disable=too-many-locals,too-many-statements
1386+ # pylint:disable=too-many-locals,too-many-statements,too-many-branches
13871387 user_prompt = "What is the weather in Seattle and San Francisco today? Please expect one tool call for Seattle and one for San Francisco"
13881388 if "anthropic.claude" in llm_model_value :
13891389 user_msg = {
@@ -2066,7 +2066,7 @@ def invoke_model_with_response_stream_tool_call(
20662066 # used only by anthropic claude
20672067 input_json_buf = ""
20682068 # used only by amazon nova
2069- tool_use = None
2069+ tool_use = {}
20702070 for event in response_0 ["body" ]:
20712071 json_bytes = event ["chunk" ].get ("bytes" , b"" )
20722072 decoded = json_bytes .decode ("utf-8" )
@@ -2173,9 +2173,6 @@ def invoke_model_with_response_stream_tool_call(
21732173 max_tokens = max_tokens ,
21742174 tools = tool_config ,
21752175 )
2176- import pprint
2177-
2178- pprint .pprint (messages [1 ])
21792176 response_1 = bedrock_runtime_client .invoke_model_with_response_stream (
21802177 body = body ,
21812178 modelId = llm_model_value ,
You can’t perform that action at this time.
0 commit comments