PraisonAI 0.0.59rc7__cp312-cp312-manylinux_2_35_x86_64.whl → 0.0.59rc9__cp312-cp312-manylinux_2_35_x86_64.whl

praisonai/cli.py

@@ -29,12 +29,13 @@ try:
 except ImportError:
     GRADIO_AVAILABLE = False
     
-def stream_subprocess(command):
+def stream_subprocess(command, env=None):
     """
     Execute a subprocess command and stream the output to the terminal in real-time.
 
     Args:
         command (list): A list containing the command and its arguments.
+        env (dict, optional): Environment variables for the subprocess.
     """
     process = subprocess.Popen(
         command,
@@ -42,7 +43,8 @@ def stream_subprocess(command):
         stderr=subprocess.STDOUT,
         text=True,
         bufsize=1,
-        universal_newlines=True
+        universal_newlines=True,
+        env=env
     )
 
     for line in iter(process.stdout.readline, ''):
@@ -148,19 +150,24 @@ class PraisonAI:
 
             try:
                 result = subprocess.check_output(['conda', 'env', 'list'])
-                if 'unsloth_env' in result.decode('utf-8'):
-                    print("Conda environment 'unsloth_env' found.")
+                if 'prasion_env' in result.decode('utf-8'):
+                    print("Conda environment 'prasion_env' found.")
                 else:
                     raise subprocess.CalledProcessError(1, 'grep')
             except subprocess.CalledProcessError:
-                print("Conda environment 'unsloth_env' not found. Setting it up...")
+                print("Conda environment 'prasion_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_script_path = os.path.join(package_root, 'train.py')
-            stream_subprocess(['conda', 'run', '--name', 'unsloth_env', 'python', train_script_path, 'train'])
+            
+            # Set environment variables
+            env = os.environ.copy()
+            env['PYTHONUNBUFFERED'] = '1'
+            
+            stream_subprocess(['conda', 'run', '--no-capture-output', '--name', 'prasion_env', 'python', '-u', train_script_path, 'train'] + train_args, env=env)
             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.59rc7 gunicorn markdown\n")
+            file.write("RUN pip install flask praisonai==0.0.59rc9 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

@@ -51,8 +51,6 @@ 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

@@ -33,7 +33,7 @@ else
 fi
 
 # Create and activate the Conda environment
-ENV_NAME="unsloth_env"
+ENV_NAME="prasion_env"
 if conda info --envs | grep -q $ENV_NAME; then
     echo "Environment $ENV_NAME already exists. Recreating..."
     conda env remove -y -n $ENV_NAME  # Remove existing environment

praisonai/train.py

@@ -3,6 +3,7 @@ import os
 import sys
 import yaml
 import torch
+import shutil
 from transformers import TextStreamer
 from unsloth import FastLanguageModel, is_bfloat16_supported
 from trl import SFTTrainer
@@ -178,7 +179,7 @@ class train:
 
     def prepare_modelfile_content(self):
         output_model = self.config["hf_model_name"]
-        return f"""FROM {output_model}/unsloth.Q5_K_M.gguf
+        return f"""FROM {output_model}/unsloth.Q4_K_M.gguf
 
 TEMPLATE \"\"\"Below are some instructions that describe some tasks. Write responses that appropriately complete each request.{{{{ if .Prompt }}}}
 

{praisonai-0.0.59rc7.dist-info → praisonai-0.0.59rc9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PraisonAI
-Version: 0.0.59rc7
+Version: 0.0.59rc9
 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.59rc7.dist-info → praisonai-0.0.59rc9.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=x6WfPbF-0vuFgehkmLlvnk2I95KZ_vz8bzpcQob4JVQ,17117
-praisonai/deploy.py,sha256=BOEhEODsStphfVLER6tfryOzkmZbb5XuKGUCsJWrBb0,6031
+praisonai/cli.py,sha256=WQf6NG1xEcyE9ySR4ahSiVDxCkedGV6A0Yaf78uYXQ0,17411
+praisonai/deploy.py,sha256=_jT6RcTHzxeWKtizSyaGmfE48FKbvw___CHn6jOuWnc,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,12 +23,12 @@ 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/config.yaml,sha256=DbfEJyvWTDACScH9XquGGsNYcyudpEwpawd_rgaPN98,1138
 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=grRG_FGYHAYMnHNyQrcZxVxSSLghSWUQuW4DHdIKmPU,2718
+praisonai/setup/setup_conda_env.sh,sha256=Ao5kCgIVamCcCKhrXWYELTUr5fQlRQmWwDYjuTxBH6s,2718
 praisonai/test.py,sha256=OL-wesjA5JTohr8rtr6kWoaS4ImkJg2l0GXJ-dUUfRU,4090
-praisonai/train.py,sha256=Kuckd3TjW6ZYTISyXpL0upB-NibsEmwk1BMhRHPXba0,9200
+praisonai/train.py,sha256=M2rpBca3k4uQvmoKh-0r_2MTlWLYL7ffKMtlU8hF7k8,9214
 praisonai/ui/chat.py,sha256=B4F1R7qP-0c-elg8WcRsYlr6-FkmHWtdunGIzU7WrDM,9321
 praisonai/ui/code.py,sha256=GcOr8lNah4AgI2RcIKmgjehzSl-KNu7x6UHrghixeaM,10095
 praisonai/ui/context.py,sha256=oWO2I_WBZb7kZnuXItf18EJX0ZQv-1nAd8rxhwhuuDU,11871
@@ -40,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.59rc7.dist-info/LICENSE,sha256=kqvFysVlnFxYOu0HxCe2HlmZmJtdmNGOxWRRkT9TsWc,1035
-praisonai-0.0.59rc7.dist-info/METADATA,sha256=hkzIsYz9KrsGHpblhJoyxfaWiCs_OGCzz8gtm_WOaKc,11151
-praisonai-0.0.59rc7.dist-info/WHEEL,sha256=HBsDV7Hj4OTiS1GX6ua7iQXUQTB9UHftbBxr7Q8Xm9c,110
-praisonai-0.0.59rc7.dist-info/entry_points.txt,sha256=jB078LEGLY3Ky_indhclomRIVVpXrPSksHjJ-tcBZ-o,133
-praisonai-0.0.59rc7.dist-info/RECORD,,
+praisonai-0.0.59rc9.dist-info/LICENSE,sha256=kqvFysVlnFxYOu0HxCe2HlmZmJtdmNGOxWRRkT9TsWc,1035
+praisonai-0.0.59rc9.dist-info/METADATA,sha256=UwVe6VMWXUoygLpeg695CvLu9vK_ikjQ0L-eZ1AE7RM,11151
+praisonai-0.0.59rc9.dist-info/WHEEL,sha256=HBsDV7Hj4OTiS1GX6ua7iQXUQTB9UHftbBxr7Q8Xm9c,110
+praisonai-0.0.59rc9.dist-info/entry_points.txt,sha256=jB078LEGLY3Ky_indhclomRIVVpXrPSksHjJ-tcBZ-o,133
+praisonai-0.0.59rc9.dist-info/RECORD,,