openapi-spec-generator 0.1.0__py3-none-any.whl

openapi_generator/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

openapi_generator/ai_client.py

@@ -0,0 +1,22 @@
+def call_ai(prompt: str, api_key: str, provider: str) -> str:
+    if provider == "claude":
+        import anthropic
+
+        client = anthropic.Anthropic(api_key=api_key)
+        with client.messages.stream(
+            model="claude-sonnet-4-6",
+            max_tokens=4096,
+            messages=[{"role": "user", "content": prompt}],
+        ) as stream:
+            return stream.get_final_message().content[0].text
+    elif provider == "openai":
+        import openai
+
+        client = openai.OpenAI(api_key=api_key)
+        response = client.chat.completions.create(
+            model="gpt-4o",
+            messages=[{"role": "user", "content": prompt}],
+        )
+        return response.choices[0].message.content
+    else:
+        raise ValueError(f"Unknown provider '{provider}'. Choose 'claude' or 'openai'.")

openapi_generator/cli.py

@@ -0,0 +1,31 @@
+import json
+import click
+from pathlib import Path
+from .generator import generate_spec
+
+
+@click.command()
+@click.argument("path", type=click.Path(exists=True, file_okay=False, path_type=Path))
+@click.option("--output", "-o", default="openapi.json", show_default=True, help="Output file path.")
+@click.option("--title", default="API", show_default=True, help="API title.")
+@click.option("--version", "api_version", default="1.0.0", show_default=True, help="API version.")
+@click.option(
+    "--api-key",
+    envvar="OPENAPI_GEN_API_KEY",
+    required=True,
+    help="AI provider API key (or set OPENAPI_GEN_API_KEY).",
+)
+@click.option(
+    "--provider",
+    type=click.Choice(["claude", "openai"]),
+    default="claude",
+    show_default=True,
+    help="AI provider to use.",
+)
+def main(path: Path, output: str, title: str, api_version: str, api_key: str, provider: str) -> None:
+    """Generate an OpenAPI specification for a REST API project directory."""
+    click.echo(f"Scanning {path} ...")
+    spec = generate_spec(path, title=title, api_version=api_version, api_key=api_key, provider=provider)
+    output_path = Path(output)
+    output_path.write_text(json.dumps(spec, indent=2), encoding="utf-8")
+    click.echo(f"Wrote JSON spec to {output_path}")

openapi_generator/file_collector.py

@@ -0,0 +1,37 @@
+import os
+from pathlib import Path
+
+_INCLUDE_EXTENSIONS = {".py", ".js", ".ts", ".go", ".java", ".rb", ".php", ".cs"}
+_SKIP_DIRS = {"node_modules", ".git", "__pycache__", "venv", ".venv", "dist", "build", ".eggs"}
+
+
+def collect_files(project_dir: Path, max_bytes: int = 400_000) -> list[tuple[str, str]]:
+    """Walk *project_dir* and return (relative_path, content) for each source file.
+
+    Stops collecting once the total accumulated size exceeds *max_bytes* to
+    avoid exceeding AI model context limits.
+    """
+    results: list[tuple[str, str]] = []
+    total = 0
+
+    for root, dirs, files in os.walk(project_dir):
+        dirs[:] = [d for d in dirs if d not in _SKIP_DIRS]
+
+        for filename in sorted(files):
+            if Path(filename).suffix not in _INCLUDE_EXTENSIONS:
+                continue
+
+            abs_path = Path(root) / filename
+            try:
+                content = abs_path.read_text(encoding="utf-8")
+            except (UnicodeDecodeError, OSError):
+                continue
+
+            total += len(content.encode("utf-8"))
+            if total > max_bytes:
+                return results
+
+            rel_path = abs_path.relative_to(project_dir).as_posix()
+            results.append((rel_path, content))
+
+    return results

openapi_generator/generator.py

@@ -0,0 +1,49 @@
+import json
+import re
+from pathlib import Path
+
+from .file_collector import collect_files
+from .ai_client import call_ai
+
+_SYSTEM_PROMPT = (
+    "You are an expert OpenAPI specification writer. "
+    "Given source files from a REST API project, produce a complete, valid "
+    "OpenAPI 3.0.3 specification in JSON. "
+    "Respond with ONLY the raw JSON — no markdown fences, no explanation, no commentary."
+)
+
+
+def _build_prompt(files: list[tuple[str, str]], title: str, api_version: str) -> str:
+    file_blocks = "\n\n".join(
+        f"### {path}\n{content}" for path, content in files
+    )
+    return (
+        f"{_SYSTEM_PROMPT}\n\n"
+        f"API title: {title}\n"
+        f"API version: {api_version}\n\n"
+        f"Source files:\n\n{file_blocks}"
+    )
+
+
+def _parse_json(raw: str) -> dict:
+    text = raw.strip()
+    text = re.sub(r"^```(?:json)?\s*", "", text)
+    text = re.sub(r"\s*```$", "", text)
+    try:
+        return json.loads(text)
+    except json.JSONDecodeError as exc:
+        raise ValueError(f"AI response was not valid JSON:\n{raw[:500]}") from exc
+
+
+def generate_spec(
+    project_dir: Path,
+    *,
+    title: str = "API",
+    api_version: str = "1.0.0",
+    api_key: str,
+    provider: str = "claude",
+) -> dict:
+    files = collect_files(project_dir)
+    prompt = _build_prompt(files, title, api_version)
+    raw = call_ai(prompt, api_key, provider)
+    return _parse_json(raw)

