78 lines
2.1 KiB
Python
78 lines
2.1 KiB
Python
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()
|