Fixed model discovery on agenthub (#10)

Author: cosminacho

CHANGELOG.md

@@ -2,6 +2,11 @@
 
 All notable changes to `uipath_llm_client` (core package) will be documented in this file.
 
+## [1.0.6]
+
+### Bug Fix
+- Fixed model discovery on AgentHub Settings.
+
 ## [1.0.5] - 2026-02-03
 
 ### Refactor

src/uipath_llm_client/__version__.py

@@ -1,3 +1,3 @@
 __titile__ = "UiPath LLM Client"
 __description__ = "A Python client for interacting with UiPath's LLM services."
-__version__ = "1.0.5"
+__version__ = "1.0.6"

src/uipath_llm_client/settings/agenthub/settings.py

@@ -48,8 +48,10 @@ class AgentHubBaseSettings(UiPathBaseSettings):
     client_secret: SecretStr | None = Field(default=None, validation_alias="UIPATH_CLIENT_SECRET")
     client_scope: str | None = Field(default=None, validation_alias="UIPATH_CLIENT_SCOPE")
 
+    # AgentHub configuration (used for discovery)
+    agenthub_config: str = Field(default="agentsruntime", validation_alias="UIPATH_AGENTHUB_CONFIG")
+
     # Tracing configuration
-    agenthub_config: str | None = Field(default=None, validation_alias="UIPATH_AGENTHUB_CONFIG")
     process_key: str | None = Field(default=None, validation_alias="UIPATH_PROCESS_KEY")
     job_key: str | None = Field(default=None, validation_alias="UIPATH_JOB_KEY")
 

tests/core/test_base_client.py

@@ -433,13 +433,13 @@ def test_build_base_url_normalized(self, agenthub_env_vars, normalized_api_confi
                 )
                 assert "agenthub_/llm/api/chat/completions" in url
 
-    def test_build_auth_headers_empty_by_default(self, agenthub_env_vars):
-        """Test build_auth_headers returns empty dict by default."""
+    def test_build_auth_headers_has_default_config(self, agenthub_env_vars):
+        """Test build_auth_headers includes default agenthub_config."""
         with patch.dict(os.environ, agenthub_env_vars, clear=True):
             with patch("uipath_llm_client.settings.agenthub.settings.AuthService"):
                 settings = AgentHubSettings()
                 headers = settings.build_auth_headers()
-                assert headers == {}
+                assert headers == {"X-UiPath-AgentHub-Config": "agentsruntime"}
 
     def test_build_auth_headers_with_tracing(self, agenthub_env_vars):
         """Test build_auth_headers includes tracing headers when set."""