TASK:lookbook上传,查询

This commit is contained in:
shahaibo
2024-10-24 15:59:36 +08:00
parent 93c37e268a
commit 07e72c1ee1
4 changed files with 111 additions and 3 deletions

View File

@@ -0,0 +1,26 @@
from fastapi import APIRouter, Query
from typing import Optional
from app.service.lookbooks.query_service import query_lookbooks_service # 引入业务逻辑
router = APIRouter()
@router.get("/query-lookbooks")
async def query_lookbooks(
tag: Optional[str] = Query(None, description="Tag to filter lookbooks"),
year: Optional[str] = Query(None, description="Year to filter lookbooks"),
n_results: int = Query(10, description="Number of results to return")
):
"""
查询向量数据库,支持按 tag 和 year 查询
:param tag: 查询过滤的标签
:param year: 查询过滤的年份
:param n_results: 返回的结果数量
:return: 查询结果
"""
try:
# 调用业务逻辑层的查询服务
result_list = await query_lookbooks_service(tag, year, n_results)
return {"status": "success", "data": result_list}
except Exception as e:
return {"status": "error", "message": str(e)}

View File

@@ -1,6 +1,7 @@
from fastapi import APIRouter
from app.api import api_test, api_outfit_matcher, api_attribute, api_similar_match, api_process_lookbooks
from app.api import api_test, api_outfit_matcher, api_attribute, api_similar_match, api_process_lookbooks, \
api_lookbooks_query
router = APIRouter()
@@ -9,3 +10,4 @@ router.include_router(api_outfit_matcher.router, tags=["outfit_matcher"], prefix
router.include_router(api_attribute.router, tags=["attribute"], prefix="/api")
router.include_router(api_similar_match.router, tags=["similar_match"], prefix="/api")
router.include_router(api_process_lookbooks.router, tags=["process-lookbooks"], prefix="/api")
router.include_router(api_lookbooks_query.router, tags=["query-lookbooks"], prefix="/api")