modelcontextprotocol / modelcontextprotocol/python-sdk

Expose `schema_generator` on `FastMCP` for tool params schema generation

Aberta
#2,582 0 comentários 0 reações 0 responsáveis Ver no GitHub

Ninguém assumiu esta issue ainda.

enhancement needs decision P3
Linguagem predominante
Python
Estrelas
24.3k
Forks
4k
Merge médio
1d 1h
PRs com merge (30d)
31

Descrição

Description
Context

Tool.from_function (in src/mcp/server/fastmcp/tools/base.py) generates the
tool parameters JSON schema via:

parameters = func_arg_metadata.arg_model.model_json_schema(by_alias=True)

This is hardcoded — no way for users to customize the generation. Pydantic 2.11+
supports union_format='primitive_type_array' which produces compact
LLM-friendly schemas like {"type": ["integer", "string"]} instead of
{"anyOf": [...]}. There's no path to enable it without monkey-patching or
post-processing.

Proposal

Add schema_generator: type[GenerateJsonSchema] | None = None to FastMCP.__init__.
When provided, plumb it to both model_json_schema calls:

# tools/base.py
parameters = func_arg_metadata.arg_model.model_json_schema(
    by_alias=True,
    schema_generator=self.schema_generator or GenerateJsonSchema,
)

# utilities/func_metadata.py
schema = model.model_json_schema(
    schema_generator=self.schema_generator or StrictJsonSchema,
)
Use case
from pydantic.json_schema import GenerateJsonSchema

class CompactUnionGen(GenerateJsonSchema):
    def __init__(self, by_alias=True, ref_template='#/$defs/{model}'):
        super().__init__(
            by_alias=by_alias,
            ref_template=ref_template,
            union_format='primitive_type_array',
        )

mcp = FastMCP("srv", schema_generator=CompactUnionGen)
Tradeoff to discuss

func_metadata.py currently uses StrictJsonSchema (warnings → errors). With a
user-supplied generator, strict semantics are lost unless the user extends
StrictJsonSchema themselves. Proposed resolution: document that custom
generators should extend StrictJsonSchema if strict output validation is
desired.

Backward compat

Default None preserves current behavior exactly.

References

No response

Guia de contribuição

Abrir o guia de contribuição

Primeiros passos

  1. Leia a issue inteira e depois o guia de contribuição do projeto.
  2. Comente na issue dizendo que vai assumir — evita que duas pessoas façam o mesmo trabalho.
  3. Faça um fork do repositório e trabalhe em uma branch.
  4. Abra um pull request que referencie o número da issue.

Direção de pesquisa

Comece por src/mcp/server/fastmcp/tools/base.py e utilities/func_metadata.py e, em seguida, rastreie como os valores de FastMCP.init chegam às duas chamadas de model_json_schema. A alteração estará concluída quando um schema_generator opcional chegar a ambos os caminhos, os valores padrão preservarem o comportamento atual e o trade-off de strict-schema for documentado ou coberto conforme proposto.

Escrita pelo modelo de indexação a partir do texto da issue.

Avaliação

Stack de tecnologia
python
Domínio
backend-api-design
Tipo de issue
Funcionalidade
Dificuldade
3/5
Tempo estimado
1-2 dias
Status de atividade
Pouca atividade
Clareza
Razoavelmente clara
Facilidade para iniciantes
55/100

Receba novas issues na sua caixa de entrada

Um resumo curto de issues do GitHub para quem está começando.