ai-mirow-projects
/
swarms
mirror of https://github.com/kyegomez/swarms
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

chore: sync

Browse Source
pull/307/head
Zack 1 year ago
parent 13c54d0b00
commit c8cb428a48
12 changed files with 68 additions and 186 deletions
Show Stats Download Patch File Download Diff File

2
multi_modal_auto_agent.py
Unescape View File

@ -10,7 +10,7 @@ img = "images/swarms.jpeg"
## Initialize the workflow
flow = Flow(
llm=llm,
max_loops='auto',
max_loops="auto",
dashboard=True,
)

4
playground/demos/positive_med/positive_med.py
Unescape View File

@ -64,8 +64,8 @@ topic_selection_task = (
" practices"
)
topics = llm(
f"Your System Instructions: {TOPIC_GENERATOR_SYSTEM_PROMPT}, Your current task:"
f" {topic_selection_task}"
f"Your System Instructions: {TOPIC_GENERATOR_SYSTEM_PROMPT}, Your current"
f" task: {topic_selection_task}"
)
dashboard = print(

1
swarms/models/__init__.py
Unescape View File

@ -55,5 +55,4 @@ __all__ = [
# "Dalle3",
# "DistilWhisperModel",
"GPT4VisionAPI",
]

6
swarms/models/anthropic.py
Unescape View File

@ -200,11 +200,9 @@ def build_extra_kwargs(
if field_name in extra_kwargs:
raise ValueError(f"Found {field_name} supplied twice.")
if field_name not in all_required_field_names:
warnings.warn(
f"""WARNING! {field_name} is not default parameter.
warnings.warn(f"""WARNING! {field_name} is not default parameter.
{field_name} was transferred to model_kwargs.
Please confirm that {field_name} is what you intended."""
)
Please confirm that {field_name} is what you intended.""")
extra_kwargs[field_name] = values.pop(field_name)
invalid_model_kwargs = all_required_field_names.intersection(

5
swarms/models/dalle3.py
Unescape View File

@ -170,9 +170,12 @@ class Dalle3:
# Handling exceptions and printing the errors details
print(
colored(
(
f"Error running Dalle3: {error} try optimizing your api"
" key and or try again"
),
"red",
)
)
raise error
@ -320,8 +323,10 @@ class Dalle3:
except Exception as error:
print(
colored(
(
f"Error running Dalle3: {error} try optimizing"
" your api key and or try again"
),
"red",
)
)

15
swarms/models/gpt4_vision_api.py
Unescape View File

@ -7,6 +7,7 @@ from dotenv import load_dotenv
load_dotenv()
openai_api_key = os.getenv("OPENAI_API_KEY")
class GPT4VisionAPI:
"""
GPT-4 Vision API
@ -37,10 +38,8 @@ class GPT4VisionAPI:
"""
def __init__(
self,
openai_api_key: str = openai_api_key
):
def __init__(self, openai_api_key: str = openai_api_key):
super().__init__()
self.openai_api_key = openai_api_key
@ -68,7 +67,9 @@ class GPT4VisionAPI:
{
"type": "image_url",
"image_url": {
"url": f"data:image/jpeg;base64,{base64_image}"
"url": (
f"data:image/jpeg;base64,{base64_image}"
)
},
},
],
@ -108,7 +109,9 @@ class GPT4VisionAPI:
{
"type": "image_url",
"image_url": {
"url": f"data:image/jpeg;base64,{base64_image}"
"url": (
f"data:image/jpeg;base64,{base64_image}"
)
},
},
],

3
swarms/models/huggingface.py
Unescape View File

@ -307,9 +307,12 @@ class HuggingfaceLLM:
except Exception as e:
print(
colored(
(
"HuggingfaceLLM could not generate text because of"
f" error: {e}, try optimizing your arguments"
),
"red",
)
)
raise

2
swarms/models/ssd_1b.py
Unescape View File

@ -144,8 +144,10 @@ class SSD1B:
# Handling exceptions and printing the errors details
print(
colored(
(
f"Error running SSD1B: {error} try optimizing your api"
" key and or try again"
),
"red",
)
);

2
swarms/prompts/autobloggen.py
Unescape View File

@ -274,5 +274,3 @@ Check Accuracy:
- Flag any bold claims that lack credible evidence for fact-checker review.
"""

8
swarms/structs/flow.py
Unescape View File

@ -489,8 +489,10 @@ class Flow:
except Exception as error:
print(
colored(
"Error activating autonomous agent. Try optimizing your"
" parameters...",
(
"Error activating autonomous agent. Try optimizing your"
" parameters..."
),
"red",
)
)
@ -657,7 +659,7 @@ class Flow:
while attempt < self.retry_attempts:
try:
response = self.llm(
task ** kwargs,
task**kwargs,
)
if self.interactive:
print(f"AI: {response}")

8
swarms/structs/sequential_workflow.py
Unescape View File

@ -533,9 +533,11 @@ class SequentialWorkflow:
except Exception as e:
print(
colored(
f"Error initializing the Sequential workflow: {e} try"
" optimizing your inputs like the flow class and task"
" description",
(
f"Error initializing the Sequential workflow: {e} try"
" optimizing your inputs like the flow class and task"
" description"
),
"red",
attrs=["bold", "underline"],
)

190
tests/models/test_gpt4_vision_api.py
Unescape View File

@ -1,12 +1,9 @@
import asyncio
import os
from unittest.mock import AsyncMock, Mock, mock_open, patch
from aiohttp import ClientResponseError
import pytest
from dotenv import load_dotenv
from unittest.mock import mock_open, patch, Mock
from requests.exceptions import RequestException
from swarms.models.gpt4_vision_api import GPT4VisionAPI
import os
from dotenv import load_dotenv
load_dotenv()
@ -26,21 +23,18 @@ def test_init(vision_api):
def test_encode_image(vision_api):
with patch(
"builtins.open",
mock_open(read_data=b"test_image_data"),
create=True,
"builtins.open", mock_open(read_data=b"test_image_data"), create=True
):
encoded_image = vision_api.encode_image(img)
encoded_image = vision_api.encode_image("test_image.jpg")
assert encoded_image == "dGVzdF9pbWFnZV9kYXRh"
def test_run_success(vision_api):
expected_response = {"This is the model's response."}
expected_response = {"choices": [{"text": "This is the model's response."}]}
with patch(
"requests.post",
return_value=Mock(json=lambda: expected_response),
"requests.post", return_value=Mock(json=lambda: expected_response)
) as mock_post:
result = vision_api.run("What is this?", img)
result = vision_api.run("What is this?", "test_image.jpg")
mock_post.assert_called_once()
assert result == "This is the model's response."
@ -50,28 +44,24 @@ def test_run_request_error(vision_api):
"requests.post", side_effect=RequestException("Request Error")
) as mock_post:
with pytest.raises(RequestException):
vision_api.run("What is this?", img)
vision_api.run("What is this?", "test_image.jpg")
def test_run_response_error(vision_api):
expected_response = {"error": "Model Error"}
with patch(
"requests.post",
return_value=Mock(json=lambda: expected_response),
"requests.post", return_value=Mock(json=lambda: expected_response)
) as mock_post:
with pytest.raises(RuntimeError):
vision_api.run("What is this?", img)
vision_api.run("What is this?", "test_image.jpg")
def test_call(vision_api):
expected_response = {
"choices": [{"text": "This is the model's response."}]
}
expected_response = {"choices": [{"text": "This is the model's response."}]}
with patch(
"requests.post",
return_value=Mock(json=lambda: expected_response),
"requests.post", return_value=Mock(json=lambda: expected_response)
) as mock_post:
result = vision_api("What is this?", img)
result = vision_api("What is this?", "test_image.jpg")
mock_post.assert_called_once()
assert result == "This is the model's response."
@ -92,12 +82,22 @@ def test_initialization_with_custom_key():
assert api.openai_api_key == custom_key
def test_run_successful_response(gpt_api):
task = "What is in the image?"
img_url = img
response_json = {"choices": [{"text": "Answer from GPT-4 Vision"}]}
mock_response = Mock()
mock_response.json.return_value = response_json
with patch("requests.post", return_value=mock_response) as mock_post:
result = gpt_api.run(task, img_url)
mock_post.assert_called_once()
assert result == response_json["choices"][0]["text"]
def test_run_with_exception(gpt_api):
task = "What is in the image?"
img_url = img
with patch(
"requests.post", side_effect=Exception("Test Exception")
):
with patch("requests.post", side_effect=Exception("Test Exception")):
with pytest.raises(Exception):
gpt_api.run(task, img_url)
@ -105,14 +105,10 @@ def test_run_with_exception(gpt_api):
def test_call_method_successful_response(gpt_api):
task = "What is in the image?"
img_url = img
response_json = {
"choices": [{"text": "Answer from GPT-4 Vision"}]
}
response_json = {"choices": [{"text": "Answer from GPT-4 Vision"}]}
mock_response = Mock()
mock_response.json.return_value = response_json
with patch(
"requests.post", return_value=mock_response
) as mock_post:
with patch("requests.post", return_value=mock_response) as mock_post:
result = gpt_api(task, img_url)
mock_post.assert_called_once()
assert result == response_json
@ -121,132 +117,6 @@ def test_call_method_successful_response(gpt_api):
def test_call_method_with_exception(gpt_api):
task = "What is in the image?"
img_url = img
with patch(
"requests.post", side_effect=Exception("Test Exception")
):
with patch("requests.post", side_effect=Exception("Test Exception")):
with pytest.raises(Exception):
gpt_api(task, img_url)
@pytest.mark.asyncio
async def test_arun_success(vision_api):
expected_response = {
"choices": [
{"message": {"content": "This is the model's response."}}
]
}
with patch(
"aiohttp.ClientSession.post",
new_callable=AsyncMock,
return_value=AsyncMock(
json=AsyncMock(return_value=expected_response)
),
) as mock_post:
result = await vision_api.arun("What is this?", img)
mock_post.assert_called_once()
assert result == "This is the model's response."
@pytest.mark.asyncio
async def test_arun_request_error(vision_api):
with patch(
"aiohttp.ClientSession.post",
new_callable=AsyncMock,
side_effect=Exception("Request Error"),
) as mock_post:
with pytest.raises(Exception):
await vision_api.arun("What is this?", img)
def test_run_many_success(vision_api):
expected_response = {
"choices": [
{"message": {"content": "This is the model's response."}}
]
}
with patch(
"requests.post",
return_value=Mock(json=lambda: expected_response),
) as mock_post:
tasks = ["What is this?", "What is that?"]
imgs = [img, img]
results = vision_api.run_many(tasks, imgs)
assert mock_post.call_count == 2
assert results == [
"This is the model's response.",
"This is the model's response.",
]
def test_run_many_request_error(vision_api):
with patch(
"requests.post", side_effect=RequestException("Request Error")
) as mock_post:
tasks = ["What is this?", "What is that?"]
imgs = [img, img]
with pytest.raises(RequestException):
vision_api.run_many(tasks, imgs)
@pytest.mark.asyncio
async def test_arun_json_decode_error(vision_api):
with patch(
"aiohttp.ClientSession.post",
new_callable=AsyncMock,
return_value=AsyncMock(
json=AsyncMock(side_effect=ValueError)
),
) as mock_post:
with pytest.raises(ValueError):
await vision_api.arun("What is this?", img)
@pytest.mark.asyncio
async def test_arun_api_error(vision_api):
error_response = {"error": {"message": "API Error"}}
with patch(
"aiohttp.ClientSession.post",
new_callable=AsyncMock,
return_value=AsyncMock(
json=AsyncMock(return_value=error_response)
),
) as mock_post:
with pytest.raises(Exception, match="API Error"):
await vision_api.arun("What is this?", img)
@pytest.mark.asyncio
async def test_arun_unexpected_response(vision_api):
unexpected_response = {"unexpected": "response"}
with patch(
"aiohttp.ClientSession.post",
new_callable=AsyncMock,
return_value=AsyncMock(
json=AsyncMock(return_value=unexpected_response)
),
) as mock_post:
with pytest.raises(Exception, match="Unexpected response"):
await vision_api.arun("What is this?", img)
@pytest.mark.asyncio
async def test_arun_retries(vision_api):
with patch(
"aiohttp.ClientSession.post",
new_callable=AsyncMock,
side_effect=ClientResponseError(None, None),
) as mock_post:
with pytest.raises(ClientResponseError):
await vision_api.arun("What is this?", img)
assert mock_post.call_count == vision_api.retries + 1
@pytest.mark.asyncio
async def test_arun_timeout(vision_api):
with patch(
"aiohttp.ClientSession.post",
new_callable=AsyncMock,
side_effect=asyncio.TimeoutError,
) as mock_post:
with pytest.raises(asyncio.TimeoutError):
await vision_api.arun("What is this?", img)

Write Preview
Loading…
Cancel
Save