James BruntonandGitHub
e10c5f6283
# Description of Changes Redesign the Python AI engine to be properly agentic and make use of `pydantic-ai` instead of `langchain` for correctness and ergonomics. This should be a good foundation for us to build our AI engine on going forwards.
20 lines
610 B
Python
20 lines
610 B
Python
from __future__ import annotations
|
|
|
|
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Depends
|
|
|
|
from stirling.agents import PdfQuestionAgent
|
|
from stirling.api.dependencies import get_pdf_question_agent
|
|
from stirling.contracts import PdfQuestionRequest, PdfQuestionResponse
|
|
|
|
router = APIRouter(prefix="/api/v1/pdf/questions", tags=["pdf-questions"])
|
|
|
|
|
|
@router.post("", response_model=PdfQuestionResponse)
|
|
async def pdf_questions(
|
|
request: PdfQuestionRequest,
|
|
agent: Annotated[PdfQuestionAgent, Depends(get_pdf_question_agent)],
|
|
) -> PdfQuestionResponse:
|
|
return await agent.handle(request)
|