From ce00a7d7abf45ebb11ed8f8901800562ed390fc9 Mon Sep 17 00:00:00 2001 From: openhands Date: Tue, 19 May 2026 19:12:48 +0000 Subject: [PATCH] Fix CI after staging rebase Relax the Ollama runtime metadata return annotation to match the provider-hook dict response and update the Google Interactions OpenAPI status expectation for the current live spec. Co-authored-by: openhands --- litellm/llms/ollama/completion/transformation.py | 5 +---- tests/test_litellm/interactions/test_openapi_compliance.py | 1 + 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/litellm/llms/ollama/completion/transformation.py b/litellm/llms/ollama/completion/transformation.py index bb88c60ae64..d3d3304e607 100644 --- a/litellm/llms/ollama/completion/transformation.py +++ b/litellm/llms/ollama/completion/transformation.py @@ -21,7 +21,6 @@ from litellm.types.llms.openai import AllMessageValues, ChatCompletionUsageBlock from litellm.types.utils import ( Delta, GenericStreamingChunk, - ModelInfoBase, ModelResponse, ModelResponseStream, ProviderField, @@ -222,9 +221,7 @@ class OllamaConfig(BaseConfig): or get_secret_str("OLLAMA_API_KEY") ) - def get_model_info( - self, model: str, api_base: Optional[str] = None - ) -> ModelInfoBase: + def get_model_info(self, model: str, api_base: Optional[str] = None) -> Any: """ curl http://localhost:11434/api/show -d '{ "name": "mistral" diff --git a/tests/test_litellm/interactions/test_openapi_compliance.py b/tests/test_litellm/interactions/test_openapi_compliance.py index 1d3b6b8ae1e..9f213c9e498 100644 --- a/tests/test_litellm/interactions/test_openapi_compliance.py +++ b/tests/test_litellm/interactions/test_openapi_compliance.py @@ -187,6 +187,7 @@ class TestResponseCompliance: "failed", "cancelled", "incomplete", + "budget_exceeded", ] assert status_prop["enum"] == expected_statuses print(f"✓ Status enum values: {expected_statuses}")