PraisonAI 0.0.59rc5__cp312-cp312-manylinux_2_35_x86_64.whl → 0.0.59rc6__cp312-cp312-manylinux_2_35_x86_64.whl

praisonai/cli.py

@@ -100,12 +100,42 @@ class PraisonAI:
             return
         
         if args.agent_file == 'train':
+            import subprocess
+            package_root = os.path.dirname(os.path.abspath(__file__))
+            config_yaml_source = os.path.join(package_root, 'setup', 'config.yaml')
+            config_yaml_destination = os.path.join(os.getcwd(), 'config.yaml')
+            
+            if not os.path.exists(config_yaml_destination):
+                try:
+                    shutil.copyfile(config_yaml_source, config_yaml_destination)
+                    print("config.yaml copied to the current directory.")
+                except FileExistsError:
+                    print("config.yaml already exists in the current directory. Skipping copy.")
+            else:
+                print("config.yaml already exists in the current directory. Skipping copy.")
+                
             if 'init' in sys.argv:
                 from praisonai.setup.setup_conda_env import main as setup_conda_main
                 setup_conda_main()
-            from .train import main as train_main
+                print("All packages installed")
+                return
+            
+            try:
+                result = subprocess.check_output(['conda', 'env', 'list'])
+                if 'unsloth_env' in result.decode('utf-8'):
+                    print("Conda environment 'unsloth_env' found.")
+                else:
+                    raise subprocess.CalledProcessError(1, 'grep')
+            except subprocess.CalledProcessError:
+                print("Conda environment 'unsloth_env' not found. Setting it up...")
+                from praisonai.setup.setup_conda_env import main as setup_conda_main
+                setup_conda_main()
+                print("All packages installed.")
+            
+            
             train_args = sys.argv[2:]  # Get all arguments after 'train' 
-            train_main()
+            train_script_path = os.path.join(package_root, 'train.py')
+            subprocess.check_call(['conda', 'run', '--name', 'unsloth_env', 'python', train_script_path])
             return
         
         invocation_cmd = "praisonai"

praisonai/deploy.py

@@ -56,7 +56,7 @@ class CloudDeployer:
             file.write("FROM python:3.11-slim\n")
             file.write("WORKDIR /app\n")
             file.write("COPY . .\n")
-            file.write("RUN pip install flask praisonai==0.0.59rc5 gunicorn markdown\n")
+            file.write("RUN pip install flask praisonai==0.0.59rc6 gunicorn markdown\n")
             file.write("EXPOSE 8080\n")
             file.write('CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]\n')
             

praisonai/setup/config.yaml

@@ -0,0 +1,58 @@
+model_name: "unsloth/Meta-Llama-3.1-8B-Instruct-bnb-4bit"
+hf_model_name: "mervinpraison/llama-3.1-tamilan-8B-test"
+max_seq_length: 2048
+load_in_4bit: true
+lora_r: 16
+lora_target_modules: 
+  - "q_proj"
+  - "k_proj"
+  - "v_proj"
+  - "o_proj"
+  - "gate_proj"
+  - "up_proj"
+  - "down_proj"
+lora_alpha: 16
+lora_dropout: 0
+lora_bias: "none"
+use_gradient_checkpointing: "unsloth"
+random_state: 3407
+use_rslora: false
+loftq_config: null
+
+dataset:
+  - name: "yahma/alpaca-cleaned"
+    split_type: "train"
+    processing_func: "format_prompts"
+    rename:
+      input: "input"
+      output: "output"
+      instruction: "instruction"
+    filter_data: false
+    filter_column_value: "id"
+    filter_value: "alpaca"
+    num_samples: 20000
+
+dataset_text_field: "text"
+dataset_num_proc: 2
+packing: false
+
+per_device_train_batch_size: 2
+gradient_accumulation_steps: 2
+warmup_steps: 5
+num_train_epochs: 1
+max_steps: 10
+learning_rate: 2.0e-4
+logging_steps: 1
+optim: "adamw_8bit"
+weight_decay: 0.01
+lr_scheduler_type: "linear"
+seed: 3407
+output_dir: "outputs"
+
+quantization_method: 
+  - "q4_k_m"
+  - "q8_0"
+  - "q5_k_m"
+
+ollama_model: "llama3.1-tamilan-test"
+model_parameters: "8b"

praisonai/setup/setup_conda_env.sh

@@ -35,8 +35,16 @@ fi
 # Create and activate the Conda environment
 ENV_NAME="unsloth_env"
 if conda info --envs | grep -q $ENV_NAME; then
