general cleaanup

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>"]

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

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

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

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