b3012b8320
- ai-service: Ollama 기반 AI 서비스 (분류, 시맨틱 검색, RAG Q&A, 패턴 분석) - AI 어시스턴트 페이지: 채팅형 Q&A, 시맨틱 검색, 패턴 분석, 분류 테스트 - 권한 시스템에 ai_assistant 페이지 등록 (기본 비활성) - 기존 페이지에 AI 기능 통합 (대시보드, 수신함, 관리함) - docker-compose, gateway, nginx 설정 업데이트 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
78 lines
2.1 KiB
Python
78 lines
2.1 KiB
Python
from fastapi import APIRouter, BackgroundTasks, Query
|
|
from pydantic import BaseModel
|
|
from services.embedding_service import (
|
|
sync_all_issues,
|
|
sync_single_issue,
|
|
sync_incremental,
|
|
search_similar_by_id,
|
|
search_similar_by_text,
|
|
)
|
|
from db.vector_store import vector_store
|
|
|
|
router = APIRouter(tags=["embeddings"])
|
|
|
|
|
|
class SyncSingleRequest(BaseModel):
|
|
issue_id: int
|
|
|
|
|
|
class SearchRequest(BaseModel):
|
|
query: str
|
|
n_results: int = 5
|
|
project_id: int | None = None
|
|
category: str | None = None
|
|
|
|
|
|
@router.post("/embeddings/sync")
|
|
async def sync_embeddings(background_tasks: BackgroundTasks):
|
|
background_tasks.add_task(sync_all_issues)
|
|
return {"status": "sync_started", "message": "전체 임베딩 동기화가 시작되었습니다"}
|
|
|
|
|
|
@router.post("/embeddings/sync-full")
|
|
async def sync_embeddings_full():
|
|
result = await sync_all_issues()
|
|
return {"status": "completed", **result}
|
|
|
|
|
|
@router.post("/embeddings/sync-single")
|
|
async def sync_single(req: SyncSingleRequest):
|
|
result = await sync_single_issue(req.issue_id)
|
|
return result
|
|
|
|
|
|
@router.post("/embeddings/sync-incremental")
|
|
async def sync_incr():
|
|
result = await sync_incremental()
|
|
return result
|
|
|
|
|
|
@router.get("/similar/{issue_id}")
|
|
async def get_similar(issue_id: int, n_results: int = Query(default=5, le=20)):
|
|
try:
|
|
results = await search_similar_by_id(issue_id, n_results)
|
|
return {"available": True, "results": results, "query_issue_id": issue_id}
|
|
except Exception as e:
|
|
return {"available": False, "results": [], "error": str(e)}
|
|
|
|
|
|
@router.post("/similar/search")
|
|
async def search_similar(req: SearchRequest):
|
|
filters = {}
|
|
if req.project_id is not None:
|
|
filters["project_id"] = str(req.project_id)
|
|
if req.category:
|
|
filters["category"] = req.category
|
|
try:
|
|
results = await search_similar_by_text(
|
|
req.query, req.n_results, filters or None
|
|
)
|
|
return {"available": True, "results": results}
|
|
except Exception as e:
|
|
return {"available": False, "results": [], "error": str(e)}
|
|
|
|
|
|
@router.get("/embeddings/stats")
|
|
async def embedding_stats():
|
|
return vector_store.stats()
|