PraisonAI 0.0.42__tar.gz → 0.0.43__tar.gz

{praisonai-0.0.42 → praisonai-0.0.43}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PraisonAI
-Version: 0.0.42
+Version: 0.0.43
 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
@@ -11,13 +11,14 @@ Classifier: Programming Language :: Python :: 3.12
 Provides-Extra: agentops
 Provides-Extra: anthropic
 Provides-Extra: api
+Provides-Extra: chat
 Provides-Extra: cohere
 Provides-Extra: google
 Provides-Extra: gradio
 Provides-Extra: openai
 Provides-Extra: ui
 Requires-Dist: agentops (>=0.2.6) ; extra == "agentops"
-Requires-Dist: chainlit (>=1.1.301,<2.0.0) ; extra == "ui"
+Requires-Dist: chainlit (>=1.1.301,<2.0.0) ; extra == "ui" or extra == "chat"
 Requires-Dist: crewai (>=0.32.0)
 Requires-Dist: flask (>=3.0.0) ; extra == "api"
 Requires-Dist: gradio (>=4.26.0) ; extra == "gradio"
@@ -25,6 +26,7 @@ Requires-Dist: langchain-anthropic (>=0.1.13) ; extra == "anthropic"
 Requires-Dist: langchain-cohere (>=0.1.4) ; extra == "cohere"
 Requires-Dist: langchain-google-genai (>=1.0.4) ; extra == "google"
 Requires-Dist: langchain-openai (>=0.1.7) ; extra == "openai"
+Requires-Dist: litellm (>=1.41.8) ; extra == "chat"
 Requires-Dist: markdown (>=3.5)
 Requires-Dist: praisonai-tools (>=0.0.7)
 Requires-Dist: pyautogen (>=0.2.19)
@@ -41,9 +43,12 @@ Description-Content-Type: text/markdown
     <img alt="PraisonAI Logo" src="docs/images/praisonai-logo-black-large.png">
   </picture>
 </p>
+<div align="center">
 
 # Praison AI
 
+</div>
+
 Praison AI, leveraging both AutoGen and CrewAI or any other agent framework, represents a low-code, centralised framework designed to simplify the creation and orchestration of multi-agent systems for various LLM applications, emphasizing ease of use, customization, and human-agent interaction.
 
 ## TL;DR

{praisonai-0.0.42 → praisonai-0.0.43}/README.md

@@ -5,9 +5,12 @@
     <img alt="PraisonAI Logo" src="docs/images/praisonai-logo-black-large.png">
   </picture>
 </p>
+<div align="center">
 
 # Praison AI
 
+</div>
+
 Praison AI, leveraging both AutoGen and CrewAI or any other agent framework, represents a low-code, centralised framework designed to simplify the creation and orchestration of multi-agent systems for various LLM applications, emphasizing ease of use, customization, and human-agent interaction.
 
 ## TL;DR

{praisonai-0.0.42 → praisonai-0.0.43}/praisonai/cli.py

@@ -84,6 +84,11 @@ class PraisonAI:
             deployer = CloudDeployer()
             deployer.run_commands()
             return
+        
+        if getattr(args, 'chat', False):
+            self.create_chainlit_chat_interface()
+            return
+        
         invocation_cmd = "praisonai"
         version_string = f"PraisonAI version {__version__}"
         
@@ -163,8 +168,30 @@ class PraisonAI:
             args.agent_file = 'agents.yaml'
         if args.agent_file == 'ui':
             args.ui = 'chainlit'
+        if args.agent_file == 'chat':
+            args.ui = 'chainlit'
+            args.chat = True
 
         return args
