Skip to content

Commit

Permalink
[easy] Set Gemini model parser id to the model id being used
Browse files Browse the repository at this point in the history
title
  • Loading branch information
Ankush Pala ankush@lastmileai.dev committed Feb 5, 2024
1 parent 2a6c66e commit 7786720
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/src/aiconfig/default_parsers/gemini.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,7 +116,7 @@ def id(self) -> str:
"""
Returns an identifier for the model (e.g. llama-2, gpt-4, etc.).
"""
return "GeminiModelParser"
return self.model_id

async def serialize(
self,
Expand Down

0 comments on commit 7786720

Please sign in to comment.