feat: add openai plugins

pull/282/head^2
Zack 1 year ago
parent d76ba69311
commit 5ce30fdef1

@ -4,6 +4,7 @@ from functools import partial
from pathlib import Path from pathlib import Path
from threading import Lock from threading import Lock
import warnings import warnings
import json
from swarms.modelui.modules.block_requests import OpenMonkeyPatch, RequestBlocker from swarms.modelui.modules.block_requests import OpenMonkeyPatch, RequestBlocker
from swarms.modelui.modules.logging_colors import logger from swarms.modelui.modules.logging_colors import logger
@ -114,9 +115,19 @@ tools_mappings = {
"walmart": "http://127.0.0.1:8079/tools/walmart", "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 = [] valid_tools_info = []
all_tools_list = [] all_tools_list = []
print(data)
print(plugin)
gr.close_all() gr.close_all()
MAX_TURNS = 30 MAX_TURNS = 30

@ -13,9 +13,20 @@ from vllm import LLM
from swarms.utils import get_logger from swarms.utils import get_logger
from pathlib import Path from pathlib import Path
from langchain.llms import VLLM from langchain.llms import VLLM
import json
logger = get_logger(__name__) 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): def load_valid_tools(tools_mappings):
tools_to_config = {} tools_to_config = {}
for key in tools_mappings: for key in tools_mappings:

Loading…
Cancel
Save