Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2025-12-17 00:40:14 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

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

Browse source 
feat(handleTools.test.js): add test for loading StructuredSD tool with functions parameter
This commit is contained in:
Daniel Avila 2023-06-14 13:23:02 -04:00 • committed by Danny Avila
parent d339c291fa
commit bffa9ad016
3 changed files with 15 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

14
api/app/langchain/tools/util/handleTools.test.js
View file

@ -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;
});
});
});