feat(tools): add structured Wolfram tool

This commit is contained in:
Daniel Avila 2023-06-15 21:06:17 -04:00 committed by Danny Avila
parent d0be2e6f4a
commit 4353d42035
3 changed files with 77 additions and 2 deletions

View file

@ -13,6 +13,7 @@ const {
AIPluginTool,
GoogleSearchAPI,
WolframAlphaAPI,
StructuredWolfram,
HttpRequestTool,
OpenAICreateImage,
StableDiffusionAPI,
@ -76,7 +77,7 @@ const loadTools = async ({ user, model, functions = null, tools = [], options =
const toolConstructors = {
calculator: Calculator,
google: GoogleSearchAPI,
wolfram: WolframAlphaAPI,
wolfram: functions ? StructuredWolfram : WolframAlphaAPI,
'dall-e': OpenAICreateImage,
'stable-diffusion': functions ? StructuredSD : StableDiffusionAPI
};