From fde09d6bd306c2b07652fc368de5da179fd3f456 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Wed, 31 Jan 2024 21:31:48 -0500
Subject: [PATCH 1/5] [EXAMPLE][Tool Agent]
---
playground/agents/tool_agent.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/playground/agents/tool_agent.py b/playground/agents/tool_agent.py
index 93e07ff3..91b576a4 100644
--- a/playground/agents/tool_agent.py
+++ b/playground/agents/tool_agent.py
@@ -33,4 +33,4 @@ agent = ToolAgent(
generated_data = agent.run(task)
# Print the generated data
-print(generated_data)
+print(f"Generated data: {generated_data}")
From 5569a5daf8ecc0d5b51d715f518da3b1b5a4c68b Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Wed, 31 Jan 2024 21:33:58 -0500
Subject: [PATCH 2/5] [BUFG][ToolAgent]
---
playground/agents/tool_agent.py | 2 +-
swarms/agents/tool_agent.py | 10 +++++-----
2 files changed, 6 insertions(+), 6 deletions(-)
diff --git a/playground/agents/tool_agent.py b/playground/agents/tool_agent.py
index 91b576a4..3cd93fbd 100644
--- a/playground/agents/tool_agent.py
+++ b/playground/agents/tool_agent.py
@@ -26,7 +26,7 @@ task = (
# Create an instance of the ToolAgent class
agent = ToolAgent(
- model=model, tokenizer=tokenizer, json_schema=json_schema
+ name="dolly-function-agent", description="Ana gent to create a child data", model=model, tokenizer=tokenizer, json_schema=json_schema
)
# Run the agent to generate the person's information
diff --git a/swarms/agents/tool_agent.py b/swarms/agents/tool_agent.py
index d9359471..f6d9f6e8 100644
--- a/swarms/agents/tool_agent.py
+++ b/swarms/agents/tool_agent.py
@@ -61,11 +61,11 @@ class ToolAgent(AbstractLLM):
def __init__(
self,
- name: str,
- description: str,
- model: Any,
- tokenizer: Any,
- json_schema: Any,
+ name: str = None,
+ description: str = None,
+ model: Any = None,
+ tokenizer: Any = None,
+ json_schema: Any = None,
max_number_tokens: int = 500,
*args,
**kwargs,
From 18b8a5d602d1249d141716ef550bff64bde8ae98 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Wed, 31 Jan 2024 21:40:37 -0500
Subject: [PATCH 3/5] [REFACTOR][ToolAgent][name and description]
---
playground/agents/tool_agent.py | 6 +++++-
swarms/agents/tool_agent.py | 4 ++--
2 files changed, 7 insertions(+), 3 deletions(-)
diff --git a/playground/agents/tool_agent.py b/playground/agents/tool_agent.py
index 3cd93fbd..71d10bb8 100644
--- a/playground/agents/tool_agent.py
+++ b/playground/agents/tool_agent.py
@@ -26,7 +26,11 @@ task = (
# Create an instance of the ToolAgent class
agent = ToolAgent(
- name="dolly-function-agent", description="Ana gent to create a child data", model=model, tokenizer=tokenizer, json_schema=json_schema
+ name="dolly-function-agent",
+ description="Ana gent to create a child data",
+ model=model,
+ tokenizer=tokenizer,
+ json_schema=json_schema,
)
# Run the agent to generate the person's information
diff --git a/swarms/agents/tool_agent.py b/swarms/agents/tool_agent.py
index f6d9f6e8..8e6adf9d 100644
--- a/swarms/agents/tool_agent.py
+++ b/swarms/agents/tool_agent.py
@@ -61,8 +61,8 @@ class ToolAgent(AbstractLLM):
def __init__(
self,
- name: str = None,
- description: str = None,
+ name: str = "Function Calling Agent",
+ description: str = "Generates a function based on the input json schema and the task",
model: Any = None,
tokenizer: Any = None,
json_schema: Any = None,
From 537f81f93272cd6e3b57ffb72e7b0b083807efde Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Wed, 31 Jan 2024 21:46:09 -0500
Subject: [PATCH 4/5] [REFACTOR][ToolAgent]
---
playground/agents/tool_agent.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/playground/agents/tool_agent.py b/playground/agents/tool_agent.py
index 71d10bb8..0a95f42c 100644
--- a/playground/agents/tool_agent.py
+++ b/playground/agents/tool_agent.py
@@ -4,7 +4,9 @@ from swarms import ToolAgent
# Load the pre-trained model and tokenizer
model = AutoModelForCausalLM.from_pretrained(
- "databricks/dolly-v2-12b"
+ "databricks/dolly-v2-12b",
+ load_in_4bit=True,
+ device_map="auto",
)
tokenizer = AutoTokenizer.from_pretrained("databricks/dolly-v2-12b")
From a552a983730c98b2b7d9da386ea4dc7053166657 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Wed, 31 Jan 2024 21:59:54 -0500
Subject: [PATCH 5/5] [EXAMPLES][Cleanup][Kosmos]
---
playground/models/kosmos2_example.py | 10 ----------
playground/models/kosmos_two_example.py | 11 -----------
pyproject.toml | 2 +-
3 files changed, 1 insertion(+), 22 deletions(-)
delete mode 100644 playground/models/kosmos2_example.py
delete mode 100644 playground/models/kosmos_two_example.py
diff --git a/playground/models/kosmos2_example.py b/playground/models/kosmos2_example.py
deleted file mode 100644
index 6fc4df02..00000000
--- a/playground/models/kosmos2_example.py
+++ /dev/null
@@ -1,10 +0,0 @@
-from swarms.models.kosmos2 import Kosmos2
-from PIL import Image
-
-
-model = Kosmos2.initialize()
-
-image = Image.open("images/swarms.jpg")
-
-detections = model(image)
-print(detections)
diff --git a/playground/models/kosmos_two_example.py b/playground/models/kosmos_two_example.py
deleted file mode 100644
index 8bf583cd..00000000
--- a/playground/models/kosmos_two_example.py
+++ /dev/null
@@ -1,11 +0,0 @@
-from swarms.models.kosmos_two import Kosmos
-
-# Initialize Kosmos
-kosmos = Kosmos()
-
-# Perform multimodal grounding
-out = kosmos.multimodal_grounding(
- "Find the red apple in the image.", "images/swarms.jpeg"
-)
-
-print(out)
diff --git a/pyproject.toml b/pyproject.toml
index c9f0c28d..f4971b47 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
[tool.poetry]
name = "swarms"
-version = "4.0.1"
+version = "4.0.2"
description = "Swarms - Pytorch"
license = "MIT"
authors = ["Kye Gomez <kye@apac.ai>"]