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

fix mypy linting errors

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-12 12:13:19 -07:00
parent d6ccaf1e63
commit 61f4080903
3 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/responses/streaming_iterator.py
View file

@ -36,7 +36,7 @@ class BaseResponsesAPIStreamingIterator:
self.logging_obj = logging_obj self.logging_obj = logging_obj
self.finished = False self.finished = False
self.responses_api_provider_config = responses_api_provider_config self.responses_api_provider_config = responses_api_provider_config
self.completed_response = None self.completed_response: Optional[ResponsesAPIStreamingResponse] = None
self.start_time = datetime.now() self.start_time = datetime.now()
def _process_chunk(self, chunk): def _process_chunk(self, chunk):
@ -102,7 +102,6 @@ class ResponsesAPIStreamingIterator(BaseResponsesAPIStreamingIterator):
): ):
super().__init__(response, model, responses_api_provider_config, logging_obj) super().__init__(response, model, responses_api_provider_config, logging_obj)
self.stream_iterator = response.aiter_lines() self.stream_iterator = response.aiter_lines()
self.completed_response: Optional[ResponsesAPIStreamingResponse] = None
def __aiter__(self): def __aiter__(self):
return self return self

4
litellm/responses/utils.py
View file

@ -1,4 +1,4 @@
from typing import Any, Dict, get_type_hints from typing import Any, Dict, cast, get_type_hints
import litellm import litellm
from litellm.llms.base_llm.responses.transformation import BaseResponsesAPIConfig from litellm.llms.base_llm.responses.transformation import BaseResponsesAPIConfig
@ -73,7 +73,7 @@ class ResponsesAPIRequestUtils:
""" """
valid_keys = get_type_hints(ResponsesAPIOptionalRequestParams).keys() valid_keys = get_type_hints(ResponsesAPIOptionalRequestParams).keys()
filtered_params = {k: v for k, v in params.items() if k in valid_keys} filtered_params = {k: v for k, v in params.items() if k in valid_keys}
return ResponsesAPIOptionalRequestParams(**filtered_params) return cast(ResponsesAPIOptionalRequestParams, filtered_params)
class ResponseAPILoggingUtils: class ResponseAPILoggingUtils:

1
litellm/types/llms/openai.py
View file

@ -35,7 +35,6 @@ from openai.types.embedding import Embedding as OpenAIEmbedding
from openai.types.fine_tuning.fine_tuning_job import FineTuningJob from openai.types.fine_tuning.fine_tuning_job import FineTuningJob
from openai.types.responses.response import ( from openai.types.responses.response import (
IncompleteDetails, IncompleteDetails,
Reasoning,
Response, Response,
ResponseOutputItem, ResponseOutputItem,
ResponseTextConfig, ResponseTextConfig,