fix: retriever_resource missing (#1317)

2023-10-12 03:37:11 +08:00
parent cbf095465c
commit 9cba1c8bf4
3 changed files with 5 additions and 3 deletions
--- a/api/core/tool/dataset_retriever_tool.py
+++ b/api/core/tool/dataset_retriever_tool.py
@@ -30,7 +30,7 @@ class DatasetRetrieverTool(BaseTool):
    dataset_id: str
    k: int = 3
    conversation_message_task: ConversationMessageTask
-    return_resource: str
+    return_resource: bool
    retriever_from: str

    @classmethod