diff --git a/portkey_ai/api_resources/utils.py b/portkey_ai/api_resources/utils.py index 860952e4..cb3ead6c 100644 --- a/portkey_ai/api_resources/utils.py +++ b/portkey_ai/api_resources/utils.py @@ -15,6 +15,7 @@ TextCompletionChunk, TextCompletion, ) +from portkey_ai.api_resources.types.feedback_type import FeedbackResponse from portkey_ai.api_resources.types.generation_type import ( PromptCompletion, PromptCompletionChunk, @@ -61,7 +62,7 @@ class CacheType(str, Enum, metaclass=MetaEnum): ResponseT = TypeVar( "ResponseT", - bound="Union[ChatCompletionChunk, ChatCompletions, TextCompletion, TextCompletionChunk, GenericResponse, PromptCompletion, PromptCompletionChunk, PromptRender, httpx.Response]", # noqa: E501 + bound="Union[ChatCompletionChunk, ChatCompletions, TextCompletion, TextCompletionChunk, GenericResponse, PromptCompletion, PromptCompletionChunk, PromptRender, FeedbackResponse, httpx.Response]", # noqa: E501 )