ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-23 14:04:18 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix(OpenAPIParser): Fix missing $defs for response schemas in experimental OpenAPI parser (#2398)

Browse source 
* fix(OpenAPIParser): Fix missing  for response schemas in experimental OpenAPI parser

* chore: run linting

---------

Co-authored-by: Christoph Netsch <netsch@alpamayo.ch>
This commit is contained in:
ChristophNetsch 2025-11-15 16:50:00 +01:00 • committed by GitHub
commit 67f9329b83
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 18 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

31
src/fastmcp/experimental/utilities/openapi/parser.py
View file

@ -630,25 +630,28 @@ class OpenAPIParser(
Returns:
Dictionary containing only the schemas needed for outputs
"""
needed_schemas = set()
if not responses or not all_schemas:
return {}
needed_schemas: set[str] = set()
# Check responses for schema references
for response in responses.values():
if response.content_schema:
for content_schema in response.content_schema.values():
# Check if this schema was originally a top-level $ref
if "x-fastmcp-top-level-schema" in content_schema:
schema_name = content_schema["x-fastmcp-top-level-schema"]
if schema_name in all_schemas:
needed_schemas.add(schema_name)
if not response.content_schema:
continue
# Extract all dependencies (transitive refs within the schema)
deps = self._extract_schema_dependencies(
content_schema, all_schemas
for content_schema in response.content_schema.values():
deps = self._extract_schema_dependencies(content_schema, all_schemas)
needed_schemas.update(deps)
schema_name = content_schema.get("x-fastmcp-top-level-schema")
if isinstance(schema_name, str) and schema_name in all_schemas:
needed_schemas.add(schema_name)
self._extract_schema_dependencies(
all_schemas[schema_name],
all_schemas,
collected=needed_schemas,
)
needed_schemas.update(deps)
# Return only the needed output schemas
return {
name: all_schemas[name] for name in needed_schemas if name in all_schemas
}