universal-mcp 0.1.13rc2__tar.gz → 0.1.13rc7__tar.gz

{universal_mcp-0.1.13rc2 → universal_mcp-0.1.13rc7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: universal-mcp
-Version: 0.1.13rc2
+Version: 0.1.13rc7
 Summary: Universal MCP acts as a middle ware for your API applications. It can store your credentials, authorize, enable disable apps on the fly and much more.
 Author-email: Manoj Bajaj <manojbajaj95@gmail.com>
 License: MIT
@@ -31,6 +31,45 @@ Requires-Dist: python-dotenv>=1.0.1; extra == 'all'
 Requires-Dist: ruff>=0.11.4; extra == 'all'
 Requires-Dist: streamlit>=1.44.1; extra == 'all'
 Requires-Dist: watchdog>=6.0.0; extra == 'all'
+Provides-Extra: applications
+Requires-Dist: universal-mcp-ahrefs; extra == 'applications'
+Requires-Dist: universal-mcp-cal-com-v2; extra == 'applications'
+Requires-Dist: universal-mcp-calendly; extra == 'applications'
+Requires-Dist: universal-mcp-clickup; extra == 'applications'
+Requires-Dist: universal-mcp-coda; extra == 'applications'
+Requires-Dist: universal-mcp-crustdata; extra == 'applications'
+Requires-Dist: universal-mcp-e2b; extra == 'applications'
+Requires-Dist: universal-mcp-elevenlabs; extra == 'applications'
+Requires-Dist: universal-mcp-falai; extra == 'applications'
+Requires-Dist: universal-mcp-figma; extra == 'applications'
+Requires-Dist: universal-mcp-firecrawl; extra == 'applications'
+Requires-Dist: universal-mcp-github; extra == 'applications'
+Requires-Dist: universal-mcp-gong; extra == 'applications'
+Requires-Dist: universal-mcp-google-calendar; extra == 'applications'
+Requires-Dist: universal-mcp-google-docs; extra == 'applications'
+Requires-Dist: universal-mcp-google-drive; extra == 'applications'
+Requires-Dist: universal-mcp-google-mail; extra == 'applications'
+Requires-Dist: universal-mcp-google-sheet; extra == 'applications'
+Requires-Dist: universal-mcp-hashnode; extra == 'applications'
+Requires-Dist: universal-mcp-heygen; extra == 'applications'
+Requires-Dist: universal-mcp-mailchimp; extra == 'applications'
+Requires-Dist: universal-mcp-markitdown; extra == 'applications'
+Requires-Dist: universal-mcp-neon; extra == 'applications'
+Requires-Dist: universal-mcp-notion; extra == 'applications'
+Requires-Dist: universal-mcp-perplexity; extra == 'applications'
+Requires-Dist: universal-mcp-reddit; extra == 'applications'
+Requires-Dist: universal-mcp-replicate; extra == 'applications'
+Requires-Dist: universal-mcp-resend; extra == 'applications'
+Requires-Dist: universal-mcp-retell; extra == 'applications'
+Requires-Dist: universal-mcp-rocketlane; extra == 'applications'
+Requires-Dist: universal-mcp-serpapi; extra == 'applications'
+Requires-Dist: universal-mcp-shortcut; extra == 'applications'
+Requires-Dist: universal-mcp-spotify; extra == 'applications'
+Requires-Dist: universal-mcp-supabase; extra == 'applications'
+Requires-Dist: universal-mcp-tavily; extra == 'applications'
+Requires-Dist: universal-mcp-wrike; extra == 'applications'
+Requires-Dist: universal-mcp-youtube; extra == 'applications'
+Requires-Dist: universal-mcp-zenquotes; extra == 'applications'
 Provides-Extra: dev
 Requires-Dist: litellm>=1.30.7; extra == 'dev'
 Requires-Dist: pyright>=1.1.398; extra == 'dev'

{universal_mcp-0.1.13rc2 → universal_mcp-0.1.13rc7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "universal-mcp"
-version = "0.1.13-rc2"
+version = "0.1.13-rc7"
 description = "Universal MCP acts as a middle ware for your API applications. It can store your credentials, authorize, enable disable apps on the fly and much more."
 readme = "README.md"
 authors = [
@@ -57,6 +57,46 @@ all = [
     "pyright>=1.1.398",
     "litellm>=1.30.7",
 ]
+applications = [
+    "universal-mcp-ahrefs",
+    "universal-mcp-cal-com-v2",
+    "universal-mcp-calendly",
+    "universal-mcp-clickup",
+    "universal-mcp-coda",
+    "universal-mcp-crustdata",
+    "universal-mcp-e2b",
+    "universal-mcp-elevenlabs",
+    "universal-mcp-falai",
+    "universal-mcp-figma",
+    "universal-mcp-firecrawl",
+    "universal-mcp-github",
+    "universal-mcp-gong",
+    "universal-mcp-google-calendar",
+    "universal-mcp-google-docs",
+    "universal-mcp-google-drive",
+    "universal-mcp-google-mail",
+    "universal-mcp-google-sheet",
+    "universal-mcp-hashnode",
+    "universal-mcp-heygen",
+    "universal-mcp-mailchimp",
+    "universal-mcp-markitdown",
+    "universal-mcp-neon",
+    "universal-mcp-notion",
+    "universal-mcp-perplexity",
+    "universal-mcp-reddit",
+    "universal-mcp-replicate",
+    "universal-mcp-resend",
+    "universal-mcp-retell",
+    "universal-mcp-rocketlane",
+    "universal-mcp-serpapi",
+    "universal-mcp-shortcut",
+    "universal-mcp-spotify",
+    "universal-mcp-supabase",
+    "universal-mcp-tavily",
+    "universal-mcp-wrike",
+    "universal-mcp-youtube",
+    "universal-mcp-zenquotes",
+]
 
 [project.scripts]
 universal_mcp = "universal_mcp.cli:app"

{universal_mcp-0.1.13rc2 → universal_mcp-0.1.13rc7}/src/playground/__main__.py

@@ -11,7 +11,9 @@ def main():
     # Ask the user if they want to run the MCP server
     run_mcp_server = input("Do you want to run the MCP server? (y/n): ")
     if run_mcp_server == "y":
-        mcp_process = subprocess.Popen(["universal_mcp", "run", "-c", "local_config.json"])
+        mcp_process = subprocess.Popen(
+            ["universal_mcp", "run", "-c", "local_config.json"]
+        )
         processes.append(mcp_process)
         time.sleep(6)  # Give MCP server time to start
         logger.info("MCP server started")

{universal_mcp-0.1.13rc2 → universal_mcp-0.1.13rc7}/src/playground/agents/react.py

@@ -12,7 +12,7 @@ async def load_tools():
     async with MultiServerMCPClient(
         {
             "agentr": {
-                "url": "http://localhost:8000/sse",
+                "url": "http://localhost:8005/sse",
                 "transport": "sse",
             },
         }

{universal_mcp-0.1.13rc2 → universal_mcp-0.1.13rc7}/src/playground/client.py

@@ -1,35 +1,29 @@
-from contextlib import asynccontextmanager
 import json
-import os
 from collections.abc import AsyncGenerator, Generator
+from contextlib import asynccontextmanager
 from typing import Any
-from uuid import UUID, uuid4
+from uuid import uuid4
 
 from langchain_core.messages import HumanMessage
 from langchain_core.runnables import RunnableConfig
+from langgraph.types import Command
 
 from playground.agents.react import create_agent
-from playground.memory import initialize_database
 from playground.schema import (
     ChatHistory,
-    ChatHistoryInput,
     ChatMessage,
-    StreamInput,
-    UserInput,
 )
 from playground.utils import (
     convert_message_content_to_string,
     langchain_to_chat_message,
     remove_tool_calls,
 )
-from langgraph.types import Command
+
 
 @asynccontextmanager
 async def create_agent_client():
-    async with create_agent() as react_agent, initialize_database() as saver:
-        await saver.setup()
+    async with create_agent() as react_agent:
         agent = react_agent
-        agent.checkpointer = saver
         client = AgentClient(agent=agent)
         yield client
 
@@ -41,10 +35,7 @@ class AgentClientError(Exception):
 class AgentClient:
     """Client for interacting with the agent directly."""
 
-    def __init__(
-        self,
-        agent
-    ) -> None:
+    def __init__(self, agent) -> None:
         """
         Initialize the client.
 
@@ -82,7 +73,9 @@ class AgentClient:
         configurable = {"thread_id": thread_id}
         if agent_config:
             if overlap := configurable.keys() & agent_config.keys():
-                raise AgentClientError(f"agent_config contains reserved keys: {overlap}")
+                raise AgentClientError(
+                    f"agent_config contains reserved keys: {overlap}"
+                )
             configurable.update(agent_config)
 
         kwargs = {
@@ -92,7 +85,7 @@ class AgentClient:
                 run_id=run_id,
             ),
         }
-        
+
         try:
             response = await self.agent.ainvoke(**kwargs)
             output = langchain_to_chat_message(response["messages"][-1])
@@ -124,7 +117,9 @@ class AgentClient:
         configurable = {"thread_id": thread_id}
         if agent_config:
             if overlap := configurable.keys() & agent_config.keys():
-                raise AgentClientError(f"agent_config contains reserved keys: {overlap}")
+                raise AgentClientError(
+                    f"agent_config contains reserved keys: {overlap}"
+                )
             configurable.update(agent_config)
 
         kwargs = {
@@ -134,7 +129,7 @@ class AgentClient:
                 run_id=run_id,
             ),
         }
-        
+
         try:
             response = self.agent.invoke(**kwargs)
             output = langchain_to_chat_message(response["messages"][-1])
@@ -146,7 +141,7 @@ class AgentClient:
     def _parse_stream_line(self, line: str) -> ChatMessage | str | None:
         """
         Parse a line from the stream.
-        
+
         This method is kept for compatibility but is no longer used
         since we're directly streaming from the agent.
         """
@@ -201,7 +196,9 @@ class AgentClient:
         configurable = {"thread_id": thread_id}
         if agent_config:
             if overlap := configurable.keys() & agent_config.keys():
-                raise AgentClientError(f"agent_config contains reserved keys: {overlap}")
+                raise AgentClientError(
+                    f"agent_config contains reserved keys: {overlap}"
+                )
             configurable.update(agent_config)
 
         kwargs = {
@@ -211,7 +208,7 @@ class AgentClient:
                 run_id=run_id,
             ),
         }
-        
+
         try:
             for event in self.agent.stream_events(**kwargs, version="v2"):
                 if not event:
@@ -226,12 +223,16 @@ class AgentClient:
                     and any(t.startswith("graph:step:") for t in event.get("tags", []))
                 ):
                     if isinstance(event["data"]["output"], Command):
-                        new_messages = event["data"]["output"].update.get("messages", [])
+                        new_messages = event["data"]["output"].update.get(
+                            "messages", []
+                        )
                     elif "messages" in event["data"]["output"]:
                         new_messages = event["data"]["output"]["messages"]
 
                 # Also yield intermediate messages from agents.utils.CustomData.adispatch().
-                if event["event"] == "on_custom_event" and "custom_data_dispatch" in event.get(
+                if event[
+                    "event"
+                ] == "on_custom_event" and "custom_data_dispatch" in event.get(
                     "tags", []
                 ):
                     new_messages = [event["data"]]
@@ -240,14 +241,11 @@ class AgentClient:
                     try:
                         chat_message = langchain_to_chat_message(message)
                         chat_message.run_id = str(run_id)
-                    except Exception as e:
+                    except Exception:
                         yield f"data: {json.dumps({'type': 'error', 'content': 'Unexpected error'})}\n\n"
                         continue
                     # LangGraph re-sends the input message, which feels weird, so drop it
-                    if (
-                        chat_message.type == "human"
-                        and chat_message.content == message
-                    ):
+                    if chat_message.type == "human" and chat_message.content == message:
                         continue
                     yield chat_message
 
@@ -296,7 +294,9 @@ class AgentClient:
         configurable = {"thread_id": thread_id}
         if agent_config:
             if overlap := configurable.keys() & agent_config.keys():
-                raise AgentClientError(f"agent_config contains reserved keys: {overlap}")
+                raise AgentClientError(
+                    f"agent_config contains reserved keys: {overlap}"
+                )
             configurable.update(agent_config)
 
         kwargs = {
@@ -306,7 +306,7 @@ class AgentClient:
                 run_id=run_id,
             ),
         }
-        
+
         try:
             async for event in self.agent.astream_events(**kwargs, version="v2"):
                 if not event:
@@ -321,12 +321,16 @@ class AgentClient:
                     and any(t.startswith("graph:step:") for t in event.get("tags", []))
                 ):
                     if isinstance(event["data"]["output"], Command):
-                        new_messages = event["data"]["output"].update.get("messages", [])
+                        new_messages = event["data"]["output"].update.get(
+                            "messages", []
+                        )
                     elif "messages" in event["data"]["output"]:
                         new_messages = event["data"]["output"]["messages"]
 
                 # Also yield intermediate messages from agents.utils.CustomData.adispatch().
-                if event["event"] == "on_custom_event" and "custom_data_dispatch" in event.get(
+                if event[
+                    "event"
+                ] == "on_custom_event" and "custom_data_dispatch" in event.get(
                     "tags", []
                 ):
                     new_messages = [event["data"]]
@@ -335,14 +339,11 @@ class AgentClient:
                     try:
                         chat_message = langchain_to_chat_message(message)
                         chat_message.run_id = str(run_id)
-                    except Exception as e:
+                    except Exception:
                         yield f"data: {json.dumps({'type': 'error', 'content': 'Unexpected error'})}\n\n"
                         continue
                     # LangGraph re-sends the input message, which feels weird, so drop it
-                    if (
-                        chat_message.type == "human"
-                        and chat_message.content == message
-                    ):
+                    if chat_message.type == "human" and chat_message.content == message:
                         continue
                     yield chat_message
 
@@ -380,9 +381,7 @@ class AgentClient:
                 )
             )
             messages = state_snapshot.values["messages"]
-            chat_messages = [
-                langchain_to_chat_message(m) for m in messages
-            ]
+            chat_messages = [langchain_to_chat_message(m) for m in messages]
             return ChatHistory(messages=chat_messages)
         except Exception as e:
             raise AgentClientError(f"Error getting history: {e}") from e

{universal_mcp-0.1.13rc2 → universal_mcp-0.1.13rc7}/src/playground/streamlit.py

@@ -6,7 +6,6 @@ from collections.abc import AsyncGenerator
 from pathlib import Path
 
 import streamlit as st
-from dotenv import load_dotenv
 from pydantic import ValidationError
 from streamlit.runtime.scriptrunner import get_script_run_ctx
 
@@ -21,6 +20,7 @@ APP_TITLE = "MCP Playground"
 APP_ICON = "🧰"
 use_streaming = True
 
+
 # --- Function to handle unique filename generation ---
 def get_unique_filepath(upload_dir: Path, filename: str) -> Path:
     """Checks if a file exists and returns a unique path if needed."""
@@ -38,6 +38,7 @@ def get_unique_filepath(upload_dir: Path, filename: str) -> Path:
             return new_filepath
         counter += 1
 
+
 async def main() -> None:
     # Configure page layout
     st.set_page_config(
@@ -49,7 +50,8 @@ async def main() -> None:
     )
 
     # Custom CSS for chat layout and styling
-    st.markdown("""
+    st.markdown(
+        """
         <style>
         /* Hide Streamlit default header and footer */
         #MainMenu, header, footer {
@@ -158,7 +160,9 @@ async def main() -> None:
             border-bottom-left-radius: 0.25rem;
         }
         </style>
-    """, unsafe_allow_html=True)
+    """,
+        unsafe_allow_html=True,
+    )
 
     # Minimal toolbar
     if st.get_option("client.toolbarMode") != "minimal":
@@ -218,7 +222,7 @@ async def main() -> None:
                 "Upload a file to process",
                 type=None,
                 key="file_uploader",
-                help="Upload any file to process with the AI assistant"
+                help="Upload any file to process with the AI assistant",
             )
 
         # Handle file upload state
@@ -234,7 +238,8 @@ async def main() -> None:
                     icon="📄",
                 )
         elif (
-            uploaded_file is None and st.session_state.get("uploaded_file_obj") is not None
+            uploaded_file is None
+            and st.session_state.get("uploaded_file_obj") is not None
         ):
             st.session_state.uploaded_file_obj = None
             st.session_state.file_processed = False
@@ -260,19 +265,21 @@ async def main() -> None:
                     yield m
 
             await draw_messages(amessage_iter())
-            st.markdown('</div>', unsafe_allow_html=True)
+            st.markdown("</div>", unsafe_allow_html=True)
 
         # Chat input area
         st.markdown('<div class="chat-input-area">', unsafe_allow_html=True)
         if user_input := st.chat_input(
-            "Enter message or upload a file and describe task...",
-            key="chat_input"
+            "Enter message or upload a file and describe task...", key="chat_input"
         ):
             final_message_content = user_input
             display_content = user_input
 
             # --- Handle File Upload Integration ---
-            if st.session_state.uploaded_file_obj and not st.session_state.file_processed:
+            if (
+                st.session_state.uploaded_file_obj
+                and not st.session_state.file_processed
+            ):
                 uploaded_file_obj = st.session_state.uploaded_file_obj
                 original_filename = uploaded_file_obj.name
                 try:
@@ -322,10 +329,11 @@ async def main() -> None:
                 st.error(
                     f"An unexpected error occurred during agent communication: {e}"
                 )
-        st.markdown('</div>', unsafe_allow_html=True)
+        st.markdown("</div>", unsafe_allow_html=True)
 
         # End chat app wrapper
-        st.markdown('</div>', unsafe_allow_html=True)
+        st.markdown("</div>", unsafe_allow_html=True)
+
 
 async def draw_messages(
     messages_agen: AsyncGenerator[ChatMessage | str, None],
@@ -419,6 +427,7 @@ async def draw_messages(
                 st.write(msg)
                 st.stop()
 
+
 async def handle_feedback() -> None:
     """Draws a feedback widget and records feedback from the user."""
     if "last_feedback" not in st.session_state:
@@ -444,6 +453,7 @@ async def handle_feedback() -> None:
         st.session_state.last_feedback = (latest_run_id, feedback)
         st.toast("Feedback recorded", icon="⭐")
 
+
 if __name__ == "__main__":
     asyncio.run(main())
 # End of Selectio

{universal_mcp-0.1.13rc2 → universal_mcp-0.1.13rc7}/src/tests/test_api_generator.py

@@ -53,7 +53,7 @@ def sample_schema(temp_dir):
 @pytest.mark.asyncio
 async def test_generate_api_without_output(sample_schema):
     """Test API generation without output file (return code only)."""
-    result =generate_api_from_schema(
+    result = generate_api_from_schema(
         schema_path=sample_schema, output_path=None, add_docstrings=False
     )
 
@@ -76,8 +76,8 @@ async def test_generate_api_with_output(sample_schema, temp_dir):
         schema_path=sample_schema, output_path=output_path, add_docstrings=True
     )
 
-    assert "app_file" is not None
-    assert "readme_file" is not None
+    assert "app_file" != None
+    assert "readme_file" != None
 
     assert app_file.exists()
     content = app_file.read_text()
@@ -138,7 +138,6 @@ async def test_generate_api_with_docstrings(sample_schema, temp_dir):
     assert "Tags:" in content
 
 
-
 @pytest.mark.asyncio
 async def test_generate_api_without_docstrings(sample_schema, temp_dir):
     """Test API generation without docstring generation."""

{universal_mcp-0.1.13rc2 → universal_mcp-0.1.13rc7}/src/tests/test_api_integration.py

@@ -1,6 +1,6 @@
 import pytest
 
-from universal_mcp.applications.perplexity.app import PerplexityApp
+from universal_mcp.applications import app_from_slug
 from universal_mcp.exceptions import NotAuthorizedError
 from universal_mcp.integrations import ApiKeyIntegration
 from universal_mcp.stores import MemoryStore
@@ -14,7 +14,7 @@ def test_perplexity_api_no_key():
     integration = ApiKeyIntegration("PERPLEXITY", store=store)
 
     # Create Perplexity app with the integration
-    app = PerplexityApp(integration=integration)
+    app = app_from_slug("perplexity")(integration=integration)
 
     # Try to make a chat request without setting API key
     with pytest.raises(NotAuthorizedError) as exc_info:

{universal_mcp-0.1.13rc2 → universal_mcp-0.1.13rc7}/src/tests/test_applications.py

@@ -59,4 +59,4 @@ def test_application(app_name):
     assert isinstance(tools[0], Callable)
     for tool in tools:
         assert tool.__name__ is not None
-        assert tool.__doc__ is not None
+        assert tool.__doc__ is not None

{universal_mcp-0.1.13rc2 → universal_mcp-0.1.13rc7}/src/tests/test_stores.py

@@ -67,6 +67,7 @@ class TestEnvironmentStore:
 
 
 # Test KeyringStore
+@pytest.mark.skip(reason="Skipping KeyringStore tests")
 class TestKeyringStore:
     @pytest.fixture
     def store(self):

{universal_mcp-0.1.13rc2 → universal_mcp-0.1.13rc7}/src/universal_mcp/applications/__init__.py

@@ -1,4 +1,5 @@
 import importlib
+import subprocess
 
 from loguru import logger
 
@@ -7,12 +8,12 @@ from universal_mcp.applications.application import (
     BaseApplication,
     GraphQLApplication,
 )
-import subprocess
 
 # Name are in the format of "app-name", eg, google-calendar
 # Folder name is "app_name", eg, google_calendar
 # Class name is NameApp, eg, GoogleCalendarApp
 
+
 def _import_class(module_path: str, class_name: str):
     """
     Helper to import a class by name from a module.
@@ -27,7 +28,10 @@ def _import_class(module_path: str, class_name: str):
         return getattr(module, class_name)
     except AttributeError as e:
         logger.error(f"Class '{class_name}' not found in module '{module_path}'")
-        raise ModuleNotFoundError(f"Class '{class_name}' not found in module '{module_path}'") from e
+        raise ModuleNotFoundError(
+            f"Class '{class_name}' not found in module '{module_path}'"
+        ) from e
+
 
 def _install_package(slug_clean: str):
     """
@@ -40,10 +44,13 @@ def _install_package(slug_clean: str):
         subprocess.check_call(cmd)
     except subprocess.CalledProcessError as e:
         logger.error(f"Installation failed for '{slug_clean}': {e}")
-        raise ModuleNotFoundError(f"Installation failed for package '{slug_clean}'") from e
+        raise ModuleNotFoundError(
+            f"Installation failed for package '{slug_clean}'"
+        ) from e
     else:
         logger.info(f"Package '{slug_clean}' installed successfully")
 
+
 def app_from_slug(slug: str):
     """
     Dynamically resolve and return the application class for the given slug.
@@ -54,19 +61,26 @@ def app_from_slug(slug: str):
     package_prefix = f"universal_mcp_{slug_clean.replace('-', '_')}"
     module_path = f"{package_prefix}.app"
 
-    logger.info(f"Resolving app for slug '{slug}' → module '{module_path}', class '{class_name}'")
+    logger.info(
+        f"Resolving app for slug '{slug}' → module '{module_path}', class '{class_name}'"
+    )
     try:
         return _import_class(module_path, class_name)
     except ModuleNotFoundError as orig_err:
-        logger.warning(f"Module '{module_path}' not found locally: {orig_err}. Installing...")
+        logger.warning(
+            f"Module '{module_path}' not found locally: {orig_err}. Installing..."
+        )
         _install_package(slug_clean)
         # Retry import after installation
         try:
             return _import_class(module_path, class_name)
         except ModuleNotFoundError as retry_err:
-            logger.error(f"Still cannot import '{module_path}' after installation: {retry_err}")
+            logger.error(
+                f"Still cannot import '{module_path}' after installation: {retry_err}"
+            )
             raise
 
+
 __all__ = [
     "app_from_slug",
     "BaseApplication",