tinymind-native-colab-handoff/bundle/model/tinymind-orchestrator/api.py

from __future__ import annotations

from typing import Any

from fastapi import FastAPI
from pydantic import BaseModel

from orchestrator import TinyMindOrchestrator


class GenerateRequest(BaseModel):
    query: str
    top_k: int = 5


app = FastAPI(title="TinyMind Orchestrator", version="1.0.0")
orch = TinyMindOrchestrator()


@app.get("/health")
def health() -> dict[str, Any]:
    return {
        "ok": True,
        "apex_retrieval": True,
        "neural_core": orch.apex.neural is not None,
        "tiny_12b_adapter_available": orch.has_12b(),
        "ollama_tinymind_apex_local_available": orch.has_ollama_model(),
    }


@app.post("/generate")
def generate(req: GenerateRequest) -> dict[str, Any]:
    return orch.generate(req.query, req.top_k)