mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-05 02:41:56 +00:00
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 <openhands@all-hands.dev>
This commit is contained in:
parent
db2ce18d5c
commit
ce00a7d7ab
2 changed files with 2 additions and 4 deletions
|
|
@ -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"
|
||||
|
|
|
|||
|
|
@ -187,6 +187,7 @@ class TestResponseCompliance:
|
|||
"failed",
|
||||
"cancelled",
|
||||
"incomplete",
|
||||
"budget_exceeded",
|
||||
]
|
||||
assert status_prop["enum"] == expected_statuses
|
||||
print(f"✓ Status enum values: {expected_statuses}")
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue