universal-mcp 0.1.8rc4__tar.gz → 0.1.10__tar.gz

{universal_mcp-0.1.8rc4 → universal_mcp-0.1.10}/PKG-INFO

@@ -1,10 +1,13 @@
 Metadata-Version: 2.4
 Name: universal-mcp
-Version: 0.1.8rc4
+Version: 0.1.10
 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>
 Requires-Python: >=3.11
+Requires-Dist: gql[all]>=3.5.2
 Requires-Dist: keyring>=25.6.0
+Requires-Dist: langchain-cerebras>=0.5.0
+Requires-Dist: langchain-google-genai>=2.1.3
 Requires-Dist: litellm>=1.30.7
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: mcp>=1.6.0
@@ -22,6 +25,8 @@ Requires-Dist: pytest>=8.3.5; extra == 'dev'
 Requires-Dist: ruff>=0.11.4; extra == 'dev'
 Provides-Extra: e2b
 Requires-Dist: e2b-code-interpreter>=1.2.0; extra == 'e2b'
+Provides-Extra: fal-ai
+Requires-Dist: fal-client>=0.5.9; extra == 'fal-ai'
 Provides-Extra: firecrawl
 Requires-Dist: firecrawl-py>=1.15.0; extra == 'firecrawl'
 Provides-Extra: markitdown

universal_mcp-0.1.10/bump_and_release.sh

@@ -0,0 +1,61 @@
+#!/bin/bash
+
+# Get the current branch name
+BRANCH=$(git branch --show-current)
+
+# Read the current version from pyproject.toml
+CURRENT_VERSION=$(grep -E '^version = "[0-9]+\.[0-9]+\.[0-9]+.*"' pyproject.toml | cut -d'"' -f2)
+
+# Split version into major, minor, patch
+IFS='.' read -r MAJOR MINOR PATCH <<< "$CURRENT_VERSION"
+
+# Remove any rc suffix from PATCH if it exists
+PATCH_NUM=$(echo $PATCH | sed 's/-rc[0-9]*//')
+
+if [ "$BRANCH" = "master" ]; then
+    # On main branch - bump patch version
+    NEW_VERSION="$MAJOR.$MINOR.$PATCH_NUM"
+else
+    # On dev branch - bump rc version
+    if [[ $PATCH == *"-rc"* ]]; then
+        # If already has rc, increment rc number
+        RC_NUM=$(echo $PATCH | grep -o 'rc[0-9]*' | sed 's/rc//')
+        NEW_VERSION="$MAJOR.$MINOR.$PATCH_NUM-rc$((RC_NUM + 1))"
+    else
+        echo "No rc suffix found, adding rc1"
+        # Increase patch number by 1
+        PATCH_NUM=$((PATCH_NUM + 1))
+        # Add rc1
+        NEW_VERSION="$MAJOR.$MINOR.$PATCH_NUM-rc1"
+    fi
+fi
+
+# Update version in pyproject.toml
+sed -i '' "s/^version = \".*\"/version = \"$NEW_VERSION\"/" pyproject.toml
+
+echo "Version bumped from $CURRENT_VERSION to $NEW_VERSION"
+
+# Stage the changed file
+git add pyproject.toml
+
+# Commit the change
+git commit -m "bump: version $CURRENT_VERSION → $NEW_VERSION"
+
+# Create and push tag if on main
+if [ "$BRANCH" = "main" ]; then
+    git tag -a "v$NEW_VERSION" -m "Release version $NEW_VERSION"
+    git push origin "v$NEW_VERSION"
+fi
+
+# Push the changes
+git push origin $BRANCH
+
+# Ask user if they want to release
+read -p "Do you want to build and publish the package? (y/N) " RELEASE_CONFIRM
+
+if [[ $RELEASE_CONFIRM =~ ^[Yy]$ ]]; then
+    # Build and publish
+    rm -rf dist && uv build && uv publish
+else
+    echo "Skipping release. Version bump committed and pushed."
+fi

