diff --git a/litellm/llms/base_llm/chat/transformation.py b/litellm/llms/base_llm/chat/transformation.py index de5fc0716a9..b649483991b 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 42eb193e531..76be0f42095 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/test_litellm/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py b/tests/test_litellm/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py index 403884ee2a3..3d4a637fc26 100644 --- a/tests/test_litellm/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py +++ b/tests/test_litellm/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