final fixes

Author: cosminacho

packages/uipath_langchain_client/src/uipath_langchain_client/factory.py

@@ -87,9 +87,10 @@ def get_chat_model(
         ValueError: If the model is not found in available models or vendor is not supported
     """
     client_settings = client_settings or get_default_client_settings()
+
     model_info = _get_model_info(model_name, client_settings, byo_connection_id)
-    is_uipath_owned = model_info.get("modelSubscriptionType") == "UiPathOwned"
 
+    is_uipath_owned = model_info.get("modelSubscriptionType") == "UiPathOwned"
     if not is_uipath_owned:
         client_settings.validate_byo_model(model_info)
 

src/uipath_llm_client/settings/agenthub/settings.py

@@ -139,4 +139,4 @@ def get_available_models(self) -> list[dict[str, Any]]:
 
     @override
     def validate_byo_model(self, model_info: dict[str, Any]) -> None:
-        raise ValueError("BYOM models are not supported for AgentHub")
+        return

src/uipath_llm_client/settings/base.py

@@ -143,6 +143,7 @@ def get_available_models(
         """
         ...
 
+    @abstractmethod
     def validate_byo_model(self, model_info: dict[str, Any]) -> None:
         """Validate that the model is a BYOM model."""
-        raise NotImplementedError("For BYOM models you must implement this method")
+        ...