同步token

app/api/v2/public_api.py

@@ -5,7 +5,8 @@ from fastapi.responses import JSONResponse
 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_REPORT_TITLE, DOCUMENT_TO_TITLE, DOCUMENT_TO_PAPER
 from app.models.base_model import get_db
 from app.models.public_api_model import DfToken, AppRegister
 from app.service.v2.api_token import DfTokenDao
@@ -18,7 +19,8 @@ public_api = APIRouter()
 
 @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_IA_QUESTIONS: df.doc, DOCUMENT_TO_REPORT_TITLE: df.report,DOCUMENT_TO_TITLE: df.title,DOCUMENT_TO_PAPER: df.paper}
 
     try:
         for api_id, token in token_dict.items():