fix failing tests

Change-Id: I0fdfc084e8c5e88bab8257eef22065ec5a6189ea

Author: MarkDaoust

google/generativeai/types/content_types.py

@@ -483,6 +483,17 @@ def strip_titles(schema):
     if items is not None:
         strip_titles(items)
 
+def strip_additional_properties(schema):
+    schema.pop("additionalProperties", None)
+
+    properties = schema.get("properties", None)
+    if properties is not None:
+        for name, value in properties.items():
+            strip_additional_properties(value)
+
+    items = schema.get("items", None)
+    if items is not None:
+        strip_additional_properties(items)
 
 def add_object_type(schema):
     properties = schema.get("properties", None)

tests/test_content.py

@@ -663,6 +663,9 @@ def test_auto_schema(self, annotation, expected):
         def fun(a: annotation):
             pass
 
+        if annotation == dict[str, Any]:
+            breakpoint()
+
         cfd = content_types.FunctionDeclaration.from_function(fun)
         got = cfd.parameters.properties["a"]
         self.assertEqual(got, expected)