feat: 添加es_search接口 (#6)

Author: longyue0521

app/service/elasticsearch.py

@@ -374,6 +374,12 @@ def search(self, parameters: SearchParameters) -> SearchResult:
             response, search_time_ms, parameters.limit, search_conditions
         )
 
+    def es_search(
+        self, index_name: str, query: dict[str, Any]
+    ) -> dict[str, Any]:
+        resp = self._client.search(index=index_name, body=query)
+        return dict(resp.body)
+
     @staticmethod
     def _classify_conditions(
         conditions: list[SearchCondition],

app/web/document.py

@@ -17,6 +17,7 @@
 import shutil
 import uuid
 from pathlib import Path
+from typing import Any
 from urllib.parse import urlparse
 
 from elasticsearch import NotFoundError
@@ -38,6 +39,7 @@
 from app.service.elasticsearch import ElasticsearchService
 from app.utils.converters import SearchConverter
 from app.web.vo import (
+    ESSearchRequest,
     FileUploadResponse,
     SaveRequest,
     SaveResponse,
@@ -116,6 +118,11 @@ def register_routes(self) -> None:
             summary="在知识库中进行搜索",
         )(self.search)
 
+        self._router.post(
+            "/es_search",
+            summary="使用传递过来的es查询语句在知识库中直接搜索",
+        )(self.es_search)
+
         self._router.post(
             "/documents/save",
             response_model=SaveResponse,
@@ -438,6 +445,21 @@ async def search(self, request: SearchRequest) -> SearchResponse:
             logger.error(f"❌ 搜索失败: {e}", exc_info=True)
             raise HTTPException(status_code=500, detail="搜索处理失败") from e
 
+    async def es_search(self, request: ESSearchRequest) -> dict[str, Any]:
+        """通过ES语法搜索文档接口"""
+        try:
+            logger.info(
+                f"🔍 收到ES搜索请求: index='{request.index}', query='{request.query}'"
+            )
+            return self._service.es_search(request.index, request.query)
+        except NotFoundError as e:
+            raise HTTPException(
+                status_code=404, detail=f"索引 {request.index} 不存在"
+            ) from e
+        except Exception as e:
+            logger.error(f"❌ 搜索失败: {e}", exc_info=True)
+            raise HTTPException(status_code=500, detail="搜索处理失败") from e
+
     async def save(self, request: SaveRequest) -> SaveResponse:
         """保存JSON格式文档到指定的Elasticsearch索引"""
         try:

app/web/vo.py

@@ -130,6 +130,11 @@ def validate_query(cls, v: Query, info: ValidationInfo) -> Query:
         return v
 
 
+class ESSearchRequest(BaseModel):
+    index: str = Field(..., min_length=1, description="ES索引名称")
+    query: dict[str, Any] = Field(..., description="符合ES语法规范的查询语句")
+
+
 class VectorHybridSearchResult(BaseModel):
     """向量+全文混合搜索结果"""