prevent title from being used in schema

minimaxir · Jul 24, 2023 · 6cfc43d · 6cfc43d
1 parent c7aa79b
commit 6cfc43d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/simpleaichat/chatgpt.py b/simpleaichat/chatgpt.py
@@ -74,6 +74,9 @@ def prepare_request(
 
     def schema_to_function(self, schema: Any):
         assert schema.__doc__, f"{schema.__name__} is missing a docstring."
+        assert (
+            "title" not in schema.__fields__.keys()
+        ), "`title` is a reserved keyword and cannot be used as a field name."
         schema_dict = schema.model_json_schema()
         remove_a_key(schema_dict, "title")