cbh/api/coach/views.py

"""
Coach views module.
"""

from cbh.api.coach import coach_router
from cbh.api.coach.schemas import InitializationResponse, QueryVectorDbRequest
from cbh.api.coach.services import generate_ephemeral_token, query_vector_db_tool
from cbh.core.wrappers import CbhResponseWrapper


@coach_router.get("/initialize")
async def generate_agent_config() -> CbhResponseWrapper[InitializationResponse]:
    """
    Generate agent config.
    """
    response = await generate_ephemeral_token()
    return CbhResponseWrapper(data=response)


@coach_router.post("/vector/db")
async def query_vector_db(
    request: QueryVectorDbRequest,
) -> CbhResponseWrapper[str]:
    """
    Query vector db.
    """
    result = await query_vector_db_tool(request.input)
    return CbhResponseWrapper(data=result)