From 3ad000bf81f2941149ad7049058e07f99cee5258 Mon Sep 17 00:00:00 2001 From: chiehw Date: Thu, 5 Jun 2025 16:11:44 +0800 Subject: [PATCH] opt: use RetrievalSourceMetadata instead of dict --- api/core/tools/entities/tool_entities.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/api/core/tools/entities/tool_entities.py b/api/core/tools/entities/tool_entities.py index ddb84b0d89..b720b88758 100644 --- a/api/core/tools/entities/tool_entities.py +++ b/api/core/tools/entities/tool_entities.py @@ -15,6 +15,7 @@ from core.plugin.entities.parameters import ( cast_parameter_value, init_frontend_parameter, ) +from core.rag.entities.citation_metadata import RetrievalSourceMetadata from core.tools.entities.common_entities import I18nObject from core.tools.entities.constants import TOOL_SELECTOR_MODEL_IDENTITY @@ -177,7 +178,7 @@ class ToolInvokeMessage(BaseModel): metadata: Optional[Mapping[str, Any]] = Field(default=None, description="The metadata of the log") class RetrieverResourceMessage(BaseModel): - retriever_resources: list[dict] = Field(..., description="retriever resources") + retriever_resources: list[RetrievalSourceMetadata] = Field(..., description="retriever resources") context: str = Field(..., description="context") class MessageType(Enum):