uipath-langchain-client 1.0.3__tar.gz → 1.0.7__tar.gz

{uipath_langchain_client-1.0.3 → uipath_langchain_client-1.0.7}/CHANGELOG.md

@@ -2,6 +2,27 @@
 
 All notable changes to `uipath_langchain_client` will be documented in this file.
 
+## [1.0.7] - 2026-02-04
+
+### Refactor
+- Refactor factory function to include byom models
+
+## [1.0.6] - 2026-02-03
+
+### Refactor
+- Updated documentation to include the new aliases for settings
+- New alias for settings and request timeout in BaseLLMClient
+
+## [1.0.5] - 2026-02-03
+
+### Bug Fix
+- Fixed retry logic on all clients
+
+## [1.0.4] - 2026-02-03
+
+### Bug Fix
+- Fix some timout issues on langchain_openai from llmgw.
+
 ## [1.0.3] - 2026-02-02
 
 ### Bug Fix

{uipath_langchain_client-1.0.3 → uipath_langchain_client-1.0.7}/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: uipath-langchain-client
-Version: 1.0.3
+Version: 1.0.7
 Summary: LangChain-compatible chat models and embeddings for UiPath's LLM services
 Requires-Python: >=3.11
 Requires-Dist: langchain>=1.2.7
-Requires-Dist: uipath-llm-client>=1.0.3
+Requires-Dist: uipath-llm-client>=1.0.6
 Provides-Extra: all
 Requires-Dist: anthropic[bedrock,vertex]>=0.77.0; extra == 'all'
 Requires-Dist: langchain-anthropic>=1.3.1; extra == 'all'
@@ -71,7 +71,7 @@ print(response.content)
 
 # Embeddings model
 embeddings = get_embedding_model(
-    model="text-embedding-3-large",
+    model_name="text-embedding-3-large",
     client_settings=settings,
 )
 vectors = embeddings.embed_documents(["Hello world"])
@@ -92,20 +92,20 @@ from uipath_langchain_client.settings import get_default_client_settings
 settings = get_default_client_settings()
 
 # OpenAI/Azure
-openai_chat = UiPathAzureChatOpenAI(model="gpt-4o-2024-11-20", client_settings=settings)
+openai_chat = UiPathAzureChatOpenAI(model="gpt-4o-2024-11-20", settings=settings)
 
 # Google Gemini
-gemini_chat = UiPathChatGoogleGenerativeAI(model="gemini-2.5-flash", client_settings=settings)
+gemini_chat = UiPathChatGoogleGenerativeAI(model="gemini-2.5-flash", settings=settings)
 
 # Anthropic Claude (via AWS Bedrock)
 claude_chat = UiPathChatAnthropic(
     model="anthropic.claude-sonnet-4-5-20250929-v1:0",
-    client_settings=settings,
+    settings=settings,
     vendor_type="awsbedrock",
 )
 
 # Normalized (provider-agnostic)
-normalized_chat = UiPathNormalizedChatModel(model="gpt-4o-2024-11-20", client_settings=settings)
+normalized_chat = UiPathNormalizedChatModel(model="gpt-4o-2024-11-20", settings=settings)
 ```
 
 ## Available Client Types

{uipath_langchain_client-1.0.3 → uipath_langchain_client-1.0.7}/README.md

@@ -41,7 +41,7 @@ print(response.content)
 
 # Embeddings model
 embeddings = get_embedding_model(
-    model="text-embedding-3-large",
+    model_name="text-embedding-3-large",
     client_settings=settings,
 )
 vectors = embeddings.embed_documents(["Hello world"])
@@ -62,20 +62,20 @@ from uipath_langchain_client.settings import get_default_client_settings
 settings = get_default_client_settings()
 
 # OpenAI/Azure
-openai_chat = UiPathAzureChatOpenAI(model="gpt-4o-2024-11-20", client_settings=settings)
+openai_chat = UiPathAzureChatOpenAI(model="gpt-4o-2024-11-20", settings=settings)
 
 # Google Gemini
-gemini_chat = UiPathChatGoogleGenerativeAI(model="gemini-2.5-flash", client_settings=settings)
+gemini_chat = UiPathChatGoogleGenerativeAI(model="gemini-2.5-flash", settings=settings)
 
 # Anthropic Claude (via AWS Bedrock)
 claude_chat = UiPathChatAnthropic(
     model="anthropic.claude-sonnet-4-5-20250929-v1:0",
-    client_settings=settings,
+    settings=settings,
     vendor_type="awsbedrock",
 )
 
 # Normalized (provider-agnostic)
-normalized_chat = UiPathNormalizedChatModel(model="gpt-4o-2024-11-20", client_settings=settings)
+normalized_chat = UiPathNormalizedChatModel(model="gpt-4o-2024-11-20", settings=settings)
 ```
 
 ## Available Client Types

