Upgrade the declarative pipeline for cleaner interface (#51)

2023-10-24 11:12:22 +07:00
parent aab982ddc4
commit 9035e25666
26 changed files with 365 additions and 169 deletions
--- a/knowledgehub/pipelines/tools/llm.py
+++ b/knowledgehub/pipelines/tools/llm.py
@@ -32,5 +32,5 @@ class LLMTool(BaseTool):
            response = self.llm(query)
        except ValueError:
            raise ToolException("LLM Tool call failed")
-        output = response.text[0]
+        output = response.text
        return output