PraisonAI 0.0.35__py3-none-any.whl → 0.0.37__py3-none-any.whl

praisonai/agents_generator.py

@@ -24,6 +24,14 @@ import importlib
 import importlib.util
 from praisonai_tools import BaseTool
 import os
+import logging
+
+agentops_exists = False
+try:
+    import agentops
+    agentops_exists = True
+except ImportError:
+    agentops_exists = False
 
 os.environ["OTEL_SDK_DISABLED"] = "true"
 
@@ -38,7 +46,7 @@ def disable_crewai_telemetry():
 disable_crewai_telemetry()
 
 class AgentsGenerator:
-    def __init__(self, agent_file, framework, config_list):
+    def __init__(self, agent_file, framework, config_list, log_level=None, agent_callback=None, task_callback=None):
         """
         Initialize the AgentsGenerator object.
 
@@ -46,15 +54,31 @@ class AgentsGenerator:
             agent_file (str): The path to the agent file.
             framework (str): The framework to be used for the agents.
             config_list (list): A list of configurations for the agents.
+            log_level (int, optional): The logging level to use. Defaults to logging.INFO.
+            agent_callback (callable, optional): A callback function to be executed after each agent step.
+            task_callback (callable, optional): A callback function to be executed after each tool run.
 
         Attributes:
             agent_file (str): The path to the agent file.
             framework (str): The framework to be used for the agents.
             config_list (list): A list of configurations for the agents.
+            log_level (int): The logging level to use.
+            agent_callback (callable, optional): A callback function to be executed after each agent step.
+            task_callback (callable, optional): A callback function to be executed after each tool run.
         """
         self.agent_file = agent_file
         self.framework = framework
         self.config_list = config_list
