PraisonAI 0.0.59rc11__cp312-cp312-manylinux_2_35_x86_64.whl → 0.0.64__cp312-cp312-manylinux_2_35_x86_64.whl

praisonai/cli.py

@@ -133,18 +133,24 @@ class PraisonAI:
             package_root = os.path.dirname(os.path.abspath(__file__))
             config_yaml_destination = os.path.join(os.getcwd(), 'config.yaml')
 
-            # Generate config.yaml using the function
-            config = generate_config(
-                model_name=args.model,
-                hf_model_name=args.hf,
-                ollama_model_name=args.ollama,
-                dataset=[{
-                    "name": args.dataset
-                }]
-            )
-            with open('config.yaml', 'w') as f:
-                yaml.dump(config, f, default_flow_style=False, indent=2) 
-
+            # Create config.yaml only if it doesn't exist or --model or --dataset is provided
+            if not os.path.exists(config_yaml_destination) or args.model or args.dataset:
+                config = generate_config(
+                    model_name=args.model,
+                    hf_model_name=args.hf,
+                    ollama_model_name=args.ollama,
+                    dataset=[{
+                        "name": args.dataset
+                    }]
+                )
+                with open('config.yaml', 'w') as f:
+                    yaml.dump(config, f, default_flow_style=False, indent=2) 
+
+            # Overwrite huggingface_save and ollama_save if --hf or --ollama are provided 
+            if args.hf:
+                config["huggingface_save"] = "true"
+            if args.ollama:
+                config["ollama_save"] = "true"
 
             if 'init' in sys.argv:
                 from praisonai.setup.setup_conda_env import main as setup_conda_main
@@ -171,7 +177,7 @@ class PraisonAI:
             env = os.environ.copy()
             env['PYTHONUNBUFFERED'] = '1'
             
-            stream_subprocess(['conda', 'run', '--no-capture-output', '--name', 'praison_env', 'python', '-u', train_script_path, 'train'] + train_args, env=env)
+            stream_subprocess(['conda', 'run', '--no-capture-output', '--name', 'praison_env', 'python', '-u', train_script_path, 'train'], 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.59rc11 gunicorn markdown\n")
+            file.write("RUN pip install flask praisonai==0.0.64 gunicorn markdown\n")
             file.write("EXPOSE 8080\n")
             file.write('CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]\n')
             

praisonai/inc/config.py

@@ -38,9 +38,9 @@ def generate_config(
     """Generates the configuration for PraisonAI with dynamic overrides."""
 
     config = {
-        "ollama_save": ollama_save or "false",
-        "huggingface_save": huggingface_save or "false",
-        "train": train or "false",
+        "ollama_save": ollama_save or "true",
+        "huggingface_save": huggingface_save or "true",
+        "train": train or "true",
 
         "model_name": model_name or "unsloth/Meta-Llama-3.1-8B-Instruct-bnb-4bit",
         "hf_model_name": hf_model_name or "mervinpraison/llama-3.1-tamilan-8B-test",

praisonai/setup/config.yaml

@@ -10,7 +10,7 @@ model_parameters: "8b"
 max_seq_length: 2048
 load_in_4bit: true
 lora_r: 16
-lora_target_modules: 
+lora_target_modules:
   - "q_proj"
   - "k_proj"
   - "v_proj"

praisonai/train.py

@@ -142,6 +142,8 @@ class train:
             ),
         )
         trainer.train()
+        self.model.save_pretrained("lora_model") # Local saving
+        self.tokenizer.save_pretrained("lora_model")
 
     def inference(self, instruction, input_text):
         FastLanguageModel.for_inference(self.model)
@@ -158,6 +160,17 @@ class train:
         inputs = self.tokenizer([alpaca_prompt.format(instruction, input_text, "")], return_tensors="pt").to("cuda")
         outputs = self.model.generate(**inputs, max_new_tokens=64, use_cache=True)
         print(self.tokenizer.batch_decode(outputs))
+        
+    def load_model(self):
+        """Loads the model and tokenizer using the FastLanguageModel library."""
+        from unsloth import FastLanguageModel
+        model, tokenizer = FastLanguageModel.from_pretrained(
+            model_name=self.config["output_dir"],
+            max_seq_length=2048,
+            dtype=None,
+            load_in_4bit=self.config["load_in_4bit"],
+        )
+        return model, tokenizer
 
     def save_model_merged(self):
         if os.path.exists(self.config["hf_model_name"]):
@@ -176,9 +189,22 @@ class train:
             quantization_method=self.config["quantization_method"],
             token=os.getenv('HF_TOKEN')
         )
+    
+    def save_model_gguf(self):
+        self.model.save_pretrained_gguf(
+            self.config["hf_model_name"],
+            self.tokenizer,
+            quantization_method="q4_k_m"
+        )
 
     def prepare_modelfile_content(self):
         output_model = self.config["hf_model_name"]