{uipath_langchain_client-1.0.3 → uipath_langchain_client-1.0.7}/demo.py

@@ -167,7 +167,7 @@ def demo_embeddings():
 
     settings = get_default_client_settings()
     embeddings = get_embedding_model(
-        model="text-embedding-3-large",
+        model_name="text-embedding-3-large",
         client_settings=settings,
     )
 
@@ -241,7 +241,7 @@ def demo_direct_client_usage():
     # Use the Azure OpenAI client directly for more control
     chat_model = UiPathAzureChatOpenAI(
         model="gpt-4o-2024-11-20",
-        client_settings=settings,
+        settings=settings,
         temperature=0.7,
     )
 

{uipath_langchain_client-1.0.3 → uipath_langchain_client-1.0.7}/pyproject.toml

@@ -6,7 +6,7 @@ readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
     "langchain>=1.2.7",
-    "uipath-llm-client>=1.0.3",
+    "uipath-llm-client>=1.0.6",
 ]
 
 [project.optional-dependencies]

{uipath_langchain_client-1.0.3 → uipath_langchain_client-1.0.7}/src/uipath_langchain_client/__init__.py

@@ -15,11 +15,11 @@ Quick Start:
     >>> settings = get_default_client_settings()
     >>>
     >>> # Chat model with auto-detected vendor
-    >>> chat = get_chat_model("gpt-4o-2024-11-20", settings)
+    >>> chat = get_chat_model(model_name="gpt-4o-2024-11-20", client_settings=settings)
     >>> response = chat.invoke("Hello!")
     >>>
     >>> # Embeddings model
-    >>> embeddings = get_embedding_model("text-embedding-3-large", settings)
+    >>> embeddings = get_embedding_model(model_name="text-embedding-3-large", client_settings=settings)
     >>> vectors = embeddings.embed_documents(["Hello world"])
 
 Settings:

{uipath_langchain_client-1.0.3 → uipath_langchain_client-1.0.7}/src/uipath_langchain_client/__version__.py

@@ -1,3 +1,3 @@
 __title__ = "UiPath LangChain Client"
 __description__ = "A Python client for interacting with UiPath's LLM services via LangChain."
-__version__ = "1.0.3"
+__version__ = "1.0.7"

{uipath_langchain_client-1.0.3 → uipath_langchain_client-1.0.7}/src/uipath_langchain_client/base_client.py

@@ -29,7 +29,7 @@ from functools import cached_property
 from typing import Any, Literal
 
 from httpx import URL, Response
-from pydantic import BaseModel, ConfigDict, Field
+from pydantic import AliasChoices, BaseModel, ConfigDict, Field
 
 from uipath_langchain_client.settings import (
     UiPathAPIConfig,
@@ -85,23 +85,26 @@ class UiPathBaseLLMClient(BaseModel):
         description="Settings for the UiPath API",
     )
     client_settings: UiPathBaseSettings = Field(
+        alias="settings",
         default_factory=get_default_client_settings,
         description="Settings for the UiPath client (defaults based on UIPATH_LLM_BACKEND env var)",
     )
+
     default_headers: Mapping[str, str] | None = Field(
         default={
-            "X-UiPath-LLMGateway-TimeoutSeconds": "30",  # server side timeout, default is 10, maximum is 300
+            "X-UiPath-LLMGateway-TimeoutSeconds": "300",  # server side timeout, default is 10, maximum is 300
             "X-UiPath-LLMGateway-AllowFull4xxResponse": "true",  # allow full 4xx responses (default is false)
         },
         description="Default request headers to include in requests",
     )