-    echo "Environment $ENV_NAME already exists. Activating..."
-    conda activate $ENV_NAME
+    echo "Environment $ENV_NAME already exists. Recreating..."
+    conda env remove -y -n $ENV_NAME  # Remove existing environment
+    if [[ "$OSTYPE" == "darwin"* ]]; then
+        # macOS (both Intel and M1/M2)
+        conda create --name $ENV_NAME python=3.10 pytorch=2.3.0 -c pytorch -y
+    elif [[ "$OSTYPE" == "linux-gnu"* ]]; then
+        # Linux
+        conda create --name $ENV_NAME python=3.10 pytorch=2.3.0 cudatoolkit=11.8 -c pytorch -c nvidia -y
+    fi
+    # conda activate $ENV_NAME
 else
     echo "Creating new environment $ENV_NAME..."
     if [[ "$OSTYPE" == "darwin"* ]]; then
@@ -45,16 +53,20 @@ else
     elif [[ "$OSTYPE" == "linux-gnu"* ]]; then
         # Linux
         conda create --name $ENV_NAME python=3.10 pytorch=2.3.0 cudatoolkit=11.8 -c pytorch -c nvidia -y
-        conda activate $ENV_NAME
-        pip install xformers==0.0.26.post1
     fi
+    # conda activate $ENV_NAME
 fi
 
-source $HOME/miniconda/bin/activate $ENV_NAME
+# source $HOME/miniconda/bin/activate $ENV_NAME
+
+# Get full path of pip
+PIP_FULL_PATH=$(conda run -n $ENV_NAME which pip)
 
-# Install other packages
-pip install --upgrade pip
-pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git@4e570be9ae4ced8cdc64e498125708e34942befc"
-pip install --no-deps "trl<0.9.0" peft accelerate bitsandbytes
+# Install other packages within the activated environment
+# Use PIP_FULL_PATH to run pip commands
+$PIP_FULL_PATH install --upgrade pip 
+$PIP_FULL_PATH install "xformers==0.0.26.post1"
+$PIP_FULL_PATH install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git@4e570be9ae4ced8cdc64e498125708e34942befc"
+$PIP_FULL_PATH install --no-deps "trl<0.9.0" peft accelerate bitsandbytes 
 
-echo "Setup completed successfully!"
+echo "Setup completed successfully!"

