universal-mcp 0.1.13rc3__tar.gz → 0.1.13rc14__tar.gz

{universal_mcp-0.1.13rc3 → universal_mcp-0.1.13rc14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: universal-mcp
-Version: 0.1.13rc3
+Version: 0.1.13rc14
 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
@@ -18,60 +18,9 @@ Requires-Dist: pydantic>=2.11.1
 Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: rich>=14.0.0
 Requires-Dist: typer>=0.15.2
-Provides-Extra: all
-Requires-Dist: langchain-mcp-adapters>=0.0.3; extra == 'all'
-Requires-Dist: langchain-openai>=0.3.12; extra == 'all'
-Requires-Dist: langgraph-checkpoint-sqlite>=2.0.6; extra == 'all'
-Requires-Dist: langgraph>=0.3.24; extra == 'all'
-Requires-Dist: litellm>=1.30.7; extra == 'all'
-Requires-Dist: pyright>=1.1.398; extra == 'all'
-Requires-Dist: pytest-asyncio>=0.26.0; extra == 'all'
-Requires-Dist: pytest>=8.3.5; extra == 'all'
-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: pre-commit>=4.2.0; extra == 'dev'
 Requires-Dist: pyright>=1.1.398; extra == 'dev'
 Requires-Dist: pytest-asyncio>=0.26.0; extra == 'dev'
 Requires-Dist: pytest>=8.3.5; extra == 'dev'

universal_mcp-0.1.13rc14/pyproject.toml

@@ -0,0 +1,127 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "universal-mcp"
+version = "0.1.13-rc14"
+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 = [
+    { name = "Manoj Bajaj", email = "manojbajaj95@gmail.com" }
+]
+requires-python = ">=3.11"
+license = { text = "MIT" }
+dependencies = [
+    "Jinja2>=3.1.3",
+    "cookiecutter>=2.6.0",
+    "gql[all]>=3.5.2",
+    "keyring>=25.6.0",
+    "litellm>=1.30.7",
+    "loguru>=0.7.3",
+    "mcp>=1.6.0",
+    "posthog>=3.24.0",
+    "pydantic>=2.11.1",
+    "pydantic-settings>=2.8.1",
+    "pyyaml>=6.0.2",
+    "rich>=14.0.0",
+    "typer>=0.15.2",
+]
+
+[project.optional-dependencies]
+dev = [
+    "litellm>=1.30.7",
+    "pre-commit>=4.2.0",
+    "pyright>=1.1.398",
+    "pytest>=8.3.5",
+    "pytest-asyncio>=0.26.0",
+    "ruff>=0.11.4",
+]
+playground = [
+    "langchain-mcp-adapters>=0.0.3",
+    "langchain-openai>=0.3.12",
+    "langgraph>=0.3.24",
+    "langgraph-checkpoint-sqlite>=2.0.6",
+    "python-dotenv>=1.0.1",
+    "streamlit>=1.44.1",
+    "watchdog>=6.0.0",
+]
+
+[project.scripts]
+universal_mcp = "universal_mcp.cli:app"
+
+# ------------------------------
+# Hatch build configuration
+# ------------------------------
+[tool.hatch.build.targets.sdist]
+include = [
+    "/src",
+    "/templates", # Important: Include the templates directory!
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/universal_mcp"]
+
+[tool.hatch.build.targets.wheel.shared-data]
+"templates" = "universal_mcp/templates"
+
+# ------------------------------
+# Tool configurations
+# ------------------------------
+[tool.ruff]
+exclude = [
+    ".bzr",
+    ".direnv",
+    ".eggs",
+    ".git",
+    ".git-rewrite",
+    ".hg",
+    ".ipynb_checkpoints",
+    ".mypy_cache",
+    ".nox",
+    ".pants.d",
+    ".pyenv",
+    ".pytest_cache",
+    ".pytype",
+    ".ruff_cache",
+    ".svn",
+    ".tox",
+    ".venv",
+    ".vscode",
+    "__pypackages__",
+    "_build",
+    "buck-out",
+    "build",
+    "dist",
+    "node_modules",
+    "site-packages",
+    "venv",
+]
+line-length = 88
+indent-width = 4
+target-version = "py312"
+
+[tool.ruff.lint]
+select = [
+    "E",    # pycodestyle
+    "F",    # Pyflakes
+    "UP",   # pyupgrade
+    "B",    # flake8-bugbear
+    "SIM",  # flake8-simplify
+    "I",    # isort
+]
+ignore = ['E501', 'B008']
+fixable = ["ALL"]
+unfixable = []
+dummy-variable-rgx = "^(_+|(_+[a-zA-Z0-9_]*[a-zA-Z0-9]+?))$"
+
+[tool.ruff.format]
+quote-style = "double"
+indent-style = "space"
+skip-magic-trailing-comma = false
+line-ending = "auto"
+docstring-code-line-length = "dynamic"
+
+[tool.pytest.ini_options]
+asyncio_mode = "strict"
+asyncio_default_fixture_loop_scope = "function"

{universal_mcp-0.1.13rc3 → universal_mcp-0.1.13rc14}/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.13rc3 → universal_mcp-0.1.13rc14}/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.13rc3 → universal_mcp-0.1.13rc14}/src/playground/streamlit.py

