# [... import ...]

class GraphQLSchema(BaseModel):
    gql_schema: str = Field(description="The schema of the graphql server")


[...crew class modified from template]
    @task
    def determine_schema(self) -> Task:
        return Task(
            config=self.tasks_config["determine_schema"],  # type: ignore[index]
            output_pydantic=GraphQLSchema,
        )

---

determine_schema:
  description: >
    Query the graphql server to get the schema.
    Use the graphql mcp tool to get the schema.
    Return any error from the server, if the query results in error.
    No other outcome is expected and allowed.
  markdown: true
  expected_output: >
    The schema of the GraphQL server, as a string.

  agent: query_manager

---

type Query {
  countries: [Country]
  country(code: String!): Country
}

type Country {
  code: String
  name: String
  native: String
  phone: String
  continent: Continent
  currency: String
  languages: [Language]
  emoji: String
}

type Continent {
  code: String
  name: String
}

type Language {
  code: String
  name: String
}

---

Traceback (most recent call last):
  File "/Users/kalfa/Source/graphql-test/graphql/src/graphql/main.py", line 24, in run
    Graphql().crew().kickoff(inputs=inputs)
    ~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^
  File "/Users/kalfa/Source/graphql-test/graphql/.venv/lib/python3.13/site-packages/crewai/crew.py", line 743, in kickoff
    result = self._run_sequential_process()
  File "/Users/kalfa/Source/graphql-test/graphql/.venv/lib/python3.13/site-packages/crewai/crew.py", line 1150, in _run_sequential_process
    return self._execute_tasks(self.tasks)
           ~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^
  File "/Users/kalfa/Source/graphql-test/graphql/.venv/lib/python3.13/site-packages/crewai/crew.py", line 1236, in _execute_tasks
    task_output = task.execute_sync(
        agent=exec_data.agent,
        context=context,
        tools=exec_data.tools,
    )
  File "/Users/kalfa/Source/graphql-test/graphql/.venv/lib/python3.13/site-packages/crewai/task.py", line 499, in execute_sync
    return self._execute_core(agent, context, tools)
           ~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/kalfa/Source/graphql-test/graphql/.venv/lib/python3.13/site-packages/crewai/task.py", line 740, in _execute_core
    raise e  # Re-raise the exception after emitting the event
    ^^^^^^^
  File "/Users/kalfa/Source/graphql-test/graphql/.venv/lib/python3.13/site-packages/crewai/task.py", line 678, in _execute_core
    pydantic_output, json_output = self._export_output(result)
                                   ~~~~~~~~~~~~~~~~~~~^^^^^^^^
  File "/Users/kalfa/Source/graphql-test/graphql/.venv/lib/python3.13/site-packages/crewai/task.py", line 968, in _export_output
    model_output = convert_to_model(
        result,
    ...<3 lines>...
        self.converter_cls,
    )
  File "/Users/kalfa/Source/graphql-test/graphql/.venv/lib/python3.13/site-packages/crewai/utilities/converter.py", line 194, in convert_to_model
    return handle_partial_json(
        result=result,
    ...<3 lines>...
        converter_cls=converter_cls,
    )
  File "/Users/kalfa/Source/graphql-test/graphql/.venv/lib/python3.13/site-packages/crewai/utilities/converter.py", line 261, in handle_partial_json
    exported_result = model.model_validate_json(match.group())
  File "/Users/kalfa/Source/graphql-test/graphql/.venv/lib/python3.13/site-packages/pydantic/main.py", line 746, in model_validate_json
    return cls.__pydantic_validator__.validate_json(
           ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^
        json_data, strict=strict, context=context, by_alias=by_alias, by_name=by_name
        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
    )
    ^
pydantic_core._pydantic_core.ValidationError: 1 validation error for GraphQLSchema
  Invalid JSON: key must be a string at line 2 column 3 [type=json_invalid, input_value='{\n  code: ID!\n  countr...ing\n  name: String!\n}', input_type=str]
    For further information visit https://errors.pydantic.dev/2.11/v/json_invalid

---

# Logic:
# 1. Objects: Must be empty {} or start with a double-quoted key {"key":
# 2. Arrays:  Must be empty [] or start with a JSON value: ", digit, -, true, false, null, [, or {
# 3. Strings: Standard JSON double-quoted strings
# 4. Primitives: true, false, null, and numbers.
json_pattern = r'(\{\s*\}|\{\s*"[^"\\]*"\s*:.*?[^"\\]\}|\[\s*\]|\[\s*(?:"|[\d\-]|true|false|null|[\[{]).*?[^"\\]\]|"(?:\\.|[^"\\])*"|\b(?:true|false|null)\b|-?\d+(?:\.\d+)?(?:[eE][+-]?\d+)?)'