PraisonAI 0.0.8__tar.gz → 0.0.9__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: praisonAI
-Version: 0.0.8
+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.8 → 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.8 → 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:
@@ -123,16 +124,24 @@ class PraisonAI:
             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()
@@ -145,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.8 → praisonai-0.0.9}/pyproject.toml

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