+        gguf_path = f"{output_model}/unsloth.Q4_K_M.gguf"
+
+        # Check if the GGUF file exists. If not, generate it ## TODO Multiple Quantisation other than Q4_K_M.gguf
+        if not os.path.exists(gguf_path):
+            self.model, self.tokenizer = self.load_model()
+            self.save_model_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 }}}}
@@ -215,10 +241,20 @@ PARAMETER stop "<|reserved_special_token_"
             self.train_model()
 
         if self.config.get("huggingface_save", "true").lower() == "true":
+            # self.model, self.tokenizer = self.load_model()
             self.save_model_merged()
 
         if self.config.get("huggingface_save_gguf", "true").lower() == "true":
+            # self.model, self.tokenizer = self.load_model()
             self.push_model_gguf()
+            
+        # if self.config.get("save_gguf", "true").lower() == "true": ## TODO
+        #     self.model, self.tokenizer = self.load_model()
+        #     self.save_model_gguf()
+        
+        # if self.config.get("save_merged", "true").lower() == "true": ## TODO
+        #     self.model, self.tokenizer = self.load_model()
+        #     self.save_model_merged()
 
         if self.config.get("ollama_save", "true").lower() == "true":
             self.create_and_push_ollama_model()

praisonai/ui/chat.py

@@ -73,11 +73,11 @@ def initialize_db():
             type TEXT NOT NULL,
             threadId UUID NOT NULL,
             parentId UUID,
-            disableFeedback BOOLEAN NOT NULL,
-            streaming BOOLEAN NOT NULL,
-            waitForAnswer BOOLEAN,
-            isError BOOLEAN,
-            metadata JSONB,
+            disableFeedback BOOLEAN NOT NULL DEFAULT 0,
+            streaming BOOLEAN NOT NULL DEFAULT 0,
+            waitForAnswer BOOLEAN DEFAULT 0,
+            isError BOOLEAN NOT NULL DEFAULT 0,
+            metadata JSONB DEFAULT '{}',
             tags TEXT[],
             input TEXT,
             output TEXT,

praisonai/ui/code.py

@@ -74,11 +74,11 @@ def initialize_db():
             type TEXT NOT NULL,
             threadId UUID NOT NULL,
             parentId UUID,
-            disableFeedback BOOLEAN NOT NULL,
-            streaming BOOLEAN NOT NULL,
-            waitForAnswer BOOLEAN,
-            isError BOOLEAN,
-            metadata JSONB,
+            disableFeedback BOOLEAN NOT NULL DEFAULT 0,
+            streaming BOOLEAN NOT NULL DEFAULT 0,
+            waitForAnswer BOOLEAN DEFAULT 0,
+            isError BOOLEAN NOT NULL DEFAULT 0,
+            metadata JSONB DEFAULT '{}',
             tags TEXT[],
             input TEXT,
             output TEXT,

