| | |
| | | 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 |
| | |
| | | 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.token import get_ragflow_token, get_bisheng_token |
| | | from app.service.service_token import get_ragflow_token, get_bisheng_token |
| | | import urllib.parse |
| | | |
| | | router = APIRouter() |
| | |
| | | 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") |