Merge pull request #411 from LlmKira/dev

Fix support for gpt-4o

app/sender/telegram/__init__.py

@@ -245,7 +245,11 @@ async def listen_login_command(message: types.Message):
             reply = await login(
                 uid=uid_make(__sender__, message.from_user.id), arg_string=_arg
             )
-            await bot.reply_to(message, text=reply)
+            await bot.reply_to(
+                message,
+                text=reply,
+                parse_mode="MarkdownV2",
+            )
 
         @bot.message_handler(commands="env", chat_types=["private"])
         async def listen_env_command(message: types.Message):

app/sender/util_func.py

@@ -169,9 +169,10 @@ async def login(uid: str, arg_string) -> str:
     """
     error = telegramify_markdown.convert(
         "🔑 **Incorrect format.**\n"
-        "You can set it via `https://<something api.openai.com>/v1$<api key>"
+        "You can set it via `/login https://<something api.openai.com>/v1$<api key>"
         "$<model such as gpt-4-turbo>$<tool_model such as gpt-3.5-turbo>` format, "
-        "or you can log in via URL using `token$https://provider.com`."
+        "or you can log in via URL using `/login token$https://provider.com`.\n"
+        "Use $ to separate the parameters."
     )
     settings = split_setting_string(arg_string)
     if not settings:

llmkira/openai/request.py

@@ -21,6 +21,8 @@
     SystemMessage,
 )
 
+VISION = ("gpt-4-vision", "gpt-4-turbo", "claude-3", "gpt-4o")
+
 
 class OpenAICredential(BaseModel):
     api_key: SecretStr
@@ -164,7 +166,7 @@ def make_url(base_url: str):
 
     @model_validator(mode="after")
     def check_vision(self):
-        if not self.model.startswith(("gpt-4-vision", "gpt-4-turbo", "claude-3")):
+        if not self.model.startswith(VISION):
             logger.info(
                 "Try to remove the image content part from the messages, because the model is not supported."
             )