Initial commit

api.py

@@ -0,0 +1,77 @@
+from __future__ import annotations
+
+import os
+from typing import Any, Dict, Optional
+
+from fastapi import FastAPI
+from pydantic import BaseModel, Field
+
+from tg_resume_db.db import connect, init_db
+from tg_resume_db.agent import agent_search
+from tg_resume_db.search import search as db_search
+
+DB_PATH = os.environ.get("CANDIDATES_DB", "./candidates.db")
+
+app = FastAPI(title="Resume Search API", version="1.0")
+
+class SearchRequest(BaseModel):
+    query: str = Field(default="")
+    limit: int = Field(default=20, ge=1, le=100)
+    offset: int = Field(default=0, ge=0)
+    remote: Optional[bool] = None
+    location: Optional[str] = None
+    experience_min: Optional[float] = None
+    salary_min: Optional[int] = None
+    salary_max: Optional[int] = None
+    english: Optional[str] = None
+    role: Optional[str] = None
+    skill: Optional[str] = None
+
+
+class AISearchRequest(BaseModel):
+    prompt: str = Field(default="")
+    limit: int = Field(default=20, ge=1, le=100)
+    ai_iters: int = Field(default=2, ge=0, le=5)
+
+
+@app.on_event("startup")
+def _startup():
+    con = connect(DB_PATH)
+    init_db(con)
+    con.close()
+
+@app.get("/health")
+def health():
+    return {"ok": True}
+
+@app.post("/search")
+def search(req: SearchRequest) -> Dict[str, Any]:
+    con = connect(DB_PATH)
+    try:
+        items = db_search(con, query=req.query, filters=req.model_dump(), limit=req.limit, offset=req.offset)
+        return {"items": items, "count": len(items)}
+    finally:
+        con.close()
+
+
+@app.post("/search/ai")
+def search_ai(req: AISearchRequest) -> Dict[str, Any]:
+    con = connect(DB_PATH)
+    try:
+        res = agent_search(
+            con,
+            user_prompt=req.prompt,
+            max_iters=req.ai_iters,
+            limit=req.limit,
+        )
+        return {
+            "ai": True,
+            "llm_used": res.get("llm_used", False),
+            "plan": res.get("plan"),
+            "history": res.get("history"),
+            "postfilter": res.get("postfilter"),
+            "items": res.get("items", []),
+            "count": int(res.get("count", 0)),
+        }
+    finally:
+        con.close()