From 13142267be1e70af57e71ae49507426aeca4be3f Mon Sep 17 00:00:00 2001 From: Wojciech-Rebisz Date: Tue, 4 Nov 2025 10:56:56 +0100 Subject: [PATCH] Fix model_limits KeyError --- .../providers/inline/agents/meta_reference/agent_instance.py | 2 +- src/llama_stack/providers/remote/inference/watsonx/watsonx.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/llama_stack/providers/inline/agents/meta_reference/agent_instance.py b/src/llama_stack/providers/inline/agents/meta_reference/agent_instance.py index 46619087c..8836a6d2d 100644 --- a/src/llama_stack/providers/inline/agents/meta_reference/agent_instance.py +++ b/src/llama_stack/providers/inline/agents/meta_reference/agent_instance.py @@ -1016,7 +1016,7 @@ async def load_data_from_url(url: str) -> str: async with httpx.AsyncClient() as client: r = await client.get(url) resp = r.text - return resp + return str(resp) raise ValueError(f"Unexpected URL: {type(url)}") diff --git a/src/llama_stack/providers/remote/inference/watsonx/watsonx.py b/src/llama_stack/providers/remote/inference/watsonx/watsonx.py index b31f1f5e8..b4dfc1f06 100644 --- a/src/llama_stack/providers/remote/inference/watsonx/watsonx.py +++ b/src/llama_stack/providers/remote/inference/watsonx/watsonx.py @@ -283,8 +283,8 @@ class WatsonXInferenceAdapter(LiteLLMOpenAIMixin): # ... provider_resource_id = f"{self.__provider_id__}/{model_spec['model_id']}" if "embedding" in functions: - embedding_dimension = model_spec["model_limits"]["embedding_dimension"] - context_length = model_spec["model_limits"]["max_sequence_length"] + embedding_dimension = model_spec.get("model_limits", {}).get("embedding_dimension", 0) + context_length = model_spec.get("model_limits", {}).get("max_sequence_length", 0) embedding_metadata = { "embedding_dimension": embedding_dimension, "context_length": context_length,