From 740145c687840f401df4e5cd7e0821388b5ab468 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Thu, 17 Aug 2023 12:16:18 -0400
Subject: [PATCH] general cleaanup

---
 pyproject.toml                | 2 +-
 swarms/__init__.py            | 2 +-
 swarms/agents/base.py         | 2 --
 swarms/agents/memory.py       | 3 +--
 swarms/workers/worker_node.py | 2 +-
 5 files changed, 4 insertions(+), 7 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 563e47cf..c403ccb6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms"
-version = "1.4.3"
+version = "1.4.5"
 description = "Swarms - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <kye@apac.ai>"]
diff --git a/swarms/__init__.py b/swarms/__init__.py
index 6fcb0b7f..2163acf5 100644
--- a/swarms/__init__.py
+++ b/swarms/__init__.py
@@ -11,4 +11,4 @@ from swarms.agents.models.anthropic import Anthropic
 from swarms.agents.models.huggingface import HuggingFaceLLM
 from swarms.agents.models.palm import GooglePalm
 from swarms.agents.models.petals import Petals
-from swarms.agents.models.openai import OpenAI
+# from swarms.agents.models.openai import OpenAI
diff --git a/swarms/agents/base.py b/swarms/agents/base.py
index 0e85f024..5a999416 100644
--- a/swarms/agents/base.py
+++ b/swarms/agents/base.py
@@ -4,8 +4,6 @@ from typing import List, Optional
 
 from langchain.chains.llm import LLMChain
 
-# from pydantic import ValidationError
-
 from swarms.agents.memory.base import VectorStoreRetriever
 from swarms.agents.memory.base_memory import BaseChatMessageHistory, ChatMessageHistory
 from swarms.agents.memory.document import Document
diff --git a/swarms/agents/memory.py b/swarms/agents/memory.py
index 4511a1fb..6880eb77 100644
--- a/swarms/agents/memory.py
+++ b/swarms/agents/memory.py
@@ -1,11 +1,10 @@
 from typing import Any, Dict, List
-# from pydantic import Field
 
 from swarms.agents.memory.base_memory import BaseChatMemory, get_prompt_input_key
 from swarms.agents.memory.base import VectorStoreRetriever
 
 class AgentMemory(BaseChatMemory):
-    retriever: VectorStoreRetriever = Field(exclude=True)
+    retriever: VectorStoreRetriever
     """VectorStoreRetriever object to connect to."""
 
     @property
diff --git a/swarms/workers/worker_node.py b/swarms/workers/worker_node.py
index 1c02d193..0860630b 100644
--- a/swarms/workers/worker_node.py
+++ b/swarms/workers/worker_node.py
@@ -1,5 +1,5 @@
 import logging
-from typing import List, Optional, Union, Any
+from typing import Any, List, Optional, Union
 
 import faiss
 from langchain.agents import Tool