tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-13 21:06:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: update test with new kwargs

Browse Source
This commit is contained in:
Krrish Dholakia
2025-06-11 22:19:17 -07:00
parent 4be0ec8e35
commit e46ef9d642
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/local_testing/test_custom_llm.py
+6 -1
View File
@@ -265,8 +265,10 @@ class MyCustomLLM(CustomLLM):
print_verbose: Callable,
logging_obj: Any,
optional_params: dict,
api_key: Optional[str] = None,
api_base: Optional[str] = None,
timeout: Optional[Union[float, httpx.Timeout]] = None,
litellm_params=None,
aembedding=None,
) -> EmbeddingResponse:
model_response.model = model
@@ -289,6 +291,9 @@ class MyCustomLLM(CustomLLM):
print_verbose: Callable,
logging_obj: Any,
optional_params: dict,
api_key: Optional[str] = None,
api_base: Optional[str] = None,
timeout: Optional[Union[float, httpx.Timeout]] = None,
litellm_params=None,
) -> EmbeddingResponse:
model_response.model = model