capstone_api.py

"""FastAPI wrapper for Internship Copilot.

Run with:
uvicorn capstone_api:app --reload --port 8000
"""

from __future__ import annotations

from typing import List, Optional

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel, Field

from agent import CHUNKS, ask, get_collection, kb_topics


app = FastAPI(
    title="Internship Copilot API",
    description="LangGraph + RAG + memory + live web search internship assistant",
    version="1.0.0",
)

# Allow Streamlit frontend to call this API
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_methods=["*"],
    allow_headers=["*"],
)


class AskRequest(BaseModel):
    question: str = Field(..., min_length=1, description="User question")
    thread_id: str = Field("default", description="Conversation thread id")
    provider: Optional[str] = Field(
        None, description="openai, groq, openrouter, or auto"
    )
    model: Optional[str] = Field(None, description="Provider-specific model override")


class AskResponse(BaseModel):
    answer: str
    route: str
    sources: List[str]
    source_urls: List[str]
    tool_type: str
    faithfulness: float
    thread_id: str


@app.get("/health")
def health() -> dict:
    collection = get_collection()
    return {
        "status": "ok",
        "docs": len(kb_topics()),
        "chunks": len(CHUNKS),
        "vector_rows": collection.count(),
    }


@app.get("/topics")
def topics() -> dict:
    return {"topics": kb_topics()}


@app.post("/ask", response_model=AskResponse)
def ask_question(request: AskRequest) -> AskResponse:
    result = ask(
        request.question,
        thread_id=request.thread_id,
        provider=request.provider,
        model=request.model,
    )
    return AskResponse(
        answer=result.get("answer", ""),
        route=result.get("route", "retrieve"),
        sources=result.get("sources", []),
        source_urls=result.get("source_urls", []),
        tool_type=result.get("tool_type", ""),
        faithfulness=float(result.get("faithfulness", 0.0) or 0.0),
        thread_id=request.thread_id,
    )