Merge pull request #1 from hienhayho/hienhayho-fix-llamaindex-result-as-answer

fix: fix pydantic validation error
This commit is contained in:
Ho Trong Hien
2024-12-17 22:29:19 +07:00
committed by GitHub

View File

@@ -18,6 +18,10 @@ class LlamaIndexTool(BaseTool):
from llama_index.core.tools import BaseTool as LlamaBaseTool
tool = cast(LlamaBaseTool, self.llama_index_tool)
if self.result_as_answer:
return tool(*args, **kwargs).content
return tool(*args, **kwargs)
@classmethod