uipath-langchain 0.0.116__tar.gz → 0.0.118__tar.gz

{uipath_langchain-0.0.116 → uipath_langchain-0.0.118}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.116
+Version: 0.0.118
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-langchain-python
@@ -24,7 +24,7 @@ Requires-Dist: langgraph>=0.2.70
 Requires-Dist: openai>=1.65.5
 Requires-Dist: pydantic-settings>=2.6.0
 Requires-Dist: python-dotenv>=1.0.1
-Requires-Dist: uipath<2.1.0,>=2.0.77
+Requires-Dist: uipath<2.1.0,>=2.0.79
 Provides-Extra: langchain
 Description-Content-Type: text/markdown
 

{uipath_langchain-0.0.116 → uipath_langchain-0.0.118}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.116"
+version = "0.0.118"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
-    "uipath>=2.0.77, <2.1.0",
+    "uipath>=2.0.79, <2.1.0",
     "langgraph>=0.2.70",
     "langchain-core>=0.3.34",
     "langgraph-checkpoint-sqlite>=2.0.3",

{uipath_langchain-0.0.116 → uipath_langchain-0.0.118}/src/uipath_langchain/_utils/_request_mixin.py

@@ -40,6 +40,8 @@ class UiPathRequestMixin(BaseModel):
 
     default_headers: Optional[Mapping[str, str]] = {
         "X-UiPath-Streaming-Enabled": "false",
+        "X-UiPath-JobKey": os.getenv("UIPATH_JOB_KEY", ""),
+        "X-UiPath-ProcessKey": os.getenv("UIPATH_PROCESS_KEY", ""),
     }
     model_name: Optional[str] = Field(
         default_factory=lambda: os.getenv("UIPATH_MODEL_NAME", "gpt-4o-2024-08-06"),

{uipath_langchain-0.0.116 → uipath_langchain-0.0.118}/src/uipath_langchain/_utils/_settings.py

@@ -1,6 +1,5 @@
 # mypy: disable-error-code="syntax"
 import os
-from enum import Enum
 from typing import Any, Optional
 
 import httpx
@@ -47,14 +46,6 @@ class UiPathClientSettings(BaseSettings):
     action_id: str = Field(default="DefaultActionId", alias="UIPATH_ACTION_ID")
 
 
-class UiPathEndpoints(Enum):
-    NORMALIZED_COMPLETION_ENDPOINT = "llmgateway_/api/chat/completions"
-    PASSTHROUGH_COMPLETION_ENDPOINT = "llmgateway_/openai/deployments/{model}/chat/completions?api-version={api_version}"
-    EMBEDDING_ENDPOINT = (
-        "llmgateway_/openai/deployments/{model}/embeddings?api-version={api_version}"
-    )
-
-
 def get_uipath_token_header(
     settings: Any = None,
 ) -> str:

{uipath_langchain-0.0.116 → uipath_langchain-0.0.118}/src/uipath_langchain/chat/models.py

@@ -1,4 +1,5 @@
 import json
+import logging
 from typing import Any, Dict, List, Literal, Optional, Union
 
 from langchain_core.callbacks import (
@@ -12,9 +13,11 @@ from langchain_core.outputs import ChatGeneration, ChatResult
 from langchain_core.runnables import Runnable
 from langchain_openai.chat_models import AzureChatOpenAI
 from pydantic import BaseModel
+from uipath.utils import EndpointManager
 
 from uipath_langchain._utils._request_mixin import UiPathRequestMixin
-from uipath_langchain._utils._settings import UiPathEndpoints
+
+logger = logging.getLogger(__name__)
 
 
 class UiPathAzureChatOpenAI(UiPathRequestMixin, AzureChatOpenAI):
@@ -71,7 +74,9 @@ class UiPathAzureChatOpenAI(UiPathRequestMixin, AzureChatOpenAI):
 
     @property
     def endpoint(self) -> str:
-        return UiPathEndpoints.PASSTHROUGH_COMPLETION_ENDPOINT.value.format(
+        endpoint = EndpointManager.get_passthrough_endpoint()
+        logger.debug("Using endpoint: %s", endpoint)
+        return endpoint.format(
             model=self.model_name, api_version=self.openai_api_version
         )
 
@@ -252,7 +257,11 @@ class UiPathChat(UiPathRequestMixin, AzureChatOpenAI):
 
     @property
     def endpoint(self) -> str:
-        return UiPathEndpoints.NORMALIZED_COMPLETION_ENDPOINT.value
+        endpoint = EndpointManager.get_normalized_endpoint()
+        logger.debug("Using endpoint: %s", endpoint)
+        return endpoint.format(
+            model=self.model_name, api_version=self.openai_api_version
+        )
 
     @property
     def is_normalized(self) -> bool:

{uipath_langchain-0.0.116 → uipath_langchain-0.0.118}/src/uipath_langchain/embeddings/embeddings.py

@@ -5,9 +5,9 @@ import httpx
 from langchain_community.callbacks.manager import openai_callback_var
 from langchain_openai.embeddings import AzureOpenAIEmbeddings, OpenAIEmbeddings
 from pydantic import Field
+from uipath.utils import EndpointManager
 
 from uipath_langchain._utils._request_mixin import UiPathRequestMixin
-from uipath_langchain._utils._settings import UiPathEndpoints
 
 
 class UiPathAzureOpenAIEmbeddings(UiPathRequestMixin, AzureOpenAIEmbeddings):
@@ -43,7 +43,8 @@ class UiPathAzureOpenAIEmbeddings(UiPathRequestMixin, AzureOpenAIEmbeddings):
 
     @property
     def endpoint(self) -> str:
-        return UiPathEndpoints.EMBEDDING_ENDPOINT.value.format(
+        endpoint = EndpointManager.get_embeddings_endpoint()
+        return endpoint.format(
             model=self.model_name, api_version=self.openai_api_version
         )
 
@@ -105,6 +106,7 @@ class UiPathOpenAIEmbeddings(UiPathRequestMixin, OpenAIEmbeddings):
 
     @property
     def endpoint(self) -> str:
-        return UiPathEndpoints.EMBEDDING_ENDPOINT.value.format(
+        endpoint = EndpointManager.get_embeddings_endpoint()
+        return endpoint.format(
             model=self.model_name, api_version=self.openai_api_version
         )