feat:visual style从配置中移除,改为prompt模板内容

This commit is contained in:
SengokuCola
2026-04-21 18:26:54 +08:00
parent 565aedf9ef
commit 363c0a77b7
8 changed files with 46 additions and 39 deletions

View File

@@ -13,7 +13,7 @@ from src.common.logger import get_logger
from src.common.database.database import get_db_session
from src.common.database.database_model import Images, ImageType
from src.common.data_models.image_data_model import MaiImage
from src.config.config import global_config
from src.prompt.prompt_manager import prompt_manager
from src.services.llm_service import LLMServiceClient
install(extra_lines=3)
@@ -374,7 +374,8 @@ class ImageManager:
logger.info(f"Cleaned mistaken registration state on {fixed_counter} image records")
async def _generate_image_description(self, image_bytes: bytes, image_format: str) -> str:
prompt = global_config.visual.visual_style
prompt_template = prompt_manager.get_prompt("image_description")
prompt = await prompt_manager.render_prompt(prompt_template)
image_base64 = base64.b64encode(image_bytes).decode("utf-8")
generation_result = await vlm.generate_response_for_image(