【ADD]Knowledge base retrieval supports file set retrieval

2026-03-04 15:28:17 +08:00
parent 4f0b653a82
commit d4c4160215
2 changed files with 5 additions and 4 deletions
--- a/api/app/schemas/chunk_schema.py
+++ b/api/app/schemas/chunk_schema.py
@@ -46,6 +46,7 @@ class ChunkUpdate(BaseModel):
 class ChunkRetrieve(BaseModel):
    query: str
    kb_ids: list[uuid.UUID]
+    file_names_filter: list[str] | None = Field(None)
    similarity_threshold: float | None = Field(None)
    vector_similarity_weight: float | None = Field(None)
    top_k: int | None = Field(None)