From b272fec78e30d1a10f3ab761684a119193391296 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期二, 19 十一月 2024 17:08:57 +0800 Subject: [PATCH] 解决冲突 --- app/api/files.py | 9 ++++++++- 1 files changed, 8 insertions(+), 1 deletions(-) diff --git a/app/api/files.py b/app/api/files.py index eed49d8..fe80f4a 100644 --- a/app/api/files.py +++ b/app/api/files.py @@ -1,7 +1,7 @@ from typing import Optional import requests -from fastapi import Depends, APIRouter, HTTPException, UploadFile, File, Query +from fastapi import Depends, APIRouter, HTTPException, UploadFile, File, Query, Form from pydantic import BaseModel from sqlalchemy.orm import Session from starlette.responses import StreamingResponse @@ -11,6 +11,7 @@ from app.models.agent_model import AgentType, AgentModel from app.models.base_model import get_db from app.models.user_model import UserModel +from app.service.basic import BasicService from app.service.bisheng import BishengService from app.service.ragflow import RagflowService from app.service.service_token import get_ragflow_token, get_bisheng_token @@ -58,6 +59,12 @@ raise HTTPException(status_code=500, detail=str(e)) result["file_name"] = file.filename return Response(code=200, msg="", data=result) + elif agent.agent_type == AgentType.BASIC: + if agent_id == "basic_excel_talk": + service = BasicService(base_url=settings.basic_base_url) + result = await service.excel_talk_upload(chat_id, file.filename, file_content) + + return Response(code=200, msg="", data=result) else: return Response(code=200, msg="Unsupported agent type") -- Gitblit v1.8.0