-
     request_timeout: int | None = Field(
+        alias="timeout",
+        validation_alias=AliasChoices("timeout", "request_timeout", "default_request_timeout"),
         default=None,
         description="Client-side request timeout in seconds",
     )
     max_retries: int = Field(
-        default=1,
+        default=0,
         description="Maximum number of retries for failed requests",
     )
     retry_config: RetryConfig | None = Field(

{uipath_langchain_client-1.0.3 → uipath_langchain_client-1.0.7}/src/uipath_langchain_client/clients/anthropic/chat_models.py

@@ -62,7 +62,7 @@ class UiPathChatAnthropic(UiPathBaseLLMClient, ChatAnthropic):
                     api_key="PLACEHOLDER",
                     base_url=str(self.uipath_sync_client.base_url),
                     default_headers=dict(self.uipath_sync_client.headers),
-                    max_retries=1,  # handled by the UiPathBaseLLMClient
+                    max_retries=0,  # handled by the UiPathBaseLLMClient
                     timeout=None,  # handled by the UiPathBaseLLMClient
                     http_client=self.uipath_sync_client,
                 )
@@ -74,7 +74,7 @@ class UiPathChatAnthropic(UiPathBaseLLMClient, ChatAnthropic):
                     base_url=str(self.uipath_sync_client.base_url),
                     default_headers=dict(self.uipath_sync_client.headers),
                     timeout=None,  # handled by the UiPathBaseLLMClient
-                    max_retries=1,  # handled by the UiPathBaseLLMClient
+                    max_retries=0,  # handled by the UiPathBaseLLMClient
                     http_client=self.uipath_sync_client,
                 )
             case "awsbedrock":
@@ -85,7 +85,7 @@ class UiPathChatAnthropic(UiPathBaseLLMClient, ChatAnthropic):
                     base_url=str(self.uipath_sync_client.base_url),
                     default_headers=dict(self.uipath_sync_client.headers),
                     timeout=None,  # handled by the UiPathBaseLLMClient
-                    max_retries=1,  # handled by the UiPathBaseLLMClient
+                    max_retries=0,  # handled by the UiPathBaseLLMClient
                     http_client=self.uipath_sync_client,
                 )
             case "anthropic":
@@ -94,7 +94,7 @@ class UiPathChatAnthropic(UiPathBaseLLMClient, ChatAnthropic):
                     base_url=str(self.uipath_sync_client.base_url),
                     default_headers=dict(self.uipath_sync_client.headers),
                     timeout=None,  # handled by the UiPathBaseLLMClient
-                    max_retries=1,  # handled by the UiPathBaseLLMClient
+                    max_retries=0,  # handled by the UiPathBaseLLMClient
                     http_client=self.uipath_sync_client,
                 )
 
@@ -108,7 +108,7 @@ class UiPathChatAnthropic(UiPathBaseLLMClient, ChatAnthropic):
                     api_key="PLACEHOLDER",
                     base_url=str(self.uipath_async_client.base_url),
                     default_headers=dict(self.uipath_async_client.headers),
-                    max_retries=1,  # handled by the UiPathBaseLLMClient
+                    max_retries=0,  # handled by the UiPathBaseLLMClient
                     timeout=None,  # handled by the UiPathBaseLLMClient
                     http_client=self.uipath_async_client,
                 )
@@ -120,7 +120,7 @@ class UiPathChatAnthropic(UiPathBaseLLMClient, ChatAnthropic):
                     base_url=str(self.uipath_async_client.base_url),
                     default_headers=dict(self.uipath_async_client.headers),
                     timeout=None,  # handled by the UiPathBaseLLMClient
-                    max_retries=1,  # handled by the UiPathBaseLLMClient
+                    max_retries=0,  # handled by the UiPathBaseLLMClient
                     http_client=self.uipath_async_client,
                 )
             case "awsbedrock":
