diff --git a/example.py b/example.py index 4ee4e7c4..cea307b0 100644 --- a/example.py +++ b/example.py @@ -20,7 +20,6 @@ llm = OpenAIChat( max_tokens=1000, ) - ## Initialize the workflow agent = Agent( llm=llm, diff --git a/pyproject.toml b/pyproject.toml index ef30e3c4..1b30a083 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api" [tool.poetry] name = "swarms" -version = "2.8.0" +version = "2.8.1" description = "Swarms - Pytorch" license = "MIT" authors = ["Kye Gomez "] diff --git a/requirements.txt b/requirements.txt index e33a7438..f020827a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,73 +1,71 @@ torch==2.1.1 -transformers>2.10 -pandas -langchain -nest_asyncio -langchain-experimental -playwright +transformers>2.10==4.35.0 +pandas==1.5.3 +langchain==0.0.333 +nest_asyncio==1.5.6 +langchain-experimental==0.0.10 +playwright==1.34.0 wget==3.2 -simpleaichat -httpx -open_clip_torch -ggl -beautifulsoup4 -google-search-results==2.4.2 -Pillow -faiss-cpu +simpleaichat==0.2.2 +httpx==0.24.1 +open_clip_torch==2.20.0 +ggl==1.1.0 +beautifulsoup4==4.11.2 +google-search-results +Pillow==9.4.0 +faiss-cpu==1.7.4 openai==0.28.0 -attrs -datasets +attrs==22.2.0 +datasets==2.10.1 pydantic==1.10.12 -soundfile +soundfile==0.12.1 arize-phoenix -weaviate-client -huggingface-hub -google-generativeai -sentencepiece -PyPDF2 -accelerate +weaviate-client==3.25.3 +huggingface-hub==0.16.4 +google-generativeai==0.3.1 +sentencepiece==0.1.98 +PyPDF2==3.0.1 +accelerate==0.22.0 vllm -chromadb -tensorflow +chromadb==0.4.14 +tensorflow==2.12.0 optimum -tiktoken -tabulate +tiktoken==0.4.0 +tabulate==0.9.0 colored addict -backoff -ratelimit +backoff==2.2.1 +ratelimit==2.2.1 albumentations basicsr -termcolor +termcolor==2.2.0 controlnet-aux -diffusers -einops -imageio -opencv-python-headless -imageio-ffmpeg +diffusers==0.17.1 +einops==0.7.0 +imageio==2.25.1 +opencv-python-headless==4.8.1.78 +imageio-ffmpeg==0.4.9 invisible-watermark kornia -safetensors -numpy -omegaconf -open_clip_torch -openai -opencv-python -prettytable -safetensors +safetensors==0.3.3 +numpy==1.25.2 +omegaconf==2.3.0 +open_clip_torch==2.20.0 +openai==0.28.0 +opencv-python==4.7.0.72 +prettytable==3.9.0 +safetensors==0.3.3 test-tube -timm +timm==0.6.13 torchmetrics webdataset -marshmallow +marshmallow==3.19.0 yapf autopep8 -cohere -torchvision -rich - - +cohere==4.24 +torchvision==0.16.1 +rich==13.5.2 mkdocs mkdocs-material mkdocs-glightbox -pre-commit +pre-commit==3.2.2 diff --git a/scripts/get_package_requirements.py b/scripts/get_package_requirements.py new file mode 100644 index 00000000..e6cd08c7 --- /dev/null +++ b/scripts/get_package_requirements.py @@ -0,0 +1,32 @@ +import pkg_resources + +def get_package_versions(requirements_path, output_path): + try: + with open(requirements_path, 'r') as file: + requirements = file.readlines() + except FileNotFoundError: + print(f"Error: The file '{requirements_path}' was not found.") + return + + package_versions = [] + + for requirement in requirements: + # Skip empty lines and comments + if requirement.strip() == '' or requirement.strip().startswith('#'): + continue + + # Extract package name + package_name = requirement.split('==')[0].strip() + try: + version = pkg_resources.get_distribution(package_name).version + package_versions.append(f"{package_name}=={version}") + except pkg_resources.DistributionNotFound: + package_versions.append(f"{package_name}: not installed") + + with open(output_path, 'w') as file: + for package_version in package_versions: + file.write(package_version + '\n') + print(f"Versions written to {output_path}") + +# Usage +get_package_versions('requirements.txt', 'installed_versions.txt')