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

fix - choices index for gemini/ provider

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-16 13:52:46 -07:00
parent e9358684fb
commit a2ef089667
2 changed files with 3 additions and 2 deletions
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: