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

fix(openai.py): fix timeout errors

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-15 18:03:07 -08:00
parent b397719709
commit 081400ecd4
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/openai.py
View file

@ -484,7 +484,7 @@ class OpenAITextCompletion(BaseLLM):
elif optional_params.get("stream", False):
return self.streaming(logging_obj=logging_obj, api_base=api_base, data=data, headers=headers, model_response=model_response, model=model)
else:
response = self._client_session.post(
response = httpx.post(
url=f"{api_base}",
json=data,
headers=headers,
@ -546,7 +546,7 @@ class OpenAITextCompletion(BaseLLM):
model_response: ModelResponse,
model: str
):
with self._client_session.stream(
with httpx.stream(
url=f"{api_base}",
json=data,
headers=headers,