技术架构

# 抽象基类定义统一接口
class BaseAgent(ABC):
    @abstractmethod
    async def chat(self, request: ChatRequest) -> UnifiedChatResponse:
        pass
    
    @abstractmethod
    async def upload_file(self, file: UploadFile, created_by: str) -> FileResponse:
        pass

# 具体适配器实现
class FastGPTAgent(BaseAgent):
    async def chat(self, request: ChatRequest) -> UnifiedChatResponse:
        # 将统一请求转换为FastGPT特定格式
        fastgpt_request = self._convert_to_fastgpt_format(request)
        # 调用FastGPT API
        response = await self._call_fastgpt_api(fastgpt_request)
        # 将FastGPT响应转换为统一格式
        return self._convert_to_unified_format(response)

class AgentRegistry:
    def __init__(self):
        self._agents = {}
    
    def register(self, name: str, agent: BaseAgent):
        self._agents[name] = agent
    
    def get_agent(self, name: str) -> BaseAgent:
        return self._agents.get(name)
    
    def list_agents(self) -> List[str]:
        return list(self._agents.keys())

class BaseAgent(ABC):
    """AI Agent抽象基类"""
    
    def __init__(self, config: Dict[str, Any]):
        self.config = config
        self._validate_config()
    
    @abstractmethod
    async def chat(self, request: ChatRequest) -> UnifiedChatResponse:
        """处理聊天请求"""
        pass
    
    @abstractmethod
    async def upload_file(self, file: UploadFile, created_by: str) -> FileResponse:
        """处理文件上传"""
        pass

class ChatRequest(BaseModel):
    query: str
    user: str
    uid: Optional[str] = None
    stream: bool = False
    detail: bool = False
    app_id: Optional[str] = None
    chat_id: Optional[str] = None
    conversation_id: Optional[str] = None
    files: Optional[List[str]] = None

class UnifiedChatResponse(BaseModel):
    id: str
    model: str
    usage: Usage
    choices: List[Choice]
    detail: Optional[Dict[str, Any]] = None

class NewPlatformAgent(BaseAgent):
    async def chat(self, request: ChatRequest) -> UnifiedChatResponse:
        # 实现具体的聊天逻辑
        pass
    
    async def upload_file(self, file: UploadFile, created_by: str) -> FileResponse:
        # 实现文件上传逻辑
        pass

# 注册到registry
registry.register("new_platform", NewPlatformAgent(config))

@app.post("/api/v1/chat")
async def chat(request: ChatRequest, agent: str = Header(...)):
    agent_instance = registry.get_agent(agent)
    if not agent_instance:
        raise HTTPException(status_code=400, detail="Unsupported agent")
    
    return await agent_instance.chat(request)

import aiohttp

class FastGPTAgent(BaseAgent):
    def __init__(self, config: Dict[str, Any]):
        super().__init__(config)
        self.session = aiohttp.ClientSession()
    
    async def _call_api(self, url: str, data: Dict):
        async with self.session.post(url, json=data) as response:
            return await response.json()

class ChatRequest(BaseModel):
    query: str = Field(..., min_length=1, max_length=10000)
    user: str = Field(..., min_length=1, max_length=100)

# 使用环境变量，避免硬编码
API_KEY = os.getenv("API_KEY")
if not API_KEY:
    raise ValueError("API_KEY environment variable is required")

ALLOWED_EXTENSIONS = {'.txt', '.pdf', '.doc', '.docx', '.jpg', '.png'}
MAX_FILE_SIZE = 10 * 1024 * 1024  # 10MB

def validate_file(file: UploadFile):
    if file.size > MAX_FILE_SIZE:
        raise HTTPException(status_code=400, detail="File too large")
    
    if not any(file.filename.endswith(ext) for ext in ALLOWED_EXTENSIONS):
        raise HTTPException(status_code=400, detail="File type not allowed")