From 059d635f02916bfec4c4bb62d7b3f4000569055b Mon Sep 17 00:00:00 2001
From: Ho Trong Hien <115549171+hienhayho@users.noreply.github.com>
Date: Tue, 17 Dec 2024 22:28:41 +0700
Subject: [PATCH] fix: fix pydantic validation error

- When passing result_as_answer=True, it will return ToolOutput so it won't pass pydantic validation as a string

- Get content of ToolOutput before return
---
 src/crewai_tools/tools/llamaindex_tool/llamaindex_tool.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/crewai_tools/tools/llamaindex_tool/llamaindex_tool.py b/src/crewai_tools/tools/llamaindex_tool/llamaindex_tool.py
index 61a747956..ba2605816 100644
--- a/src/crewai_tools/tools/llamaindex_tool/llamaindex_tool.py
+++ b/src/crewai_tools/tools/llamaindex_tool/llamaindex_tool.py
@@ -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