From b66c9ac1369f7f50b610b10e11022f2c95eb73d3 Mon Sep 17 00:00:00 2001 From: F4ria Date: Wed, 10 Jul 2024 20:24:16 +0800 Subject: [PATCH] fix: update image_to_data_uri to use dynamic content type --- handlers/__init__.py | 3 ++- handlers/useful.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/handlers/__init__.py b/handlers/__init__.py index d8def7b..b2f9da6 100644 --- a/handlers/__init__.py +++ b/handlers/__init__.py @@ -220,9 +220,10 @@ def enrich_text_with_urls(text: str) -> str: def image_to_data_uri(file_path): + content_type = guess_type(file_path)[0] with open(file_path, "rb") as image_file: encoded_image = base64.b64encode(image_file.read()).decode("utf-8") - return f"data:image/png;base64,{encoded_image}" + return f"data:{content_type};base64,{encoded_image}" import json diff --git a/handlers/useful.py b/handlers/useful.py index 17d9228..239dd2e 100644 --- a/handlers/useful.py +++ b/handlers/useful.py @@ -389,7 +389,7 @@ def answer_it_handler(message: Message, bot: TeleBot) -> None: #### Complete Messages #### if CHATGPT_COMPLETE and CHATGPT_API_KEY: full_answer += complete_chatgpt_future.result() - if CLADUE_COMPLETE and ANTHROPIC_API_KEY and not local_image_path: + if CLADUE_COMPLETE and ANTHROPIC_API_KEY: full_answer += complete_claude_future.result() if COHERE_COMPLETE and COHERE_API_KEY and not local_image_path: full_answer += complete_cohere_future.result()