jinchan
9e6854c3d0
- 新增 local_api_client.py，提供统一同步HTTP调用类和API端点定义 - 新增 local_api_models.py，定义OpenAI图像生成及响应数据模型 - 新增 local_api_utils.py，包含图像张量处理、下载重试及API响应验证等工具 - 优化 node.py，移除旧实现，改为导入拆分后的模块，提高代码可维护性 - .gitignore中添加doc/目录忽略规则
38 lines
977 B
Python
38 lines
977 B
Python
from typing import Any, List, Optional
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
|
|
class OpenAIImageGenerationRequest(BaseModel):
|
|
model_config = ConfigDict(extra="allow")
|
|
|
|
model: str
|
|
prompt: str
|
|
quality: Optional[str] = None
|
|
background: Optional[str] = None
|
|
n: Optional[int] = 1
|
|
seed: Optional[int] = None
|
|
size: Optional[str] = None
|
|
|
|
|
|
class OpenAIImageEditRequest(OpenAIImageGenerationRequest):
|
|
pass
|
|
|
|
|
|
class OpenAIImageResponseItem(BaseModel):
|
|
model_config = ConfigDict(extra="allow")
|
|
|
|
b64_json: Optional[str] = None
|
|
url: Optional[str] = None
|
|
revised_prompt: Optional[str] = None
|
|
|
|
|
|
class OpenAIImageGenerationResponse(BaseModel):
|
|
model_config = ConfigDict(extra="allow")
|
|
|
|
created: Optional[int] = None
|
|
data: List[OpenAIImageResponseItem] = Field(default_factory=list)
|
|
success: Optional[bool] = None
|
|
message: Optional[str] = None
|
|
raw: Optional[Any] = None
|