openapi_spec_generator-0.1.0.dist-info/METADATA

@@ -0,0 +1,98 @@
+Metadata-Version: 2.4
+Name: openapi-spec-generator
+Version: 0.1.0
+Summary: Generate OpenAPI specifications for any REST API project directory.
+License: MIT
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: click>=8.0
+Provides-Extra: claude
+Requires-Dist: anthropic>=0.25; extra == "claude"
+Provides-Extra: openai
+Requires-Dist: openai>=1.0; extra == "openai"
+Provides-Extra: all
+Requires-Dist: anthropic>=0.25; extra == "all"
+Requires-Dist: openai>=1.0; extra == "all"
+Provides-Extra: dev
+Requires-Dist: pytest>=8.0; extra == "dev"
+Requires-Dist: pytest-cov>=5.0; extra == "dev"
+
+# OpenAPI Generator
+
+A pip-installable CLI tool that uses AI to generate OpenAPI 3.0.3 specifications for any REST API project directory.
+
+## Overview
+
+Point this tool at any REST API project and it will scan the source files, send them to an AI model (Claude or OpenAI), and produce a valid OpenAPI 3.0.3 JSON specification.
+
+## Installation
+
+```bash
+# With Claude support (default)
+pip install "openapi-generator[claude]"
+
+# With OpenAI support
+pip install "openapi-generator[openai]"
+
+# Both providers
+pip install "openapi-generator[all]"
+```
+
+## Usage
+
+```bash
+openapi-gen <path-to-project> --api-key <your-api-key>
+```
+
+The API key can also be provided via the `OPENAPI_GEN_API_KEY` environment variable:
+
+```bash
+export OPENAPI_GEN_API_KEY=sk-ant-...
+openapi-gen <path-to-project>
+```
+
+### Options
+
+| Option | Default | Description |
+|---|---|---|
+| `<path>` | _(required)_ | Path to the REST API project directory |
+| `--api-key` | `OPENAPI_GEN_API_KEY` env var | AI provider API key |
+| `--provider` | `claude` | AI provider: `claude` or `openai` |
+| `--output`, `-o` | `openapi.json` | Output file path |
+| `--title` | `API` | API title in the spec |
+| `--version` | `1.0.0` | API version in the spec |
+
+### Examples
+
+```bash
+# Generate a spec using Claude (default)
+export OPENAPI_GEN_API_KEY=sk-ant-...
+openapi-gen ./my-api
+
+# Generate a spec using OpenAI
+openapi-gen ./my-api --provider openai --api-key sk-...
+
+# Specify title, version, and output path
+openapi-gen ./my-api --title "My REST API" --version "2.0.0" --output ./docs/openapi.json
+```
+
+## Output
+
+The generated file follows the [OpenAPI 3.0.3 specification](https://spec.openapis.org/oas/v3.0.3) and can be used directly with tools like Swagger UI, Postman, or any OpenAPI-compatible client generator.
+
+## Supported Languages
+
+The tool scans files with the following extensions: `.py`, `.js`, `.ts`, `.go`, `.java`, `.rb`, `.php`, `.cs`
+
+## Development
+
+```bash
+git clone https://github.com/your-org/openapi-generator
+cd openapi-generator
+pip install -e ".[dev]"
+pytest
+```
+
+## License
+
+MIT

openapi_spec_generator-0.1.0.dist-info/RECORD

@@ -0,0 +1,10 @@
+openapi_generator/__init__.py,sha256=kUR5RAFc7HCeiqdlX36dZOHkUI5wI6V_43RpEcD8b-0,22
+openapi_generator/ai_client.py,sha256=LaQx54jXZRDJauXQ8vFrt8e8FYemcnTXAfU6XIwTO0g,820
+openapi_generator/cli.py,sha256=TuNjNwLauYYKm0QPA1kY1KAENvRR81NCbezPECsOULs,1257
+openapi_generator/file_collector.py,sha256=k74-ZtjyrB7Ojz-nYyVKMYCBJCuIm-n_fqBvGkXp3Ig,1268
+openapi_generator/generator.py,sha256=myz-MHVHXzka7jnrKb6PFmTeroFCJznrb-G2Wb8gteM,1408
+openapi_spec_generator-0.1.0.dist-info/METADATA,sha256=LafjnZWoBsW6AlG7z-jOj1aI2Q763Gb6ptLPx_kxmcE,2773
+openapi_spec_generator-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+openapi_spec_generator-0.1.0.dist-info/entry_points.txt,sha256=hFqC1pfJAr_WhteyW7DzVwhWdMfvo8g3Qb7p-u5udCE,59
+openapi_spec_generator-0.1.0.dist-info/top_level.txt,sha256=YODEqCx34asgokWpD2p-Sm7tbt4Fw0Vk8W3eKfcvPAw,18
+openapi_spec_generator-0.1.0.dist-info/RECORD,,

openapi_spec_generator-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

openapi_spec_generator-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+openapi-gen = openapi_generator.cli:main

openapi_spec_generator-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+openapi_generator