| | |
|---|
| | | from app.api import Response, get_current_user, ResponseList |
|---|
| | | from app.config.config import settings |
|---|
| | | from app.config.const import DOCUMENT_TO_REPORT, IMAGE_TO_TEXT, DOCUMENT_TO_REPORT_TITLE, DOCUMENT_IA_QUESTIONS, \ |
|---|
| | | DOCUMENT_TO_PAPER |
|---|
| | | DOCUMENT_TO_PAPER, DOCUMENT_IA_QUESTIONS_DS |
|---|
| | | from app.models import MenuCapacityModel |
|---|
| | | from app.models.agent_model import AgentType, AgentModel |
|---|
| | | from app.models.base_model import get_db |
|---|
| | |
|---|
| | | data = await dify_service.upload(token, file.filename, file_content, current_user.id) |
|---|
| | | except Exception as e: |
|---|
| | | raise HTTPException(status_code=500, detail=str(e)) |
|---|
| | | elif agent.chat_type == "reportWorkflow" or agent.chat_type == "documentIa" or agent.chat_type == "paperTalk": |
|---|
| | | elif agent.chat_type == "reportWorkflow" or agent.chat_type == "documentIa" or agent.chat_type == "paperTalk" or agent.chat_type == "documentIaDs": |
|---|
| | | token_dict = { |
|---|
| | | "reportWorkflow": DOCUMENT_TO_REPORT_TITLE, |
|---|
| | | "documentIa": DOCUMENT_IA_QUESTIONS, |
|---|
| | | "paperTalk": DOCUMENT_TO_PAPER, |
|---|
| | | "documentIaDs": DOCUMENT_IA_QUESTIONS_DS, |
|---|
| | | } |
|---|
| | | token = DfTokenDao(db).get_token_by_id(token_dict[agent.chat_type]) |
|---|
| | | if not token: |
|---|
