BTh,ddlZddlmZddlmZmZmZddlmZddl Z ddl m Z ddl m Z ddlmZGdd eZGd d eZy) N)abstractmethod)TupleTypeUnion)mock)BaseTool) SecretStr)BaseStandardTestsceZdZdZeedeeeeffdZ ede fdZ ede fdZ e jdefdZy) ToolsTestsz :private: Base class for testing tools. This won't show in the documentation, but the docstrings will be inherited by subclasses. returncy)zE Returns a class or instance of a tool to be tested. Nselfs r/var/www/catia.catastroantioquia-mas.com/valormas/lib/python3.12/site-packages/langchain_tests/unit_tests/tools.pytool_constructorzToolsTests.tool_constructors ciS)zU Returns a dictionary of parameters to pass to the tool constructor. rrs rtool_constructor_paramsz"ToolsTests.tool_constructor_paramss  rciS)z Returns a dictionary representing the "args" of an example tool call. This should NOT be a ToolCall dict - it should not have {"name", "id", "args"} keys. rrs rtool_invoke_params_examplez%ToolsTests.tool_invoke_params_example#s  rct|jtr(|jik7r d}t ||jS|jdi|jS)z :private: zUIf tool_constructor is an instance of BaseTool, tool_constructor_params must be emptyr) isinstancerrr ValueError)rmsgs rtoolzToolsTests.tool-s` d++X 6++r1<!o%(( ($t$$Dt'C'CDDrN)__name__ __module__ __qualname____doc__propertyrrrrrdictrrpytestfixturerrrrr r s  %X(@"A   D ^^ Eh E Err creZdZdZedeeeeffdZd dZd dZ de ddfdZ de ddfd Z de ddfd Z y) ToolsUnitTestsz* Base class for tools unit tests. r c iiifS)z_Return env vars, init args, and expected instance attrs for initializing from env vars.rrs rinit_from_env_paramsz#ToolsUnitTests.init_from_env_paramsBs2rzrNct|jtr |j}n|jdi|j}|Jy)z Test that the tool can be initialized with :attr:`tool_constructor` and :attr:`tool_constructor_params`. If this fails, check that the keyword args defined in :attr:`tool_constructor_params` are valid. Nr)rrrrrrs r test_initzToolsUnitTests.test_initHsF d++X 6((D(4((H4+G+GHDrcr|j\}}}|rtjjtj |5|j di|}dddJ|jD]8\}}t||}t|tr|j}||k(r8Jyy#1swYZxYw)Nr) r)rpatchr#osenvironritemsgetattrrr get_secret_value)r env_params tools_paramsexpected_attrsrkexpectedactuals rtest_init_from_envz!ToolsUnitTests.test_init_from_envTs373L3L0 L. Z8 =,t,,<|< =# ##-335 * 8 q)fi0#446F)))  *  = =s B--B6rc |jsJy)z Tests that the tool has a name attribute to pass to chat models. If this fails, add a `name` parameter to your tool. N)namer+s r test_has_namezToolsUnitTests.test_has_name`s yyyrc(|jsJy)aS Tests that the tool has an input schema. If this fails, add an `args_schema` to your tool. See `this guide `_ and see how `CalculatorInput` is configured in the `CustomCalculatorTool.args_schema` attribute N)get_input_schemar+s rtest_has_input_schemaz$ToolsUnitTests.test_has_input_schemahs$$&&&rcL|j}|di|jsJy)z Tests that the provided example params match the declared input schema. If this fails, update the `tool_invoke_params_example` attribute to match the input schema (`args_schema`) of the tool. Nr)r?r)rr input_schemas r'test_input_schema_matches_invoke_paramsz6ToolsUnitTests.test_input_schema_matches_invoke_paramsus),,. >d==>>>r)r N)rrr r!r"rr#r)r,r:rr=r@rCrrrr'r'=sqeD$,<&=   *(t '( 't ' ?H ? ?rr')r/abcrtypingrrrunittestrr$langchain_core.toolsrpydanticr langchain_tests.baser r r'rrrrJs> %% )2-E"-E`B?ZB?r