PraisonAI 0.0.59rc6__tar.gz → 0.0.59rc8__tar.gz

{praisonai-0.0.59rc6 → praisonai-0.0.59rc8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PraisonAI
-Version: 0.0.59rc6
+Version: 0.0.59rc8
 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.59rc6 → praisonai-0.0.59rc8}/praisonai/cli.py

@@ -13,6 +13,7 @@ from .auto import AutoGenerator
 from .agents_generator import AgentsGenerator
 from .inbuilt_tools import *
 import shutil
+import subprocess
 import logging
 logging.basicConfig(level=os.environ.get('LOGLEVEL', 'INFO'), format='%(asctime)s - %(levelname)s - %(message)s')
 
@@ -27,6 +28,34 @@ try:
     GRADIO_AVAILABLE = True
 except ImportError:
     GRADIO_AVAILABLE = False
+    
+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,
+        stdout=subprocess.PIPE,
+        stderr=subprocess.STDOUT,
+        text=True,
+        bufsize=1,
+        universal_newlines=True,
+        env=env
+    )
+
+    for line in iter(process.stdout.readline, ''):
+        print(line, end='')
+        sys.stdout.flush()  # Ensure output is flushed immediately
+
+    process.stdout.close()
+    return_code = process.wait()
+
+    if return_code != 0:
+        raise subprocess.CalledProcessError(return_code, command)
 
 class PraisonAI:
     def __init__(self, agent_file="agents.yaml", framework="", auto=False, init=False, agent_yaml=None):
@@ -100,11 +129,10 @@ 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)
@@ -113,29 +141,33 @@ class PraisonAI:
                     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()
                 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.")
+                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')
             
+            # Set environment variables
+            env = os.environ.copy()
+            env['PYTHONUNBUFFERED'] = '1'
             
-            train_args = sys.argv[2:]  # Get all arguments after 'train' 
-            train_script_path = os.path.join(package_root, 'train.py')
-            subprocess.check_call(['conda', 'run', '--name', 'unsloth_env', 'python', train_script_path])
+            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-0.0.59rc6 → praisonai-0.0.59rc8}/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.59rc6 gunicorn markdown\n")
+            file.write("RUN pip install flask praisonai==0.0.59rc8 gunicorn markdown\n")
             file.write("EXPOSE 8080\n")
             file.write('CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]\n')
             

{praisonai-0.0.59rc6 → praisonai-0.0.59rc8}/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-0.0.59rc6 → praisonai-0.0.59rc8}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "PraisonAI"
-version = "0.0.59rc6"
+version = "0.0.59rc8"
 description = "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."
 authors = ["Mervin Praison"]
 license = ""