28 lines
579 B
Python
28 lines
579 B
Python
"""
|
|
数据模型定义
|
|
"""
|
|
from pydantic import BaseModel
|
|
from typing import Dict, List, Optional, Any
|
|
|
|
|
|
class ChatMessage(BaseModel):
|
|
role: str
|
|
content: str
|
|
images: Optional[List[str]] = None
|
|
files: Optional[List[str]] = None
|
|
|
|
|
|
class ChatRequest(BaseModel):
|
|
model: str = "qwen-plus"
|
|
messages: List[Dict[str, Any]]
|
|
stream: bool = True
|
|
temperature: Optional[float] = 0.7
|
|
max_tokens: Optional[int] = 2000
|
|
|
|
|
|
class ModelInfo(BaseModel):
|
|
id: str
|
|
name: str
|
|
description: str
|
|
maxTokens: int
|
|
provider: str |