diff --git a/litellm/llms/base_llm/chat/transformation.py b/litellm/llms/base_llm/chat/transformation.py index 7d6109c84bc..2e04c25cf24 100644 --- a/litellm/llms/base_llm/chat/transformation.py +++ b/litellm/llms/base_llm/chat/transformation.py @@ -1,3 +1,4 @@ +from collections.abc import Mapping """ Common base config for all LLM providers """ diff --git a/litellm/main.py b/litellm/main.py index 4f328438245..d7dd02e4e90 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -1,3 +1,4 @@ +from pydantic import TypeAdapter # LiteLLM main module: public completion, embedding, streaming, and moderation entrypoints. # # +-----------------------------------------------+ diff --git a/litellm/types/completion.py b/litellm/types/completion.py index ba5c0a2721d..18a1642633c 100644 --- a/litellm/types/completion.py +++ b/litellm/types/completion.py @@ -1,4 +1,5 @@ from __future__ import annotations +from collections.abc import Mapping from collections.abc import Callable, Coroutine, Iterable from dataclasses import dataclass diff --git a/tests/integration/_support/client.py b/tests/integration/_support/client.py index e07cbe6b2a3..d9210eb1351 100644 --- a/tests/integration/_support/client.py +++ b/tests/integration/_support/client.py @@ -3,7 +3,7 @@ from __future__ import annotations import os import time import uuid -from collections.abc import Callable, Iterator, Mapping +from collections.abc import Sequence, Callable, Iterator, Mapping from contextlib import ExitStack, contextmanager from dataclasses import dataclass from hashlib import sha256 @@ -22,6 +22,11 @@ def object_value(value: JsonValue) -> dict[str, JsonValue]: return JSON_OBJECT.validate_python(value) +def list_value(value: JsonValue) -> Sequence[JsonValue]: + assert isinstance(value, list), f"Expected a list, received {type(value).__name__}" + return value + + def string_value(value: JsonValue) -> str: assert isinstance(value, str), f"Expected a string, received {type(value).__name__}" return value diff --git a/tests/unit/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py b/tests/unit/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py index 403884ee2a3..3d4a637fc26 100644 --- a/tests/unit/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py +++ b/tests/unit/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py @@ -1,3 +1,4 @@ +from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler import asyncio import base64 import copy