scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-24 05:45:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 72

fix(v1m): remove Any annotations and add explicit init return type for strict ruff

Browse source
This commit is contained in:
Rick Sanchez 2026-10-13 09:12:11 +03:30
parent b076d11060
commit e0875bb74c
1 changed files with 7 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
litellm/llms/v1m.py
View file

@ -1,13 +1,12 @@
import json
import time
from typing import Any
from litellm.llms.base_llm.chat.transformation import BaseLLMException
from litellm.llms.openai.chat.gpt_transformation import OpenAIGPTConfig
class V1MError(BaseLLMException):
def __init__(self, status_code: int, message: str, headers: Any = None):
def __init__(self, status_code: int, message: str, headers: object = None) -> None:
super().__init__(status_code=status_code, message=message, headers=headers)
@ -29,11 +28,11 @@ class V1MConfig(OpenAIGPTConfig):
def transform_request(
self,
model: str,
messages: list[dict[str, Any]] | None = None,
state: str | dict[str, Any] | None = None,
questions: dict[str, Any] | None = None,
**kwargs: Any,
) -> dict[str, Any]:
messages: list[dict] | None = None,
state: str | dict | None = None,
questions: dict | None = None,
**kwargs,
) -> dict:
clean_model = model.replace("v1m/", "") if model.startswith("v1m/") else model
if state is not None and questions is not None:
@ -70,7 +69,7 @@ class V1MConfig(OpenAIGPTConfig):
},
}
def transform_response(self, raw_resp: dict[str, Any], model: str) -> dict[str, Any]:
def transform_response(self, raw_resp: dict, model: str) -> dict:
evals = raw_resp.get("evaluations", raw_resp)
return {
"id": f"v1m-{int(time.time())}",