From b14c2d33b37737f46f7ad90c00c1255cf4f1fe63 Mon Sep 17 00:00:00 2001 From: Jason Date: Wed, 13 Sep 2023 22:10:18 -0400 Subject: [PATCH] fix bug --- instructor/function_calls.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/instructor/function_calls.py b/instructor/function_calls.py index 97938c1..c759aae 100644 --- a/instructor/function_calls.py +++ b/instructor/function_calls.py @@ -31,7 +31,7 @@ def _remove_a_key(d, remove_key) -> None: """Remove a key from a dictionary recursively""" if isinstance(d, dict): for key in list(d.keys()): - if key == remove_key: + if key == remove_key and "type" in d.keys(): del d[key] else: _remove_a_key(d[key], remove_key) @@ -200,6 +200,7 @@ class OpenAISchema(BaseModel): f"the required parameters with correct types" ) + _remove_a_key(parameters, "title") _remove_a_key(parameters, "additionalProperties") return { "name": schema["title"],