phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix typing for aresponses

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-11 22:03:48 -07:00
parent 8ada5a469d
commit 20e3398476
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/responses/main.py
View file

@ -15,6 +15,7 @@ from litellm.types.llms.openai import (
ResponseIncludable, ResponseIncludable,
ResponseInputParam, ResponseInputParam,
ResponsesAPIOptionalRequestParams, ResponsesAPIOptionalRequestParams,
ResponsesAPIResponse,
ResponseTextConfigParam, ResponseTextConfigParam,
ToolChoice, ToolChoice,
ToolParam, ToolParam,
@ -22,6 +23,8 @@ from litellm.types.llms.openai import (
from litellm.types.router import GenericLiteLLMParams from litellm.types.router import GenericLiteLLMParams
from litellm.utils import ProviderConfigManager, client from litellm.utils import ProviderConfigManager, client
from .streaming_iterator import ResponsesAPIStreamingIterator
####### ENVIRONMENT VARIABLES ################### ####### ENVIRONMENT VARIABLES ###################
# Initialize any necessary instances or variables here # Initialize any necessary instances or variables here
base_llm_http_handler = BaseLLMHTTPHandler() base_llm_http_handler = BaseLLMHTTPHandler()
@ -72,7 +75,7 @@ async def aresponses(
extra_body: Optional[Dict[str, Any]] = None, extra_body: Optional[Dict[str, Any]] = None,
timeout: Optional[Union[float, httpx.Timeout]] = None, timeout: Optional[Union[float, httpx.Timeout]] = None,
**kwargs, **kwargs,
): ) -> Union[ResponsesAPIResponse, ResponsesAPIStreamingIterator]:
litellm_logging_obj: LiteLLMLoggingObj = kwargs.get("litellm_logging_obj") # type: ignore litellm_logging_obj: LiteLLMLoggingObj = kwargs.get("litellm_logging_obj") # type: ignore
litellm_call_id: Optional[str] = kwargs.get("litellm_call_id", None) litellm_call_id: Optional[str] = kwargs.get("litellm_call_id", None)
@ -131,8 +134,6 @@ async def aresponses(
) )
return response return response
pass
def responses( def responses(
input: Union[str, ResponseInputParam], input: Union[str, ResponseInputParam],