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

refactor: update method signature

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-12 15:23:38 -07:00
parent 738c0b873d
commit 88e9edf7db
23 changed files with 35 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/llms/replicate/chat/handler.py
View file

@ -169,7 +169,10 @@ def completion(
) # for pricing this must remain right before calling api
prediction_url = replicate_config.get_complete_url(
api_base=api_base, model=model, optional_params=optional_params
api_base=api_base,
model=model,
optional_params=optional_params,
litellm_params=litellm_params,
)
## COMPLETION CALL
@ -243,7 +246,10 @@ async def async_completion(
) -> Union[ModelResponse, CustomStreamWrapper]:
prediction_url = replicate_config.get_complete_url(
api_base=api_base, model=model, optional_params=optional_params
api_base=api_base,
model=model,
optional_params=optional_params,
litellm_params=litellm_params,
)
async_handler = get_async_httpx_client(
llm_provider=litellm.LlmProviders.REPLICATE,