PraisonAI 0.0.7__tar.gz → 0.0.9__tar.gz

{praisonai-0.0.7 → praisonai-0.0.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: praisonAI
-Version: 0.0.7
+Version: 0.0.9
 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.7 → praisonai-0.0.9}/praisonai/__main__.py

@@ -2,6 +2,9 @@
 
 from .cli import PraisonAI
 
-if __name__ == "__main__":
+def main():
     praison_ai = PraisonAI()
-    praison_ai.main()
+    praison_ai.main()
+
+if __name__ == "__main__":
+    main()

{praisonai-0.0.7 → praisonai-0.0.9}/praisonai/auto.py

@@ -70,8 +70,6 @@ class AutoGenerator:
             "roles": {},
             "dependencies": []
         }
-        print(self.topic)
-        print(self.framework)
 
         for role_id, role_details in json_data['roles'].items():
             yaml_data['roles'][role_id] = {
@@ -131,5 +129,5 @@ The final role you create should satisfy the provided task: """ + self.topic + "
         return user_content
 
     
-generator = AutoGenerator(framework="crewai", topic="Create a snake game in python")
-print(generator.generate())
+# generator = AutoGenerator(framework="crewai", topic="Create a snake game in python")
+# print(generator.generate())

{praisonai-0.0.7 → praisonai-0.0.9}/praisonai/cli.py

@@ -13,7 +13,7 @@ import argparse
 from .auto import AutoGenerator
 
 class PraisonAI:
-    def __init__(self, agent_file="agents.yaml", framework="crewai", auto=False):
+    def __init__(self, agent_file="agents.yaml", framework="crewai", auto=False, init=False):
         self.config_list = [
             {
                 'model': os.environ.get("OPENAI_MODEL_NAME", "gpt-4-turbo-preview"),
@@ -23,6 +23,7 @@ class PraisonAI:
         self.agent_file = agent_file
         self.framework = framework
         self.auto = auto
+        self.init = init
 
     def generate_crew_and_kickoff(self):
         with open(self.agent_file, 'r') as f:
@@ -35,7 +36,7 @@ class PraisonAI:
         tasks = []
         if framework == "autogen":
             # Load the LLM configuration dynamically
-            print(self.config_list)
+            # print(self.config_list)
             llm_config = {"config_list": self.config_list}
             
             for role, details in config['roles'].items():
@@ -114,21 +115,33 @@ class PraisonAI:
         ui = args.ui  # Default UI flag
 
         if args.agent_file:
-            self.agent_file = args.agent_file
+            if args.agent_file == "tests.test": # Argument used for testing purposes
+                full_path = os.path.abspath("agents.yaml")
+            else:
+                full_path = os.path.abspath(args.agent_file)
+            self.agent_file = full_path
         else:
             full_path = os.path.abspath(self.agent_file)
             self.filename = full_path
         
-        if args.auto:
-            self.topic = ' '.join(args.auto)
-        elif self.auto:  # Use the auto attribute if args.auto is not provided
+        if args.auto or args.init:
+            temp_topic = ' '.join(args.auto) if args.auto else ' '.join(args.init)
+            self.topic = temp_topic
+        elif self.auto or self.init:  # Use the auto attribute if args.auto is not provided
             self.topic = self.auto
             
         if args.auto or self.auto:
+            self.filename = "test.yaml"
             generator = AutoGenerator(topic=self.topic , framework=self.framework)
             self.agent_file = generator.generate()
             result = self.generate_crew_and_kickoff()
             return result
+        elif args.init or self.init:
+            self.filename = "agents.yaml"
+            generator = AutoGenerator(topic=self.topic , framework=self.framework, filename=self.filename)
+            self.agent_file = generator.generate()
+            print("File {} created successfully".format(self.agent_file))
+            return "File {} created successfully".format(self.agent_file)
         
         if ui:
             self.create_gradio_interface()
@@ -141,6 +154,7 @@ class PraisonAI:
         parser.add_argument("--framework", choices=["crewai", "autogen"], default="crewai", help="Specify the framework")
         parser.add_argument("--ui", action="store_true", help="Enable UI mode")
         parser.add_argument("--auto", nargs=argparse.REMAINDER, help="Enable auto mode and pass arguments for it")
+        parser.add_argument("--init", nargs=argparse.REMAINDER, help="Enable auto mode and pass arguments for it")
         parser.add_argument("agent_file", nargs="?", help="Specify the agent file")
 
         return parser.parse_args()

{praisonai-0.0.7 → praisonai-0.0.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "praisonAI"
-version = "0.0.7"
+version = "0.0.9"
 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 = ""