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