{praisonai-0.0.59rc11.dist-info → praisonai-0.0.64.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PraisonAI
-Version: 0.0.59rc11
+Version: 0.0.64
 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
@@ -78,6 +78,11 @@ Praison AI, leveraging both AutoGen and CrewAI or any other agent framework, rep
 | **Chat** | Chat with 100+ LLMs, single AI Agent | [https://docs.praison.ai/ui/chat](https://docs.praison.ai/ui/chat) |
 | **Code** | Chat with entire Codebase, single AI Agent | [https://docs.praison.ai/ui/code](https://docs.praison.ai/ui/code) |
 
+| Other Features | Description | Docs |
+|---|---|---|
+| **Train** | Fine-tune LLMs using your custom data | [https://docs.praison.ai/train](https://docs.praison.ai/train) |
+
+
 ## Google Colab Multi Agents
 
 |               | Cookbook        | Open in Colab                                                                                                                                                                                                                                  |
@@ -87,9 +92,12 @@ Praison AI, leveraging both AutoGen and CrewAI or any other agent framework, rep
 
 ## Install
 
-| PraisonAI | PraisonAI Code | PraisonAI Chat |
-| --- | --- | --- |
-| `pip install praisonai` | `pip install "praisonai[code]"` | `pip install "praisonai[chat]"` |
+|  | Installation |
+|---|---|
+| **PraisonAI** | `pip install praisonai` |
+| **PraisonAI Code** | `pip install "praisonai[code]"` |
+| **PraisonAI Chat** | `pip install "praisonai[chat]"` |
+| **PraisonAI Train** | `pip install "praisonai[train]"` |
 
 ## Key Features
 

{praisonai-0.0.59rc11.dist-info → praisonai-0.0.64.dist-info}/RECORD

@@ -3,12 +3,12 @@ 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=Cz_oegYqZBRwWLpzbUiTVQRZqJJg38UgTBtnHFjk_HE,17648
-praisonai/deploy.py,sha256=9qkDTiytzxjHKlso9sBSiMVPZTr9wJZQgdxZh8mhzd0,6032
+praisonai/cli.py,sha256=ZDQF9OXgaquu_89cLc-jsx5D9zWxYg4ChPhlVv66drk,18047
+praisonai/deploy.py,sha256=8eoDMlgK7klsNEr1T6kyTsmuRI90wUFHMi-cgmDNVsI,6028
 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
-praisonai/inc/config.py,sha256=ZjmgY9Bh3PjA74Gwdal-MvPj50CKITiKny0aoY7XWN4,3336
+praisonai/inc/config.py,sha256=up2-841ruK7MCUUT3xkWBA5S6WsY0sFODNfcT6Q4Wms,3333
 praisonai/inc/models.py,sha256=1kwP9o56AvN8L38x7eeAzudjAvstN0uWu-woQkgxAe4,5449
 praisonai/public/android-chrome-192x192.png,sha256=ENJEqhDE3XEQViRhKNDezQKRiOiuHOUj5nzRN43fz50,6535
 praisonai/public/android-chrome-512x512.png,sha256=4txEwB0cJkxFVarRdvFGJZR1DtWJ2h-L_2cUEjBXHAc,15244
@@ -24,14 +24,14 @@ 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=pjThU3Mzp5ce2ZFb1GHnR3RviE9dhpwRz3ONd9JnnwE,1212
+praisonai/setup/config.yaml,sha256=sr_D1RIvv3LQ_eueOMZV0rAUiWTR-n2xuE1RhKK6b34,1211
 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=te7s0KHsTi7XM-vkNvE0dKC1HeU2tXxqE-sPUScV6fY,2718
 praisonai/test.py,sha256=OL-wesjA5JTohr8rtr6kWoaS4ImkJg2l0GXJ-dUUfRU,4090
-praisonai/train.py,sha256=S4YPXXMypR_d6K_m0dcrDxFNanaOvXZq8tM2IenpmPs,9522
-praisonai/ui/chat.py,sha256=B4F1R7qP-0c-elg8WcRsYlr6-FkmHWtdunGIzU7WrDM,9321
-praisonai/ui/code.py,sha256=GcOr8lNah4AgI2RcIKmgjehzSl-KNu7x6UHrghixeaM,10095
+praisonai/train.py,sha256=DvORlrwKOD-2v4r_z84eV3LsfzpNs-WnPKb5cQB3_t4,11071
+praisonai/ui/chat.py,sha256=V4kG2jog8FK0zHbcG_gTS58JzeriG_ZTorCjEnjJz38,9383
+praisonai/ui/code.py,sha256=MaegqaKknkPup1AVoVIkVLa7YUp25kytqq_udm0xtIg,10157
 praisonai/ui/context.py,sha256=oWO2I_WBZb7kZnuXItf18EJX0ZQv-1nAd8rxhwhuuDU,11871
 praisonai/ui/public/fantasy.svg,sha256=4Gs3kIOux-pjGtw6ogI_rv5_viVJxnE5gRwGilsSg0o,1553
 praisonai/ui/public/game.svg,sha256=y2QMaA01m8XzuDjTOBWzupOC3-TpnUl9ah89mIhviUw,2406
@@ -41,8 +41,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.59rc11.dist-info/LICENSE,sha256=kqvFysVlnFxYOu0HxCe2HlmZmJtdmNGOxWRRkT9TsWc,1035
-praisonai-0.0.59rc11.dist-info/METADATA,sha256=oPMDxC8HHl-S91_BkN6RunO0LyP8SXnKSIjuHHF0Frc,11152
-praisonai-0.0.59rc11.dist-info/WHEEL,sha256=HBsDV7Hj4OTiS1GX6ua7iQXUQTB9UHftbBxr7Q8Xm9c,110
-praisonai-0.0.59rc11.dist-info/entry_points.txt,sha256=jB078LEGLY3Ky_indhclomRIVVpXrPSksHjJ-tcBZ-o,133
-praisonai-0.0.59rc11.dist-info/RECORD,,
+praisonai-0.0.64.dist-info/LICENSE,sha256=kqvFysVlnFxYOu0HxCe2HlmZmJtdmNGOxWRRkT9TsWc,1035
+praisonai-0.0.64.dist-info/METADATA,sha256=bA_r7YzM5exBiLdr1CGXBpT7xgZ8UMIvYuUF4cHz-zQ,11406
+praisonai-0.0.64.dist-info/WHEEL,sha256=HBsDV7Hj4OTiS1GX6ua7iQXUQTB9UHftbBxr7Q8Xm9c,110
+praisonai-0.0.64.dist-info/entry_points.txt,sha256=jB078LEGLY3Ky_indhclomRIVVpXrPSksHjJ-tcBZ-o,133
+praisonai-0.0.64.dist-info/RECORD,,