refactor(langchain/tools): move availableTools import to tools/index.js

This commit is contained in:
Daniel Avila 2023-06-14 13:09:49 -04:00 committed by Danny Avila
parent a42ef2944c
commit d339c291fa
4 changed files with 6 additions and 6 deletions

View file

@ -6,8 +6,10 @@ const StructuredSD = require('./structured/StableDiffusion');
const StableDiffusionAPI = require('./StableDiffusion'); const StableDiffusionAPI = require('./StableDiffusion');
const WolframAlphaAPI = require('./Wolfram'); const WolframAlphaAPI = require('./Wolfram');
const SelfReflectionTool = require('./SelfReflection'); const SelfReflectionTool = require('./SelfReflection');
const availableTools = require('./manifest.json');
module.exports = { module.exports = {
availableTools,
GoogleSearchAPI, GoogleSearchAPI,
HttpRequestTool, HttpRequestTool,
AIPluginTool, AIPluginTool,

View file

@ -9,6 +9,7 @@ const { ChatOpenAI } = require('langchain/chat_models/openai');
const { Calculator } = require('langchain/tools/calculator'); const { Calculator } = require('langchain/tools/calculator');
const { WebBrowser } = require('langchain/tools/webbrowser'); const { WebBrowser } = require('langchain/tools/webbrowser');
const { const {
availableTools,
AIPluginTool, AIPluginTool,
GoogleSearchAPI, GoogleSearchAPI,
WolframAlphaAPI, WolframAlphaAPI,
@ -17,7 +18,6 @@ const {
StableDiffusionAPI, StableDiffusionAPI,
StructuredSD, StructuredSD,
} = require('../'); } = require('../');
const availableTools = require('../manifest.json');
const validateTools = async (user, tools = []) => { const validateTools = async (user, tools = []) => {
try { try {

View file

@ -20,11 +20,11 @@ jest.mock('../../../../models/User', () => {
jest.mock('../../../../server/services/PluginService', () => mockPluginService); jest.mock('../../../../server/services/PluginService', () => mockPluginService);
const User = require('../../../../models/User'); const User = require('../../../../models/User');
const { validateTools, loadTools, availableTools } = require('./'); const { validateTools, loadTools } = require('./');
const PluginService = require('../../../../server/services/PluginService'); const PluginService = require('../../../../server/services/PluginService');
const { BaseChatModel } = require('langchain/chat_models/openai'); const { BaseChatModel } = require('langchain/chat_models/openai');
const { Calculator } = require('langchain/tools/calculator'); const { Calculator } = require('langchain/tools/calculator');
const { OpenAICreateImage, GoogleSearchAPI } = require('../'); const { availableTools, OpenAICreateImage, GoogleSearchAPI } = require('../');
describe('Tool Handlers', () => { describe('Tool Handlers', () => {
let fakeUser; let fakeUser;

View file

@ -1,8 +1,6 @@
const availableTools = require('../manifest.json');
const { validateTools, loadTools } = require('./handleTools'); const { validateTools, loadTools } = require('./handleTools');
module.exports = { module.exports = {
validateTools, validateTools,
loadTools, loadTools
availableTools
}; };