@@ -131,7 +131,7 @@ class UiPathChatAnthropic(UiPathBaseLLMClient, ChatAnthropic):
                     base_url=str(self.uipath_async_client.base_url),
                     default_headers=dict(self.uipath_async_client.headers),
                     timeout=None,  # handled by the UiPathBaseLLMClient
-                    max_retries=1,  # handled by the UiPathBaseLLMClient
+                    max_retries=0,  # handled by the UiPathBaseLLMClient
                     http_client=self.uipath_async_client,
                 )
             case _:
@@ -140,7 +140,7 @@ class UiPathChatAnthropic(UiPathBaseLLMClient, ChatAnthropic):
                     base_url=str(self.uipath_async_client.base_url),
                     default_headers=dict(self.uipath_async_client.headers),
                     timeout=None,  # handled by the UiPathBaseLLMClient
-                    max_retries=1,  # handled by the UiPathBaseLLMClient
+                    max_retries=0,  # handled by the UiPathBaseLLMClient
                     http_client=self.uipath_async_client,
                 )
 

{uipath_langchain_client-1.0.3 → uipath_langchain_client-1.0.7}/src/uipath_langchain_client/clients/bedrock/chat_models.py

@@ -59,5 +59,5 @@ class UiPathChatBedrock(UiPathBaseLLMClient, ChatBedrock):
     def _as_converse(self) -> UiPathChatBedrockConverse:
         return UiPathChatBedrockConverse(
             model=self.model_name,
-            client_settings=self.client_settings,
+            settings=self.client_settings,
         )

{uipath_langchain_client-1.0.3 → uipath_langchain_client-1.0.7}/src/uipath_langchain_client/clients/normalized/chat_models.py

