Update _rm_titles function

langchain-ai · Mar 13, 2024 · b743d92 · b743d92
1 parent 6adea20
commit b743d92
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/libs/core/langchain_core/utils/function_calling.py b/libs/core/langchain_core/utils/function_calling.py
@@ -51,10 +51,14 @@ class ToolDescription(TypedDict):
     function: FunctionDescription
 
 
-def _rm_titles(kv: dict) -> dict:
+def _rm_titles(
+    kv: dict
+) -> dict:
+
     new_kv = {}
     for k, v in kv.items():
-        if k == "title":
+        # print(kv.keys())
+        if k == "title" and 'type' in kv.keys():
             continue
         elif isinstance(v, dict):
             new_kv[k] = _rm_titles(v)