clean up

Former-commit-id: e9ee3055e8

setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 setup(
   name = 'swarms',
   packages = find_packages(exclude=[]),
-  version = '1.1.0',
+  version = '1.1.1',
   license='MIT',
   description = 'Swarms - Pytorch',
   author = 'Kye Gomez',

swarms/agents/memory/base.py

@@ -7,7 +7,6 @@ import math
 import warnings
 from abc import ABC, abstractmethod
 from functools import partial
-
 from typing import (
     Any,
     Callable,
@@ -22,8 +21,6 @@ from typing import (
     TypeVar,
 )
 
-from pydantic import Field, root_validator
-
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForRetrieverRun,
     CallbackManagerForRetrieverRun,
@@ -31,6 +28,7 @@ from langchain.callbacks.manager import (
 from langchain.docstore.document import Document
 from langchain.embeddings.base import Embeddings
 from langchain.schema import BaseRetriever
+from pydantic import Field, root_validator
 
 VST = TypeVar("VST", bound="VectorStore")
 

swarms/agents/prompts/chains/llm.py

@@ -4,8 +4,7 @@ from __future__ import annotations
 import warnings
 from typing import Any, Dict, List, Optional, Sequence, Tuple, Union
 
-from pydantic import Extra, Field
+from langchain.base_language import BaseLanguageModel
-
 from langchain.callbacks.manager import (
     AsyncCallbackManager,
     AsyncCallbackManagerForChainRun,
@@ -24,7 +23,7 @@ from langchain.schema import (
     NoOpOutputParser,
     PromptValue,
 )
-from langchain.base_language import BaseLanguageModel
+from pydantic import Extra, Field
 
 
 class LLMChain(Chain):