From 5ce30fdef161ce23305bb420bcfb8f03d71cd64c Mon Sep 17 00:00:00 2001 From: Zack Date: Fri, 8 Dec 2023 03:56:54 -0800 Subject: [PATCH] feat: add openai plugins --- app.py | 11 +++++++++++ swarms/tools/openai_plugins.py | 1 + swarms/tools/tools_controller.py | 11 +++++++++++ 3 files changed, 23 insertions(+) create mode 100644 swarms/tools/openai_plugins.py diff --git a/app.py b/app.py index 5b1751f0..f2b55bed 100644 --- a/app.py +++ b/app.py @@ -4,6 +4,7 @@ from functools import partial from pathlib import Path from threading import Lock import warnings +import json from swarms.modelui.modules.block_requests import OpenMonkeyPatch, RequestBlocker from swarms.modelui.modules.logging_colors import logger @@ -114,9 +115,19 @@ tools_mappings = { "walmart": "http://127.0.0.1:8079/tools/walmart", } +# Load the JSON file +with open('swarms/tools/openai.json', 'r') as f: + data = json.load(f) +for plugin in data: + url = plugin['manifest']['api']['url'].replace('/.well-known/openapi.yaml', '') + tool_name = plugin['namespace'] + tools_mappings[tool_name] = url valid_tools_info = [] all_tools_list = [] +print(data) +print(plugin) + gr.close_all() MAX_TURNS = 30 diff --git a/swarms/tools/openai_plugins.py b/swarms/tools/openai_plugins.py new file mode 100644 index 00000000..8b137891 --- /dev/null +++ b/swarms/tools/openai_plugins.py @@ -0,0 +1 @@ + diff --git a/swarms/tools/tools_controller.py b/swarms/tools/tools_controller.py index c3efff0f..070d7dfb 100644 --- a/swarms/tools/tools_controller.py +++ b/swarms/tools/tools_controller.py @@ -13,9 +13,20 @@ from vllm import LLM from swarms.utils import get_logger from pathlib import Path from langchain.llms import VLLM +import json logger = get_logger(__name__) +# with open('/openai.json', 'r') as f: +# data = json.load(f) + +# tools_mappings = {} + +# for plugin in data: +# url = plugin['manifest']['api']['url'].replace('/.well-known/openapi.yaml', '') +# tool_name = plugin['namespace'] +# tools_mappings[tool_name] = url + def load_valid_tools(tools_mappings): tools_to_config = {} for key in tools_mappings: