mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-02 17:30:23 +08:00
refactor: 移除 Python 后端源码,全面迁移至 Rust gateway 架构
- 删除全部 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)
This commit is contained in:
165
_deprecated_py_src/models/gemini.py
Normal file
165
_deprecated_py_src/models/gemini.py
Normal file
@@ -0,0 +1,165 @@
|
||||
"""
|
||||
Google Gemini API 请求/响应模型
|
||||
|
||||
支持 Gemini API 的请求/响应格式
|
||||
作为 API 网关,采用宽松类型定义以支持 API 新特性透传
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any
|
||||
|
||||
from pydantic import BaseModel, ConfigDict, Field
|
||||
|
||||
|
||||
class BaseModelWithExtras(BaseModel):
|
||||
"""允许额外字段的基础模型"""
|
||||
|
||||
model_config = ConfigDict(extra="allow")
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# 内容定义 - 使用宽松类型以支持透传
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
|
||||
class GeminiContent(BaseModelWithExtras):
|
||||
"""
|
||||
Gemini 消息内容
|
||||
|
||||
使用宽松类型定义,parts 接受任意字典列表以支持 API 新特性
|
||||
"""
|
||||
|
||||
role: str | None = None
|
||||
parts: list[dict[str, Any]]
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# 请求模型 - 只定义网关需要的字段,其余透传
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
|
||||
class GeminiRequest(BaseModelWithExtras):
|
||||
"""
|
||||
Gemini 统一请求模型
|
||||
|
||||
内部使用,统一处理 generateContent 和 streamGenerateContent
|
||||
|
||||
注意: Gemini API 通过 URL 端点区分流式/非流式请求:
|
||||
- generateContent - 非流式
|
||||
- streamGenerateContent - 流式
|
||||
请求体中不应包含 stream 字段
|
||||
|
||||
采用宽松类型定义,除必要字段外全部透传
|
||||
"""
|
||||
|
||||
model: str | None = Field(default=None, description="模型名称,从 URL 路径提取(内部使用)")
|
||||
contents: list[GeminiContent]
|
||||
# 以下字段全部使用 dict[str, Any] 透传,不做结构验证
|
||||
system_instruction: dict[str, Any] | None = Field(default=None, alias="systemInstruction")
|
||||
tools: list[dict[str, Any]] | None = None
|
||||
tool_config: dict[str, Any] | None = Field(default=None, alias="toolConfig")
|
||||
safety_settings: list[dict[str, Any]] | None = Field(default=None, alias="safetySettings")
|
||||
generation_config: dict[str, Any] | None = Field(default=None, alias="generationConfig")
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# 响应模型 - 用于解析上游响应提取必要信息(如 usage)
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
|
||||
class GeminiUsageMetadata(BaseModelWithExtras):
|
||||
"""Token 使用量 - 用于计费统计"""
|
||||
|
||||
prompt_token_count: int = Field(default=0, alias="promptTokenCount")
|
||||
candidates_token_count: int = Field(default=0, alias="candidatesTokenCount")
|
||||
total_token_count: int = Field(default=0, alias="totalTokenCount")
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# 文件 API 模型
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
|
||||
class GeminiFileMetadata(BaseModelWithExtras):
|
||||
"""
|
||||
Gemini 文件元数据
|
||||
|
||||
用于上传文件时指定的元数据信息
|
||||
"""
|
||||
|
||||
display_name: str | None = Field(default=None, alias="displayName")
|
||||
|
||||
|
||||
class GeminiFileUploadRequest(BaseModelWithExtras):
|
||||
"""
|
||||
Gemini 文件上传请求
|
||||
|
||||
用于 media.upload API 的请求体
|
||||
"""
|
||||
|
||||
file: GeminiFileMetadata | None = None
|
||||
|
||||
|
||||
class GeminiFile(BaseModelWithExtras):
|
||||
"""
|
||||
Gemini 文件资源
|
||||
|
||||
表示已上传到 Gemini API 的文件
|
||||
"""
|
||||
|
||||
name: str | None = None # 文件名,格式:files/xxx
|
||||
display_name: str | None = Field(default=None, alias="displayName")
|
||||
mime_type: str | None = Field(default=None, alias="mimeType")
|
||||
size_bytes: str | None = Field(default=None, alias="sizeBytes")
|
||||
create_time: str | None = Field(default=None, alias="createTime")
|
||||
update_time: str | None = Field(default=None, alias="updateTime")
|
||||
expiration_time: str | None = Field(default=None, alias="expirationTime")
|
||||
sha256_hash: str | None = Field(default=None, alias="sha256Hash")
|
||||
uri: str | None = None # 文件 URI,用于在请求中引用
|
||||
download_uri: str | None = Field(default=None, alias="downloadUri")
|
||||
state: str | None = None # PROCESSING, ACTIVE, FAILED
|
||||
error: dict[str, Any] | None = None
|
||||
# 视频文件元数据
|
||||
video_metadata: dict[str, Any] | None = Field(default=None, alias="videoMetadata")
|
||||
|
||||
|
||||
class GeminiFileListResponse(BaseModelWithExtras):
|
||||
"""
|
||||
Gemini 文件列表响应
|
||||
|
||||
用于 files.list API 的响应体
|
||||
"""
|
||||
|
||||
files: list["GeminiFile"] | None = None
|
||||
next_page_token: str | None = Field(default=None, alias="nextPageToken")
|
||||
|
||||
|
||||
class GeminiFileUploadResponse(BaseModelWithExtras):
|
||||
"""
|
||||
Gemini 文件上传响应
|
||||
|
||||
用于 media.upload API 的响应体
|
||||
"""
|
||||
|
||||
file: GeminiFile | None = None
|
||||
|
||||
|
||||
class GeminiFilePart(BaseModelWithExtras):
|
||||
"""
|
||||
Gemini 文件引用部分
|
||||
|
||||
用于在请求内容中引用已上传的文件
|
||||
使用 file_data 字段引用文件 URI
|
||||
"""
|
||||
|
||||
file_data: dict[str, Any] | None = Field(default=None, alias="fileData")
|
||||
# fileData 格式:{"mimeType": "...", "fileUri": "..."}
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Thought Signature 常量
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
# 用于从其他模型迁移对话时绕过签名验证
|
||||
DUMMY_THOUGHT_SIGNATURE = "context_engineering_is_the_way_to_go"
|
||||
Reference in New Issue
Block a user