From d2136a7157f4246f7d1bef82cb938718f31ecf1f Mon Sep 17 00:00:00 2001 From: Kye Date: Thu, 27 Jul 2023 21:32:06 -0400 Subject: [PATCH] clean up Former-commit-id: e9ee3055e8992d434cfcb44ee21e64d6ff640822 --- setup.py | 2 +- swarms/agents/memory/base.py | 4 +--- swarms/agents/prompts/chains/llm.py | 5 ++--- 3 files changed, 4 insertions(+), 7 deletions(-) diff --git a/setup.py b/setup.py index 30f97a65..dcfd92de 100644 --- a/setup.py +++ b/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', diff --git a/swarms/agents/memory/base.py b/swarms/agents/memory/base.py index 4b14a1af..d892eee2 100644 --- a/swarms/agents/memory/base.py +++ b/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") diff --git a/swarms/agents/prompts/chains/llm.py b/swarms/agents/prompts/chains/llm.py index b9a5fea3..4326bc79 100644 --- a/swarms/agents/prompts/chains/llm.py +++ b/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):