mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-08 12:30:14 +00:00
feat(core): Support higher-order operators (#1984)
Co-authored-by: 谨欣 <echo.cmy@antgroup.com>
This commit is contained in:
@@ -64,6 +64,7 @@ class KnowledgeSpaceRetrieverResource(RetrieverResource):
|
||||
"""Knowledge Space retriever resource."""
|
||||
|
||||
def __init__(self, name: str, space_name: str, context: Optional[dict] = None):
|
||||
# TODO: Build the retriever in a thread pool, it will block the event loop
|
||||
retriever = KnowledgeSpaceRetriever(
|
||||
space_id=space_name,
|
||||
top_k=context.get("top_k", None) if context else 4,
|
||||
|
Reference in New Issue
Block a user