| | |
|---|
| | | file_type: Optional[str] = Query(None, description="Optional file type for basic agents"), |
|---|
| | | db: Session = Depends(get_db) |
|---|
| | | ): |
|---|
| | | agent = db.query(AgentModel).filter(AgentModel.id == agent_id).first() |
|---|
| | | # agent = db.query(AgentModel).filter(AgentModel.id == agent_id).first() |
|---|
| | | agent = db.query(MenuCapacityModel).filter(MenuCapacityModel.chat_id == agent_id).first() |
|---|
| | | |
|---|
| | | if not agent: |
|---|
| | | return Response(code=404, msg="Agent not found") |
|---|
| | | |
|---|
| | | if agent.agent_type == AgentType.BISHENG: |
|---|
| | | agent_type = int(agent.capacity_type) |
|---|
| | | if agent_type == AgentType.BISHENG: |
|---|
| | | url = urllib.parse.unquote(url) |
|---|
| | | # 从 URL 中提取文件名 |
|---|
| | | parsed_url = urllib.parse.urlparse(url) |
|---|
| | | filename = urllib.parse.unquote(parsed_url.path.split('/')[-1]) |
|---|
| | | url = url.replace("http://minio:9000", settings.sgb_base_url) |
|---|
| | | elif agent.agent_type == AgentType.RAGFLOW: |
|---|
| | | elif agent_type == AgentType.RAGFLOW: |
|---|
| | | if not doc_id: |
|---|
| | | return Response(code=400, msg="doc_id is required") |
|---|
| | | url = f"{settings.fwr_base_url}/v1/document/get/{doc_id}" |
|---|
| | | filename = doc_name |
|---|
| | | elif agent.agent_type == AgentType.BASIC: |
|---|
| | | elif agent_type == AgentType.BASIC: |
|---|
| | | if agent_id == "basic_excel_talk": |
|---|
| | | return await download_basic_file(file_id, file_type) |
|---|
| | | |
|---|