{praisonai-0.0.59rc5.dist-info → praisonai-0.0.59rc6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PraisonAI
-Version: 0.0.59rc5
+Version: 0.0.59rc6
 Summary: PraisonAI application combines AutoGen and CrewAI or similar frameworks into a low-code solution for building and managing multi-agent LLM systems, focusing on simplicity, customization, and efficient human-agent collaboration.
 Author: Mervin Praison
 Requires-Python: >=3.10,<3.13

{praisonai-0.0.59rc5.dist-info → praisonai-0.0.59rc6.dist-info}/RECORD

@@ -3,8 +3,8 @@ praisonai/__main__.py,sha256=MVgsjMThjBexHt4nhd760JCqvP4x0IQcwo8kULOK4FQ,144
 praisonai/agents_generator.py,sha256=8d1WRbubvEkBrW1HZ7_xnGyqgJi0yxmXa3MgTIqef1c,19127
 praisonai/auto.py,sha256=9spTXqj47Hmmqv5QHRYE_RzSVHH_KoPbaZjskUj2UcE,7895
 praisonai/chainlit_ui.py,sha256=bNR7s509lp0I9JlJNvwCZRUZosC64qdvlFCt8NmFamQ,12216
-praisonai/cli.py,sha256=Ff4FyjfAh_QcL7wMCiW4RxKNOgqEorsNtR5zvBtuGvA,14875
-praisonai/deploy.py,sha256=E2YIHDRQ8nb6nQiSghTXxxskynM_2g1-ms9jhmekDt8,6031
+praisonai/cli.py,sha256=C8RwY-r1-86UQ4w9Z2dQj9fhMR0WJ56r6KFyBB5-c3U,16477
+praisonai/deploy.py,sha256=S5qpzGQkevxEiYjfqfDoV7H620qsnSpfF6dgjF-pgyM,6031
 praisonai/inbuilt_tools/__init__.py,sha256=mUKnbL6Gram9c9f2m8wJwEzURBLmPEOcHzwySBH89YA,74
 praisonai/inbuilt_tools/autogen_tools.py,sha256=svYkM2N7DVFvbiwgoAS7U_MqTOD8rHf8VD3BaFUV5_Y,14907
 praisonai/inc/__init__.py,sha256=sPDlYBBwdk0VlWzaaM_lG0_LD07lS2HRGvPdxXJFiYg,62
@@ -23,9 +23,10 @@ praisonai/public/movie.svg,sha256=aJ2EQ8vXZusVsF2SeuAVxP4RFJzQ14T26ejrGYdBgzk,12
 praisonai/public/thriller.svg,sha256=2dYY72EcgbEyTxS4QzjAm37Y4srtPWEW4vCMFki98ZI,3163
 praisonai/setup/__init__.py  ,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 praisonai/setup/build.py,sha256=NyTAXQ_UZ8vKo_KwCINp8ctmauZyCMDkw1rys3ay0ec,646
+praisonai/setup/config.yaml,sha256=t2WtEDT4JE4yVMAg7hKD18XER3qz41-V36HHYp07yEo,1162
 praisonai/setup/post_install.py,sha256=hXukn_7bL64vE582SZcS-9MiZGeJj6hN7upoR1oJ-Bo,576
 praisonai/setup/setup_conda_env.py,sha256=4QiWrqgEObivzOMwfJgWaCPpUEpB68cQ6lFwVwFoufk,816
-praisonai/setup/setup_conda_env.sh,sha256=akFPm3hZq_b6C17zLhLD22owJruzdQYgCqjJmDPr7kE,2126
+praisonai/setup/setup_conda_env.sh,sha256=grRG_FGYHAYMnHNyQrcZxVxSSLghSWUQuW4DHdIKmPU,2718
 praisonai/test.py,sha256=OL-wesjA5JTohr8rtr6kWoaS4ImkJg2l0GXJ-dUUfRU,4090
 praisonai/train.py,sha256=Kuckd3TjW6ZYTISyXpL0upB-NibsEmwk1BMhRHPXba0,9200
 praisonai/ui/chat.py,sha256=B4F1R7qP-0c-elg8WcRsYlr6-FkmHWtdunGIzU7WrDM,9321
@@ -39,8 +40,8 @@ praisonai/ui/public/movie.svg,sha256=aJ2EQ8vXZusVsF2SeuAVxP4RFJzQ14T26ejrGYdBgzk
 praisonai/ui/public/thriller.svg,sha256=2dYY72EcgbEyTxS4QzjAm37Y4srtPWEW4vCMFki98ZI,3163
 praisonai/ui/sql_alchemy.py,sha256=HsyeRq-G9qbQobHWpTJHHKQiT4FvYw_7iuv-2PNh0IU,27419
 praisonai/version.py,sha256=ugyuFliEqtAwQmH4sTlc16YXKYbFWDmfyk87fErB8-8,21
-praisonai-0.0.59rc5.dist-info/LICENSE,sha256=kqvFysVlnFxYOu0HxCe2HlmZmJtdmNGOxWRRkT9TsWc,1035
-praisonai-0.0.59rc5.dist-info/METADATA,sha256=K_WXer8aL_M5VM8eQn-YWkRC0Re1BMMjb21kh3xJ9Ts,11151
-praisonai-0.0.59rc5.dist-info/WHEEL,sha256=HBsDV7Hj4OTiS1GX6ua7iQXUQTB9UHftbBxr7Q8Xm9c,110
-praisonai-0.0.59rc5.dist-info/entry_points.txt,sha256=jB078LEGLY3Ky_indhclomRIVVpXrPSksHjJ-tcBZ-o,133
-praisonai-0.0.59rc5.dist-info/RECORD,,
+praisonai-0.0.59rc6.dist-info/LICENSE,sha256=kqvFysVlnFxYOu0HxCe2HlmZmJtdmNGOxWRRkT9TsWc,1035
+praisonai-0.0.59rc6.dist-info/METADATA,sha256=vNj7yWobbpL8E-o0faxY1SXp-ruDlybRgRag5e1Hb_Q,11151
+praisonai-0.0.59rc6.dist-info/WHEEL,sha256=HBsDV7Hj4OTiS1GX6ua7iQXUQTB9UHftbBxr7Q8Xm9c,110
+praisonai-0.0.59rc6.dist-info/entry_points.txt,sha256=jB078LEGLY3Ky_indhclomRIVVpXrPSksHjJ-tcBZ-o,133
+praisonai-0.0.59rc6.dist-info/RECORD,,