From 2f3fd3e2f07e94584cba41937d8e57ce977bb1aa Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 11 May 2024 19:42:14 -0700 Subject: [PATCH] fix(anthropic.py): fix linting error --- litellm/llms/anthropic.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/llms/anthropic.py b/litellm/llms/anthropic.py index f3e2e2d700..4a9aabc22a 100644 --- a/litellm/llms/anthropic.py +++ b/litellm/llms/anthropic.py @@ -3,7 +3,7 @@ import json from enum import Enum import requests, copy # type: ignore import time -from typing import Callable, Optional, List +from typing import Callable, Optional, List, Union from litellm.utils import ModelResponse, Usage, map_finish_reason, CustomStreamWrapper import litellm from .prompt_templates.factory import prompt_factory, custom_prompt @@ -154,7 +154,7 @@ class AnthropicChatCompletion(BaseLLM): def process_streaming_response( self, model: str, - response: requests.Response | httpx.Response, + response: Union[requests.Response, httpx.Response], model_response: ModelResponse, stream: bool, logging_obj: litellm.utils.Logging,