{universal_mcp-0.1.8rc4 → universal_mcp-0.1.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "universal-mcp"
-version = "0.1.8rc4"
+version = "0.1.10"
 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 = [
@@ -18,6 +18,9 @@ dependencies = [
     "rich>=14.0.0",
     "posthog>=3.24.0",
     "mcp>=1.6.0",
+    "gql[all]>=3.5.2",
+    "langchain-cerebras>=0.5.0",
+    "langchain-google-genai>=2.1.3",
 ]
 
 [project.optional-dependencies]
@@ -50,6 +53,9 @@ serpapi = [
 markitdown = [
     "markitdown[all]>=0.1.1",
 ]
+fal-ai = [
+    "fal-client>=0.5.9",
+]
 [project.scripts]
 universal_mcp = "universal_mcp.cli:app"
 

{universal_mcp-0.1.8rc4 → universal_mcp-0.1.10}/src/tests/test_api_generator.py

@@ -9,7 +9,7 @@ from universal_mcp.utils.api_generator import generate_api_from_schema
 @pytest.fixture
 def temp_dir(tmp_path):
     """Create a temporary directory for test files."""
-    return Path("/tmp/test_api_generator")
+    return tmp_path
 
 
 # Fixture for a sample OpenAPI schema
@@ -139,9 +139,7 @@ async def test_generate_api_with_docstrings(sample_schema, temp_dir):
     assert "from universal_mcp.integrations import Integration" in content
     assert "def test_operation" in content
     assert '"""' in content  # Basic check for docstring presence
-    # TODO: Fix this test
-    # assert "Args:" in content or "Parameters:" in content
-    assert "Returns:" in content
+    assert "Tags:" in content
 
     # Verify temporary file was cleaned up
     assert not output_path.exists()

{universal_mcp-0.1.8rc4 → universal_mcp-0.1.10}/src/tests/test_applications.py

@@ -21,6 +21,7 @@ from universal_mcp.applications import app_from_slug
         "firecrawl",
         "serpapi",
         "notion",
+        "hashnode",
     ],
 )
 def test_application(app_name):

universal_mcp-0.1.10/src/tests/test_tool.py

@@ -0,0 +1,302 @@
+from typing import Annotated
+
+from pydantic import Field
+
+from universal_mcp.tools.func_metadata import FuncMetadata
+from universal_mcp.utils.docstring_parser import parse_docstring
+
+
+def test_func_metadata_annotated():
+    def func(a: Annotated[int, Field(title="First integer")], b: int):
+        """Test function with annotated and typed args"""
+        return a + b
+
+    meta = FuncMetadata.func_metadata(func)
+    assert meta.arg_model.model_json_schema() == {
+        "type": "object",
+        "title": "funcArguments",
+        "properties": {
+            "a": {"type": "integer", "title": "First integer"},
+            "b": {"type": "integer", "title": "B"},
+        },
+        "required": ["a", "b"],
+    }
+
+
+def test_func_metadata_no_args():
+    def func():
+        """Test function with no args"""
+        return 42
+
+    meta = FuncMetadata.func_metadata(func)
+    assert meta.arg_model.model_json_schema() == {
+        "type": "object",
+        "title": "funcArguments",
+        "properties": {},
+        # "required": [],
+    }
+
+
+def test_func_metadata_untyped():
+    def func(a, b=10):
+        """Test function with untyped args"""
+        return a + b
+
+    meta = FuncMetadata.func_metadata(func)
+    assert meta.arg_model.model_json_schema() == {
+        "type": "object",
+        "title": "funcArguments",
+        "properties": {
+            "a": {"type": "string", "title": "a"},
+            "b": {"type": "string", "title": "b", "default": 10},
+        },
+        "required": ["a"],
+    }
+
+
+def test_func_metadata_required():
+    def func(a: int, b: str, c: float = 1.0):
+        """Test function with required and optional args"""
+        return f"{a} {b} {c}"
+
+    meta = FuncMetadata.func_metadata(func)
+    assert meta.arg_model.model_json_schema() == {
+        "type": "object",
+        "title": "funcArguments",
+        "properties": {
+            "a": {"type": "integer", "title": "A"},
+            "b": {"type": "string", "title": "B"},
+            "c": {"type": "number", "title": "C", "default": 1.0},
+        },
+        "required": ["a", "b"],
+    }
+
+
+def test_func_metadata_none_type():
+    def func(a: None = None):
+        """Test function with None type"""
+        return a
+
+    meta = FuncMetadata.func_metadata(func)
+    assert meta.arg_model.model_json_schema() == {
+        "type": "object",
+        "title": "funcArguments",
+        "properties": {
+            "a": {"type": "null", "title": "A", "default": None},
+        },
+    }
+
+
+def test_parse_docstring_empty_none():
+    """Test with None input."""
+    docstring = None
+    expected = {"summary": "", "args": {}, "returns": "", "raises": {}, "tags": []}
+    assert parse_docstring(docstring) == expected
+
+
+def test_parse_docstring_empty_string():
+    """Test with an empty string input."""
+    docstring = ""
+    expected = {"summary": "", "args": {}, "returns": "", "raises": {}, "tags": []}
+    assert parse_docstring(docstring) == expected
+
+
+def test_parse_docstring_whitespace_string():
+    """Test with a whitespace-only string input."""
+    docstring = "   \n  "
+    expected = {"summary": "", "args": {}, "returns": "", "raises": {}, "tags": []}
+    assert parse_docstring(docstring) == expected
+
+
+def test_parse_docstring_summary_only():
+    """Test with only a summary and no sections."""
+    docstring = """
+    This is a short summary.
+    It spans multiple lines.
+    """
+    expected = {
+        "summary": "This is a short summary. It spans multiple lines.",
+        "args": {},
+        "returns": "",
+        "raises": {},
+        "tags": [],
+    }
+    assert parse_docstring(docstring) == expected
+
+
+def test_parse_docstring_summary_and_args():
+    """Test with summary and an Args section."""
+    docstring = """
+    Function to add two numbers.
+
+    Args:
+        a: The first number.
+        b: The second number.
+    """
+    expected = {
+        "summary": "Function to add two numbers.",
+        "args": {"a": "The first number.", "b": "The second number."},
+        "returns": "",
+        "raises": {},
+        "tags": [],
+    }
+    assert parse_docstring(docstring) == expected
+
+
+def test_parse_docstring_summary_and_returns():
+    """Test with summary and a simple Returns section."""
+    docstring = """
+    Calculates the sum.
+
+    Returns:
+        The sum of a and b.
+    """
+    expected = {
+        "summary": "Calculates the sum.",
+        "args": {},
+        "returns": "The sum of a and b.",
+        "raises": {},
+        "tags": [],
+    }
+    assert parse_docstring(docstring) == expected
+
+
+def test_parse_docstring_summary_and_raises():
+    """Test with summary and a simple Raises section."""
+    docstring = """
+    Divides two numbers.
+
+    Raises:
+        ZeroDivisionError: If the denominator is zero.
+    """
+    expected = {
+        "summary": "Divides two numbers.",
+        "args": {},
+        "returns": "",
+        "raises": {"ZeroDivisionError": "If the denominator is zero."},
+        "tags": [],
+    }
+    assert parse_docstring(docstring) == expected
+
+
+def test_parse_docstring_summary_and_tags():
+    """Test with summary and a simple Tags section."""
+    docstring = """
+    Processes data.
+
+    Tags:
+        data, processing, important
+    """
+    expected = {
+        "summary": "Processes data.",
+        "args": {},
+        "returns": "",
+        "raises": {},
+        "tags": ["data", "processing", "important"],
+    }
+    assert parse_docstring(docstring) == expected
+
+
+def test_parse_docstring_multiple_sections_single_line():
+    """Test with summary and multiple sections with single-line content."""
+    docstring = """
+    Performs a basic operation.
+
+    Args:
+        input: Input value.
+    Returns:
+        Output value.
+    Raises:
+        Exception: If something goes wrong.
+    Tags:
+        basic, test
+    """
+    expected = {
+        "summary": "Performs a basic operation.",
+        "args": {"input": "Input value."},
+        "returns": "Output value.",
+        "raises": {"Exception": "If something goes wrong."},
+        "tags": ["basic", "test"],
+    }
+    assert parse_docstring(docstring) == expected
+
+
+def test_parse_docstring_args_multi_line():
+    """Test with Args section containing a multi-line description."""
+    docstring = """
+    Processes complex input.
+
+    Args:
+        config: Configuration object.
+                It contains settings for the processing job,
+                including connection details and parameters.
+    """
+    expected = {
+        "summary": "Processes complex input.",
+        "args": {
+            "config": "Configuration object. It contains settings for the processing job, including connection details and parameters."
+        },
+        "returns": "",
+        "raises": {},
+        "tags": [],
+    }
+    assert parse_docstring(docstring) == expected
+
+
+def test_parse_docstring_returns_multi_line():
+    """Test with Returns section spanning multiple lines."""
+    docstring = """
+    Fetches detailed information.
+
+    Returns:
+        A dictionary containing comprehensive details about the fetched data,
+        including timestamps, source, and processing status. This is a detailed response.
+    """
+    expected = {
+        "summary": "Fetches detailed information.",
+        "args": {},
+        "returns": "A dictionary containing comprehensive details about the fetched data, including timestamps, source, and processing status. This is a detailed response.",
+        "raises": {},
+        "tags": [],
+    }
+    assert parse_docstring(docstring) == expected
+
+
+def test_parse_docstring_raises_multi_line():
+    """Test with Raises section containing a multi-line description for an exception."""
+    docstring = """
+    Performs a critical action.
+
+    Raises:
+        CriticalError: This error indicates a major failure
+                       during the critical action execution.
+                       Further details are logged separately.
+    """
+    expected = {
+        "summary": "Performs a critical action.",
+        "args": {},
+        "returns": "",
+        "raises": {
+            "CriticalError": "This error indicates a major failure during the critical action execution. Further details are logged separately."
+        },
+        "tags": [],
+    }
+    assert parse_docstring(docstring) == expected
+
+
+def test_parse_docstring_no_summary():
+    """Test a docstring that starts immediately with a section."""
+    docstring = """
+    Args:
+        data: The data to process.
+    Returns:
+        Processed data.
+    """
+    expected = {
+        "summary": "",
+        "args": {"data": "The data to process."},
+        "returns": "Processed data.",
+        "raises": {},
+        "tags": [],
+    }
+    assert parse_docstring(docstring) == expected

{universal_mcp-0.1.8rc4 → universal_mcp-0.1.10}/src/universal_mcp/applications/__init__.py

@@ -2,7 +2,11 @@ import importlib
 
 from loguru import logger
 
-from universal_mcp.applications.application import APIApplication, Application
+from universal_mcp.applications.application import (
+    APIApplication,
+    BaseApplication,
+    GraphQLApplication,
+)
 
 # Name are in the format of "app-name", eg, google-calendar
 # Folder name is "app_name", eg, google_calendar
@@ -21,6 +25,7 @@ def app_from_slug(slug: str):
 
 __all__ = [
     "app_from_slug",
-    "Application",
+    "BaseApplication",
     "APIApplication",
+    "GraphQLApplication",
 ]