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

Merge pull request #3693 from BerriAI/litellm_fix_gemini_responses

Browse source 
[Fix] AI Studio (Gemini API) returns invalid 1 index instead of 0 when "stream": false
This commit is contained in:
Ishaan Jaff 2024-05-16 14:21:55 -07:00 • committed by GitHub
commit 2179598d1d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/gemini.py
View file

@ -260,7 +260,7 @@ def completion(
message_obj = Message(content=item.content.parts[0].text)
else:
message_obj = Message(content=None)
choice_obj = Choices(index=idx + 1, message=message_obj)
choice_obj = Choices(index=idx, message=message_obj)
choices_list.append(choice_obj)
model_response["choices"] = choices_list
except Exception as e:
@ -352,7 +352,7 @@ async def async_completion(
message_obj = Message(content=item.content.parts[0].text)
else:
message_obj = Message(content=None)
choice_obj = Choices(index=idx + 1, message=message_obj)
choice_obj = Choices(index=idx, message=message_obj)
choices_list.append(choice_obj)
model_response["choices"] = choices_list
except Exception as e:

1
litellm/tests/test_completion.py
View file

@ -3246,6 +3246,7 @@ def test_completion_gemini():
response = completion(model=model_name, messages=messages)
# Add any assertions,here to check the response
print(response)
assert response.choices[0]["index"] == 0
except litellm.APIError as e:
pass
except Exception as e: