PraisonAI 2.0.65__cp313-cp313-manylinux_2_39_x86_64.whl → 2.0.67__cp313-cp313-manylinux_2_39_x86_64.whl

praisonai/cli.py

@@ -25,7 +25,6 @@ CALL_MODULE_AVAILABLE = False
 CREWAI_AVAILABLE = False
 AUTOGEN_AVAILABLE = False
 PRAISONAI_AVAILABLE = False
-TRAIN_AVAILABLE = False
 try:
     # Create necessary directories and set CHAINLIT_APP_ROOT
     if "CHAINLIT_APP_ROOT" not in os.environ:
@@ -72,12 +71,6 @@ try:
 except ImportError:
     pass
 
-try:
-    import accelerate
-    TRAIN_AVAILABLE = True
-except ImportError:
-    pass
-
 logging.basicConfig(level=os.environ.get('LOGLEVEL', 'INFO'), format='%(asctime)s - %(levelname)s - %(message)s')
 logging.getLogger('alembic').setLevel(logging.ERROR)
 logging.getLogger('gradio').setLevel(logging.ERROR)
@@ -399,13 +392,8 @@ class PraisonAI:
                 sys.exit(0)
 
             elif args.command == 'train':
-                if not TRAIN_AVAILABLE:
-                    print("[red]ERROR: Train feature is not installed. Install with:[/red]")
-                    print("\npip install \"praisonai[train]\"\n")
-                    sys.exit(1)
                 package_root = os.path.dirname(os.path.abspath(__file__))
                 config_yaml_destination = os.path.join(os.getcwd(), 'config.yaml')
-                
 
             elif args.command == 'ui':
                 if not CHAINLIT_AVAILABLE:

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==2.0.65 gunicorn markdown\n")
+            file.write("RUN pip install flask praisonai==2.0.67 gunicorn markdown\n")
             file.write("EXPOSE 8080\n")
             file.write('CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]\n')
             

praisonai/train.py

@@ -312,26 +312,41 @@ class TrainModel:
 
     def prepare_modelfile_content(self):
         output_model = self.config["hf_model_name"]
+        # Determine stop tokens from config or infer based on model name
+        model_name = self.config["model_name"].lower()
+        if "phi" in model_name:
+            inferred_stop_tokens = ["<|end|>", "<|user|>", "<|assistant|>"]
+        elif "llava" in model_name:
+            inferred_stop_tokens = ["</s>", "USER:", "ASSSISTANT:"]
+        elif "mistral" in model_name:
+            inferred_stop_tokens = ["[INST]", "[/INST]"]
+        elif "qwen" in model_name:
+            inferred_stop_tokens = ["<|endoftext|>"]
+        elif "deepseek" in model_name:
+            inferred_stop_tokens = ["<｜begin▁of▁sentence｜>", "<｜end▁of▁sentence｜>", "<｜User｜>", "<｜Assistant｜>"]
+        else:
+            inferred_stop_tokens = ["<|start_header_id|>", "<|end_header_id|>", "<|eot_id|>"]
+        # Use stop_tokens from config if provided, otherwise use inferred
+        model_stop_tokens = self.config.get("stop_tokens", inferred_stop_tokens)
+
         gguf_path = f"{output_model}/unsloth.Q4_K_M.gguf"
         if not os.path.exists(gguf_path):
             self.model, self.hf_tokenizer = self.load_model()
             self.save_model_gguf()
+        stop_parameters = "\n".join([f'PARAMETER stop "{token}"' for token in model_stop_tokens])
         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 }}}}
+    TEMPLATE \"\"\"Below are some instructions that describe some tasks. Write responses that appropriately complete each request.{{{{ if .Prompt }}}}
 
-### Instruction:
-{{{{ .Prompt }}}}
+    ### Instruction:
+    {{{{ .Prompt }}}}
 
-{{{{ end }}}}### Response:
-{{{{ .Response }}}}\"\"\"
+    {{{{ end }}}}### Response:
+    {{{{ .Response }}}}\"\"\"
+
+    {stop_parameters}
+    """
 
-PARAMETER stop ""
-PARAMETER stop ""
-PARAMETER stop ""
-PARAMETER stop ""
-PARAMETER stop "<|reserved_special_token_"
-"""
 
     def create_and_push_ollama_model(self):
         modelfile_content = self.prepare_modelfile_content()

