From 6b4093952e555e1eb2713bd85133a5f697cda1e0 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 31 十二月 2024 14:21:12 +0800
Subject: [PATCH] 文档报告支持生成小标题功能
---
app/api/public_api.py | 5 +++--
1 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/app/api/public_api.py b/app/api/public_api.py
index 06a37dc..9186342 100644
--- a/app/api/public_api.py
+++ b/app/api/public_api.py
@@ -5,7 +5,8 @@
from Log import logger
from app.api import Response
from app.api.auth import login
-from app.config.const import IMAGE_TO_TEXT, DOCUMENT_TO_CLEANING, DOCUMENT_TO_REPORT, DIFY, BISHENG, RAGFLOW
+from app.config.const import IMAGE_TO_TEXT, DOCUMENT_TO_CLEANING, DOCUMENT_TO_REPORT, DIFY, BISHENG, RAGFLOW, \
+ DOCUMENT_IA_QUESTIONS, DOCUMENT_TO_TITLE, DOCUMENT_TO_REPORT_TITLE
from app.models.base_model import get_db
from app.models.public_api_model import DfToken, AppRegister
from app.service.common.api_token import DfTokenDao
@@ -17,7 +18,7 @@
@public_api.post("/sync/df_token", response_model=Response)
async def sync_df_token(df: DfToken, db=Depends(get_db)):
- token_dict = {IMAGE_TO_TEXT: df.image, DOCUMENT_TO_CLEANING: df.clean, DOCUMENT_TO_REPORT: df.report}
+ token_dict = {IMAGE_TO_TEXT: df.image, DOCUMENT_TO_CLEANING: df.clean, DOCUMENT_TO_REPORT: df.report, DOCUMENT_TO_REPORT_TITLE: df.reoprt_title, DOCUMENT_TO_TITLE: df.title, DOCUMENT_IA_QUESTIONS: df.question}
try:
for api_id, token in token_dict.items():
--
Gitblit v1.8.0