clean up of bingchapgpt, revgpt, cookies, json, schemas in swarms

kyegomez · Oct 22, 2023 · ccb239d · ccb239d
1 parent f5f95f3
commit ccb239d
Show file tree

Hide file tree

Showing 22 changed files with 1,553 additions and 138 deletions.
diff --git a/bingchat.py b/bingchat.py
diff --git a/cookies.json b/cookies.json
diff --git a/playground/agents/revgpt_agent.py b/playground/agents/revgpt_agent.py
@@ -10,15 +10,12 @@
     "plugin_ids": [os.getenv("REVGPT_PLUGIN_IDS")],
     "disable_history": os.getenv("REVGPT_DISABLE_HISTORY") == "True",
     "PUID": os.getenv("REVGPT_PUID"),
-    "unverified_plugin_domains": [os.getenv("REVGPT_UNVERIFIED_PLUGIN_DOMAINS")]
+    "unverified_plugin_domains": [os.getenv("REVGPT_UNVERIFIED_PLUGIN_DOMAINS")],
 }
 
 llm = RevChatGPTModel(access_token=os.getenv("ACCESS_TOKEN"), **config)
 
-worker = Worker(
-    ai_name="Optimus Prime",
-    llm=llm
-)
+worker = Worker(ai_name="Optimus Prime", llm=llm)
 
 task = "What were the winning boston marathon times for the past 5 years (ending in 2022)? Generate a table of the year, name, country of origin, and times."
 response = worker.run(task)

diff --git a/playground/models/bingchat.py b/playground/models/bingchat.py
@@ -2,18 +2,20 @@
 from swarms.workers.worker import Worker
 from swarms.tools.autogpt import EdgeGPTTool, tool
 from swarms.models import OpenAIChat
-import os 
+import os
 
 api_key = os.getenv("OPENAI_API_KEY")
 
 # Initialize the EdgeGPTModel
 edgegpt = BingChat(cookies_path="./cookies.txt")
 
+
 @tool
 def edgegpt(task: str = None):
     """A tool to run infrence on the EdgeGPT Model"""
     return EdgeGPTTool.run(task)
 
+
 # Initialize the language model,
 # This model can be swapped out with Anthropic, ETC, Huggingface Models like Mistral, ETC
 llm = OpenAIChat(
@@ -22,11 +24,7 @@ def edgegpt(task: str = None):
 )
 
 # Initialize the Worker with the custom tool
-worker = Worker(
-    llm=llm,
-    ai_name="EdgeGPT Worker",
-    external_tools=[edgegpt]
-)
+worker = Worker(llm=llm, ai_name="EdgeGPT Worker", external_tools=[edgegpt])
 
 # Use the worker to process a task
 task = "Hello, my name is ChatGPT"

diff --git a/revgpt.py → playground/models/revgpt.py b/revgpt.py → playground/models/revgpt.py
@@ -14,7 +14,7 @@
     "plugin_ids": [os.getenv("REVGPT_PLUGIN_IDS")],
     "disable_history": os.getenv("REVGPT_DISABLE_HISTORY") == "True",
     "PUID": os.getenv("REVGPT_PUID"),
-    "unverified_plugin_domains": [os.getenv("REVGPT_UNVERIFIED_PLUGIN_DOMAINS")]
+    "unverified_plugin_domains": [os.getenv("REVGPT_UNVERIFIED_PLUGIN_DOMAINS")],
 }
 
 # For v1 model