mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-02 01:10:23 +08:00
1d9c77522a
- 删除全部 Python 源码 (src/) 及 Alembic 迁移脚本，归档至 _deprecated_py_src/ - 重构 Rust gateway ai_pipeline: 拆分 planner/finalize 模块，新增 contracts/adaptation 层 - 重组 handlers 模块为 admin/public/proxy/internal/shared 子模块结构 - 新增 executor 模块，引入 Rust 原生数据库迁移 (aether-data/migrations) - 简化 CI/Docker 构建流程，移除 base image 二级构建，统一为单一 app image - 移除 Python 相关基础设施文件 (entrypoint.sh, gunicorn_conf.py, Dockerfile.base)
154 lines
3.7 KiB
Python
154 lines
3.7 KiB
Python
"""
|
|
OpenAI API 数据模型定义
|
|
"""
|
|
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, ConfigDict
|
|
|
|
|
|
# 配置允许额外字段，以支持 API 的新特性
|
|
class BaseModelWithExtras(BaseModel):
|
|
model_config = ConfigDict(extra="allow")
|
|
|
|
|
|
class OpenAIMessage(BaseModelWithExtras):
|
|
"""OpenAI消息模型"""
|
|
|
|
role: str
|
|
content: str | list[dict[str, Any]] | None = None
|
|
tool_calls: list[dict[str, Any]] | None = None
|
|
tool_call_id: str | None = None
|
|
name: str | None = None
|
|
|
|
|
|
class OpenAIFunction(BaseModelWithExtras):
|
|
"""OpenAI函数定义"""
|
|
|
|
name: str
|
|
description: str | None = None
|
|
parameters: dict[str, Any]
|
|
|
|
|
|
class OpenAITool(BaseModelWithExtras):
|
|
"""OpenAI工具定义"""
|
|
|
|
type: str = "function"
|
|
function: OpenAIFunction
|
|
|
|
|
|
class OpenAIRequest(BaseModelWithExtras):
|
|
"""OpenAI请求模型"""
|
|
|
|
model: str
|
|
messages: list[OpenAIMessage]
|
|
max_tokens: int | None = None
|
|
temperature: float | None = 1.0
|
|
top_p: float | None = None
|
|
stream: bool | None = False
|
|
stop: str | list[str] | None = None
|
|
tools: list[OpenAITool] | None = None
|
|
tool_choice: str | dict[str, Any] | None = None
|
|
presence_penalty: float | None = None
|
|
frequency_penalty: float | None = None
|
|
n: int | None = None
|
|
seed: int | None = None
|
|
response_format: dict[str, Any] | None = None
|
|
logit_bias: dict[str, float] | None = None
|
|
logprobs: bool | None = None
|
|
top_logprobs: int | None = None
|
|
user: str | None = None
|
|
|
|
|
|
class ResponsesInputMessage(BaseModelWithExtras):
|
|
"""Responses API 输入消息"""
|
|
|
|
type: str = "message"
|
|
role: str
|
|
content: list[dict[str, Any]]
|
|
|
|
|
|
class ResponsesReasoningConfig(BaseModelWithExtras):
|
|
"""Responses API 推理配置"""
|
|
|
|
effort: str = "high" # low, medium, high
|
|
summary: str = "auto" # auto, off
|
|
|
|
|
|
class ResponsesRequest(BaseModelWithExtras):
|
|
"""OpenAI Responses API 请求模型（用于 Claude Code 等客户端）"""
|
|
|
|
model: str
|
|
instructions: str | None = None
|
|
input: list[ResponsesInputMessage]
|
|
tools: list[dict[str, Any]] | None = None
|
|
tool_choice: str | dict[str, Any] | None = "auto"
|
|
parallel_tool_calls: bool | None = False
|
|
reasoning: ResponsesReasoningConfig | None = None
|
|
store: bool | None = False
|
|
stream: bool | None = True
|
|
include: list[str] | None = None
|
|
prompt_cache_key: str | None = None
|
|
# 其他参数
|
|
max_tokens: int | None = None
|
|
temperature: float | None = None
|
|
top_p: float | None = None
|
|
stop: str | list[str] | None = None
|
|
|
|
|
|
class OpenAIUsage(BaseModelWithExtras):
|
|
"""OpenAI使用统计"""
|
|
|
|
prompt_tokens: int
|
|
completion_tokens: int
|
|
total_tokens: int
|
|
|
|
|
|
class OpenAIChoice(BaseModelWithExtras):
|
|
"""OpenAI选择结果"""
|
|
|
|
index: int
|
|
message: OpenAIMessage
|
|
finish_reason: str | None = None
|
|
logprobs: dict[str, Any] | None = None
|
|
|
|
|
|
class OpenAIResponse(BaseModelWithExtras):
|
|
"""OpenAI响应模型"""
|
|
|
|
id: str
|
|
object: str = "chat.completion"
|
|
created: int
|
|
model: str
|
|
choices: list[OpenAIChoice]
|
|
usage: OpenAIUsage | None = None
|
|
system_fingerprint: str | None = None
|
|
|
|
|
|
class OpenAIStreamDelta(BaseModelWithExtras):
|
|
"""OpenAI流式响应增量"""
|
|
|
|
role: str | None = None
|
|
content: str | None = None
|
|
tool_calls: list[dict[str, Any]] | None = None
|
|
|
|
|
|
class OpenAIStreamChoice(BaseModelWithExtras):
|
|
"""OpenAI流式响应选择"""
|
|
|
|
index: int
|
|
delta: OpenAIStreamDelta
|
|
finish_reason: str | None = None
|
|
logprobs: dict[str, Any] | None = None
|
|
|
|
|
|
class OpenAIStreamResponse(BaseModelWithExtras):
|
|
"""OpenAI流式响应模型"""
|
|
|
|
id: str
|
|
object: str = "chat.completion.chunk"
|
|
created: int
|
|
model: str
|
|
choices: list[OpenAIStreamChoice]
|
|
system_fingerprint: str | None = None
|