LCORE-974: proper types checked by Pydantic

Author: tisnik

tests/unit/models/config/test_tls_configuration.py

@@ -36,6 +36,7 @@ def test_tls_configuration_in_service_configuration() -> None:
         access_log=True,
         workers=10,
         auth_enabled=True,
+        root_path="/",
     )
     assert cfg is not None
     assert cfg.tls_config is not None

tests/unit/models/responses/test_successful_responses.py

@@ -3,7 +3,7 @@
 """Unit tests for all successful response models."""
 
 import pytest
-from pydantic import AnyUrl, ValidationError, ConfigDict
+from pydantic import AnyUrl, AnyHttpUrl, ValidationError, ConfigDict
 from pydantic_core import SchemaError
 
 from models.config import (
@@ -955,9 +955,10 @@ def test_constructor(self) -> None:
                 workers=10,
                 color_log=True,
                 access_log=True,
+                root_path="/",
             ),
             llama_stack=LlamaStackConfiguration(
-                url="http://localhost:8321",
+                url=AnyHttpUrl("http://localhost:8321"),
                 use_as_library_client=False,
                 api_key=None,
                 library_client_config_path=None,

tests/unit/test_client.py

@@ -4,6 +4,8 @@
 
 import json
 
+from pydantic import AnyHttpUrl
+
 import pytest
 from client import AsyncLlamaStackClientHolder
 from models.config import LlamaStackConfiguration
@@ -54,7 +56,7 @@ async def test_get_async_llama_stack_library_client() -> None:
 async def test_get_async_llama_stack_remote_client() -> None:
     """Test the initialization of asynchronous Llama Stack client in server mode."""
     cfg = LlamaStackConfiguration(
-        url="http://localhost:8321",
+        url=AnyHttpUrl("http://localhost:8321"),
         api_key=None,
         use_as_library_client=False,
         library_client_config_path="./tests/configuration/minimal-stack.yaml",
@@ -90,7 +92,7 @@ async def test_get_async_llama_stack_wrong_configuration() -> None:
 async def test_update_provider_data_service_client() -> None:
     """Test that update_provider_data updates headers for service clients."""
     cfg = LlamaStackConfiguration(
-        url="http://localhost:8321",
+        url=AnyHttpUrl("http://localhost:8321"),
         api_key=None,
         use_as_library_client=False,
         library_client_config_path=None,