{praisonai-2.0.65.dist-info → praisonai-2.0.67.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: PraisonAI
-Version: 2.0.65
+Version: 2.0.67
 Summary: PraisonAI is an AI Agents Framework with Self Reflection. PraisonAI application combines PraisonAI Agents, AutoGen, and CrewAI into a low-code solution for building and managing multi-agent LLM systems, focusing on simplicity, customisation, and efficient human-agent collaboration.
 Author: Mervin Praison
 Requires-Python: >=3.10,<3.13

{praisonai-2.0.65.dist-info → praisonai-2.0.67.dist-info}/RECORD

@@ -4,8 +4,8 @@ praisonai/agents_generator.py,sha256=j8lYudAr3wlVBQLng3iYL6mfRqx2i9M6wlryxIVRzDA
 praisonai/api/call.py,sha256=krOfTCZM_bdbsNuWQ1PijzCHECkDvEi9jIvvZaDQUUU,11035
 praisonai/auto.py,sha256=uLDm8CU3L_3amZsd55yzf9RdBF1uW-BGSx7nl9ctNZ4,8680
 praisonai/chainlit_ui.py,sha256=bNR7s509lp0I9JlJNvwCZRUZosC64qdvlFCt8NmFamQ,12216
-praisonai/cli.py,sha256=C4MoIXaLHldRNYPqXpHPpD8DjJU2qUJqZMHzbem7te4,25254
-praisonai/deploy.py,sha256=VOFYtqPTNJ25WrgRTI8wAqlS2DL5SjJGNDZp8kEsan8,6028
+praisonai/cli.py,sha256=wKf17OzAmDhCWXtP_HDF4UrULCiWV1CI5yTEtCPUi28,24898
+praisonai/deploy.py,sha256=Txysvn5rP7d-pqpUoIvGGSJtFHeleQU1nbcUrSP5Hlc,6028
 praisonai/inbuilt_tools/__init__.py,sha256=fai4ZJIKz7-iOnGZv5jJX0wmT77PKa4x2jqyaJddKFA,569
 praisonai/inbuilt_tools/autogen_tools.py,sha256=kJdEv61BTYvdHOaURNEpBcWq8Rs-oC03loNFTIjT-ak,4687
 praisonai/inc/__init__.py,sha256=sPDlYBBwdk0VlWzaaM_lG0_LD07lS2HRGvPdxXJFiYg,62
@@ -33,7 +33,7 @@ praisonai/setup/setup_conda_env.py,sha256=4QiWrqgEObivzOMwfJgWaCPpUEpB68cQ6lFwVw
 praisonai/setup/setup_conda_env.sh,sha256=aD5Tb_T4UeVtJD5ibYTBB50YWCKh7INdiei2LN1fBig,4081
 praisonai/setup.py,sha256=0jHgKnIPCtBZiGYaYyTz3PzrJI6nBy55VXk2UctXlDo,373
 praisonai/test.py,sha256=OL-wesjA5JTohr8rtr6kWoaS4ImkJg2l0GXJ-dUUfRU,4090
-praisonai/train.py,sha256=m3-N2XWRernBWRg_xRr7bilxhq7WvFJYwxfvjROIjEw,15541
+praisonai/train.py,sha256=5ZR3FAZELIwe8Fv2Rt4ID3bjVvuVQEd5fWZ-3n-KUpo,16534
 praisonai/ui/README.md,sha256=QG9yucvBieVjCjWFzu6hL9xNtYllkoqyJ_q1b0YYAco,1124
 praisonai/ui/agents.py,sha256=1qsWE2yCaQKhuc-1uLHdMfZJeOXzBtp4pe5q7bk2EuA,32813
 praisonai/ui/callbacks.py,sha256=V4_-GjxmjDFmugUZGfQHKtNSysx7rT6i1UblbM_8lIM,1968
@@ -82,8 +82,8 @@ praisonai/ui/realtimeclient/tools.py,sha256=IJOYwVOBW5Ocn5_iV9pFkmSKR3WU3YpX3kwF
 praisonai/ui/sql_alchemy.py,sha256=oekZOXlRGMJ2SuC-lmgMMIzAmvbMg2DWeGTSpOzbVBM,29674
 praisonai/ui/tools.md,sha256=Ad3YH_ZCLMWlz3mDXllQnQ_S5l55LWqLdcZSh-EXrHI,3956
 praisonai/version.py,sha256=ugyuFliEqtAwQmH4sTlc16YXKYbFWDmfyk87fErB8-8,21
-praisonai-2.0.65.dist-info/LICENSE,sha256=kqvFysVlnFxYOu0HxCe2HlmZmJtdmNGOxWRRkT9TsWc,1035
-praisonai-2.0.65.dist-info/METADATA,sha256=ynQM9tQOLA8E6kocdiTfj2RDLiEwqGwIthkWjM5vG9U,21942
-praisonai-2.0.65.dist-info/WHEEL,sha256=OiNztsphQWM3l0xJ9BHQRElMnxzHbt1M68r2N60f8T8,110
-praisonai-2.0.65.dist-info/entry_points.txt,sha256=I_xc6a6MNTTfLxYmAxe0rgey0G-_hbY07oFW-ZDnkw4,135
-praisonai-2.0.65.dist-info/RECORD,,
+praisonai-2.0.67.dist-info/LICENSE,sha256=kqvFysVlnFxYOu0HxCe2HlmZmJtdmNGOxWRRkT9TsWc,1035
+praisonai-2.0.67.dist-info/METADATA,sha256=Ocu8ITZ5PpCjQM23ILnYEnLqLukAGFIVMmhLzDmPvtI,21942
+praisonai-2.0.67.dist-info/WHEEL,sha256=OiNztsphQWM3l0xJ9BHQRElMnxzHbt1M68r2N60f8T8,110
+praisonai-2.0.67.dist-info/entry_points.txt,sha256=I_xc6a6MNTTfLxYmAxe0rgey0G-_hbY07oFW-ZDnkw4,135
+praisonai-2.0.67.dist-info/RECORD,,