Use argument() instead of [function][arguments]

hemang1729 · hemang1729 · commit b476f13a3a3b · 2024-12-10T16:57:25.000+01:00
diff --git a/docs/testing/Examples/langgraph.md b/docs/testing/Examples/langgraph.md
@@ -116,7 +116,7 @@ def test_weather_agent_with_sf_and_nyc(weather_agent):
         find_weather_tool_calls = trace.tool_calls(name="_find_weather")
         assert_true(len(find_weather_tool_calls) == 2)
         find_weather_tool_call_args = str(
-            F.map(lambda x: x["function"]["arguments"], find_weather_tool_calls)
+            F.map(lambda x: x.argument(), find_weather_tool_calls)
         )
         assert_true(
             "San Francisco" in find_weather_tool_call_args

Original file line number	Diff line number	Diff line change
`@@ -116,7 +116,7 @@ def test_weather_agent_with_sf_and_nyc(weather_agent):`
`116`	`116`	`find_weather_tool_calls = trace.tool_calls(name="_find_weather")`
`117`	`117`	`assert_true(len(find_weather_tool_calls) == 2)`
`118`	`118`	`find_weather_tool_call_args = str(`
`119`		`- F.map(lambda x: x["function"]["arguments"], find_weather_tool_calls)`
	`119`	`+ F.map(lambda x: x.argument(), find_weather_tool_calls)`
`120`	`120`	`)`
`121`	`121`	`assert_true(`
`122`	`122`	`"San Francisco" in find_weather_tool_call_args`