+    
+    def create_chainlit_chat_interface(self):
+        """
+        Create a Chainlit interface for the chat application.
+
+        This function sets up a Chainlit application that listens for messages.
+        When a message is received, it runs PraisonAI with the provided message as the topic.
+        The generated agents are then used to perform tasks.
+
+        Returns:
+            None: This function does not return any value. It starts the Chainlit application.
+        """
+        if CHAINLIT_AVAILABLE:
+            import praisonai
+            os.environ["CHAINLIT_PORT"] = "8084"
+            chat_ui_path = os.path.join(os.path.dirname(praisonai.__file__), 'ui', 'chat.py')
+            chainlit_run([chat_ui_path])
+        else:
+            print("ERROR: Chat UI is not installed. Please install it with 'pip install \"praisonai\[chat]\"' to use the chat UI.")
 
     def create_gradio_interface(self):
         """
@@ -247,7 +274,7 @@ class PraisonAI:
             chainlit_ui_path = os.path.join(os.path.dirname(praisonai.__file__), 'chainlit_ui.py')
             chainlit_run([chainlit_ui_path])
         else:
-            print("ERROR: Chainlit is not installed. Please install it with 'pip install chainlit' to use the UI.")        
+            print("ERROR: Chainlit is not installed. Please install it with 'pip install \"praisonai\[ui]\"' to use the UI.")        
 
 if __name__ == "__main__":
     praison_ai = PraisonAI()

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

praisonai-0.0.43/praisonai/ui/chat.py

@@ -0,0 +1,308 @@
+import chainlit as cl
+from chainlit.input_widget import TextInput
+from chainlit.types import ThreadDict
+from litellm import acompletion
+import os
+import sqlite3
+from datetime import datetime
+from typing import Dict, List, Optional
+from dotenv import load_dotenv
+load_dotenv()
+import chainlit.data as cl_data
+from chainlit.step import StepDict
+from literalai.helper import utc_now
+
+now = utc_now()
+
+create_step_counter = 0
+
+import json
+
+DB_PATH = "threads.db"
+
+def initialize_db():
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS threads (
+            id TEXT PRIMARY KEY,
+            name TEXT,
+            createdAt TEXT,
+            userId TEXT,
+            userIdentifier TEXT
+        )
+    ''')
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS steps (
+            id TEXT PRIMARY KEY,
+            threadId TEXT,
+            name TEXT,
+            createdAt TEXT,
+            type TEXT,
+            output TEXT,
+            FOREIGN KEY (threadId) REFERENCES threads (id)
+        )
+    ''')
+    conn.commit()
+    conn.close()
+
+def save_thread_to_db(thread):
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
+    cursor.execute('''
+        INSERT OR REPLACE INTO threads (id, name, createdAt, userId, userIdentifier)
+        VALUES (?, ?, ?, ?, ?)
+    ''', (thread['id'], thread['name'], thread['createdAt'], thread['userId'], thread['userIdentifier']))
+    
+    # No steps to save as steps are empty in the provided thread data
+    conn.commit()
+    conn.close()
+    print("saved")
+
+def update_thread_in_db(thread):
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
+
+    # Insert or update the thread
+    cursor.execute('''
+        INSERT OR REPLACE INTO threads (id, name, createdAt, userId, userIdentifier)
+        VALUES (?, ?, ?, ?, ?)
+    ''', (thread['id'], thread['name'], thread['createdAt'], thread['userId'], thread['userIdentifier']))
+
+    # Fetch message_history from metadata
+    message_history = thread['metadata']['message_history']
+
+    # Ensure user messages come first followed by assistant messages
+    user_messages = [msg for msg in message_history if msg['role'] == 'user']
+    assistant_messages = [msg for msg in message_history if msg['role'] == 'assistant']
+    ordered_steps = [val for pair in zip(user_messages, assistant_messages) for val in pair]
+
+    # Generate steps from ordered message_history
+    steps = []
+    for idx, message in enumerate(ordered_steps):
+        step_id = f"{thread['id']}-step-{idx}"
+        step_type = 'user_message' if message['role'] == 'user' else 'assistant_message'
+        step_name = 'user' if message['role'] == 'user' else 'assistant'
+        created_at = message.get('createdAt', thread['createdAt'])  # Use thread's createdAt if no timestamp in message
+        steps.append({
+            'id': step_id,
+            'threadId': thread['id'],
+            'name': step_name,
+            'createdAt': created_at,
+            'type': step_type,
+            'output': message['content']
+        })
+
+    # Insert all steps into the database
+    for step in steps:
+        cursor.execute('''
+            INSERT OR REPLACE INTO steps (id, threadId, name, createdAt, type, output)
+            VALUES (?, ?, ?, ?, ?, ?)
+        ''', (step['id'], step['threadId'], step['name'], step['createdAt'], step['type'], step['output']))
+    
+    conn.commit()
+    conn.close()
+
+def load_threads_from_db():
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
+    cursor.execute('SELECT * FROM threads')
+    thread_rows = cursor.fetchall()
+    threads = []
+    for thread_row in thread_rows:
+        cursor.execute('SELECT * FROM steps WHERE threadId = ?', (thread_row[0],))
+        step_rows = cursor.fetchall()
+        steps = []
+        for step_row in step_rows:
+            steps.append({
+                "id": step_row[0],
+                "threadId": step_row[1],
+                "name": step_row[2],
+                "createdAt": step_row[3],
+                "type": step_row[4],
+                "output": step_row[5]
+            })
+        threads.append({
+            "id": thread_row[0],
+            "name": thread_row[1],
+            "createdAt": thread_row[2],
+            "userId": thread_row[3],
+            "userIdentifier": thread_row[4],
+            "steps": steps
+        })
+    conn.close()
+    return threads
+
+# Initialize the database
+initialize_db()
+thread_history = load_threads_from_db()
+
+deleted_thread_ids = []  # type: List[str]
+
+class TestDataLayer(cl_data.BaseDataLayer):
+    async def get_user(self, identifier: str):
+        return cl.PersistedUser(id="test", createdAt=now, identifier=identifier)
+
+    async def create_user(self, user: cl.User):
+        return cl.PersistedUser(id="test", createdAt=now, identifier=user.identifier)
+
+    async def update_thread(
+        self,
+        thread_id: str,
+        name: Optional[str] = None,
+        user_id: Optional[str] = None,
+        metadata: Optional[Dict] = None,
+        tags: Optional[List[str]] = None,
+    ):
+        thread = next((t for t in thread_history if t["id"] == thread_id), None)
+        if thread:
+            if name:
+                thread["name"] = name
+            if metadata:
+                thread["metadata"] = metadata
+            if tags:
+                thread["tags"] = tags
+            update_thread_in_db(thread)
+            cl.user_session.set("message_history", thread['metadata']['message_history'])
+            cl.user_session.set("thread_id", thread["id"])
+            print("Updated")
+            
+        else:
+            thread_history.append(
+                {
+                    "id": thread_id,
+                    "name": name,
+                    "metadata": metadata,
+                    "tags": tags,
+                    "createdAt": utc_now(),
+                    "userId": user_id,
+                    "userIdentifier": "admin",
+                    "steps": [],
+                }
+            )
+            thread = {
+                "id": thread_id,
+                "name": name,
+                "metadata": metadata,
+                "tags": tags,
+                "createdAt": utc_now(),
+                "userId": user_id,
+                "userIdentifier": "admin",
+                "steps": [],
+            }
+            save_thread_to_db(thread)
+
+    @cl_data.queue_until_user_message()
+    async def create_step(self, step_dict: StepDict):
+        global create_step_counter
+        create_step_counter += 1
+
+        thread = next(
+            (t for t in thread_history if t["id"] == step_dict.get("threadId")), None
+        )
+        if thread:
+            thread["steps"].append(step_dict)
+
+    async def get_thread_author(self, thread_id: str):
+        return "admin"
+
+    async def list_threads(
+        self, pagination: cl_data.Pagination, filters: cl_data.ThreadFilter
+    ) -> cl_data.PaginatedResponse[cl_data.ThreadDict]:
+        return cl_data.PaginatedResponse(
+            data=[t for t in thread_history if t["id"] not in deleted_thread_ids],
+            pageInfo=cl_data.PageInfo(
+                hasNextPage=False, startCursor=None, endCursor=None
+            ),
+        )
+
+    async def get_thread(self, thread_id: str):
+        thread_history = load_threads_from_db()
+        return next((t for t in thread_history if t["id"] == thread_id), None)
+
+    async def delete_thread(self, thread_id: str):
+        deleted_thread_ids.append(thread_id)
+
+cl_data._data_layer = TestDataLayer()
+
+@cl.on_chat_start
+async def start():
+    initialize_db()
+    await cl.ChatSettings(
+        [
+            TextInput(
+                id="model_name",
+                label="Enter the Model Name",
+                placeholder="e.g., gpt-3.5-turbo"
+            )
+        ]
+    ).send()
+
+@cl.on_settings_update
+async def setup_agent(settings):
+    model_name = settings["model_name"]
+    cl.user_session.set("model_name", model_name)
+
+@cl.on_message
+async def main(message: cl.Message):
+    model_name = cl.user_session.get("model_name", "gpt-3.5-turbo")
+    message_history = cl.user_session.get("message_history", [])
+    message_history.append({"role": "user", "content": message.content})
+
+    msg = cl.Message(content="")
+    await msg.send()
+
+    response = await acompletion(
+        model=model_name,
+        messages=message_history,
+        stream=True,
+        temperature=0.7,
+        max_tokens=500,
+        top_p=1
+    )
+
+    full_response = ""
+    async for part in response:
+        if token := part['choices'][0]['delta']['content']:
+            await msg.stream_token(token)
+            full_response += token
+    print(full_response)
+    message_history.append({"role": "assistant", "content": full_response})
+    print(message_history)
+    cl.user_session.set("message_history", message_history)
+    await msg.update()
+
+username = os.getenv("CHAINLIT_USERNAME", "admin")  # Default to "admin" if not found
+password = os.getenv("CHAINLIT_PASSWORD", "admin")  # Default to "admin" if not found
+
+@cl.password_auth_callback
+def auth_callback(username: str, password: str):
+    if (username, password) == (username, password):
+        return cl.User(
+            identifier=username, metadata={"role": "ADMIN", "provider": "credentials"}
+        )
+    else:
+        return None
+
+async def send_count():
+    await cl.Message(
+        f"Create step counter: {create_step_counter}", disable_feedback=True
+    ).send()
+
+@cl.on_chat_resume
+async def on_chat_resume(thread: cl_data.ThreadDict):
+    thread_id = thread["id"]
+    cl.user_session.set("thread_id", thread["id"])
+    message_history = cl.user_session.get("message_history", [])
+    steps = thread["steps"]
+
+    for message in steps:
+        msg_type = message.get("type")
+        if msg_type == "user_message":
+            message_history.append({"role": "user", "content": message.get("output", "")})
+        elif msg_type == "assistant_message":
+            message_history.append({"role": "assistant", "content": message.get("output", "")})
+        else:
+            print(f"Message without type: {message}")
+
+        cl.user_session.set("message_history", message_history)

{praisonai-0.0.42 → praisonai-0.0.43}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "PraisonAI"
-version = "0.0.42"
+version = "0.0.43"
 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 = ""
@@ -29,6 +29,7 @@ langchain-google-genai = {version = ">=1.0.4", optional = true}
 langchain-anthropic = {version = ">=0.1.13", optional = true}
 langchain-openai = {version = ">=0.1.7", optional = true}
 langchain-cohere = {version = ">=0.1.4", optional = true}
+litellm = {version = ">=1.41.8", optional = true}
 
 [tool.poetry.group.docs.dependencies]
 mkdocs = "*"
@@ -75,6 +76,7 @@ langchain-google-genai = ">=1.0.4"
 langchain-anthropic = ">=0.1.13"
 langchain-openai = ">=0.1.7"
 langchain-cohere = ">=0.1.4"
+litellm = ">=1.41.8"
 
 [build-system]
 requires = ["poetry-core"]
@@ -92,4 +94,5 @@ agentops = ["agentops"]
 google = ["langchain-google-genai"]
 openai = ["langchain-openai"]
 anthropic = ["langchain-anthropic"]
-cohere = ["langchain-cohere"]
+cohere = ["langchain-cohere"]
+chat = ["chainlit", "litellm"]