feat: graph search api support metadata filter

pingcap · Jan 7, 2025 · 555aaa3 · 555aaa3
1 parent c1efd7f
commit 555aaa3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/backend/app/api/admin_routes/knowledge_base/graph/models.py b/backend/app/api/admin_routes/knowledge_base/graph/models.py
@@ -33,6 +33,7 @@ class GraphSearchRequest(BaseModel):
     include_meta: bool = True
     depth: int = 2
     with_degree: bool = True
+    relationship_meta_filters: dict = {}
 
 
 class KnowledgeRequest(BaseModel):

diff --git a/backend/app/api/admin_routes/knowledge_base/graph/routes.py b/backend/app/api/admin_routes/knowledge_base/graph/routes.py
@@ -207,7 +207,7 @@ def search_graph(session: SessionDep, kb_id: int, request: GraphSearchRequest):
             request.include_meta,
             request.with_degree,
             False,
-            {},
+            request.relationship_meta_filters,
         )
         return {
             "entities": entities,