From 09f8cb7e6d72660c2cabb380e857ac6d35272224 Mon Sep 17 00:00:00 2001 From: Woody Date: Thu, 23 Apr 2026 13:26:20 +0800 Subject: [PATCH] refactor(backend): update Pydantic models for ingestion and query Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus --- backend/app/models/common.py | 8 ++++++++ backend/app/models/ingest.py | 20 -------------------- backend/app/models/query.py | 15 +++++++++++++++ 3 files changed, 23 insertions(+), 20 deletions(-) create mode 100644 backend/app/models/common.py create mode 100644 backend/app/models/query.py diff --git a/backend/app/models/common.py b/backend/app/models/common.py new file mode 100644 index 0000000..f5500c7 --- /dev/null +++ b/backend/app/models/common.py @@ -0,0 +1,8 @@ +from pydantic import BaseModel + + +class SourceMetadata(BaseModel): + filename: str + upload_date: str + content_summary: str + chunk_index: int diff --git a/backend/app/models/ingest.py b/backend/app/models/ingest.py index 1deda9d..3b54531 100644 --- a/backend/app/models/ingest.py +++ b/backend/app/models/ingest.py @@ -1,27 +1,7 @@ -from datetime import datetime -from typing import List - from pydantic import BaseModel -class SourceMetadata(BaseModel): - filename: str - upload_date: str - content_summary: str - chunk_index: int - - class IngestResponse(BaseModel): document_id: str chunk_count: int filename: str - - -class QueryRequest(BaseModel): - question: str - - -class QueryResponse(BaseModel): - keywords: List[str] - answer: str - sources: List[SourceMetadata] diff --git a/backend/app/models/query.py b/backend/app/models/query.py new file mode 100644 index 0000000..f7ec6ab --- /dev/null +++ b/backend/app/models/query.py @@ -0,0 +1,15 @@ +from typing import List + +from pydantic import BaseModel + +from app.models.common import SourceMetadata + + +class QueryRequest(BaseModel): + question: str + + +class QueryResponse(BaseModel): + keywords: List[str] + answer: str + sources: List[SourceMetadata]