aimd-api 0.9.2__py3-none-any.whl

aimd_api/__init__.py

@@ -0,0 +1,22 @@
+"""HTTP API package for aimd."""
+
+from importlib import import_module
+
+__all__ = [
+    "EngineCapabilityResponse",
+    "EnginesResponse",
+    "HealthResponse",
+    "ProcessRequest",
+    "ProcessResponse",
+    "app",
+    "create_app",
+    "main",
+]
+
+
+def __getattr__(name: str):
+    if name in __all__:
+        app_module = import_module(".app", __name__)
+
+        return getattr(app_module, name)
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

aimd_api/app.py

@@ -0,0 +1,158 @@
+"""FastAPI service package for aimd."""
+
+from importlib.metadata import version
+import os
+
+from fastapi import FastAPI, HTTPException
+from logly import logger
+from pydantic import BaseModel, Field
+
+from aimd.application.bootstrap import build_container
+from aimd.application.models import TaskType
+from aimd.application.services.interface_payloads import (
+    build_process_input,
+    engine_capabilities_payload,
+    get_request_temp_dir,
+    process_result_payload,
+)
+from aimd.application.services.output_writer import persist_result_output_if_requested
+from aimd.errors import AimdError
+
+
+class HealthResponse(BaseModel):
+    status: str = "ok"
+
+
+class EngineCapabilityResponse(BaseModel):
+    name: str
+    available: bool
+    reason: str | None = None
+    fix_hint: str | None = None
+    selected_by_auto: bool = False
+
+
+class EnginesResponse(BaseModel):
+    auto_selected_engine: str | None = None
+    engines: list[EngineCapabilityResponse]
+
+
+class ProcessRequest(BaseModel):
+    input_source: str = Field(
+        ..., description="Audio/video file path, video URL, or document file path."
+    )
+    output_file: str | None = Field(
+        default=None, description="Optional path to write resulting markdown output."
+    )
+    transcribe_engine: str = Field(
+        default="auto",
+        description="Transcription engine: auto, mlx, qwen.",
+    )
+    model: str | None = Field(
+        default=None,
+        description="Model for transcription. mlx defaults to mlx-community/Qwen3-ASR-1.7B-4bit "
+        "and also supports other documented mlx-audio STT model IDs. "
+        "qwen supports Qwen/Qwen3-ASR-1.7B (default) or Qwen/Qwen3-ASR-0.6B.",
+    )
+    language: str | None = Field(
+        default=None, description="Language code for transcription, e.g. zh, en, ja."
+    )
+    save_original: str | None = Field(
+        default=None,
+        description="Optional path to persist downloaded audio from URL processing.",
+    )
+    cookies: str | None = Field(
+        default=None,
+        description="Path to Netscape-format cookies file for URL extraction.",
+    )
+    cookies_from_browser: str | None = Field(
+        default=None,
+        description="Browser cookie source for URL extraction, e.g. chrome:default.",
+    )
+    raw_transcript: bool = Field(
+        default=False,
+        description="Preserve original subtitle formatting (SRT/VTT timestamps). "
+        "By default, subtitles are simplified to plain text.",
+    )
+
+
+class ProcessResponse(BaseModel):
+    task_type: TaskType
+    title: str
+    chunk_list: list[str]
+    split_header_level: int | None = None
+    platform: str | None = None
+    output_file: str | None = None
+    output_dir: str | None = None
+
+
+def create_app() -> FastAPI:
+    """Build FastAPI app instance."""
+    app = FastAPI(
+        title="aimd API",
+        description="Context preparation API for LLM workflows",
+        version=version("aimd-cli"),
+    )
+    container = build_container()
+
+    @app.get("/healthz", response_model=HealthResponse)
+    async def healthz() -> HealthResponse:
+        return HealthResponse()
+
+    @app.get("/v1/engines", response_model=EnginesResponse)
+    async def engines() -> EnginesResponse:
+        result = container.list_engines_use_case.execute()
+        return EnginesResponse(**engine_capabilities_payload(result))
+
+    @app.post("/v1/process", response_model=ProcessResponse)
+    async def process(request: ProcessRequest) -> ProcessResponse:
+        try:
+            temp_dir = get_request_temp_dir()
+
+            result = await container.process_input_use_case.execute(
+                build_process_input(
+                    input_source=request.input_source,
+                    transcribe_engine=request.transcribe_engine,
+                    model=request.model,
+                    language=request.language,
+                    save_original=request.save_original,
+                    cookies=request.cookies,
+                    cookies_from_browser=request.cookies_from_browser,
+                    temp_dir=temp_dir,
+                    raw_transcript=request.raw_transcript,
+                )
+            )
+
+            persisted = persist_result_output_if_requested(result, request.output_file)
+            if persisted.ignored_output_file:
+                logger.warning(
+                    "Ignoring output_file for book conversions; output is a directory."
+                )
+
+            return ProcessResponse(
+                **process_result_payload(
+                    result,
+                    output_file=persisted.output_file,
+                    output_dir=persisted.output_dir,
+                )
+            )
+        except AimdError as exc:
+            raise HTTPException(status_code=exc.status_code, detail=str(exc)) from exc
+        except ValueError as exc:
+            raise HTTPException(status_code=400, detail=str(exc)) from exc
+        except Exception as exc:
+            logger.error(f"API processing failed: {exc}")
+            raise HTTPException(status_code=500, detail=str(exc)) from exc
+
+    return app
+
+
+app = create_app()
+
+
+def main() -> None:
+    """Run FastAPI service via uvicorn."""
+    import uvicorn
+
+    host = os.getenv("AIMD_API_HOST", "127.0.0.1")
+    port = int(os.getenv("AIMD_API_PORT", "8000"))
+    uvicorn.run("aimd_api:app", host=host, port=port, reload=False)

aimd_api-0.9.2.dist-info/METADATA

@@ -0,0 +1,10 @@
+Metadata-Version: 2.3
+Name: aimd-api
+Version: 0.9.2
+Summary: FastAPI service for aimd.
+Author: Shu Li
+Author-email: Shu Li <zetarylee@gmail.com>
+Requires-Dist: aimd-cli
+Requires-Dist: fastapi>=0.116.1
+Requires-Dist: uvicorn>=0.35.0
+Requires-Python: >=3.10, <3.13

aimd_api-0.9.2.dist-info/RECORD

@@ -0,0 +1,6 @@
+aimd_api/__init__.py,sha256=YLH1Wgf_cKn7nOlQpiUbKptSnUXY2AZxxrKVcc3z41A,472
+aimd_api/app.py,sha256=u1KeAOgFWfy4ecxwiF0BBiad9clDniRbeIsxsvFnHBY,5338
+aimd_api-0.9.2.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+aimd_api-0.9.2.dist-info/entry_points.txt,sha256=sgB6NImUHyK7NwaqyFQmV_fUdxN05bQiroJC-HqwFJw,44
+aimd_api-0.9.2.dist-info/METADATA,sha256=J99mmz97dmM2MphEcEaCPAgBPDVUOyPeHRnIt-pc2ss,263
+aimd_api-0.9.2.dist-info/RECORD,,

aimd_api-0.9.2.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: uv 0.8.24
+Root-Is-Purelib: true
+Tag: py3-none-any

aimd_api-0.9.2.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+aimd-api = aimd_api:main
+