@@ -18,7 +18,7 @@ Example:
     >>> settings = get_default_client_settings()
     >>> chat = UiPathNormalizedChatModel(
     ...     model="gpt-4o-2024-11-20",
-    ...     client_settings=settings,
+    ...     settings=settings,
     ... )
     >>> response = chat.invoke("Hello!")
 """
@@ -88,7 +88,7 @@ class UiPathNormalizedChatModel(UiPathBaseLLMClient, BaseChatModel):
     Example:
         >>> chat = UiPathNormalizedChatModel(
         ...     model="gpt-4o-2024-11-20",
-        ...     client_settings=settings,
+        ...     settings=settings,
         ...     temperature=0.7,
         ...     max_tokens=1000,
         ... )

{uipath_langchain_client-1.0.3 → uipath_langchain_client-1.0.7}/src/uipath_langchain_client/clients/openai/chat_models.py

@@ -43,13 +43,13 @@ class UiPathChatOpenAI(UiPathBaseLLMClient, ChatOpenAI):  # type: ignore[overrid
         self.root_client = OpenAI(
             api_key="PLACEHOLDER",
             timeout=None,  # handled by the UiPath client
-            max_retries=1,  # handled by the UiPath client
+            max_retries=0,  # handled by the UiPath client
             http_client=self.uipath_sync_client,
         )
         self.root_async_client = AsyncOpenAI(
             api_key="PLACEHOLDER",
             timeout=None,  # handled by the UiPath client
-            max_retries=1,  # handled by the UiPath client
+            max_retries=0,  # handled by the UiPath client
             http_client=self.uipath_async_client,
         )
         self.client = self.root_client.chat.completions
@@ -86,7 +86,7 @@ class UiPathAzureChatOpenAI(UiPathBaseLLMClient, AzureChatOpenAI):  # type: igno
             api_version="PLACEHOLDER",
             api_key="PLACEHOLDER",
             timeout=None,  # handled by the UiPath client
-            max_retries=1,  # handled by the UiPath client
+            max_retries=0,  # handled by the UiPath client
             http_client=self.uipath_sync_client,
         )
         self.root_async_client = AsyncAzureOpenAI(
@@ -94,7 +94,7 @@ class UiPathAzureChatOpenAI(UiPathBaseLLMClient, AzureChatOpenAI):  # type: igno
             api_version="PLACEHOLDER",
             api_key="PLACEHOLDER",
             timeout=None,  # handled by the UiPath client
-            max_retries=1,  # handled by the UiPath client
+            max_retries=0,  # handled by the UiPath client
             http_client=self.uipath_async_client,
         )
         self.client = self.root_client.chat.completions

{uipath_langchain_client-1.0.3 → uipath_langchain_client-1.0.7}/src/uipath_langchain_client/clients/openai/embeddings.py

@@ -36,13 +36,13 @@ class UiPathOpenAIEmbeddings(UiPathBaseLLMClient, OpenAIEmbeddings):
         self.client = OpenAI(
             api_key="PLACEHOLDER",
             timeout=None,  # handled by the UiPath client
-            max_retries=1,  # handled by the UiPath client
+            max_retries=0,  # handled by the UiPath client
             http_client=self.uipath_sync_client,
         ).embeddings
         self.async_client = AsyncOpenAI(
             api_key="PLACEHOLDER",
             timeout=None,  # handled by the UiPath client
-            max_retries=1,  # handled by the UiPath client
+            max_retries=0,  # handled by the UiPath client
             http_client=self.uipath_async_client,
         ).embeddings
         return self
@@ -70,7 +70,7 @@ class UiPathAzureOpenAIEmbeddings(UiPathBaseLLMClient, AzureOpenAIEmbeddings):
             api_version="PLACEHOLDER",
             api_key="PLACEHOLDER",
             timeout=None,  # handled by the UiPath client
-            max_retries=1,  # handled by the UiPath client
+            max_retries=0,  # handled by the UiPath client
             http_client=self.uipath_sync_client,
         ).embeddings
         self.async_client = AsyncAzureOpenAI(
@@ -78,7 +78,7 @@ class UiPathAzureOpenAIEmbeddings(UiPathBaseLLMClient, AzureOpenAIEmbeddings):
             api_version="PLACEHOLDER",
             api_key="PLACEHOLDER",
             timeout=None,  # handled by the UiPath client
-            max_retries=1,  # handled by the UiPath client
+            max_retries=0,  # handled by the UiPath client
             http_client=self.uipath_async_client,
         ).embeddings
         return self

{uipath_langchain_client-1.0.3 → uipath_langchain_client-1.0.7}/src/uipath_langchain_client/clients/vertexai/chat_models.py

@@ -36,7 +36,7 @@ class UiPathChatAnthropicVertex(UiPathBaseLLMClient, ChatAnthropicVertex):  # ty
             base_url=str(self.uipath_sync_client.base_url),
             default_headers=self.uipath_sync_client.headers,
             timeout=None,  # handled by the UiPath client
-            max_retries=1,  # handled by the UiPath client
+            max_retries=0,  # handled by the UiPath client
             http_client=self.uipath_sync_client,
         )
         self.async_client = AsyncAnthropicVertex(
@@ -46,7 +46,7 @@ class UiPathChatAnthropicVertex(UiPathBaseLLMClient, ChatAnthropicVertex):  # ty
             base_url=str(self.uipath_async_client.base_url),
             default_headers=self.uipath_async_client.headers,
             timeout=None,  # handled by the UiPath client
-            max_retries=1,  # handled by the UiPath client
+            max_retries=0,  # handled by the UiPath client
             http_client=self.uipath_async_client,
         )
         return self

{uipath_langchain_client-1.0.3 → uipath_langchain_client-1.0.7}/src/uipath_langchain_client/factory.py

@@ -16,8 +16,8 @@ Example:
     >>> settings = get_default_client_settings()
     >>>
     >>> # Auto-detect vendor from model name
-    >>> chat = get_chat_model("gpt-4o-2024-11-20", settings)
-    >>> embeddings = get_embedding_model("text-embedding-3-large", settings)
+    >>> chat = get_chat_model(model_name="gpt-4o-2024-11-20", client_settings=settings)
+    >>> embeddings = get_embedding_model(model_name="text-embedding-3-large", client_settings=settings)
 """
 
 from typing import Any, Literal
@@ -47,7 +47,14 @@ def _get_model_info(
         ]
 
     if not byo_connection_id and len(matching_models) > 1:
