Merge pull request #334 from UiPath/chore/update-vertex-async-client

chore: use ChatGoogleGenerativeAI instead of deprecated Vertex

Author: radugheo

.github/workflows/integration_tests.yml

@@ -36,6 +36,7 @@ jobs:
   integration-tests:
     needs: [discover-testcases]
     runs-on: ubuntu-latest
+    timeout-minutes: 10
     container:
       image: ghcr.io/astral-sh/uv:python3.12-bookworm
       env:

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.1.25"
+version = "0.1.26"
 description = "Python SDK that enables developers to build and deploy LangGraph agents to the UiPath Cloud Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
@@ -33,9 +33,8 @@ maintainers = [
 
 [project.optional-dependencies]
 vertex = [
-    "langchain-google-vertexai>=2.1.2",
-    "langchain-community>=0.4.1",
-    "types-protobuf>=6.32.1.20251105",
+    "langchain-google-genai>=2.0.0",
+    "google-generativeai>=0.8.0",
 ]
 bedrock = [
     "langchain-aws>=0.2.35",
@@ -84,7 +83,7 @@ select = ["E", "F", "B", "I"]
 [tool.ruff.lint.per-file-ignores]
 "*" = ["E501"]
 "src/uipath_langchain/chat/bedrock.py" = ["E402"]
-"src/uipath_langchain/chat/gemini.py" = ["E402"]
+"src/uipath_langchain/chat/vertex.py" = ["E402"]
 
 [tool.ruff.format]
 quote-style = "double"

src/uipath_langchain/chat/gemini.py

@@ -4,6 +4,7 @@
 
 import httpx
 from langchain_openai import AzureChatOpenAI
+from uipath._utils._ssl_context import get_httpx_client_kwargs
 from uipath.utils import EndpointManager
 
 from .supported_models import OpenAIModels
@@ -87,11 +88,11 @@ def __init__(
             default_headers=self._build_headers(token),
             http_async_client=httpx.AsyncClient(
                 transport=UiPathURLRewriteTransport(verify=True),
-                verify=True,
+                **get_httpx_client_kwargs(),
             ),
             http_client=httpx.Client(
                 transport=UiPathSyncURLRewriteTransport(verify=True),
-                verify=True,
+                **get_httpx_client_kwargs(),
             ),
             api_key=token,
             api_version=api_version,