Agent开发指南

from abc import ABC, abstractmethod
from typing import Dict, Any
from fastapi import UploadFile

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 NewPlatformAgent(BaseAgent):
    def __init__(self, config: Dict[str, Any]):
        self.config = config
        self._validate_config()
    
    async def chat(self, request: ChatRequest) -> UnifiedChatResponse:
        # 1. 转换请求格式
        platform_request = self._convert_request(request)
        
        # 2. 调用平台API
        response = await self._call_platform_api(platform_request)
        
        # 3. 转换响应格式
        return self.format_response(response)
    
    async def upload_file(self, file: UploadFile, created_by: str) -> FileResponse:
        # 实现文件上传逻辑
        pass

from .registry import registry

def register_new_platform_agent():
    config = {
        "base_url": os.getenv("NEW_PLATFORM_BASE_URL"),
        "api_key": os.getenv("NEW_PLATFORM_API_KEY")
    }
    
    if config["base_url"] and config["api_key"]:
        agent = NewPlatformAgent(config)
        registry.register("new_platform", agent)

# 自动注册
register_new_platform_agent()

NEW_PLATFORM_BASE_URL=https://api.newplatform.com
NEW_PLATFORM_API_KEY=your-api-key

def _convert_request(self, request: ChatRequest) -> Dict[str, Any]:
    return {
        "message": request.query,
        "user_id": request.user,
        "stream": request.stream
    }

def format_response(self, response: Any) -> UnifiedChatResponse:
    return UnifiedChatResponse(
        id=response.get("id"),
        model=response.get("model"),
        usage=response.get("usage"),
        choices=response.get("choices", [])
    )

async def chat(self, request: ChatRequest) -> UnifiedChatResponse:
    try:
        # 实现聊天逻辑
        pass
    except Exception as e:
        logger.error(f"Chat error: {e}")
        raise HTTPException(status_code=500, detail="Internal error")

def test_config_validation(self):
    """测试配置验证"""
    with pytest.raises(ValueError):
        NewPlatformAgent({})

async def test_chat(self):
    """测试聊天功能"""
    request = ChatRequest(query="Hello", user="test_user")
    response = await self.agent.chat(request)
    assert response.id is not None

async def test_api_integration():
    """测试API集成"""
    response = await client.post(
        "/api/v1/chat",
        headers={"agent": "new_platform"},
        json={"query": "Hello", "user": "test_user"}
    )
    assert response.status_code == 200