No duplicate schemas

Author: g-eoj

pydantic_ai_slim/pydantic_ai/_output.py

@@ -385,7 +385,7 @@ def _build_processor(
 
         return UnionOutputProcessor(outputs=outputs, strict=strict, name=name, description=description)
 
-    def build_json_schema(self) -> JsonSchema:
+    def build_json_schema(self) -> JsonSchema:  # noqa: C901
         # allow any output with {'type': 'string'} if no constraints
         if not any([self.allows_deferred_tools, self.allows_image, self.object_def, self.toolset]):
             return TypeAdapter(str).json_schema()
@@ -407,19 +407,23 @@ def build_json_schema(self) -> JsonSchema:
                 json_schema = tool_processor.object_def.json_schema
                 if k := tool_processor.outer_typed_dict_key:
                     json_schema = json_schema['properties'][k]
-                json_schemas.append(json_schema)
+                if json_schema not in json_schemas:
+                    json_schemas.append(json_schema)
 
         elif self.allows_text:
             json_schema = TypeAdapter(str).json_schema()
             json_schemas.append(json_schema)
 
         if self.allows_deferred_tools:
             json_schema = TypeAdapter(DeferredToolRequests).json_schema(mode='serialization')
-            json_schemas.append(json_schema)
+            if json_schema not in json_schemas:
+                json_schemas.append(json_schema)
+
         if self.allows_image:
             json_schema = TypeAdapter(_messages.BinaryImage).json_schema()
             json_schema = {k: v for k, v in json_schema['properties'].items() if k in ['data', 'media_type']}
-            json_schemas.append(json_schema)
+            if json_schema not in json_schemas:
+                json_schemas.append(json_schema)
 
         if len(json_schemas) == 1:
             return json_schemas[0]

tests/test_agent_output_schemas.py

@@ -101,9 +101,7 @@ async def test_tool_output_json_schema():
         'test',
         output_type=[ToolOutput(bool), ToolOutput(bool), ToolOutput(bool)],
     )
-    assert agent.output_json_schema() == snapshot(
-        {'anyOf': [{'type': 'boolean'}, {'type': 'boolean'}, {'type': 'boolean'}]}
-    )
+    assert agent.output_json_schema() == snapshot({'type': 'boolean'})
 
 
 async def test_native_output_json_schema():