[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
-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

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