[CLEANUP][Feat][Swarms Memory]

kyegomez · Jul 8, 2024 · b8f3127 · b8f3127
1 parent bbbbc9e
commit b8f3127
Show file tree

Hide file tree

Showing 17 changed files with 95 additions and 375 deletions.
diff --git a/README.md b/README.md
diff --git a/docs/swarms/structs/index.md b/docs/swarms/structs/index.md
@@ -72,7 +72,7 @@ agent.run("Generate a 10,000 word blog on health and wellness.")
 
 ```python
 from swarms import Agent, OpenAIChat
-from playground.memory.chromadb_example import ChromaDB # Copy and paste the code and put it in your own local directory.
+from swarms_memory import ChromaDB # Copy and paste the code and put it in your own local directory.
 
 # Making an instance of the ChromaDB class
 memory = ChromaDB(

diff --git a/playground/agents/agent_with_long_term_memory.py b/playground/agents/agent_with_long_term_memory.py
@@ -1,5 +1,5 @@
 from swarms import Agent, OpenAIChat
-from playground.memory.chromadb_example import ChromaDB
+from swarms_memory import ChromaDB
 from swarms.models.tiktoken_wrapper import TikTokenizer
 
 # Initialize the agent

diff --git a/playground/agents/agent_with_longterm_memory.py b/playground/agents/agent_with_longterm_memory.py
@@ -4,7 +4,7 @@
 
 # Import the OpenAIChat model and the Agent struct
 from swarms import Agent, OpenAIChat
-from playground.memory.chromadb_example import ChromaDB
+from swarms_memory import ChromaDB
 
 # Load the environment variables
 load_dotenv()

diff --git a/playground/agents/new_perplexity_agent.py b/playground/agents/new_perplexity_agent.py
@@ -1,6 +1,6 @@
 from swarms import Agent
 from swarms.models.llama3_hosted import llama3Hosted
-from playground.memory.chromadb_example import ChromaDB
+from swarms_memory import ChromaDB
 from swarms.tools.prebuilt.bing_api import fetch_web_articles_bing_api
 
 # Define the research system prompt

diff --git a/playground/agents/perplexity_agent.py b/playground/agents/perplexity_agent.py
@@ -10,7 +10,7 @@
 """
 
 from swarms import Agent, OpenAIChat
-from playground.memory.chromadb_example import ChromaDB
+from swarms_memory import ChromaDB
 from swarms.tools.prebuilt.bing_api import fetch_web_articles_bing_api
 import os
 from dotenv import load_dotenv

diff --git a/playground/demos/patient_question_assist/main.py b/playground/demos/patient_question_assist/main.py
@@ -1,6 +1,6 @@
 from swarms import Agent, OpenAIChat
 from typing import List
-from playground.memory.chromadb_example import ChromaDB
+from swarms_memory import ChromaDB
 
 memory = ChromaDB(
     metric="cosine",

diff --git a/playground/demos/swarm_mechanic/swarm_mechanic_example.py b/playground/demos/swarm_mechanic/swarm_mechanic_example.py
@@ -15,7 +15,7 @@
 
 from swarms import Agent, llama3Hosted, AgentRearrange
 from pydantic import BaseModel
-from playground.memory.chromadb_example import ChromaDB
+from swarms_memory import ChromaDB
 
 # Initialize the language model agent (e.g., GPT-3)
 llm = llama3Hosted(max_tokens=3000)

diff --git a/playground/demos/swarm_of_complaince/compliance_swarm.py b/playground/demos/swarm_of_complaince/compliance_swarm.py
@@ -11,7 +11,7 @@
 
 from swarms import Agent
 from swarms.models.llama3_hosted import llama3Hosted
-from playground.memory.chromadb_example import ChromaDB
+from swarms_memory import ChromaDB
 
 
 # Model

diff --git a/playground/structs/multi_agent_collaboration/mixture_of_agents/agent_ops_moa.py b/playground/structs/multi_agent_collaboration/mixture_of_agents/agent_ops_moa.py
@@ -1,6 +1,6 @@
 from swarms import Agent, OpenAIChat
 from swarms.structs.mixture_of_agents import MixtureOfAgents
-from playground.memory.chromadb_example import ChromaDB
+from swarms_memory import ChromaDB
 
 
 SEC_DATA = """

diff --git a/playground/structs/multi_agent_collaboration/mixture_of_agents/moa_with_scp.py b/playground/structs/multi_agent_collaboration/mixture_of_agents/moa_with_scp.py
@@ -1,6 +1,6 @@
 from swarms import Agent, OpenAIChat
 from swarms.structs.mixture_of_agents import MixtureOfAgents
-from playground.memory.chromadb_example import ChromaDB
+from swarms_memory import ChromaDB
 
 
 SEC_DATA = """

diff --git a/playground/structs/multi_agent_collaboration/swarm_network_example.py b/playground/structs/multi_agent_collaboration/swarm_network_example.py
@@ -6,7 +6,7 @@
     OpenAIChat,
     TogetherLLM,
 )
-from playground.memory.chromadb_example import ChromaDB
+from swarms_memory import ChromaDB
 from dotenv import load_dotenv
 
 # load the environment variables

diff --git a/playground/swarms/movers_swarm.py b/playground/swarms/movers_swarm.py
@@ -10,7 +10,7 @@
 """
 
 from swarms import Agent, OpenAIChat
-from playground.memory.chromadb_example import ChromaDB
+from swarms_memory import ChromaDB
 from swarms.tools.prebuilt.bing_api import fetch_web_articles_bing_api
 import os
 from dotenv import load_dotenv

diff --git a/playground/swarms_example.ipynb b/playground/swarms_example.ipynb
@@ -96,7 +96,7 @@
    "outputs": [],
    "source": [
     "from swarms import Agent, OpenAIChat\n",
-    "from playground.memory.chromadb_example import ChromaDB\n",
+    "from swarms_memory import ChromaDB\n",
     "\n",
     "# Making an instance of the ChromaDB class\n",
     "memory = ChromaDB(\n",

diff --git a/pyproject.toml b/pyproject.toml
@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms"
-version = "5.3.2"
+version = "5.3.3"
 description = "Swarms - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <kye@apac.ai>"]
@@ -55,6 +55,7 @@ openai = ">=1.30.1,<2.0"
 termcolor = "*"
 tiktoken = "*"
 networkx = "*"
+swarms-memory = "*"
 
 
 

diff --git a/requirements.txt b/requirements.txt
@@ -28,4 +28,5 @@ pytest>=8.1.1
 termcolor>=2.4.0
 pandas>=2.2.2
 fastapi>=0.110.1
-networkx
+networkx
+swarms-memory
diff --git a/json_log_cleanup.py → scripts/cleanup/json_log_cleanup.py b/json_log_cleanup.py → scripts/cleanup/json_log_cleanup.py