pydantic-mcp

Instructions

Implementation Reference

• src/pydantic_mcp/tools.py:239-268 (handler)

  The main handler function for migrate_v1_to_v2 which processes code or target model for migration issues.

  def migrate_v1_to_v2(
      code: str | None = None,
      target: str | None = None,
      apply_fixes: bool = False,
  ) -> ToolResponse:
      """Analyze a snippet or model source for common Pydantic v1-to-v2 migration issues."""
      if code is None and target is None:
          return make_response(
              diagnostics=[
                  Diagnostic(
                      level="error",
                      message="Provide either `code` or `target`.",
                      code="missing_input",
                  )
              ],
              result={"findings": [], "risk_level": "none", "updated_code": None},
          )
      if code is None and target is not None:
          runtime_target = resolve_target(
              target,
              registry=REGISTRY,
              settings=SERVER_SETTINGS,
          )
          code = inspect.getsource(
              runtime_target.model_class or runtime_target.annotation
          )
          response = migration_report(code, apply_fixes=apply_fixes)
          response.resolved_target = runtime_target.resolved
          return response
      return migration_report(code or "", apply_fixes=apply_fixes)

• src/pydantic_mcp/server.py:20-50 (registration)

  Registration of the migrate_v1_to_v2 tool in the server module.

  migrate_v1_to_v2 = _tools.migrate_v1_to_v2
  parse_partial_json = _tools.parse_partial_json
  generate_model_from_json = _tools.generate_model_from_json
  server_capabilities = _resources.server_capabilities
  models_index = _resources.models_index
  model_metadata = _resources.model_metadata
  model_schema = _resources.model_schema
  model_examples = _resources.model_examples
  reference_overview = _resources.reference_overview
  project_settings = _resources.project_settings
  project_import_roots = _resources.project_import_roots
  recent_errors = _resources.recent_errors
  migration_rules = _resources.migration_rules
  changed_models = _resources.changed_models
  
  __all__ = [
      "SERVER_NAME",
      "SERVER_SETTINGS",
      "SERVER_VERSION",
      "build_capabilities",
      "changed_models",
      "compare_validation_modes",
      "create_example_payload",
      "explain_model",
      "generate_json_schema",
      "generate_model_from_json",
      "inspect_type",
      "list_models",
      "main",
      "mcp",
      "migrate_v1_to_v2",