Merge pull request #129 from UiPath/fix/uipath_chat_endpoint

fix(chat): fix uipath chat endpoint

Author: ionmincu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.117"
+version = "0.0.118"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"

src/uipath_langchain/chat/models.py

@@ -1,6 +1,5 @@
 import json
 import logging
-from os import environ as env
 from typing import Any, Dict, List, Literal, Optional, Union
 
 from langchain_core.callbacks import (
@@ -85,15 +84,6 @@ def endpoint(self) -> str:
 class UiPathChat(UiPathRequestMixin, AzureChatOpenAI):
     """Custom LLM connector for LangChain integration with UiPath Normalized."""
 
-    def __init__(self, *args: Any, **kwargs: Any):
-        """Initialize the UiPath Azure Chat OpenAI model."""
-
-        super().__init__(*args, **kwargs)
-        self.default_headers = {
-            "X-UiPath-JobKey": env.get("UIPATH_JOB_KEY", ""),
-            "X-UiPath-ProcessKey": env.get("UIPATH_PROCESS_KEY", ""),
-        }
-
     def _create_chat_result(
         self,
         response: Union[Dict[str, Any], BaseModel],
@@ -267,7 +257,7 @@ def with_structured_output(
 
     @property
     def endpoint(self) -> str:
-        endpoint = EndpointManager.get_passthrough_endpoint()
+        endpoint = EndpointManager.get_normalized_endpoint()
         logger.debug("Using endpoint: %s", endpoint)
         return endpoint.format(
             model=self.model_name, api_version=self.openai_api_version