[UPDATED requirements.txt with verison numbers]

example.py

@@ -20,7 +20,6 @@ llm = OpenAIChat(
     max_tokens=1000,
 )
 
-
 ## Initialize the workflow
 agent = Agent(
     llm=llm,

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 <kye@apac.ai>"]

requirements.txt

@@ -1,73 +1,71 @@
 torch==2.1.1
-transformers>2.10
+transformers>2.10==4.35.0
-pandas
+pandas==1.5.3
-langchain
+langchain==0.0.333
-nest_asyncio
+nest_asyncio==1.5.6
-langchain-experimental
+langchain-experimental==0.0.10
-playwright
+playwright==1.34.0
 wget==3.2
-simpleaichat
+simpleaichat==0.2.2
-httpx
+httpx==0.24.1
-open_clip_torch
+open_clip_torch==2.20.0
-ggl
+ggl==1.1.0
-beautifulsoup4
+beautifulsoup4==4.11.2
-google-search-results==2.4.2
+google-search-results
-Pillow
+Pillow==9.4.0
-faiss-cpu
+faiss-cpu==1.7.4
 openai==0.28.0
-attrs
+attrs==22.2.0
-datasets
+datasets==2.10.1
 pydantic==1.10.12
-soundfile
+soundfile==0.12.1
 arize-phoenix
-weaviate-client 
+weaviate-client==3.25.3
-huggingface-hub
+huggingface-hub==0.16.4
-google-generativeai
+google-generativeai==0.3.1
-sentencepiece
+sentencepiece==0.1.98
-PyPDF2
+PyPDF2==3.0.1
-accelerate
+accelerate==0.22.0
 vllm
-chromadb
+chromadb==0.4.14
-tensorflow
+tensorflow==2.12.0
 optimum
-tiktoken
+tiktoken==0.4.0
-tabulate 
+tabulate==0.9.0
 colored
 addict
-backoff
+backoff==2.2.1
-ratelimit
+ratelimit==2.2.1
 albumentations
 basicsr
-termcolor
+termcolor==2.2.0
 controlnet-aux
-diffusers
+diffusers==0.17.1
-einops
+einops==0.7.0
-imageio
+imageio==2.25.1
-opencv-python-headless
+opencv-python-headless==4.8.1.78
-imageio-ffmpeg
+imageio-ffmpeg==0.4.9
 invisible-watermark
 kornia
-safetensors
+safetensors==0.3.3
-numpy
+numpy==1.25.2
-omegaconf
+omegaconf==2.3.0
-open_clip_torch
+open_clip_torch==2.20.0
-openai
+openai==0.28.0
-opencv-python
+opencv-python==4.7.0.72
-prettytable
+prettytable==3.9.0
-safetensors
+safetensors==0.3.3
 test-tube
-timm
+timm==0.6.13
 torchmetrics
 webdataset
-marshmallow
+marshmallow==3.19.0
 yapf
 autopep8
-cohere
+cohere==4.24
-torchvision
+torchvision==0.16.1
-rich
+rich==13.5.2
-
-
 mkdocs
 mkdocs-material
 mkdocs-glightbox
-pre-commit
+pre-commit==3.2.2

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')