yunqueai/baijiahao_text_to_image
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Gemini生图时自动补充文章标签到ai_image_tags表

Browse Source 
- 新增default_tag_id和default_tag_name字段，与tag_id和tag_name保持一致
- 设置status字段为'generate'，确保生成的图片可被后续匹配使用
- 确保标签数据的一致性和完整性
This commit is contained in:
shengyudong
2026-02-05 20:59:58 +08:00
parent 97dcff8c8b
commit 2830908dae
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
article_auto_image_matching.py
View File

@@ -994,18 +994,20 @@ class ArticleImageMatcher:
image_id = cursor.lastrowid image_id = cursor.lastrowid
logger.info(f"图片信息已插入ai_images表，image_id: {image_id}") logger.info(f"图片信息已插入ai_images表，image_id: {image_id}")
# 插入ai_image_tags表 # 插入ai_image_tags表（default_tag_id和default_tag_name与tag_id和tag_name保持一致）
insert_tag_query = """ insert_tag_query = """
INSERT INTO ai_image_tags INSERT INTO ai_image_tags
(image_id, image_name, image_url, image_thumb_url, tag_id, tag_name, (image_id, image_name, image_url, image_thumb_url, tag_id, tag_name,
keywords_id, keywords_name, department_id, department_name, keywords_id, keywords_name, department_id, department_name,
image_source, created_user_id, image_attached_article_count) default_tag_id, default_tag_name, image_source, created_user_id, image_attached_article_count, status)
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
""" """
cursor.execute(insert_tag_query, ( cursor.execute(insert_tag_query, (
image_id, image_name, image_url, '', tag_id, tag_name, image_id, image_name, image_url, '', tag_id, tag_name,
keywords_id, keywords, department_id, department, keywords_id, keywords, department_id, department,
3, 1, 0 # image_source: 3表示AI生成 tag_id, tag_name, # default_tag_id和default_tag_name与tag_id和tag_name一致
3, 1, 0, # image_source: 3表示AI生成
'generate' # status字段设置为generate
)) ))
tag_image_id = cursor.lastrowid tag_image_id = cursor.lastrowid
logger.info(f"图片标签信息已插入ai_image_tags表，tag_image_id: {tag_image_id}") logger.info(f"图片标签信息已插入ai_image_tags表，tag_image_id: {tag_image_id}")