38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
|
# 简化版本的搜索端点，只支持GET方法
|
|||
|
|
@router.get("/{kb_id}/search")
|
|||
|
|
async def search_knowledge_base(
|
|||
|
|
kb_id: int,
|
|||
|
|
query: str,
|
|||
|
|
limit: int = 5,
|
|||
|
|
db: Session = Depends(get_db),
|
|||
|
|
current_user: User = Depends(AuthService.get_current_user)
|
|||
|
|
):
|
|||
|
|
"""Search documents in a knowledge base."""
|
|||
|
|
try:
|
|||
|
|
# Verify knowledge base exists and user has access
|
|||
|
|
kb_service = KnowledgeBaseService(db)
|
|||
|
|
kb = kb_service.get_knowledge_base(kb_id)
|
|||
|
|
if not kb:
|
|||
|
|
raise HTTPException(
|
|||
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|||
|
|
detail="Knowledge base not found"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# Perform search
|
|||
|
|
doc_service = DocumentService(db)
|
|||
|
|
results = doc_service.search_documents(kb_id, query, limit)
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
"knowledge_base_id": kb_id,
|
|||
|
|
"query": query,
|
|||
|
|
"results": results,
|
|||
|
|
"total_results": len(results)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
except HTTPException:
|
|||
|
|
raise
|
|||
|
|
except Exception as e:
|
|||
|
|
raise HTTPException(
|
|||
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|||
|
|
detail=f"Failed to search knowledge base: {str(e)}"
|
|||
|
|
)
|