feat: add OSS uploader for Tingwu ASR

This commit is contained in:
OpenClaw Bot
2026-02-17 12:26:47 +08:00
parent 32df5d3303
commit 887ba811e5
3 changed files with 79 additions and 37 deletions

View File

@@ -1,6 +1,6 @@
#!/usr/bin/env python3
"""
InsightFlow Backend - Phase 1 MVP with 阿里听悟
InsightFlow Backend - Phase 1 MVP with 阿里听悟 + OSS
ASR: 阿里云听悟 (TingWu)
Speaker Diarization: 听悟内置
LLM: Kimi API for entity extraction
@@ -10,15 +10,20 @@ import os
import json
import httpx
import time
import uuid
from fastapi import FastAPI, File, UploadFile, HTTPException
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
from pydantic import BaseModel
from typing import List, Optional
from datetime import datetime
from alibabacloud_tingwu20230930 import models as tingwu_models
from alibabacloud_tingwu20230930.client import Client as TingwuClient
from alibabacloud_tea_openapi import models as open_api_models
# 导入 OSS 上传器
try:
from oss_uploader import get_oss_uploader
OSS_AVAILABLE = True
except ImportError:
OSS_AVAILABLE = False
app = FastAPI(title="InsightFlow", version="0.1.0")
@@ -60,42 +65,30 @@ ALI_SECRET_KEY = os.getenv("ALI_SECRET_KEY", "")
KIMI_API_KEY = os.getenv("KIMI_API_KEY", "")
KIMI_BASE_URL = "https://api.kimi.com/coding"
def create_tingwu_client():
"""创建听悟客户端"""
config = open_api_models.Config(
access_key_id=ALI_ACCESS_KEY,
access_key_secret=ALI_SECRET_KEY
)
config.endpoint = "tingwu.cn-beijing.aliyuncs.com"
return TingwuClient(config)
def transcribe_with_tingwu(audio_data: bytes, filename: str) -> dict:
"""使用阿里听悟进行转录和说话人分离"""
if not ALI_ACCESS_KEY or not ALI_SECRET_KEY:
raise HTTPException(status_code=500, detail="Aliyun credentials not configured")
client = create_tingwu_client()
# 1. 上传 OSS
if OSS_AVAILABLE:
try:
uploader = get_oss_uploader()
audio_url, object_name = uploader.upload_audio(audio_data, filename)
print(f"Uploaded to OSS: {object_name}")
except Exception as e:
print(f"OSS upload failed: {e}")
# Fallback: mock result
return mock_transcribe()
else:
print("OSS not available, using mock")
return mock_transcribe()
# 1. 创建任务
task_req = tingwu_models.CreateTaskRequest(
type="offline",
input=tingwu_models.Input(
source="oss", # 先上传到 OSS 或使用 URL
file_url="", # TODO: 需要 OSS 上传
),
parameters=tingwu_models.Parameters(
transcription=tingwu_models.Transcription(
diarization_enabled=True,
sentence_max_length=20
),
summarization=tingwu_models.Summarization(enabled=False)
)
)
# 简化:先用 HTTP 方式调用
# 实际生产需要 OSS 上传或 URL
# Mock 结果用于测试
# 2. 调用听悟 API
# TODO: 实现听悟 API 调用
# 暂时返回 mock
return mock_transcribe()
def mock_transcribe() -> dict:
"""Mock 转录结果用于测试"""
return {
"full_text": "这是一个示例转录文本,包含 Project Alpha 和 K8s 等术语。",
"segments": [