estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: Invalid expected metadata value type error (#22793)

Browse Source
This commit is contained in:
Will
2025-07-22 22:28:16 +08:00
committed by GitHub
parent 366d42d9c0
commit 095bae01b2
3 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api/core/tools/utils/dataset_retriever/dataset_retriever_base_tool.py
View File

@@ -1,5 +1,5 @@
from abc import abstractmethod from abc import abstractmethod
from typing import Any, Optional from typing import Optional
from msal_extensions.persistence import ABC # type: ignore from msal_extensions.persistence import ABC # type: ignore
from pydantic import BaseModel, ConfigDict from pydantic import BaseModel, ConfigDict
@@ -21,11 +21,7 @@ class DatasetRetrieverBaseTool(BaseModel, ABC):
model_config = ConfigDict(arbitrary_types_allowed=True) model_config = ConfigDict(arbitrary_types_allowed=True)
@abstractmethod @abstractmethod
def _run( def _run(self, query: str) -> str:
self,
*args: Any,
**kwargs: Any,
) -> Any:
"""Use the tool. """Use the tool.
Add run_manager: Optional[CallbackManagerForToolRun] = None Add run_manager: Optional[CallbackManagerForToolRun] = None

2
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
View File

@@ -462,7 +462,7 @@ class KnowledgeRetrievalNode(BaseNode):
expected_value = self.graph_runtime_state.variable_pool.convert_template( expected_value = self.graph_runtime_state.variable_pool.convert_template(
expected_value expected_value
).value[0] ).value[0]
if expected_value.value_type == "number": # type: ignore if expected_value.value_type in {"number", "integer", "float"}: # type: ignore
expected_value = expected_value.value # type: ignore expected_value = expected_value.value # type: ignore
elif expected_value.value_type == "string": # type: ignore elif expected_value.value_type == "string": # type: ignore
expected_value = re.sub(r"[\r\n\t]+", " ", expected_value.text).strip() # type: ignore expected_value = re.sub(r"[\r\n\t]+", " ", expected_value.text).strip() # type: ignore

2
api/core/workflow/nodes/llm/node.py
View File

@@ -565,7 +565,7 @@ class LLMNode(BaseNode):
retriever_resources=original_retriever_resource, context=context_str.strip() retriever_resources=original_retriever_resource, context=context_str.strip()
) )
def _convert_to_original_retriever_resource(self, context_dict: dict): def _convert_to_original_retriever_resource(self, context_dict: dict) -> RetrievalSourceMetadata | None:
if ( if (
"metadata" in context_dict "metadata" in context_dict
and "_source" in context_dict["metadata"] and "_source" in context_dict["metadata"]