| | |
| | | from app.models.user_model import UserModel |
| | | from app.service.basic import BasicService |
| | | from app.service.bisheng import BishengService |
| | | from app.service.common.api_token import DfTokenDao |
| | | from app.service.v2.api_token import DfTokenDao |
| | | from app.service.difyService import DifyService |
| | | from app.service.ragflow import RagflowService |
| | | from app.service.service_token import get_ragflow_token, get_bisheng_token |
| | |
| | | return Response(code=400, msg=str(e)) |
| | | |
| | | if agent.agent_type == AgentType.RAGFLOW: |
| | | token = get_ragflow_token(db, current_user.id) |
| | | token = await get_ragflow_token(db, current_user.id) |
| | | ragflow_service = RagflowService(base_url=settings.fwr_base_url) |
| | | # 查询会话是否存在,不存在先创建会话 |
| | | history = await ragflow_service.get_session_history(token, chat_id) |
| | |
| | | elif agent.agent_type == AgentType.BISHENG: |
| | | bisheng_service = BishengService(base_url=settings.sgb_base_url) |
| | | try: |
| | | token = get_bisheng_token(db, current_user.id) |
| | | token = await get_bisheng_token(db, current_user.id) |
| | | result = await bisheng_service.upload(token, file.filename, file_content) |
| | | except Exception as e: |
| | | raise HTTPException(status_code=500, detail=str(e)) |