Files
conai/backend/app/schemas/rag.py
sinmb79 2a4950d8a0 feat: CONAI Phase 1 MVP 초기 구현
소형 건설업체(100억 미만)를 위한 AI 기반 토목공사 통합관리 플랫폼

Backend (FastAPI):
- SQLAlchemy 모델 13개 (users, projects, wbs, tasks, daily_reports, reports, inspections, quality, weather, permits, rag, settings)
- API 라우터 11개 (auth, projects, tasks, daily_reports, reports, inspections, weather, rag, kakao, permits, settings)
- Services: Claude AI 래퍼, CPM Gantt 계산, 기상청 API, RAG(pgvector), 카카오 Skill API
- Alembic 마이그레이션 (pgvector 포함)
- pytest 테스트 (CPM, 날씨 경보)

Frontend (Next.js 15):
- 11개 페이지 (대시보드, 프로젝트, Gantt, 일보, 검측, 품질, 날씨, 인허가, RAG, 설정)
- TanStack Query + Zustand + Tailwind CSS

인프라:
- Docker Compose (PostgreSQL pgvector + backend + frontend)
- 한국어 README 및 설치 가이드

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-03-24 20:06:36 +09:00

45 lines
1.0 KiB
Python

import uuid
from datetime import datetime
from pydantic import BaseModel
from app.models.rag import RagSourceType
class RagAskRequest(BaseModel):
question: str
source_types: list[RagSourceType] | None = None # Filter by source type
top_k: int = 5
class RagSource(BaseModel):
id: uuid.UUID
title: str
source_type: RagSourceType
chunk_content: str
relevance_score: float
model_config = {"from_attributes": True}
class RagAskResponse(BaseModel):
question: str
answer: str
sources: list[RagSource]
disclaimer: str = "이 답변은 참고용이며 법률 자문이 아닙니다. 중요 사항은 전문가에게 확인하세요."
class RagSourceCreate(BaseModel):
title: str
source_type: RagSourceType
source_url: str | None = None
class RagSourceResponse(BaseModel):
id: uuid.UUID
title: str
source_type: RagSourceType
source_url: str | None
chunk_count: int = 0
created_at: datetime
model_config = {"from_attributes": True}