scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-05 08:07:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 136

ran black

Browse source
This commit is contained in:
Sameer Kankute 2026-04-02 19:24:51 +05:30
parent 9fe534aff2
commit f7c24ae395
No known key found for this signature in database
2 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/llms/bedrock/chat/invoke_transformations/base_invoke_transformation.py
View file

@ -2,8 +2,8 @@ import copy
import json
import time
from functools import partial
from urllib.parse import quote
from typing import TYPE_CHECKING, Any, List, Optional, Tuple, Union, cast, get_args
from urllib.parse import quote
import httpx
@ -186,9 +186,9 @@ class AmazonInvokeConfig(BaseConfig, BaseAWSLLM):
config = litellm.AmazonCohereConfig.get_config()
self._apply_config_to_params(config, inference_params)
if stream is True:
inference_params[
"stream"
] = True # cohere requires stream = True in inference params
inference_params["stream"] = (
True # cohere requires stream = True in inference params
)
request_data = {"prompt": prompt, **inference_params}
elif provider == "anthropic":
transformed_request = (

8
litellm/llms/bedrock/common_utils.py
View file

@ -1109,9 +1109,11 @@ class CommonBatchFilesUtils:
return (
dict(prepped.headers),
request_data.encode("utf-8")
if isinstance(request_data, str)
else request_data,
(
request_data.encode("utf-8")
if isinstance(request_data, str)
else request_data
),
)
def generate_unique_job_name(self, model: str, prefix: str = "litellm") -> str: