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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: universal-mcp
-Version: 0.1.13rc3
+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

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

@@ -1,6 +1,6 @@
 [project]
 name = "universal-mcp"
-version = "0.1.13-rc3"
+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 = [
@@ -199,43 +199,3 @@ docstring-code-line-length = "dynamic"
 [tool.pytest.ini_options]
 asyncio_mode = "strict"
 asyncio_default_fixture_loop_scope = "function"
-
-[tool.uv.sources]
-universal-mcp-ahrefs = { git = "https://github.com/universal-mcp/ahrefs" }
-universal-mcp-cal-com-v2 = { git = "https://github.com/universal-mcp/cal-com-v2" }
-universal-mcp-calendly = { git = "https://github.com/universal-mcp/calendly" }
-universal-mcp-clickup = { git = "https://github.com/universal-mcp/clickup" }
-universal-mcp-coda = { git = "https://github.com/universal-mcp/coda" }
-universal-mcp-crustdata = { git = "https://github.com/universal-mcp/crustdata" }
-universal-mcp-e2b = { git = "https://github.com/universal-mcp/e2b" }
-universal-mcp-elevenlabs = { git = "https://github.com/universal-mcp/elevenlabs" }
-universal-mcp-falai = { git = "https://github.com/universal-mcp/falai" }
-universal-mcp-figma = { git = "https://github.com/universal-mcp/figma" }
-universal-mcp-firecrawl = { git = "https://github.com/universal-mcp/firecrawl" }
-universal-mcp-github = { git = "https://github.com/universal-mcp/github" }
-universal-mcp-gong = { git = "https://github.com/universal-mcp/gong" }
-universal-mcp-google-calendar = { git = "https://github.com/universal-mcp/google-calendar" }
-universal-mcp-google-docs = { git = "https://github.com/universal-mcp/google-docs" }
-universal-mcp-google-drive = { git = "https://github.com/universal-mcp/google-drive" }
-universal-mcp-google-mail = { git = "https://github.com/universal-mcp/google-mail" }
-universal-mcp-google-sheet = { git = "https://github.com/universal-mcp/google-sheet" }
-universal-mcp-hashnode = { git = "https://github.com/universal-mcp/hashnode" }
-universal-mcp-heygen = { git = "https://github.com/universal-mcp/heygen" }
-universal-mcp-mailchimp = { git = "https://github.com/universal-mcp/mailchimp" }
-universal-mcp-markitdown = { git = "https://github.com/universal-mcp/markitdown" }
-universal-mcp-neon = { git = "https://github.com/universal-mcp/neon" }
-universal-mcp-notion = { git = "https://github.com/universal-mcp/notion" }
-universal-mcp-perplexity = { git = "https://github.com/universal-mcp/perplexity" }
-universal-mcp-reddit = { git = "https://github.com/universal-mcp/reddit" }
-universal-mcp-replicate = { git = "https://github.com/universal-mcp/replicate" }
-universal-mcp-resend = { git = "https://github.com/universal-mcp/resend" }
-universal-mcp-retell = { git = "https://github.com/universal-mcp/retell" }
-universal-mcp-rocketlane = { git = "https://github.com/universal-mcp/rocketlane" }
-universal-mcp-serpapi = { git = "https://github.com/universal-mcp/serpapi" }
-universal-mcp-shortcut = { git = "https://github.com/universal-mcp/shortcut" }
-universal-mcp-spotify = { git = "https://github.com/universal-mcp/spotify" }
-universal-mcp-supabase = { git = "https://github.com/universal-mcp/supabase" }
-universal-mcp-tavily = { git = "https://github.com/universal-mcp/tavily" }
-universal-mcp-wrike = { git = "https://github.com/universal-mcp/wrike" }
-universal-mcp-youtube = { git = "https://github.com/universal-mcp/youtube" }
-universal-mcp-zenquotes = { git = "https://github.com/universal-mcp/zenquotes" }

{universal_mcp-0.1.13rc3 → 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.13rc3 → 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.13rc3 → universal_mcp-0.1.13rc7}/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.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.13rc3 → 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.13rc3 → 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.13rc3 → 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.13rc3 → 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",

{universal_mcp-0.1.13rc3 → universal_mcp-0.1.13rc7}/src/universal_mcp/cli.py

@@ -1,11 +1,9 @@
-import asyncio
-import os
+import re
 from pathlib import Path
 
 import typer
 from rich import print as rprint
 from rich.panel import Panel
-import re
 
 from universal_mcp.utils.installation import (
     get_supported_apps,
@@ -41,9 +39,9 @@ def generate(
     try:
         # Run the async function in the event loop
         result = generate_api_from_schema(
-                schema_path=schema_path,
-                output_path=output_path,
-            )
+            schema_path=schema_path,
+            output_path=output_path,
+        )
 
         if not output_path:
             # Print to stdout if no output path
@@ -68,7 +66,6 @@ def docgen(
         "-m",
         help="Model to use for generating docstrings",
     ),
-    
 ):
     """Generate docstrings for Python files using LLMs.
 
@@ -154,6 +151,7 @@ def install(app_name: str = typer.Argument(..., help="Name of app to install")):
         typer.echo(f"Error installing app: {e}", err=True)
         raise typer.Exit(1) from e
 
+
 @app.command()
 def init(
     output_dir: Path | None = typer.Option(
@@ -162,13 +160,13 @@ def init(
         "-o",
         help="Output directory for the project (must exist)",
     ),
-    app_name: str|None = typer.Option(
+    app_name: str | None = typer.Option(
         None,
         "--app-name",
         "-a",
         help="App name (letters, numbers, hyphens, underscores only)",
     ),
-    integration_type: str|None = typer.Option(
+    integration_type: str | None = typer.Option(
         None,
         "--integration-type",
         "-i",
@@ -195,7 +193,7 @@ def init(
         app_name = typer.prompt(
             "Enter the app name",
             default="app_name",
-            prompt_suffix=" (e.g., reddit, youtube): "
+            prompt_suffix=" (e.g., reddit, youtube): ",
         ).strip()
     validate_pattern(app_name, "app name")
 
@@ -203,10 +201,10 @@ def init(
         path_str = typer.prompt(
             "Enter the output directory for the project",
             default=str(Path.cwd()),
-            prompt_suffix=": "
+            prompt_suffix=": ",
         ).strip()
         output_dir = Path(path_str)
-    
+
     if not output_dir.exists():
         try:
             output_dir.mkdir(parents=True, exist_ok=True)
@@ -219,7 +217,7 @@ def init(
                 f"❌ Failed to create output directory '{output_dir}': {e}",
                 fg=typer.colors.RED,
             )
-            raise typer.Exit(code=1)
+            raise typer.Exit(code=1) from e
     elif not output_dir.is_dir():
         typer.secho(
             f"❌ Output path '{output_dir}' exists but is not a directory.",
@@ -232,7 +230,7 @@ def init(
         integration_type = typer.prompt(
             "Choose the integration type",
             default="agentr",
-            prompt_suffix=" (api_key, oauth, agentr, none): "
+            prompt_suffix=" (api_key, oauth, agentr, none): ",
         ).lower()
     if integration_type not in ("api_key", "oauth", "agentr", "none"):
         typer.secho(
@@ -240,7 +238,6 @@ def init(
             fg=typer.colors.RED,
         )
         raise typer.Exit(code=1)
-    
 
     typer.secho("🚀 Generating project using cookiecutter...", fg=typer.colors.BLUE)
     try:
@@ -255,10 +252,11 @@ def init(
         )
     except Exception as exc:
         typer.secho(f"❌ Project generation failed: {exc}", fg=typer.colors.RED)
-        raise typer.Exit(code=1)
+        raise typer.Exit(code=1) from exc
 
     project_dir = output_dir / f"universal-mcp-{app_name}"
     typer.secho(f"✅ Project created at {project_dir}", fg=typer.colors.GREEN)
 
+
 if __name__ == "__main__":
     app()

{universal_mcp-0.1.13rc3 → universal_mcp-0.1.13rc7}/src/universal_mcp/servers/server.py

@@ -1,8 +1,6 @@
-import os
 from abc import ABC, abstractmethod
 from collections.abc import Callable
 from typing import Any
-from urllib.parse import urlparse
 
 import httpx
 from loguru import logger

{universal_mcp-0.1.13rc3 → universal_mcp-0.1.13rc7}/src/universal_mcp/stores/store.py

@@ -11,11 +11,13 @@ class StoreError(Exception):
 
     pass
 
+
 class KeyNotFoundError(StoreError):
     """Exception raised when a key is not found in the store."""
 
     pass
 
+
 class BaseStore(ABC):
     """
     Abstract base class defining the interface for credential stores.

{universal_mcp-0.1.13rc3 → universal_mcp-0.1.13rc7}/src/universal_mcp/tools/tools.py

@@ -260,9 +260,7 @@ class ToolManager:
         try:
             available_tool_functions = app.list_tools()
         except TypeError as e:
-            logger.error(
-                f"Error calling list_tools for app '{app.name}'. Error: {e}"
-            )
+            logger.error(f"Error calling list_tools for app '{app.name}'. Error: {e}")
             return
         except Exception as e:
             logger.error(f"Failed to get tool list from app '{app.name}': {e}")

{universal_mcp-0.1.13rc3 → universal_mcp-0.1.13rc7}/src/universal_mcp/utils/agentr.py

@@ -1,9 +1,13 @@
-from loguru import logger
 import os
+
 import httpx
+from loguru import logger
+
 from universal_mcp.config import AppConfig
+from universal_mcp.exceptions import NotAuthorizedError
 from universal_mcp.utils.singleton import Singleton
 
+
 class AgentrClient(metaclass=Singleton):
     """Helper class for AgentR API operations.
 
@@ -22,7 +26,9 @@ class AgentrClient(metaclass=Singleton):
                 "API key for AgentR is missing. Please visit https://agentr.dev to create an API key, then set it as AGENTR_API_KEY environment variable."
             )
             raise ValueError("AgentR API key required - get one at https://agentr.dev")
-        self.base_url = (base_url or os.getenv("AGENTR_BASE_URL", "https://api.agentr.dev")).rstrip("/")
+        self.base_url = (
+            base_url or os.getenv("AGENTR_BASE_URL", "https://api.agentr.dev")
+        ).rstrip("/")
 
     def get_credentials(self, integration_name: str) -> dict:
         """Get credentials for an integration from the AgentR API.
@@ -87,4 +93,3 @@ class AgentrClient(metaclass=Singleton):
         response.raise_for_status()
         data = response.json()
         return [AppConfig.model_validate(app) for app in data]
-

{universal_mcp-0.1.13rc3 → universal_mcp-0.1.13rc7}/src/universal_mcp/utils/api_generator.py

@@ -1,10 +1,11 @@
+import importlib.util
 import inspect
 import os
-from pathlib import Path
-from loguru import logger
 import shutil
-import importlib.util
+from pathlib import Path
+
 from jinja2 import Environment, FileSystemLoader, TemplateError, select_autoescape
+from loguru import logger
 
 from universal_mcp.utils.openapi import generate_api_client, load_schema
 
@@ -26,6 +27,7 @@ def validate_and_load_schema(schema_path: Path) -> dict:
         echo(f"Error loading schema: {e}", err=True)
         raise
 
+
 def get_class_info(module: any) -> tuple[str | None, any]:
     """Find the main class in the generated module."""
     for name, obj in inspect.getmembers(module):
@@ -33,19 +35,18 @@ def get_class_info(module: any) -> tuple[str | None, any]:
             return name, obj
     return None, None
 
-def generate_readme(
-    app_dir: Path, folder_name: str, tools: list
-) -> Path:
+
+def generate_readme(app_dir: Path, folder_name: str, tools: list) -> Path:
     """Generate README.md with API documentation.
-    
+
     Args:
         app_dir: Directory where the README will be generated
         folder_name: Name of the application folder
         tools: List of Function objects from the OpenAPI schema
-        
+
     Returns:
         Path to the generated README file
-        
+
     Raises:
         FileNotFoundError: If the template directory doesn't exist
         TemplateError: If there's an error rendering the template
@@ -69,23 +70,19 @@ def generate_readme(
 
     try:
         env = Environment(
-            loader=FileSystemLoader(template_dir),
-            autoescape=select_autoescape()
+            loader=FileSystemLoader(template_dir), autoescape=select_autoescape()
         )
         template = env.get_template("README.md.j2")
     except Exception as e:
         logger.error(f"Error loading template: {e}")
-        raise TemplateError(f"Error loading template: {e}")
+        raise TemplateError(f"Error loading template: {e}") from e
 
     # Render the template
     try:
-        readme_content = template.render(
-            name=app,
-            tools=formatted_tools
-        )
+        readme_content = template.render(name=app, tools=formatted_tools)
     except Exception as e:
         logger.error(f"Error rendering template: {e}")
-        raise TemplateError(f"Error rendering template: {e}")
+        raise TemplateError(f"Error rendering template: {e}") from e
 
     # Write the README file
     readme_file = app_dir / "README.md"
@@ -95,10 +92,11 @@ def generate_readme(
         logger.info(f"Documentation generated at: {readme_file}")
     except Exception as e:
         logger.error(f"Error writing README file: {e}")
-        raise IOError(f"Error writing README file: {e}")
+        raise OSError(f"Error writing README file: {e}") from e
 
     return readme_file
 
+
 def test_correct_output(gen_file: Path):
     # Check file is non-empty
     if gen_file.stat().st_size == 0:
@@ -137,7 +135,6 @@ def generate_api_from_schema(
     """
     # Local imports for logging and file operations
 
-
     logger.info("Starting API generation for schema: %s", schema_path)
 
     # 1. Parse and validate schema
@@ -174,10 +171,15 @@ def generate_api_from_schema(
         f.write(code)
 
     if not test_correct_output(gen_file):
-        logger.error("Generated code validation failed for '%s'. Aborting generation.", gen_file)
+        logger.error(
+            "Generated code validation failed for '%s'. Aborting generation.", gen_file
+        )
         logger.info("Next steps:")
         logger.info(" 1) Review your OpenAPI schema for potential mismatches.")
-        logger.info(" 2) Inspect '%s' for syntax or logic errors in the generated code.", gen_file)
+        logger.info(
+            " 2) Inspect '%s' for syntax or logic errors in the generated code.",
+            gen_file,
+        )
         logger.info(" 3) Correct the issues and re-run the command.")
         return {"error": "Validation failed. See logs above for detailed instructions."}
 
@@ -210,13 +212,14 @@ def generate_api_from_schema(
             client = cls()
             tools = client.list_tools()
         except Exception as e:
-            logger.warning("Failed to instantiate '%s' or list tools: %s", class_name, e)
+            logger.warning(
+                "Failed to instantiate '%s' or list tools: %s", class_name, e
+            )
     else:
         logger.warning("No generated class found in module 'temp_module'")
     readme_file = generate_readme(target_dir, output_path.stem, tools)
     logger.info("README generated at: %s", readme_file)
 
-
     # Cleanup intermediate file
     try:
         os.remove(gen_file)

{universal_mcp-0.1.13rc3 → universal_mcp-0.1.13rc7}/src/universal_mcp/utils/openapi.py

@@ -1,15 +1,16 @@
 import json
 import re
+from dataclasses import dataclass
 from pathlib import Path
-from typing import Any, Dict, List, Literal
-from loguru import logger
+from typing import Any, Literal
 
 import yaml
 from jinja2 import Environment, FileSystemLoader, select_autoescape
-from dataclasses import dataclass
+from loguru import logger
+
 
 def convert_to_snake_case(identifier: str) -> str:
-    """ 
+    """
     Convert a camelCase or PascalCase identifier to snake_case.
 
     Args:
@@ -140,15 +141,13 @@ def generate_api_client(schema):
     # Set up Jinja2 environment
     env = Environment(
         loader=FileSystemLoader(Path(__file__).parent.parent / "templates"),
-        autoescape=select_autoescape()
+        autoescape=select_autoescape(),
     )
     template = env.get_template("api_client.py.j2")
 
     # Render the template
     class_code = template.render(
-        class_name=class_name,
-        base_url=base_url,
-        methods=methods
+        class_name=class_name, base_url=base_url, methods=methods
     )
 
     return class_code
@@ -158,10 +157,10 @@ def generate_api_client(schema):
 class Function:
     name: str
     type: Literal["get", "post", "put", "delete", "patch", "options", "head"]
-    args: Dict[str, str]
+    args: dict[str, str]
     return_type: str
     description: str
-    tags: List[str]
+    tags: list[str]
     implementation: str
 
     @property
@@ -171,10 +170,7 @@ class Function:
 
 
 def generate_method_code(
-    path: str,
-    method: str,
-    operation: dict[str, Any],
-    full_schema: dict[str, Any]
+    path: str, method: str, operation: dict[str, Any], full_schema: dict[str, Any]
 ) -> Function:
     """
     Generate a Function object for a single API method.
@@ -205,7 +201,6 @@ def generate_method_code(
             return "dict[str, Any]"
         return "Any"
 
-
     # Determine function name
     if op_id := operation.get("operationId"):
         cleaned_id = op_id.replace(".", "_").replace("-", "_")
@@ -240,10 +235,12 @@ def generate_method_code(
     # Analyze requestBody
     has_body = "requestBody" in operation
     body_required = bool(has_body and operation["requestBody"].get("required"))
-    content = (operation.get("requestBody", {}) or {}).get("content", {}) if has_body else {}
+    content = (
+        (operation.get("requestBody", {}) or {}).get("content", {}) if has_body else {}
+    )
     is_array_body = False
-    request_props: Dict[str, Any] = {}
-    required_fields: List[str] = []
+    request_props: dict[str, Any] = {}
+    required_fields: list[str] = []
     if has_body and content:
         for mime, info in content.items():
             if not mime.startswith("application/json") or "schema" not in info:
@@ -258,10 +255,12 @@ def generate_method_code(
                 request_props = schema.get("properties", {}) or {}
                 for name, prop_schema in list(request_props.items()):
                     if pre := prop_schema.get("$ref"):
-                        request_props[name] = resolve_schema_reference(pre, full_schema) or prop_schema
+                        request_props[name] = (
+                            resolve_schema_reference(pre, full_schema) or prop_schema
+                        )
 
     # Build function arguments with Annotated[type, description]
-    arg_defs: Dict[str, str] = {}
+    arg_defs: dict[str, str] = {}
     for p in path_params:
         name = p["name"]
         ty = map_type(p.get("schema", {}))
@@ -304,7 +303,7 @@ def generate_method_code(
     # Assemble description
     summary = operation.get("summary", "")
     operation_desc = operation.get("description", "")
-    desc_parts: List[str] = []
+    desc_parts: list[str] = []
     if summary:
         desc_parts.append(summary)
     if operation_desc:
@@ -315,25 +314,33 @@ def generate_method_code(
 
     # Generate implementation code
     implementation_lines = []
-    
+
     # Add parameter validation for required fields
     for param in path_params + query_params:
         if param.get("required"):
             name = param["name"]
             implementation_lines.append(f"if {name} is None:")
-            implementation_lines.append(f"    raise ValueError(\"Missing required parameter '{name}'\")")
-    
+            implementation_lines.append(
+                f"    raise ValueError(\"Missing required parameter '{name}'\")"
+            )
+
     if has_body and body_required:
         if is_array_body:
             implementation_lines.append("if items is None:")
-            implementation_lines.append("    raise ValueError(\"Missing required parameter 'items'\")")
+            implementation_lines.append(
+                "    raise ValueError(\"Missing required parameter 'items'\")"
+            )
         elif request_props:
             for prop in required_fields:
                 implementation_lines.append(f"if {prop} is None:")
-                implementation_lines.append(f"    raise ValueError(\"Missing required parameter '{prop}'\")")
+                implementation_lines.append(
+                    f"    raise ValueError(\"Missing required parameter '{prop}'\")"
+                )
         else:
             implementation_lines.append("if request_body is None:")
-            implementation_lines.append("    raise ValueError(\"Missing required parameter 'request_body'\")")
+            implementation_lines.append(
+                "    raise ValueError(\"Missing required parameter 'request_body'\")"
+            )
 
     # Build request body
     if has_body:
@@ -342,35 +349,43 @@ def generate_method_code(
         elif request_props:
             implementation_lines.append("request_body = {")
             for prop in request_props:
-                implementation_lines.append(f"    \"{prop}\": {prop},")
+                implementation_lines.append(f'    "{prop}": {prop},')
             implementation_lines.append("}")
-            implementation_lines.append("request_body = {k: v for k, v in request_body.items() if v is not None}")
+            implementation_lines.append(
+                "request_body = {k: v for k, v in request_body.items() if v is not None}"
+            )
         else:
             implementation_lines.append("request_body = request_body")
 
     # Build URL with path parameters
-    path = "/".join([path_params["name"] for path_params in path_params]) or '\"\"'
-    url = '\"{self.base_url}{path}\"'
-    implementation_lines.append(f'path = {path}')
-    implementation_lines.append(f'url = f{url}')
+    path = "/".join([path_params["name"] for path_params in path_params]) or '""'
+    url = '"{self.base_url}{path}"'
+    implementation_lines.append(f"path = {path}")
+    implementation_lines.append(f"url = f{url}")
 
     # Build query parameters
     if query_params:
         implementation_lines.append("query_params = {")
         for param in query_params:
             name = param["name"]
-            implementation_lines.append(f"        \"{name}\": {name},")
+            implementation_lines.append(f'        "{name}": {name},')
         implementation_lines.append("    }")
-        implementation_lines.append("query_params = {k: v for k, v in query_params.items() if v is not None}")
+        implementation_lines.append(
+            "query_params = {k: v for k, v in query_params.items() if v is not None}"
+        )
     else:
         implementation_lines.append("query_params = {}")
 
     # Make the request using the appropriate method
     http_method = method.lower()
     if has_body:
-        implementation_lines.append(f"response = self._{http_method}(url, data=request_body, params=query_params)")
+        implementation_lines.append(
+            f"response = self._{http_method}(url, data=request_body, params=query_params)"
+        )
     else:
-        implementation_lines.append(f"response = self._{http_method}(url, params=query_params)")
+        implementation_lines.append(
+            f"response = self._{http_method}(url, params=query_params)"
+        )
 
     # Handle response
     implementation_lines.append("response.raise_for_status()")
@@ -386,12 +401,13 @@ def generate_method_code(
         return_type=return_type,
         description=description_text,
         tags=tags,
-        implementation=implementation
+        implementation=implementation,
     )
 
     logger.debug(f"Generated function: {function}")
     return function
 
+
 # Example usage
 if __name__ == "__main__":
     # Sample OpenAPI schema

universal_mcp-0.1.13rc3/src/playground/memory/__init__.py

@@ -1,14 +0,0 @@
-from langgraph.checkpoint.base import BaseCheckpointSaver
-
-from playground.memory.sqlite import get_sqlite_saver
-
-
-def initialize_database() -> BaseCheckpointSaver:
-    """
-    Initialize the appropriate database checkpointer based on configuration.
-    Returns an initialized AsyncCheckpointer instance.
-    """
-    return get_sqlite_saver()
-
-
-__all__ = ["initialize_database"]

universal_mcp-0.1.13rc3/src/playground/memory/sqlite.py

@@ -1,9 +0,0 @@
-from langgraph.checkpoint.base import BaseCheckpointSaver
-from langgraph.checkpoint.sqlite.aio import AsyncSqliteSaver
-
-from playground.settings import settings
-
-
-def get_sqlite_saver() -> BaseCheckpointSaver:
-    """Initialize and return a SQLite saver instance."""
-    return AsyncSqliteSaver.from_conn_string(settings.SQLITE_DB_PATH)

{universal_mcp-0.1.13rc3 → universal_mcp-0.1.13rc7}/src/universal_mcp/integrations/__init__.py

@@ -1,9 +1,9 @@
 from universal_mcp.config import IntegrationConfig
 from universal_mcp.integrations.integration import (
+    AgentRIntegration,
     ApiKeyIntegration,
     Integration,
     OAuthIntegration,
-    AgentRIntegration,
 )
 from universal_mcp.stores.store import BaseStore