Merge pull request #173 from UiPath/feat/google-adk-uipath-llm-providers

feat: add UiPath LLM gateway providers for Google ADK

Author: cristipufu

packages/uipath-google-adk/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-google-adk"
-version = "0.0.2"
+version = "0.0.3"
 description = "Python SDK that enables developers to build and deploy Google ADK agents to the UiPath Cloud Platform"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -21,6 +21,9 @@ maintainers = [
     { name = "Cristian Pufu", email = "cristian.pufu@uipath.com" }
 ]
 
+[project.optional-dependencies]
+anthropic = ["anthropic>=0.43.0"]
+
 [project.entry-points."uipath.middlewares"]
 register = "uipath_google_adk.middlewares:register_middleware"
 
@@ -93,6 +96,10 @@ module = "openinference.*"
 ignore_missing_imports = true
 ignore_errors = true
 
+[[tool.mypy.overrides]]
+module = "anthropic.*"
+ignore_missing_imports = true
+
 [tool.pytest.ini_options]
 testpaths = ["tests"]
 python_files = "test_*.py"

packages/uipath-google-adk/samples/quickstart-agent/.env.example

@@ -1,6 +1,11 @@
 import httpx
 from google.adk.agents import Agent
 
+from uipath_google_adk.chat import UiPathGemini
+
+# from uipath_google_adk.chat import UiPathOpenAI
+# from uipath_google_adk.chat import UiPathAnthropic
+
 
 def get_weather(location: str) -> str:
     """Get the current weather for a location using the Open-Meteo API.
@@ -55,7 +60,9 @@ def get_weather(location: str) -> str:
 
 agent = Agent(
     name="weather_agent",
-    model="gemini-2.5-flash",
+    model=UiPathGemini(model="gemini-2.5-flash"),
+    # model=UiPathOpenAI(model="gpt-4o-mini-2024-07-18"),
+    # model=UiPathAnthropic(model="anthropic.claude-haiku-4-5-20251001-v1:0"),
     instruction="You are a helpful weather assistant. Use the get_weather tool to provide weather information.",
     tools=[get_weather],
 )

packages/uipath-google-adk/samples/quickstart-agent/main.py

@@ -5,7 +5,7 @@ description = "Quickstart Google ADK agent example"
 readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
-    "uipath-google-adk>=0.0.1, <0.1.0",
+    "uipath-google-adk[anthropic]>=0.0.1, <0.1.0",
     "google-adk>=1.25.0",
     "uipath>=2.8.18, <2.9.0",
 ]
@@ -17,3 +17,7 @@ dev = [
 
 [tool.uv]
 override-dependencies = ["opentelemetry-sdk>=1.39.0,<1.40.0"]
+
+[tool.uv.sources]
+uipath-dev = { path = "../../../../../uipath-dev-python", editable = true }
+uipath-google-adk = { path = "../../", editable = true }

packages/uipath-google-adk/samples/quickstart-agent/pyproject.toml

@@ -1,7 +1,7 @@
 {
   "$schema": "https://cloud.uipath.com/draft/2024-12/uipath",
   "runtimeOptions": {
-    "isConversational": false
+    "isConversational": true
   },
   "packOptions": {
     "fileExtensionsIncluded": [],
@@ -11,4 +11,4 @@
     "includeUvLock": true
   },
   "functions": {}
-}
+}

packages/uipath-google-adk/samples/quickstart-agent/uipath.json

@@ -0,0 +1 @@
+GOOGLE_API_KEY=your_gemini_api_key_here

packages/uipath-google-adk/samples/typed-agent/env.example

@@ -1 +1,16 @@
 """UiPath Google ADK Runtime Integration."""
+
+
+def __getattr__(name):
+    if name in ("UiPathOpenAI", "UiPathGemini", "UiPathAnthropic"):
+        from . import chat
+
+        return getattr(chat, name)
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+__all__ = [
+    "UiPathOpenAI",
+    "UiPathGemini",
+    "UiPathAnthropic",
+]

packages/uipath-google-adk/src/uipath_google_adk/__init__.py

@@ -0,0 +1,32 @@
+"""
+UiPath Google ADK Chat module.
+
+Provides LLM implementations that route through UiPath's LLM Gateway (AgentHub)
+for use with Google ADK agents.
+
+NOTE: This module uses lazy imports via __getattr__ to avoid loading heavy
+dependencies (httpx, anthropic SDK) at import time.
+"""
+
+
+def __getattr__(name):
+    if name == "UiPathOpenAI":
+        from .openai import UiPathOpenAI
+
+        return UiPathOpenAI
+    if name == "UiPathGemini":
+        from .gemini import UiPathGemini
+
+        return UiPathGemini
+    if name == "UiPathAnthropic":
+        from .anthropic import UiPathAnthropic
+
+        return UiPathAnthropic
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+__all__ = [
+    "UiPathOpenAI",
+    "UiPathGemini",
+    "UiPathAnthropic",
+]

packages/uipath-google-adk/src/uipath_google_adk/chat/__init__.py

@@ -0,0 +1,76 @@
+"""Shared utilities for UiPath Google ADK chat providers."""
+
+import os
+from typing import Optional
+
+from uipath.utils import EndpointManager
+
+
+def get_uipath_config() -> tuple[str, str]:
+    """Read UiPath configuration from environment variables.
+
+    Returns:
+        Tuple of (uipath_url, access_token).
+
+    Raises:
+        ValueError: If required environment variables are not set.
+    """
+    uipath_url = os.getenv("UIPATH_URL")
+    if not uipath_url:
+        raise ValueError("UIPATH_URL environment variable is required")
+
+    access_token = os.getenv("UIPATH_ACCESS_TOKEN")
+    if not access_token:
+        raise ValueError("UIPATH_ACCESS_TOKEN environment variable is required")
+
+    return uipath_url, access_token
+
+
+def get_uipath_headers(token: str) -> dict[str, str]:
+    """Build HTTP headers for UiPath Gateway requests.
+
+    Args:
+        token: UiPath access token.
+
+    Returns:
+        Dictionary of HTTP headers.
+    """
+    headers = {
+        "Authorization": f"Bearer {token}",
+    }
+    if job_key := os.getenv("UIPATH_JOB_KEY"):
+        headers["X-UiPath-JobKey"] = job_key
+    if process_key := os.getenv("UIPATH_PROCESS_KEY"):
+        headers["X-UiPath-ProcessKey"] = process_key
+    return headers
+
+
+def build_gateway_url(
+    vendor: str,
+    model: str,
+    uipath_url: Optional[str] = None,
+) -> str:
+    """Build the full URL for the UiPath LLM Gateway.
+
+    Args:
+        vendor: The LLM vendor (e.g., "openai", "vertexai", "anthropic").
+        model: The model name.
+        uipath_url: Optional UiPath URL. If not provided, reads from UIPATH_URL env var.
+
+    Returns:
+        The full gateway URL.
+
+    Raises:
+        ValueError: If UIPATH_URL is not set.
+    """
+    if not uipath_url:
+        uipath_url = os.getenv("UIPATH_URL")
+    if not uipath_url:
+        raise ValueError("UIPATH_URL environment variable is required")
+
+    vendor_endpoint = EndpointManager.get_vendor_endpoint()
+    formatted_endpoint = vendor_endpoint.format(
+        vendor=vendor,
+        model=model,
+    )
+    return f"{uipath_url.rstrip('/')}/{formatted_endpoint}"

packages/uipath-google-adk/src/uipath_google_adk/chat/_common.py

@@ -0,0 +1,132 @@
+"""UiPath Anthropic LLM for Google ADK agents.
+
+Wraps the Google ADK ``AnthropicLlm`` class, overriding the
+``_anthropic_client`` cached property to create an ``AsyncAnthropic``
+client that routes all requests through UiPath's LLM Gateway using
+the Bedrock invoke format.
+"""
+
+from __future__ import annotations
+
+import json
+import logging
+from functools import cached_property
+
+import httpx
+from google.adk.models.anthropic_llm import AnthropicLlm
+from typing_extensions import override
+from uipath._utils._ssl_context import get_httpx_client_kwargs
+
+from ._common import build_gateway_url, get_uipath_config, get_uipath_headers
+
+logger = logging.getLogger(__name__)
+
+
+def _check_anthropic_dependency() -> None:
+    """Check that the anthropic SDK is installed."""
+    import importlib.util
+
+    if importlib.util.find_spec("anthropic") is None:
+        raise ImportError(
+            "The 'anthropic' package is required to use UiPathAnthropic.\n"
+            "Install it with:\n\n"
+            "  pip install 'uipath-google-adk[anthropic]'\n\n"
+            "  # or with uv:\n"
+            "  uv add 'uipath-google-adk[anthropic]'\n"
+        )
+
+
+class _AsyncUrlRewriteTransport(httpx.AsyncHTTPTransport):
+    """Async transport that rewrites Anthropic SDK requests to UiPath gateway.
+
+    Intercepts the Anthropic SDK's ``/v1/messages`` endpoint and:
+    1. Rewrites the URL to the UiPath gateway
+    2. Transforms the body to Bedrock invoke format (removes ``model``,
+       adds ``anthropic_version``)
+
+    Extends ``httpx.AsyncHTTPTransport`` directly (proven pattern from
+    ``uipath-openai-agents``) and mutates ``request.url`` in-place.
+    """
+
+    def __init__(self, gateway_url: str, **kwargs):
+        self.gateway_url = gateway_url
+        super().__init__(**kwargs)
+
+    async def handle_async_request(self, request: httpx.Request) -> httpx.Response:
+        url_str = str(request.url)
+        # Intercept Anthropic SDK's /v1/messages endpoint
+        if "/v1/messages" in url_str:
+            gateway_parsed = httpx.URL(self.gateway_url)
+
+            # Transform body to Bedrock invoke format
+            body = json.loads(request.content)
+            body.pop("model", None)
+            body.setdefault("anthropic_version", "bedrock-2023-05-31")
+            is_streaming = body.get("stream", False)
+            new_content = json.dumps(body).encode()
+
+            headers = dict(request.headers)
+            headers["host"] = gateway_parsed.host
+            headers["content-length"] = str(len(new_content))
+            if is_streaming:
+                headers["X-UiPath-Streaming-Enabled"] = "true"
+
+            # Create new request with modified URL and body
+            request = httpx.Request(
+                method=request.method,
+                url=gateway_parsed,
+                headers=headers,
+                content=new_content,
+                extensions=request.extensions,
+            )
+
+        return await super().handle_async_request(request)
+
+
+class UiPathAnthropic(AnthropicLlm):
+    """Anthropic LLM that routes requests through UiPath's LLM Gateway.
+
+    Subclasses Google ADK's ``AnthropicLlm`` and overrides
+    ``_anthropic_client`` to use the Bedrock invoke format through
+    the UiPath gateway (matching the ``uipath-llamaindex`` implementation).
+
+    Uses Bedrock model names (``anthropic.claude-*``) with the ``awsbedrock``
+    vendor and ``invoke`` API flavor.
+
+    Example::
+
+        from uipath_google_adk.chat import UiPathAnthropic
+        from google.adk.agents import Agent
+
+        agent = Agent(
+            name="assistant",
+            model=UiPathAnthropic(model="anthropic.claude-haiku-4-5-20251001-v1:0"),
+            instruction="You are a helpful assistant.",
+        )
+    """
+
+    @cached_property
+    @override
+    def _anthropic_client(self):
+        _check_anthropic_dependency()
+
+        from anthropic import AsyncAnthropic  # type: ignore[import-not-found]
+
+        uipath_url, token = get_uipath_config()
+        effective_model = self.model
+        gateway_url = build_gateway_url("awsbedrock", effective_model, uipath_url)
+        auth_headers = get_uipath_headers(token)
+        auth_headers["X-UiPath-LlmGateway-ApiFlavor"] = "invoke"
+        client_kwargs = get_httpx_client_kwargs()
+        verify = client_kwargs.get("verify", True)
+
+        http_client = httpx.AsyncClient(
+            transport=_AsyncUrlRewriteTransport(gateway_url, verify=verify),
+            **client_kwargs,
+        )
+
+        return AsyncAnthropic(
+            api_key="uipath-gateway",
+            default_headers=auth_headers,
+            http_client=http_client,
+        )