youhua

zhaoqingang

2024-11-21 95d2318cd8fe7f39a9786fe9a8088440b081b295

 app/api/files.py

@@ -68,6 +68,12 @@

            return Response(code=200, msg="", data=result)

        elif agent_id == "basic_paper_agent":
            service = BasicService(base_url=settings.basic_paper_url)
            result = await service.paper_file_upload(chat_id, file.filename, file_content)

            return Response(code=200, msg="", data=result)

    else:
        return Response(code=200, msg="Unsupported agent type")

@@ -101,6 +107,8 @@
        if agent_id == "basic_excel_talk":
            return await download_basic_file(file_id, file_type)

        elif agent_id == "basic_question_talk":
            return await download_basic_file(file_id, file_type)
    else:
        return Response(code=400, msg="Unsupported agent type")

@@ -137,5 +145,12 @@
            media_type=mimetype,
            headers={"Content-Disposition": f"attachment; filename={filename}"}
        )
    elif file_type == "word":
        content, filename, mimetype = await service.questions_talk_word_download(file_id)
        return StreamingResponse(
            io.BytesIO(content),
            media_type=mimetype,
            headers={"Content-Disposition": f"attachment; filename={filename}"}
        )
    else:
        return Response(code=400, msg="Unsupported file type")