Added a correct way to get a custom tool's description (#68)

* Fixed support for custom tool
langchain-ai · Mar 19, 2024 · df1814e · df1814e
1 parent 90bcf52
commit df1814e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/vertexai/langchain_google_vertexai/functions_utils.py b/libs/vertexai/langchain_google_vertexai/functions_utils.py
@@ -67,7 +67,7 @@ def _format_tools_to_vertex_tool(
         elif isinstance(tool, dict):
             func = {
                 "name": tool["name"],
-                "description": tool.pop("description"),
+                "description": tool["description"],
                 "parameters": _get_parameters_from_schema(tool["parameters"]),
             }
         else: