diff --git a/playground/agents/revgpt_agent.py b/playground/agents/revgpt_agent.py deleted file mode 100644 index 16a720e8..00000000 --- a/playground/agents/revgpt_agent.py +++ /dev/null @@ -1,28 +0,0 @@ -import os -from dotenv import load_dotenv -from swarms.models.revgptV4 import RevChatGPTModel -from swarms.workers.worker import Worker - -load_dotenv() - -config = { - "model": os.getenv("REVGPT_MODEL"), - "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") - ], -} - -llm = RevChatGPTModel(access_token=os.getenv("ACCESS_TOKEN"), **config) - -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) -print(response) diff --git a/pyproject.toml b/pyproject.toml index eea95362..dcad7e7e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api" [tool.poetry] name = "swarms" -version = "2.4.0" +version = "2.4.1" description = "Swarms - Pytorch" license = "MIT" authors = ["Kye Gomez "] diff --git a/swarms/models/__init__.py b/swarms/models/__init__.py index deaa5fde..deac803f 100644 --- a/swarms/models/__init__.py +++ b/swarms/models/__init__.py @@ -25,7 +25,7 @@ from swarms.models.idefics import Idefics # noqa: E402 from swarms.models.vilt import Vilt # noqa: E402 from swarms.models.nougat import Nougat # noqa: E402 from swarms.models.layoutlm_document_qa import LayoutLMDocumentQA # noqa: E402 -from swarms.models.gpt4_vision_api import GPT4VisionAPI # noqa: E40 +from swarms.models.gpt4_vision_api import GPT4VisionAPI # noqa: E402 # from swarms.models.gpt4v import GPT4Vision # from swarms.models.dalle3 import Dalle3 diff --git a/swarms/models/gpt4_vision_api.py b/swarms/models/gpt4_vision_api.py index 87436c55..835384ec 100644 --- a/swarms/models/gpt4_vision_api.py +++ b/swarms/models/gpt4_vision_api.py @@ -2,6 +2,7 @@ import base64 import os import requests from dotenv import load_dotenv +import concurrent.futures # Load environment variables load_dotenv() @@ -135,4 +136,5 @@ class GPT4VisionAPI: except Exception as error: print(f"Error with the request: {error}") raise error - # Function to handle vision tasks \ No newline at end of file + # Function to handle vision tasks +