fix tests

This commit is contained in:
geekan 2023-12-28 18:06:02 +08:00
parent 637f04dd2a
commit 4e32ee120c
5 changed files with 24 additions and 24 deletions

View file

@ -58,7 +58,7 @@ class GeminiGPTAPI(BaseLLM):
genai.configure(api_key=config.gemini_api_key)
def _user_msg(self, msg: str) -> dict[str, str]:
# Not to change BaseGPTAPI default functions but update with Gemini's conversation format.
# Not to change BaseLLM default functions but update with Gemini's conversation format.
# You should follow the format.
return {"role": "user", "parts": [msg]}