-        matching_models = [m for m in matching_models if m.get("byomDetails") is None]
+        matching_models = [
+            m
+            for m in matching_models
+            if (
+                (m.get("modelSubscriptionType", "") == "UiPathOwned")
+                or (m.get("byomDetails") is None)
+            )
+        ]
 
     if not matching_models:
         raise ValueError(
@@ -87,67 +94,115 @@ def get_chat_model(
             UiPathNormalizedChatModel,
         )
 
-        return UiPathNormalizedChatModel(model=model_name, **model_kwargs)
+        return UiPathNormalizedChatModel(
+            model=model_name,
+            settings=client_settings,
+            byo_connection_id=byo_connection_id,
+            **model_kwargs,
+        )
 
     vendor_type = model_info["vendor"].lower()
+    is_uipath_owned = model_info.get("modelSubscriptionType") == "UiPathOwned"
     match vendor_type:
         case "openai":
-            if "gpt" in model_name:
+            if is_uipath_owned:
                 from uipath_langchain_client.clients.openai.chat_models import (
                     UiPathAzureChatOpenAI,
                 )
 
                 return UiPathAzureChatOpenAI(
                     model=model_name,
-                    client_settings=client_settings,
+                    settings=client_settings,
                     **model_kwargs,
                 )
             else:
-                raise ValueError(f"Invalid model name: {model_name} for vendor: {vendor_type}")
-        case "vertexai":
-            if "gemini" in model_name:
-                from uipath_langchain_client.clients.google.chat_models import (
-                    UiPathChatGoogleGenerativeAI,
+                from uipath_langchain_client.clients.openai.chat_models import (
+                    UiPathChatOpenAI,
                 )
 
-                return UiPathChatGoogleGenerativeAI(
+                return UiPathChatOpenAI(
                     model=model_name,
-                    client_settings=client_settings,
+                    settings=client_settings,
+                    byo_connection_id=byo_connection_id,
                     **model_kwargs,
                 )
-            elif "claude" in model_name:
-                from uipath_langchain_client.clients.anthropic.chat_models import (
-                    UiPathChatAnthropic,
+        case "vertexai":
+            if is_uipath_owned:
+                if "claude" in model_name:
+                    from uipath_langchain_client.clients.vertexai.chat_models import (
+                        UiPathChatAnthropicVertex,
+                    )
+
+                    return UiPathChatAnthropicVertex(
+                        model=model_name,
+                        settings=client_settings,
+                        **model_kwargs,
+                    )
+                elif "gemini" in model_name:
+                    from uipath_langchain_client.clients.google.chat_models import (
+                        UiPathChatGoogleGenerativeAI,
+                    )
+
+                    return UiPathChatGoogleGenerativeAI(
+                        model=model_name,
+                        settings=client_settings,
+                        **model_kwargs,
+                    )
+                else:
+                    raise ValueError(
+                        f"We don't have a client that currently supports this model: {model_name} on vendor: {vendor_type}"
+                    )
+            else:
+                from uipath_langchain_client.clients.google.chat_models import (
+                    UiPathChatGoogleGenerativeAI,
                 )
 
-                return UiPathChatAnthropic(
+                return UiPathChatGoogleGenerativeAI(
                     model=model_name,
-                    client_settings=client_settings,
-                    vendor_type="vertexai",
+                    settings=client_settings,
+                    byo_connection_id=byo_connection_id,
                     **model_kwargs,
                 )
-            else:
-                raise ValueError(f"Invalid model name: {model_name} for vendor: {vendor_type}")
         case "awsbedrock":
-            if "claude" in model_name:
-                from uipath_langchain_client.clients.anthropic.chat_models import (
-                    UiPathChatAnthropic,
+            if is_uipath_owned:
+                if "claude" in model_name:
+                    from uipath_langchain_client.clients.anthropic.chat_models import (
+                        UiPathChatAnthropic,
+                    )
+
+                    return UiPathChatAnthropic(
+                        model=model_name,
+                        settings=client_settings,
+                        **model_kwargs,
+                    )
+                else:
+                    from uipath_langchain_client.clients.bedrock.chat_models import (
+                        UiPathChatBedrock,
+                    )
+
+                    return UiPathChatBedrock(
+                        model=model_name,
+                        settings=client_settings,
+                        **model_kwargs,
+                    )
+            else:
+                from uipath_langchain_client.clients.bedrock.chat_models import (
+                    UiPathChatBedrockConverse,
                 )
 
-                return UiPathChatAnthropic(
+                return UiPathChatBedrockConverse(
                     model=model_name,
-                    client_settings=client_settings,
-                    vendor_type="awsbedrock",
+                    settings=client_settings,
                     **model_kwargs,
                 )
-            else:
-                raise ValueError(f"Invalid model name: {model_name} for vendor: {vendor_type}")
         case _:
-            raise ValueError(f"Invalid UiPath vendor type: {vendor_type}")
+            raise ValueError(
+                f"Invalid vendor type: {vendor_type}, we don't currently have clients that support that api type"
+            )
 
 
 def get_embedding_model(
-    model: str,
+    model_name: str,
     byo_connection_id: str | None = None,
     client_settings: UiPathBaseSettings | None = None,
     client_type: Literal["passthrough", "normalized"] = "passthrough",
@@ -172,11 +227,11 @@ def get_embedding_model(
 
     Example:
         >>> settings = get_default_client_settings()
-        >>> embeddings = get_embedding_model("text-embedding-3-large", settings)
+        >>> embeddings = get_embedding_model(model_name="text-embedding-3-large", client_settings=settings)
         >>> vectors = embeddings.embed_documents(["Hello world"])
     """
     client_settings = client_settings or get_default_client_settings()
-    model_info = _get_model_info(model, client_settings, byo_connection_id)
+    model_info = _get_model_info(model_name, client_settings, byo_connection_id)
 
     if client_type == "normalized":
         from uipath_langchain_client.clients.normalized.embeddings import (
@@ -184,26 +239,45 @@ def get_embedding_model(
         )
 
         return UiPathNormalizedEmbeddings(
-            model=model, client_settings=client_settings, **model_kwargs
+            model=model_name,
+            settings=client_settings,
+            byo_connection_id=byo_connection_id,
+            **model_kwargs,
         )
 
     vendor_type = model_info["vendor"].lower()
+    is_uipath_owned = model_info.get("modelSubscriptionType") == "UiPathOwned"
     match vendor_type:
         case "openai":
-            from uipath_langchain_client.clients.openai.embeddings import (
-                UiPathAzureOpenAIEmbeddings,
-            )
+            if is_uipath_owned:
+                from uipath_langchain_client.clients.openai.embeddings import (
+                    UiPathAzureOpenAIEmbeddings,
+                )
 
-            return UiPathAzureOpenAIEmbeddings(
-                model=model, client_settings=client_settings, **model_kwargs
-            )
+                return UiPathAzureOpenAIEmbeddings(
+                    model=model_name, settings=client_settings, **model_kwargs
+                )
+            else:
+                from uipath_langchain_client.clients.openai.embeddings import (
+                    UiPathOpenAIEmbeddings,
+                )
+
+                return UiPathOpenAIEmbeddings(
+                    model=model_name,
+                    settings=client_settings,
+                    byo_connection_id=byo_connection_id,
+                    **model_kwargs,
+                )
         case "vertexai":
             from uipath_langchain_client.clients.google.embeddings import (
                 UiPathGoogleGenerativeAIEmbeddings,
             )
 
             return UiPathGoogleGenerativeAIEmbeddings(
-                model=model, client_settings=client_settings, **model_kwargs
+                model=model_name,
+                settings=client_settings,
+                byo_connection_id=byo_connection_id,
+                **model_kwargs,
             )
         case "awsbedrock":
             from uipath_langchain_client.clients.bedrock.embeddings import (
@@ -211,7 +285,12 @@ def get_embedding_model(
             )
 
             return UiPathBedrockEmbeddings(
-                model=model, client_settings=client_settings, **model_kwargs
+                model=model_name,
+                settings=client_settings,
+                byo_connection_id=byo_connection_id,
+                **model_kwargs,
             )
         case _:
-            raise ValueError(f"Invalid UiPath Embeddings provider: {vendor_type}")
+            raise ValueError(
+                f"We don't currently have clients that support this provider: {vendor_type}"
+            )