fixup! feat(tools): add llm and calculator tools

Signed-off-by: Lukáš Janeček <[email protected]>
i-am-bee · Dec 13, 2024 · 910c5a6 · 910c5a6
1 parent 0cbb53a
commit 910c5a6
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 24 deletions.
diff --git a/package.json b/package.json
@@ -48,7 +48,7 @@
     "@zilliz/milvus2-sdk-node": "^2.4.4",
     "ajv": "^8.17.1",
     "axios": "^1.7.7",
-    "bee-agent-framework": "0.0.52",
+    "bee-agent-framework": "0.0.53",
     "bullmq": "^5.32.0",
     "bullmq-otel": "^1.0.1",
     "cache-manager": "^5.7.6",

diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
diff --git a/src/runs/execution/tools/helpers.ts b/src/runs/execution/tools/helpers.ts
@@ -35,7 +35,6 @@ import { SearchToolOptions, SearchToolOutput } from 'bee-agent-framework/tools/s
 import { PromptTemplate } from 'bee-agent-framework/template';
 import { CalculatorTool } from 'bee-agent-framework/tools/calculator';
 import { LLMTool } from 'bee-agent-framework/tools/llm';
-import { BaseMessage } from 'bee-agent-framework/llms/primitives/message';
 
 import { AgentContext } from '../execute.js';
 import { getRunVectorStores } from '../helpers.js';
@@ -138,8 +137,7 @@ export async function getTools(run: LoadedRun, context: AgentContext): Promise<F
   if (llmUsage)
     tools.push(
       new LLMTool({
-        llm: createChatLLM({ model: getDefaultModel() }),
-        transform: (text) => [new BaseMessage('assistant', text)]
+        llm: createChatLLM({ model: getDefaultModel() })
       })
     );
 

diff --git a/src/tools/tools.service.ts b/src/tools/tools.service.ts
@@ -29,7 +29,6 @@ import { OpenMeteoTool } from 'bee-agent-framework/tools/weather/openMeteo';
 import { ArXivTool } from 'bee-agent-framework/tools/arxiv';
 import { PythonTool } from 'bee-agent-framework/tools/python/python';
 import { parse } from 'yaml';
-import { BaseMessage } from 'bee-agent-framework/llms/primitives/message';
 
 import { Tool as ToolDto } from './dtos/tool.js';
 import { AnyTool, Tool, ToolExecutor, ToolType } from './entities/tool/tool.entity.js';
@@ -468,8 +467,7 @@ function getSystemTools() {
   const fileSearch = new FileSearchTool({ vectorStores: [], maxNumResults: 0 });
   const readFile = new ReadFileTool({ files: [], fileSize: 0 });
   const llmTool = new LLMTool({
-    llm: createChatLLM({ model: getDefaultModel() }),
-    transform: (text) => [new BaseMessage('assistant', text)]
+    llm: createChatLLM({ model: getDefaultModel() })
   });
   const calculatorTool = new CalculatorTool();