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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: universal-mcp
-Version: 0.1.13rc7
+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.13rc7 → 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,8 +70,8 @@ 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" != None
@@ -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,40 +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
@@ -203,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()
@@ -231,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.13rc7 → 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
+        if "important" in tool.__doc__:
+            important_tools.append(tool.__name__)
+    assert len(important_tools) > 0

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

@@ -1,5 +1,7 @@
 import importlib
+import os
 import subprocess
+import sys
 
 from loguru import logger
 
@@ -9,8 +11,16 @@ from universal_mcp.applications.application import (
     GraphQLApplication,
 )
 
+UNIVERSAL_MCP_HOME = os.path.join(os.path.expanduser("~"), ".universal-mcp", "packages")
+
+if not os.path.exists(UNIVERSAL_MCP_HOME):
+    os.makedirs(UNIVERSAL_MCP_HOME)
+
+# set python path to include the universal-mcp home directory
+sys.path.append(UNIVERSAL_MCP_HOME)
+
+
 # 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
 
 
@@ -38,7 +48,7 @@ def _install_package(slug_clean: str):
     Helper to install a package via pip from the universal-mcp GitHub repository.
     """
     repo_url = f"git+https://github.com/universal-mcp/{slug_clean}"
-    cmd = ["uv", "pip", "install", repo_url]
+    cmd = ["uv", "pip", "install", repo_url, "--target", UNIVERSAL_MCP_HOME]
     logger.info(f"Installing package '{slug_clean}' with command: {' '.join(cmd)}")
     try:
         subprocess.check_call(cmd)

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

@@ -23,6 +23,12 @@ def generate(
         "-o",
         help="Output file path - should match the API name (e.g., 'twitter.py' for Twitter API)",
     ),
+    class_name: str = typer.Option(
+        None,
+        "--class-name",
+        "-c",
+        help="Class name to use for the API client",
+    ),
 ):
     """Generate API client from OpenAPI schema with optional docstring generation.
 
@@ -38,25 +44,35 @@ def generate(
 
     try:
         # Run the async function in the event loop
-        result = generate_api_from_schema(
+        app_file = generate_api_from_schema(
             schema_path=schema_path,
             output_path=output_path,
+            class_name=class_name,
         )
-
-        if not output_path:
-            # Print to stdout if no output path
-            print(result["code"])
-        else:
-            typer.echo("API client successfully generated and installed.")
-            if "app_file" in result:
-                typer.echo(f"Application file: {result['app_file']}")
-            if "readme_file" in result and result["readme_file"]:
-                typer.echo(f"Documentation: {result['readme_file']}")
+        typer.echo("API client successfully generated and installed.")
+        typer.echo(f"Application file: {app_file}")
     except Exception as e:
         typer.echo(f"Error generating API client: {e}", err=True)
         raise typer.Exit(1) from e
 
 
+@app.command()
+def readme(
+    file_path: Path = typer.Argument(..., help="Path to the Python file to process"),
+    class_name: str = typer.Option(
+        None,
+        "--class-name",
+        "-c",
+        help="Class name to use for the API client",
+    ),
+):
+    """Generate a README.md file for the API client."""
+    from universal_mcp.utils.readme import generate_readme
+
+    readme_file = generate_readme(file_path, class_name)
+    typer.echo(f"README.md file generated at: {readme_file}")
+
+
 @app.command()
 def docgen(
     file_path: Path = typer.Argument(..., help="Path to the Python file to process"),

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

@@ -4,7 +4,6 @@ import os
 import shutil
 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
@@ -36,67 +35,6 @@ def get_class_info(module: any) -> tuple[str | None, any]:
     return None, None
 
 
-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
-        IOError: If there's an error writing the README file
-    """
-    app = folder_name.replace("_", " ").title()
-    logger.info(f"Generating README for {app} in {app_dir}")
-
-    # Format tools into (name, description) tuples
-    formatted_tools = []
-    for tool in tools:
-        name = tool.__name__
-        description = tool.__doc__.strip().split("\n")[0]
-        formatted_tools.append((name, description))
-
-    # Set up Jinja2 environment
-    template_dir = Path(__file__).parent.parent / "templates"
-    if not template_dir.exists():
-        logger.error(f"Template directory not found: {template_dir}")
-        raise FileNotFoundError(f"Template directory not found: {template_dir}")
-
-    try:
-        env = Environment(
-            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}") from e
-
-    # Render the template
-    try:
-        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}") from e
-
-    # Write the README file
-    readme_file = app_dir / "README.md"
-    try:
-        with open(readme_file, "w") as f:
-            f.write(readme_content)
-        logger.info(f"Documentation generated at: {readme_file}")
-    except Exception as e:
-        logger.error(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:
@@ -120,7 +58,7 @@ def test_correct_output(gen_file: Path):
 def generate_api_from_schema(
     schema_path: Path,
     output_path: Path | None = None,
-    add_docstrings: bool = True,
+    class_name: str | None = None,
 ) -> tuple[Path, Path]:
     """
     Generate API client from OpenAPI schema and write to app.py with a README.
@@ -147,7 +85,7 @@ def generate_api_from_schema(
 
     # 2. Generate client code
     try:
-        code = generate_api_client(schema)
+        code = generate_api_client(schema, class_name)
         logger.info("API client code generated.")
     except Exception as e:
         logger.error("Code generation failed: %s", e)
@@ -192,34 +130,6 @@ def generate_api_from_schema(
     shutil.copy(gen_file, app_file)
     logger.info("App file written to: %s", app_file)
 
-    # 6. Collect tools and generate README
-    import importlib.util
-    import sys
-
-    # Load the generated module as "temp_module"
-    spec = importlib.util.spec_from_file_location("temp_module", str(app_file))
-    module = importlib.util.module_from_spec(spec)
-    sys.modules["temp_module"] = module
-    spec.loader.exec_module(module)
-
-    # Retrieve the generated API class
-    class_name, cls = get_class_info(module)
-
-    # Instantiate client and collect its tools
-    tools = []
-    if cls:
-        try:
-            client = cls()
-            tools = client.list_tools()
-        except Exception as 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)
@@ -227,4 +137,4 @@ def generate_api_from_schema(
     except Exception as e:
         logger.warning("Could not remove intermediate file %s: %s", gen_file, e)
 
-    return app_file, readme_file
+    return app_file

{universal_mcp-0.1.13rc7 → universal_mcp-0.1.13rc14}/src/universal_mcp/utils/docgen.py

@@ -176,7 +176,7 @@ def extract_json_from_text(text):
 
 
 def generate_docstring(
-    function_code: str, model: str = "perplexity/sonar-pro"
+    function_code: str, model: str = "perplexity/sonar"
 ) -> DocstringOutput:
     """
     Generate a docstring for a Python function using litellm with structured output.
@@ -509,7 +509,7 @@ def insert_docstring_into_function(function_code: str, docstring: str) -> str:
         return function_code
 
 
-def process_file(file_path: str, model: str = "perplexity/sonar-pro") -> int:
+def process_file(file_path: str, model: str = "perplexity/sonar") -> int:
     """
     Process a Python file and add docstrings to all functions in it.