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}")