+        self.log_level = log_level
+        self.agent_callback = agent_callback
+        self.task_callback = task_callback
+        self.log_level = log_level or logging.getLogger().getEffectiveLevel()
+        if self.log_level == logging.NOTSET:
+            self.log_level = os.environ.get('LOGLEVEL', 'INFO').upper()
+        
+        logging.basicConfig(level=self.log_level, format='%(asctime)s - %(levelname)s - %(message)s')
+        self.logger = logging.getLogger(__name__)
+        self.logger.setLevel(self.log_level)
         
     def is_function_or_decorated(self, obj):
         """
@@ -183,10 +207,10 @@ class AgentsGenerator:
         
         if os.path.isfile(tools_py_path):
             tools_dict.update(self.load_tools_from_module_class(tools_py_path))
-            # print("tools.py exists in the root directory. Loading tools.py and skipping tools folder.")
+            self.logger.debug("tools.py exists in the root directory. Loading tools.py and skipping tools folder.")
         elif tools_dir_path.is_dir():
             tools_dict.update(self.load_tools_from_module_class(tools_dir_path))
-            # print("tools folder exists in the root directory")
+            self.logger.debug("tools folder exists in the root directory")
         
         framework = self.framework or config.get('framework')
 
@@ -197,6 +221,8 @@ class AgentsGenerator:
             # print(self.config_list)
             llm_config = {"config_list": self.config_list}
             
+            if agentops_exists:
+                agentops.init(os.environ.get("AGENTOPS_API_KEY"), tags=["autogen"])
             # Assuming the user proxy agent is set up as per your requirements
             user_proxy = autogen.UserProxyAgent(
                 name="User",
@@ -240,10 +266,13 @@ class AgentsGenerator:
                         # Additional fields like carryover can be added based on dependencies
                     }
                     tasks.append(chat_task)
-
             response = user_proxy.initiate_chats(tasks)
             result = "### Output ###\n"+response[-1].summary if hasattr(response[-1], 'summary') else ""
+            if agentops_exists:
+                agentops.end_session("Success")
         else:
+            if agentops_exists:
+                agentops.init(os.environ.get("AGENTOPS_API_KEY"), tags=["crewai"])
             for role, details in config['roles'].items():
                 role_filled = details['role'].format(topic=topic)
                 goal_filled = details['goal'].format(topic=topic)
@@ -252,6 +281,11 @@ class AgentsGenerator:
                 # Adding tools to the agent if exists
                 agent_tools = [tools_dict[tool] for tool in details.get('tools', []) if tool in tools_dict]
                 agent = Agent(role=role_filled, goal=goal_filled, backstory=backstory_filled, tools=agent_tools, allow_delegation=False)
+                
+                # Set agent callback if provided
+                if self.agent_callback:
+                    agent.step_callback = self.agent_callback
+
                 agents[role] = agent
 
                 for task_name, task_details in details.get('tasks', {}).items():
@@ -259,13 +293,26 @@ class AgentsGenerator:
                     expected_output_filled = task_details['expected_output'].format(topic=topic)
 
                     task = Task(description=description_filled, expected_output=expected_output_filled, agent=agent)
+                    
+                    # Set tool callback if provided
+                    if self.task_callback:
+                        task.callback = self.task_callback
+
                     tasks.append(task)
+            
             crew = Crew(
                 agents=list(agents.values()),
                 tasks=tasks,
                 verbose=2
             )
+            
+            self.logger.debug("Final Crew Configuration:")
+            self.logger.debug(f"Agents: {crew.agents}")
+            self.logger.debug(f"Tasks: {crew.tasks}")
 
             response = crew.kickoff()
             result = f"### Task Output ###\n{response}"
+            if agentops_exists:
+                agentops.end_session("Success")
         return result
+

praisonai/chainlit_ui.py

@@ -8,6 +8,8 @@ from chainlit.input_widget import Select, TextInput
 from typing import Optional
 from dotenv import load_dotenv
 load_dotenv()
+from contextlib import redirect_stdout
+from io import StringIO
 
 framework = "crewai"
 config_list = [
@@ -178,10 +180,57 @@ async def on_chat_resume(thread: ThreadDict):
 # async def tool(data: Optional[str] = None, language: Optional[str] = None):
 #     return cl.Message(content=data, language=language)
 
+@cl.step(type="tool", show_input=False)
+async def run_agents(agent_file: str, framework: str):
+    """Runs the agents and returns the result."""
+    agents_generator = AgentsGenerator(agent_file, framework, config_list)
+    current_step = cl.context.current_step
+    print("Current Step:", current_step)
+
+    stdout_buffer = StringIO()
+    with redirect_stdout(stdout_buffer):
+        result = agents_generator.generate_crew_and_kickoff()
+        
+    complete_output = stdout_buffer.getvalue()
+
+    async with cl.Step(name="gpt4", type="llm", show_input=True) as step:
+        step.input = ""
+        
+        for line in stdout_buffer.getvalue().splitlines():
+            print(line)
+            await step.stream_token(line)
+            
+        tool_res = await output(complete_output)
+
+    yield result
+
+@cl.step(type="tool", show_input=False, language="yaml")
+async def output(output):
+    return output
+
+@cl.step(type="tool", show_input=False, language="yaml")
+def agent(output):
+    return(f"""
+        Agent Step Completed!
+        Output: {output}
+    """)
+
+@cl.step(type="tool", show_input=False, language="yaml")
+def task(output):
+    return(f"""
+        Task Completed!
+        Task: {output.description}
+        Output: {output.raw_output}
+        {output}
+    """)
+
 @cl.on_message
 async def main(message: cl.Message):
     """Run PraisonAI with the provided message as the topic."""
     message_history = cl.user_session.get("message_history")
+    if message_history is None:
+        message_history = []
+        cl.user_session.set("message_history", message_history)
     message_history.append({"role": "user", "content": message.content})
     topic = message.content
     chat_profile = cl.user_session.get("chat_profile")
@@ -189,9 +238,22 @@ async def main(message: cl.Message):
     if chat_profile == "Auto":
         agent_file = "agents.yaml"
         generator = AutoGenerator(topic=topic, agent_file=agent_file, framework=framework, config_list=config_list)
+        await cl.sleep(2)
         agent_file = generator.generate()
-        agents_generator = AgentsGenerator(agent_file, framework, config_list)
-        result = agents_generator.generate_crew_and_kickoff()
+        agents_generator = AgentsGenerator(
+            agent_file, 
+            framework, 
+            config_list, 
+            # agent_callback=agent,
+            # task_callback=task
+        )
+        # Capture stdout
+        stdout_buffer = StringIO()
+        with redirect_stdout(stdout_buffer):
+            result = agents_generator.generate_crew_and_kickoff()
+            
+        complete_output = stdout_buffer.getvalue()
+        tool_res = await output(complete_output)
         msg = cl.Message(content=result)
         await msg.send()
         message_history.append({"role": "assistant", "content": message.content})

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

praisonai/inbuilt_tools/autogen_tools.py

@@ -14,6 +14,8 @@ from pathlib import Path
 import os
 import inspect
 import sys
+import logging
+logging.basicConfig(level=os.environ.get('LOGLEVEL', 'INFO'), format='%(asctime)s - %(levelname)s - %(message)s')
 
 def create_autogen_tool_function(tool_name):
     def autogen_tool(assistant, user_proxy):
@@ -51,10 +53,10 @@ tools_dir_path = Path(root_directory) / 'tools'
 tools_module = None
 
 if os.path.isfile(tools_py_path):
-    print(f"{tools_py_path} exists in the root directory. Loading {tools_py_path} and skipping tools folder.")
+    logging.info(f"{tools_py_path} exists in the root directory. Loading {tools_py_path} and skipping tools folder.")
     tools_module = importlib.import_module("tools")
 elif tools_dir_path.is_dir():
-    print(f"tools folder exists in the root directory. Loading {tool_name} from tools/{tool_name}.py.")
+    logging.info(f"tools folder exists in the root directory. Loading {tool_name} from tools/{tool_name}.py.")
     tools_module = importlib.import_module(f"tools.{tool_name}")
 
 # Create autogen_TOOL_NAME_HERE function for each tool

{praisonai-0.0.35.dist-info → praisonai-0.0.37.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PraisonAI
-Version: 0.0.35
+Version: 0.0.37
 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
@@ -8,9 +8,11 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Provides-Extra: agentops
 Provides-Extra: api
 Provides-Extra: gradio
 Provides-Extra: ui
+Requires-Dist: agentops (==0.2.3) ; extra == "agentops"
 Requires-Dist: chainlit (>=1.1.301,<2.0.0) ; extra == "ui"
 Requires-Dist: crewai (>=0.32.0)
 Requires-Dist: flask (>=3.0.0) ; extra == "api"
@@ -297,6 +299,30 @@ gcloud run deploy praisonai-service \
     --set-env-vars OPENAI_MODEL_NAME=${OPENAI_MODEL_NAME},OPENAI_API_KEY=${OPENAI_API_KEY},OPENAI_API_BASE=${OPENAI_API_BASE}
 ```
 
+### Commands to Install Dependencies:
+
+1. **Install all dependencies, including dev dependencies:**
+   ```sh
+   poetry install
+   ```
+
+2. **Install only documentation dependencies:**
+   ```sh
+   poetry install --with docs
+   ```
+
+3. **Install only test dependencies:**
+   ```sh
+   poetry install --with test
+   ```
+
+4. **Install only dev dependencies:**
+   ```sh
+   poetry install --with dev
+   ```
+
+This configuration ensures that your development dependencies are correctly categorized and installed as needed.
+
 ## Other Models
 
 ```bash

{praisonai-0.0.35.dist-info → praisonai-0.0.37.dist-info}/RECORD

@@ -1,20 +1,20 @@
 praisonai/__init__.py,sha256=JrgyPlzZfLlozoW7SHZ1nVJ63rLPR3ki2k5ZPywYrnI,175
 praisonai/__main__.py,sha256=MVgsjMThjBexHt4nhd760JCqvP4x0IQcwo8kULOK4FQ,144
-praisonai/agents_generator.py,sha256=RWjA9COp9TLRW70uA-7PTDYQJal2_2FbVBqzNLzrHtw,12790
+praisonai/agents_generator.py,sha256=K4I5o7UliVhIb9nSFoGsCsJpPe7Cxtsg4Mflgtp2pHE,15030
 praisonai/auto.py,sha256=CpE9zUbdCP1wR0aLZWKVLxiLU9JOQprRNBK9WKDZVQo,7758
-praisonai/chainlit_ui.py,sha256=qJ40aN7oxaI_8IpwPJhf4YPGM2-mBVcy_8bwtR0aCJU,10243
+praisonai/chainlit_ui.py,sha256=uXo7KTuOH09cu_rLU5RU5JLVCKSc9FM5IcpuW_SVG90,12079
 praisonai/cli.py,sha256=W_PL7bWnEEuFYBT24mh8171f7TiE_LdxVfrM16fp7Pc,10868
-praisonai/deploy.py,sha256=Xz5C_pf9iDcP1LNhBfanrV15ye479bs1CbgbtrpVii4,6031
+praisonai/deploy.py,sha256=mQkAIR4X3ZWQCgX-EIJ3rKPeft9bH4cnCj5IriHHG9o,6031
 praisonai/inbuilt_tools/__init__.py,sha256=mUKnbL6Gram9c9f2m8wJwEzURBLmPEOcHzwySBH89YA,74
-praisonai/inbuilt_tools/autogen_tools.py,sha256=XZoQHIZMjgXvVwoQ5-Hds3TFneeqd5mMSou2bEvgAuc,14764
+praisonai/inbuilt_tools/autogen_tools.py,sha256=svYkM2N7DVFvbiwgoAS7U_MqTOD8rHf8VD3BaFUV5_Y,14907
 praisonai/public/fantasy.svg,sha256=4Gs3kIOux-pjGtw6ogI_rv5_viVJxnE5gRwGilsSg0o,1553
 praisonai/public/game.svg,sha256=y2QMaA01m8XzuDjTOBWzupOC3-TpnUl9ah89mIhviUw,2406
 praisonai/public/movie.svg,sha256=aJ2EQ8vXZusVsF2SeuAVxP4RFJzQ14T26ejrGYdBgzk,1289
 praisonai/public/thriller.svg,sha256=2dYY72EcgbEyTxS4QzjAm37Y4srtPWEW4vCMFki98ZI,3163
 praisonai/test.py,sha256=RZKq3UEFb6AnFFiHER3zBXfNmlteSLBlrTmOvnpnZLo,4092
 praisonai/version.py,sha256=ugyuFliEqtAwQmH4sTlc16YXKYbFWDmfyk87fErB8-8,21
-praisonai-0.0.35.dist-info/LICENSE,sha256=kqvFysVlnFxYOu0HxCe2HlmZmJtdmNGOxWRRkT9TsWc,1035
-praisonai-0.0.35.dist-info/METADATA,sha256=ttszmWEJm4ywwgN8ZR42xEfSHM56NbKt3MGl6eWkHHw,9764
-praisonai-0.0.35.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-praisonai-0.0.35.dist-info/entry_points.txt,sha256=Qg41eW3A1-dvdV5tF7LqChfYof8Rihk2rN1fiEE3vnk,53
-praisonai-0.0.35.dist-info/RECORD,,
+praisonai-0.0.37.dist-info/LICENSE,sha256=kqvFysVlnFxYOu0HxCe2HlmZmJtdmNGOxWRRkT9TsWc,1035
+praisonai-0.0.37.dist-info/METADATA,sha256=Xpz7nJpDWwWqRC69QYlUVUwv2ecyxdATFgwkhjxhYw0,10359
+praisonai-0.0.37.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+praisonai-0.0.37.dist-info/entry_points.txt,sha256=Qg41eW3A1-dvdV5tF7LqChfYof8Rihk2rN1fiEE3vnk,53
+praisonai-0.0.37.dist-info/RECORD,,