Skip to content

agent_script_spec.generate

agent_script_spec.generate

generate_schemas()

Generate JSON schemas from Pydantic models.

Source code in packages/agent-script-spec/src/agent_script_spec/generate.py
Python
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
def generate_schemas():
    """Generate JSON schemas from Pydantic models."""
    package_dir = Path(__file__).parent.parent.parent
    output_dir = package_dir / "dist"
    output_dir.mkdir(exist_ok=True)

    # Get version from environment or default
    version = os.environ.get("SCHEMA_VERSION", "0.1.0")

    # Generate schema for main model
    schema = AgentScriptSpecification.model_json_schema()

    # Add metadata with version
    schema["$id"] = (
        f"https://github.com/theagenticguy/agent-script-spec/releases/download/schema-v{version}/agent_script_spec.json"
    )
    schema["title"] = "Agent Script Specification Schema"
    schema["version"] = version

    output_file = output_dir / "agent_script_spec.json"
    with open(output_file, "w") as f:
        json.dump(schema, f, indent=2)

    print(f"Generated schema: {output_file} (version: {version})")
    return