From 1409bb30dfd08437ae3686a4a4d9e4545691f43a Mon Sep 17 00:00:00 2001 From: Stephen Hu Date: Wed, 6 Aug 2025 11:42:25 +0800 Subject: [PATCH] Refactor:Improve the logic so that it does not decode base 64 for the test image each time (#9264) ### What problem does this PR solve? Improve the logic so that it does not decode base 64 for the test image each time ### Type of change - [x] Refactoring - [x] Performance Improvement --------- Co-authored-by: Kevin Hu --- api/apps/llm_app.py | 5 ++--- api/utils/base64_image.py | 2 ++ 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/api/apps/llm_app.py b/api/apps/llm_app.py index 8214bead0..9f44fe290 100644 --- a/api/apps/llm_app.py +++ b/api/apps/llm_app.py @@ -15,7 +15,6 @@ # import logging import json -import base64 from flask import request from flask_login import login_required, current_user from api.db.services.llm_service import LLMFactoriesService, TenantLLMService, LLMService @@ -24,7 +23,7 @@ from api.utils.api_utils import server_error_response, get_data_error_result, va from api.db import StatusEnum, LLMType from api.db.db_models import TenantLLM from api.utils.api_utils import get_json_result -from api.utils.base64_image import test_image_base64 +from api.utils.base64_image import test_image from rag.llm import EmbeddingModel, ChatModel, RerankModel, CvModel, TTSModel @@ -256,7 +255,7 @@ def add_llm(): base_url=llm["api_base"] ) try: - image_data = base64.b64decode(test_image_base64) + image_data = test_image m, tc = mdl.describe(image_data) if not m and not tc: raise Exception(m) diff --git a/api/utils/base64_image.py b/api/utils/base64_image.py index b602b8087..d9dd4cde9 100644 --- a/api/utils/base64_image.py +++ b/api/utils/base64_image.py @@ -1 +1,3 @@ +import base64 test_image_base64 = "iVBORw0KGgoAAAANSUhEUgAAAGQAAABkCAIAAAD/gAIDAAAA6ElEQVR4nO3QwQ3AIBDAsIP9d25XIC+EZE8QZc18w5l9O+AlZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBWYFZgVmBT+IYAHHLHkdEgAAAABJRU5ErkJggg==" +test_image = base64.b64decode(test_image_base64) \ No newline at end of file