phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix getting params

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-10 13:54:42 -07:00
parent 72dd21dc47
commit 68ce46c4b8
2 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/bedrock/chat.py
View file

@ -1500,7 +1500,9 @@ class BedrockConverseLLM(BaseAWSLLM):
if isinstance(timeout, float) or isinstance(timeout, int): if isinstance(timeout, float) or isinstance(timeout, int):
timeout = httpx.Timeout(timeout) timeout = httpx.Timeout(timeout)
_params["timeout"] = timeout _params["timeout"] = timeout
client = get_async_httpx_client(_params, llm_provider=litellm.LlmProviders.BEDROCK) # type: ignore client = get_async_httpx_client(
params=_params, llm_provider=litellm.LlmProviders.BEDROCK
)
else: else:
client = client # type: ignore client = client # type: ignore

4
litellm/llms/bedrock/embed/embedding.py
View file

@ -130,7 +130,9 @@ class BedrockEmbedding(BaseAWSLLM):
if isinstance(timeout, float) or isinstance(timeout, int): if isinstance(timeout, float) or isinstance(timeout, int):
timeout = httpx.Timeout(timeout) timeout = httpx.Timeout(timeout)
_params["timeout"] = timeout _params["timeout"] = timeout
client = get_async_httpx_client(_params, llm_provider=litellm.LlmProviders.BEDROCK) # type: ignore client = get_async_httpx_client(
params=_params, llm_provider=litellm.LlmProviders.BEDROCK
)
else: else:
client = client client = client