Feat: add BYOM API flavor constants and resolve them in factories

BYOM models from the discovery endpoint return new PascalCase API flavors
(e.g. OpenAiChatCompletions, GeminiGenerateContent, AwsBedrockConverse).
Add these as constants and automatically resolve them to the correct
routing-level flavors and vendor types so clients are instantiated correctly.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: cosminacho

CHANGELOG.md

@@ -2,6 +2,14 @@
 
 All notable changes to `uipath_llm_client` (core package) will be documented in this file.
 
+## [1.9.0] - 2026-04-16
+
+### Added
+- BYOM API flavor constants for discovery endpoint: `OpenAiChatCompletions`, `OpenAiResponses`, `OpenAiEmbeddings`, `GeminiGenerateContent`, `GeminiEmbeddings`, `AwsBedrockInvoke`, `AwsBedrockConverse`
+- `BYOM_TO_ROUTING_FLAVOR` mapping to resolve BYOM discovery flavors to routing-level API flavors
+- Extended `API_FLAVOR_TO_VENDOR_TYPE` with BYOM flavor entries for automatic vendor resolution
+- LiteLLM client now resolves BYOM discovery flavors to correct routing flavors and litellm providers
+
 ## [1.8.2] - 2026-04-13
 
 ### Fixed

packages/uipath_langchain_client/CHANGELOG.md

@@ -2,6 +2,11 @@
 
 All notable changes to `uipath_langchain_client` will be documented in this file.
 
+## [1.9.0] - 2026-04-16
+
+### Added
+- Factory functions (`get_chat_model`, `get_embedding_model`) now automatically resolve BYOM discovery API flavors to the correct client and routing flavor
+
 ## [1.8.2] - 2026-04-13
 
 ### Changed

packages/uipath_langchain_client/pyproject.toml

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

packages/uipath_langchain_client/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.8.2"
+__version__ = "1.9.0"

packages/uipath_langchain_client/src/uipath_langchain_client/factory.py

@@ -28,6 +28,7 @@
 )
 from uipath_langchain_client.settings import (
     API_FLAVOR_TO_VENDOR_TYPE,
+    BYOM_TO_ROUTING_FLAVOR,
     ApiFlavor,
     RoutingMode,
     UiPathBaseSettings,
@@ -160,6 +161,10 @@ def get_chat_model(
         raise ValueError("No vendor type or api flavor found in model info")
     discovered_vendor_type = discovered_vendor_type.lower()
 
+    # For BYOM models, derive routing api_flavor from the discovered BYOM flavor
+    if api_flavor is None and discovered_api_flavor is not None:
+        api_flavor = BYOM_TO_ROUTING_FLAVOR.get(discovered_api_flavor)
+
     match discovered_vendor_type:
         case VendorType.OPENAI:
             if api_flavor == ApiFlavor.RESPONSES:

packages/uipath_langchain_client/src/uipath_langchain_client/settings.py

@@ -24,6 +24,7 @@
 )
 from uipath.llm_client.settings.constants import (
     API_FLAVOR_TO_VENDOR_TYPE,
+    BYOM_TO_ROUTING_FLAVOR,
     ApiFlavor,
     ApiType,
     RoutingMode,
@@ -41,4 +42,5 @@
     "ApiFlavor",
     "VendorType",
     "API_FLAVOR_TO_VENDOR_TYPE",
+    "BYOM_TO_ROUTING_FLAVOR",
 ]

src/uipath/llm_client/__version__.py

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

src/uipath/llm_client/clients/litellm/client.py

@@ -27,6 +27,7 @@
 from uipath.llm_client.settings.base import UiPathAPIConfig
 from uipath.llm_client.settings.constants import (
     API_FLAVOR_TO_VENDOR_TYPE,
+    BYOM_TO_ROUTING_FLAVOR,
     ApiFlavor,
     ApiType,
     RoutingMode,
@@ -87,6 +88,14 @@
     "converse": "bedrock",
     "invoke": "bedrock",
     "anthropic-claude": "vertex_ai",
+    # BYOM discovery flavors (mapped via routing flavor resolution)
+    "OpenAiChatCompletions": "openai",
+    "OpenAiResponses": "openai",
+    "OpenAiEmbeddings": "openai",
+    "GeminiGenerateContent": "gemini",
+    "GeminiEmbeddings": "gemini",
+    "AwsBedrockInvoke": "bedrock",
+    "AwsBedrockConverse": "bedrock",
 }
 
 _ANTHROPIC_FAMILY = "anthropicclaude"
@@ -212,6 +221,10 @@ def _discover_and_build_api_config(
         resolved_vendor = str(vendor_type or discovered_vendor).lower()
         resolved_flavor = str(api_flavor) if api_flavor is not None else discovered_flavor
 
+        # Resolve BYOM discovery flavors to routing-level flavors
+        if resolved_flavor is not None:
+            resolved_flavor = BYOM_TO_ROUTING_FLAVOR.get(resolved_flavor, resolved_flavor)
+
         # OpenAI defaults to chat-completions when no flavor is discovered
         if resolved_flavor is None and resolved_vendor in ("openai", "azure"):
             resolved_flavor = ApiFlavor.CHAT_COMPLETIONS

src/uipath/llm_client/settings/constants.py

@@ -20,19 +20,47 @@ class VendorType(StrEnum):
 
 
 class ApiFlavor(StrEnum):
+    # Routing-level flavors (used in X-UiPath-LlmGateway-ApiFlavor header)
     CHAT_COMPLETIONS = "chat-completions"
     RESPONSES = "responses"
     GENERATE_CONTENT = "generate-content"
     CONVERSE = "converse"
     INVOKE = "invoke"
     ANTHROPIC_CLAUDE = "anthropic-claude"
 
+    # BYOM discovery flavors (returned by discovery endpoint for BYOM models)
+    OPENAI_CHAT_COMPLETIONS = "OpenAiChatCompletions"
+    OPENAI_RESPONSES = "OpenAiResponses"
+    OPENAI_EMBEDDINGS = "OpenAiEmbeddings"
+    GEMINI_GENERATE_CONTENT = "GeminiGenerateContent"
+    GEMINI_EMBEDDINGS = "GeminiEmbeddings"
+    AWS_BEDROCK_INVOKE = "AwsBedrockInvoke"
+    AWS_BEDROCK_CONVERSE = "AwsBedrockConverse"
+
 
 API_FLAVOR_TO_VENDOR_TYPE: dict[ApiFlavor, VendorType] = {
+    # Routing flavors
     ApiFlavor.CHAT_COMPLETIONS: VendorType.OPENAI,
     ApiFlavor.RESPONSES: VendorType.OPENAI,
     ApiFlavor.GENERATE_CONTENT: VendorType.VERTEXAI,
     ApiFlavor.ANTHROPIC_CLAUDE: VendorType.VERTEXAI,
     ApiFlavor.CONVERSE: VendorType.AWSBEDROCK,
     ApiFlavor.INVOKE: VendorType.AWSBEDROCK,
+    # BYOM discovery flavors
+    ApiFlavor.OPENAI_CHAT_COMPLETIONS: VendorType.OPENAI,
+    ApiFlavor.OPENAI_RESPONSES: VendorType.OPENAI,
+    ApiFlavor.OPENAI_EMBEDDINGS: VendorType.OPENAI,
+    ApiFlavor.GEMINI_GENERATE_CONTENT: VendorType.VERTEXAI,
+    ApiFlavor.GEMINI_EMBEDDINGS: VendorType.VERTEXAI,
+    ApiFlavor.AWS_BEDROCK_INVOKE: VendorType.AWSBEDROCK,
+    ApiFlavor.AWS_BEDROCK_CONVERSE: VendorType.AWSBEDROCK,
+}
+
+
+BYOM_TO_ROUTING_FLAVOR: dict[str, ApiFlavor] = {
+    ApiFlavor.OPENAI_CHAT_COMPLETIONS: ApiFlavor.CHAT_COMPLETIONS,
+    ApiFlavor.OPENAI_RESPONSES: ApiFlavor.RESPONSES,
+    ApiFlavor.GEMINI_GENERATE_CONTENT: ApiFlavor.GENERATE_CONTENT,
+    ApiFlavor.AWS_BEDROCK_INVOKE: ApiFlavor.INVOKE,
+    ApiFlavor.AWS_BEDROCK_CONVERSE: ApiFlavor.CONVERSE,
 }