@@ -20,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."""
@@ -37,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(
@@ -48,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 {
@@ -157,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":
@@ -217,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
@@ -233,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
@@ -259,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:
@@ -321,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],
@@ -418,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:
@@ -443,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.13rc3 → universal_mcp-0.1.13rc14}/src/tests/test_api_generator.py

@@ -53,9 +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(
-        schema_path=sample_schema, output_path=None, add_docstrings=False
-    )
+    result = generate_api_from_schema(schema_path=sample_schema, output_path=None)
 
     assert "code" in result
     assert isinstance(result["code"], str)
@@ -72,12 +70,12 @@ async def test_generate_api_with_output(sample_schema, temp_dir):
     """Test API generation with output file."""
     output_path = temp_dir / "test.py"
 
-    app_file, readme_file = generate_api_from_schema(
-        schema_path=sample_schema, output_path=output_path, add_docstrings=True
+    app_file = generate_api_from_schema(
+        schema_path=sample_schema, output_path=output_path
     )
 
-    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()
@@ -87,14 +85,6 @@ async def test_generate_api_with_output(sample_schema, temp_dir):
     assert "def test_operation" in content
     assert "def list_tools" in content
 
-    # Verify README exists and contains expected content
-    if readme_file:
-        assert readme_file.exists()
-        readme_content = readme_file.read_text()
-        assert "Test MCP Server" in readme_content
-        assert "Tool List" in readme_content
-        assert "test_operation" in readme_content
-
 
 @pytest.mark.asyncio
 async def test_generate_api_invalid_schema(temp_dir):
@@ -115,41 +105,16 @@ async def test_generate_api_nonexistent_schema():
         )
 
 
-@pytest.mark.asyncio
-async def test_generate_api_with_docstrings(sample_schema, temp_dir):
-    """Test API generation with docstring generation."""
-    output_path = temp_dir / "test_with_docs.py"
-
-    app_file, readme_file = generate_api_from_schema(
-        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.exists()
-
-    # Check if docstrings were added
-    content = app_file.read_text()
-    # Check for required imports and class structure
-    assert "from universal_mcp.applications import APIApplication" in content
-    assert "from universal_mcp.integrations import Integration" in content
-    assert "def test_operation" in content
-    assert '"""' in content  # Basic check for docstring presence
-    assert "Tags:" in content
-
-
-
 @pytest.mark.asyncio
 async def test_generate_api_without_docstrings(sample_schema, temp_dir):
     """Test API generation without docstring generation."""
     output_path = temp_dir / "test_without_docs.py"
 
-    app_file, readme_file = generate_api_from_schema(
-        schema_path=sample_schema, output_path=output_path, add_docstrings=False
+    app_file = generate_api_from_schema(
+        schema_path=sample_schema, output_path=output_path
     )
 
     assert app_file is not None
-    assert readme_file is not None
     assert app_file.exists()
 
     # Verify the app was generated
@@ -204,12 +169,11 @@ async def test_generate_api_with_complex_schema(temp_dir):
         json.dump(schema, f)
 
     output_path = temp_dir / "complex.py"
-    app_file, readme_file = generate_api_from_schema(
-        schema_path=schema_file, output_path=output_path, add_docstrings=True
+    app_file = generate_api_from_schema(
+        schema_path=schema_file, output_path=output_path
     )
 
     assert app_file is not None
-    assert readme_file is not None
     assert app_file.exists()
 
     content = app_file.read_text()
@@ -232,13 +196,3 @@ async def test_generate_api_with_complex_schema(temp_dir):
 
     # Check for proper typing imports
     assert "from typing import" in content
-
-    # Verify README was generated
-    readme_content = readme_file.read_text()
-
-    # Check README content
-    assert "Complex MCP Server" in readme_content
-    assert "Tool List" in readme_content
-    assert "list_users" in readme_content
-    assert "create_user" in readme_content
-    assert "get_user" in readme_content

{universal_mcp-0.1.13rc3 → universal_mcp-0.1.13rc14}/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.13rc3 → universal_mcp-0.1.13rc14}/src/tests/test_applications.py

@@ -57,6 +57,10 @@ def test_application(app_name):
     logger.info(f"Tools for {app_name}: {tools}")
     assert len(tools) > 0
     assert isinstance(tools[0], Callable)
+    important_tools = []
     for tool in tools:
         assert tool.__name__ is not None
-        assert tool.__doc__ is not None
+        assert tool.__doc__ is not None
+        if "important" in tool.__doc__:
+            important_tools.append(tool.__name__)
+    assert len(important_tools) > 0