PraisonAI 0.1.6__cp312-cp312-manylinux_2_35_x86_64.whl → 0.1.8__cp312-cp312-manylinux_2_35_x86_64.whl

praisonai/agents_generator.py

@@ -288,7 +288,7 @@ class AgentsGenerator:
                 backstory_filled = details['backstory'].format(topic=topic)
                 
                 # Adding tools to the agent if exists
-                agent_tools = [tools_dict[tool] for tool in details.get('tools', []) if tool in tools_dict]
+                agent_tools = [tools_dict.get(tool) for tool in details.get('tools', [])]
                 
                 llm_model = details.get('llm')  # Get the llm configuration
                 if llm_model:

praisonai/api/call.py

@@ -12,6 +12,8 @@ import uvicorn
 from pyngrok import ngrok, conf
 from rich import print
 import argparse
+import logging
+import importlib.util
 
 load_dotenv()
 
@@ -23,8 +25,10 @@ PUBLIC = os.getenv('PUBLIC', 'false').lower() == 'true'
 SYSTEM_MESSAGE = (
     "You are a helpful and bubbly AI assistant who loves to chat about "
     "anything the user is interested in and is prepared to offer them facts. "
+    "Keep your responses short and to the point. "
     "You have a penchant for dad jokes, owl jokes, and rickrolling – subtly. "
     "Always stay positive, but work in a joke when appropriate."
+    "Start your conversation by saying 'Hi! I'm Praison AI. How can I help you today?'"
 )
 VOICE = 'alloy'
 LOG_EVENT_TYPES = [
@@ -38,6 +42,46 @@ app = FastAPI()
 if not OPENAI_API_KEY:
     raise ValueError('Missing the OpenAI API key. Please set it in the .env file.')
 
+# Set up logging
+logger = logging.getLogger(__name__)
+log_level = os.getenv("LOGLEVEL", "INFO").upper()
+logger.handlers = []
+
+# Try to import tools from the root directory
+tools = []
+tools_path = os.path.join(os.getcwd(), 'tools.py')
+logger.info(f"Tools path: {tools_path}")
+
+def import_tools_from_file(file_path):
+    spec = importlib.util.spec_from_file_location("custom_tools", file_path)
+    custom_tools_module = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(custom_tools_module)
+    logger.debug(f"Imported tools from {file_path}")
+    return custom_tools_module
+
+try:
+    if os.path.exists(tools_path):
+        # tools.py exists in the root directory, import from file
+        custom_tools_module = import_tools_from_file(tools_path)
+        logger.info("Successfully imported custom tools from root tools.py")
+    else:
+        logger.info("No custom tools.py file found in the root directory")
+        custom_tools_module = None
+
+    if custom_tools_module:
+        # Update the tools list with custom tools
+        if hasattr(custom_tools_module, 'tools') and isinstance(custom_tools_module.tools, list):
+            tools.extend(custom_tools_module.tools)
+        else:
+            for name, obj in custom_tools_module.__dict__.items():
+                if callable(obj) and not name.startswith("__"):
+                    tool_definition = getattr(obj, 'definition', None)
+                    if tool_definition:
+                        tools.append(tool_definition)
+
+except Exception as e:
+    logger.warning(f"Error importing custom tools: {str(e)}. Continuing without custom tools.")
+
 @app.get("/status", response_class=HTMLResponse)
 async def index_page():
     return """
@@ -57,7 +101,7 @@ async def handle_incoming_call(request: Request):
     response = VoiceResponse()
     response.say("")
     response.pause(length=1)
-    response.say("O.K. you can start talking!")
+    # response.say("")
     host = request.url.hostname
     connect = Connect()
     connect.stream(url=f'wss://{host}/media-stream')
@@ -110,6 +154,10 @@ async def handle_media_stream(websocket: WebSocket):
                         print(f"Received event: {response['type']}", response)
                     if response['type'] == 'session.updated':
                         print("Session updated successfully:", response)
+                    
+                    if response['type'] == 'response.done':
+                        await handle_response_done(response, openai_ws)
+                    
                     if response['type'] == 'response.audio.delta' and response.get('delta'):
                         # Audio from OpenAI
                         try:
@@ -125,12 +173,68 @@ async def handle_media_stream(websocket: WebSocket):
                         except Exception as e:
                             print(f"Error processing audio data: {e}")
             except Exception as e:
-                print(f"Error in send_to_twilio: {e}")
+                print(f"Error in Sending to Phone: {e}")
 
         await asyncio.gather(receive_from_twilio(), send_to_twilio())
 
+async def handle_response_done(response, openai_ws):
+    """Handle the response.done event and process any function calls."""
+    print("Handling response.done:", response)
+    output_items = response.get('response', {}).get('output', [])
+    for item in output_items:
+        if item.get('type') == 'function_call':
+            await process_function_call(item, openai_ws)
+
+async def process_function_call(item, openai_ws):
+    """Process a function call item and send the result back to OpenAI."""
+    function_name = item.get('name')
+    arguments = json.loads(item.get('arguments', '{}'))
+    call_id = item.get('call_id')
+
+    print(f"Processing function call: {function_name}")
+    print(f"Arguments: {arguments}")
+
+    result = await call_tool(function_name, arguments)
+
+    # Send the function call result back to OpenAI
+    await openai_ws.send(json.dumps({
+        "type": "conversation.item.create",
+        "item": {
+            "type": "function_call_output",
+            "call_id": call_id,
+            "output": json.dumps(result)
+        }
+    }))
+
+    # Create a new response after sending the function call result
+    await openai_ws.send(json.dumps({
+        "type": "response.create"
+    }))
+
+async def call_tool(function_name, arguments):
+    """Call the appropriate tool function and return the result."""
+    tool = next((t for t in tools if t[0]['name'] == function_name), None)
+    if not tool:
+        return {"error": f"Function {function_name} not found"}
+    
+    try:
+        # Assuming the tool function is the second element in the tuple
+        result = await tool[1](**arguments)
+        return result
+    except Exception as e:
+        return {"error": str(e)}
+
 async def send_session_update(openai_ws):
     """Send session update to OpenAI WebSocket."""
+    global tools
+    print(f"Formatted tools: {tools}")
+    
+    use_tools = [
+        {**tool[0], "type": "function"}
+        for tool in tools
+        if isinstance(tool, tuple) and len(tool) > 0 and isinstance(tool[0], dict)
+    ]
+    
     session_update = {
         "type": "session.update",
         "session": {
@@ -142,10 +246,8 @@ async def send_session_update(openai_ws):
             },
             "input_audio_format": "g711_ulaw",
             "output_audio_format": "g711_ulaw",
-            # "input_audio_transcription": { "model": 'whisper-1' },
-            # "transcription_models": [{"model": "whisper-1"}],
             "voice": VOICE,
-            "tools": [],
+            "tools": use_tools,
             "tool_choice": "auto",
             "instructions": SYSTEM_MESSAGE,
             "modalities": ["text", "audio"],

praisonai/cli.py

@@ -17,7 +17,12 @@ import shutil
 import subprocess
 import logging
 import importlib
-import praisonai.api.call as call_module
+try:
+    import praisonai.api.call as call_module
+    CALL_MODULE_AVAILABLE = True
+except ImportError:
+    CALL_MODULE_AVAILABLE = False
+    
 logging.basicConfig(level=os.environ.get('LOGLEVEL', 'INFO'), format='%(asctime)s - %(levelname)s - %(message)s')
 
 try:

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

{praisonai-0.1.6.dist-info → praisonai-0.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PraisonAI
-Version: 0.1.6
+Version: 0.1.8
 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.1.6.dist-info → praisonai-0.1.8.dist-info}/RECORD

@@ -1,11 +1,11 @@
 praisonai/__init__.py,sha256=JrgyPlzZfLlozoW7SHZ1nVJ63rLPR3ki2k5ZPywYrnI,175
 praisonai/__main__.py,sha256=MVgsjMThjBexHt4nhd760JCqvP4x0IQcwo8kULOK4FQ,144
-praisonai/agents_generator.py,sha256=8d1WRbubvEkBrW1HZ7_xnGyqgJi0yxmXa3MgTIqef1c,19127
-praisonai/api/call.py,sha256=6yVJqczuCnVYk3EIKsPzLYjSLC79IwD5o57mpGz9dxo,7277
+praisonai/agents_generator.py,sha256=SltYtjFrADEkYkbzNrWHT6tIxpyjAMBbwH2czZOGPhg,19109
+praisonai/api/call.py,sha256=iHdAlgIH_oTsEbjaGGu1Jjo6DTfMR-SfFdtSxnOLCeY,11032
 praisonai/auto.py,sha256=9spTXqj47Hmmqv5QHRYE_RzSVHH_KoPbaZjskUj2UcE,7895
 praisonai/chainlit_ui.py,sha256=bNR7s509lp0I9JlJNvwCZRUZosC64qdvlFCt8NmFamQ,12216
-praisonai/cli.py,sha256=BlVEsQW45pbYoqAtm-EiNpoHt8BFKX5frsbJ9s-tko0,20667
-praisonai/deploy.py,sha256=tXQBqMtNxOYuTKBeUkhz6UTAfJMhYJ1CBtw5W8Jc-pM,6027
+praisonai/cli.py,sha256=8q8nfvc9ztKW3AQ0NpFnby739MCcvaqF3Y14BrpuCXo,20768
+praisonai/deploy.py,sha256=MY4PCgp8iSA-T1ovph00zZ8dTjIL7gZ7sjNENOOZ5sg,6027
 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
@@ -46,8 +46,8 @@ praisonai/ui/realtimeclient/realtimedocs.txt,sha256=hmgd8Uwy2SkjSndyyF_-ZOaNxiyH
 praisonai/ui/realtimeclient/tools.py,sha256=IJOYwVOBW5Ocn5_iV9pFkmSKR3WU3YpX3kwF0I3jikQ,7855
 praisonai/ui/sql_alchemy.py,sha256=kf025P_37C505YDDJZ-dPSmN_d62J2DCrkxbDAzXyrM,29884
 praisonai/version.py,sha256=ugyuFliEqtAwQmH4sTlc16YXKYbFWDmfyk87fErB8-8,21
-praisonai-0.1.6.dist-info/LICENSE,sha256=kqvFysVlnFxYOu0HxCe2HlmZmJtdmNGOxWRRkT9TsWc,1035
-praisonai-0.1.6.dist-info/METADATA,sha256=8_nLIQ4kHTff3Kmb3RMWJdf3bkKDw7sOWifoC5jVF2I,13291
-praisonai-0.1.6.dist-info/WHEEL,sha256=Ie8mbC-etDUh6aDhzdvvvp_A-4mQQX7whlOFBnSJhcE,110
-praisonai-0.1.6.dist-info/entry_points.txt,sha256=LT8Ie95dy4JkUlpgS0YyvD4Tf9zTstjaXISJnijDxoM,172
-praisonai-0.1.6.dist-info/RECORD,,
+praisonai-0.1.8.dist-info/LICENSE,sha256=kqvFysVlnFxYOu0HxCe2HlmZmJtdmNGOxWRRkT9TsWc,1035
+praisonai-0.1.8.dist-info/METADATA,sha256=C2u3l914yI9enMAZLYuaaAY72fRXXwZ8e--VkGn3KdQ,13291
+praisonai-0.1.8.dist-info/WHEEL,sha256=Ie8mbC-etDUh6aDhzdvvvp_A-4mQQX7whlOFBnSJhcE,110
+praisonai-0.1.8.dist-info/entry_points.txt,sha256=LT8Ie95dy4JkUlpgS0YyvD4Tf9zTstjaXISJnijDxoM,172
+praisonai-0.1.8.dist-info/RECORD,,