uipath-langchain 0.0.74__tar.gz → 0.0.76__tar.gz

{uipath_langchain-0.0.74 → uipath_langchain-0.0.76}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.74
+Version: 0.0.76
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-python
@@ -25,5 +25,5 @@ Requires-Dist: pydantic-settings>=2.6.0
 Requires-Dist: python-dotenv>=1.0.1
 Requires-Dist: requests>=2.23.3
 Requires-Dist: types-requests>=2.32.0.20241016
-Requires-Dist: uipath-sdk>=0.0.99
+Requires-Dist: uipath-sdk>=0.0.101
 Provides-Extra: langchain

{uipath_langchain-0.0.74 → uipath_langchain-0.0.76}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.74"
+version = "0.0.76"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.9"
 dependencies = [
-    "uipath-sdk>=0.0.99",
+    "uipath-sdk>=0.0.101",
     "langgraph>=0.2.70",
     "langchain-core>=0.3.34",
     "langgraph-checkpoint-sqlite>=2.0.3",

{uipath_langchain-0.0.74 → uipath_langchain-0.0.76}/uipath_langchain/_cli/_runtime/_output.py

@@ -15,7 +15,7 @@ from uipath_sdk._cli._runtime._contracts import (
     UiPathRuntimeResult,
     UiPathRuntimeStatus,
 )
-from uipath_sdk._models._interrupt_models.invoke_process import InvokeProcess
+from uipath_sdk._models import InvokeProcess
 from uipath_sdk._models.actions import Action
 
 from ._context import LangGraphRuntimeContext

{uipath_langchain-0.0.74 → uipath_langchain-0.0.76}/uipath_langchain/_cli/cli_run.py

@@ -4,6 +4,7 @@ from os import environ as env
 from typing import Optional
 
 from dotenv import load_dotenv
+from uipath_sdk._cli._runtime._contracts import UiPathTraceContext
 from uipath_sdk._cli.middlewares import MiddlewareResult
 
 from ._runtime._context import LangGraphRuntimeContext
@@ -40,6 +41,17 @@ def langgraph_run_middleware(
             context.job_id = env.get("UIPATH_JOB_KEY")
             context.trace_id = env.get("UIPATH_TRACE_ID")
             context.tracing_enabled = env.get("UIPATH_TRACING_ENABLED", True)
+            context.trace_context = UiPathTraceContext(
+                enabled=env.get("UIPATH_TRACING_ENABLED", True),
+                trace_id=env.get("UIPATH_TRACE_ID"),
+                parent_span_id=env.get("UIPATH_PARENT_SPAN_ID"),
+                root_span_id=env.get("UIPATH_ROOT_SPAN_ID"),
+                job_id=env.get("UIPATH_JOB_KEY"),
+                org_id=env.get("UIPATH_ORGANIZATION_ID"),
+                tenant_id=env.get("UIPATH_TENANT_ID"),
+                process_key=env.get("UIPATH_PROCESS_UUID"),
+                folder_key=env.get("UIPATH_FOLDER_KEY"),
+            )
             context.langsmith_tracing_enabled = env.get("LANGSMITH_TRACING", False)
 
             # Add default env variables

{uipath_langchain-0.0.74/uipath_langchain/utils → uipath_langchain-0.0.76/uipath_langchain/_utils}/_request_mixin.py

@@ -18,12 +18,12 @@ from tenacity import (
     wait_exponential_jitter,
 )
 
-from uipath_langchain.utils._settings import (
+from uipath_langchain._utils._settings import (
     UiPathClientFactorySettings,
     UiPathClientSettings,
     get_uipath_token_header,
 )
-from uipath_langchain.utils._sleep_policy import before_sleep_log
+from uipath_langchain._utils._sleep_policy import before_sleep_log
 
 
 def get_from_uipath_url():
@@ -425,9 +425,14 @@ class UiPathRequestMixin(BaseModel):
     @property
     def url(self) -> str:
         if not self._url:
-            self._url = (
-                f"{self.base_url}/{self.org_id}/{self.tenant_id}/{self.endpoint}"
-            )
+            env_uipath_url = os.getenv("UIPATH_URL")
+
+            if env_uipath_url:
+                self._url = f"{env_uipath_url.rstrip('/')}/{self.endpoint}"
+            else:
+                self._url = (
+                    f"{self.base_url}/{self.org_id}/{self.tenant_id}/{self.endpoint}"
+                )
         return self._url
 
     @property

{uipath_langchain-0.0.74 → uipath_langchain-0.0.76}/uipath_langchain/chat/models.py

@@ -13,8 +13,8 @@ from langchain_core.runnables import Runnable
 from langchain_openai.chat_models import AzureChatOpenAI
 from pydantic import BaseModel
 
-from uipath_langchain.utils._request_mixin import UiPathRequestMixin
-from uipath_langchain.utils._settings import UiPathEndpoints
+from uipath_langchain._utils._request_mixin import UiPathRequestMixin
+from uipath_langchain._utils._settings import UiPathEndpoints
 
 
 class UiPathAzureChatOpenAI(UiPathRequestMixin, AzureChatOpenAI):

{uipath_langchain-0.0.74 → uipath_langchain-0.0.76}/uipath_langchain/embeddings/embeddings.py

@@ -6,8 +6,8 @@ from langchain_community.callbacks.manager import openai_callback_var
 from langchain_openai.embeddings import AzureOpenAIEmbeddings, OpenAIEmbeddings
 from pydantic import Field
 
-from uipath_langchain.utils._request_mixin import UiPathRequestMixin
-from uipath_langchain.utils._settings import UiPathEndpoints
+from uipath_langchain._utils._request_mixin import UiPathRequestMixin
+from uipath_langchain._utils._settings import UiPathEndpoints
 
 
 class UiPathAzureOpenAIEmbeddings(UiPathRequestMixin, AzureOpenAIEmbeddings):