refactor(handleTools.js): change loadTools function signature to include functions parameter

feat(handleTools.test.js): add test for loading StructuredSD tool with functions parameter

Author: danny-avila

api/app/langchain/ChatAgent.js

@@ -451,6 +451,7 @@ Only respond with your conversational reply to the following User Message:
       user,
       model,
       tools: this.options.tools,
+      functions: this.functionsAgent,
       options: {
         openAIApiKey: this.openAIApiKey
       }

api/app/langchain/tools/util/handleTools.js

@@ -72,8 +72,7 @@ const loadToolWithAuth = async (user, authFields, ToolConstructor, options = {})
   };
 };
 
-const loadTools = async ({ user, model, tools = [], options = {} }) => {
-  const { functions } = options;
+const loadTools = async ({ user, model, functions = null, tools = [], options = {} }) => {
   const toolConstructors = {
     calculator: Calculator,
     google: GoogleSearchAPI,

api/app/langchain/tools/util/handleTools.test.js

@@ -24,7 +24,7 @@ const { validateTools, loadTools } = require('./');
 const PluginService = require('../../../../server/services/PluginService');
 const { BaseChatModel } = require('langchain/chat_models/openai');
 const { Calculator } = require('langchain/tools/calculator');
-const { availableTools, OpenAICreateImage, GoogleSearchAPI } = require('../');
+const { availableTools, OpenAICreateImage, GoogleSearchAPI, StructuredSD } = require('../');
 
 describe('Tool Handlers', () => {
   let fakeUser;
@@ -174,5 +174,17 @@ describe('Tool Handlers', () => {
       });
       expect(toolFunctions).toEqual({});
     });
+    it('should return the StructuredTool version when using functions', async () => {
+      process.env.SD_WEBUI_URL = mockCredential;
+      toolFunctions = await loadTools({
+        user: fakeUser._id,
+        model: BaseChatModel,
+        tools: ['stable-diffusion'],
+        functions: true
+      });
+      const structuredTool = await toolFunctions['stable-diffusion']();
+      expect(structuredTool).toBeInstanceOf(StructuredSD);
+      delete process.env.SD_WEBUI_URL;
+    });
   });
 });