mirror of
https://github.com/fawney19/Aether.git
synced 2026-10-04 16:37:46 +08:00
fawney19
5d96d6673b
- 新增 aether-ai-pipeline 和 aether-data-contracts crate，将 pipeline 逻辑与数据契约从 gateway 中解耦 - 重构 admin handlers：拆分单体模块为 auth/billing/endpoint/features/model/observability/provider/system 等独立子模块 - 合并 chat/cli 重复代码路径：精简 conversion、finalize、planner 中的 sync/chat/cli 分支 - 重构 scheduler/executor/data 层，引入 facade 模式降低模块间耦合 - 移除冗余的 intent 模块，将 plan_fallback/policy/stream_path/sync_path 迁移至 executor - 前端适配：调整 admin API 调用和 provider 模型测试对话框
61 lines
2.4 KiB
Rust
61 lines
2.4 KiB
Rust
use std::collections::BTreeMap;
|
|
use std::sync::{Arc, Mutex};
|
|
use std::time::{SystemTime, UNIX_EPOCH};
|
|
|
|
use aether_crypto::{
|
|
decrypt_python_fernet_ciphertext, encrypt_python_fernet_plaintext, DEVELOPMENT_ENCRYPTION_KEY,
|
|
};
|
|
use aether_data::repository::auth::{
|
|
InMemoryAuthApiKeySnapshotRepository, StoredAuthApiKeySnapshot,
|
|
};
|
|
use aether_data::repository::auth_modules::{
|
|
InMemoryAuthModuleReadRepository, StoredLdapModuleConfig, StoredOAuthProviderModuleConfig,
|
|
};
|
|
use aether_data::repository::management_tokens::{
|
|
InMemoryManagementTokenRepository, ManagementTokenReadRepository, StoredManagementToken,
|
|
StoredManagementTokenUserSummary, StoredManagementTokenWithUser,
|
|
};
|
|
use aether_data::repository::oauth_providers::{
|
|
InMemoryOAuthProviderRepository, OAuthProviderReadRepository, StoredOAuthProviderConfig,
|
|
};
|
|
use aether_data::repository::provider_catalog::InMemoryProviderCatalogReadRepository;
|
|
use aether_data::repository::proxy_nodes::{
|
|
InMemoryProxyNodeRepository, ProxyNodeReadRepository, StoredProxyNode, StoredProxyNodeEvent,
|
|
};
|
|
use aether_data::repository::quota::InMemoryProviderQuotaRepository;
|
|
use aether_data::repository::wallet::InMemoryWalletRepository;
|
|
use aether_data_contracts::repository::{
|
|
candidates::{RequestCandidateStatus, StoredRequestCandidate},
|
|
global_models::{
|
|
GlobalModelReadRepository, StoredAdminGlobalModel, StoredAdminProviderModel,
|
|
StoredProviderActiveGlobalModel, StoredProviderModelStats, StoredPublicGlobalModel,
|
|
},
|
|
provider_catalog::{
|
|
ProviderCatalogReadRepository, StoredProviderCatalogEndpoint, StoredProviderCatalogKey,
|
|
StoredProviderCatalogProvider,
|
|
},
|
|
quota::StoredProviderQuotaSnapshot,
|
|
};
|
|
use axum::body::{to_bytes, Body, Bytes};
|
|
use axum::response::Response;
|
|
use axum::routing::{any, post};
|
|
use axum::{extract::Request, Json, Router};
|
|
use http::header::{HeaderName, HeaderValue};
|
|
use http::{HeaderMap, StatusCode};
|
|
use serde_json::json;
|
|
|
|
mod admin;
|
|
mod helpers;
|
|
mod internal;
|
|
mod proxy;
|
|
|
|
use super::{
|
|
build_router, build_router_with_execution_runtime_override, build_router_with_state,
|
|
build_state_with_execution_runtime_override, start_server, wait_until, AppState,
|
|
FrontdoorCorsConfig, FrontdoorUserRpmConfig, GatewayFallbackMetricKind, GatewayFallbackReason,
|
|
Infallible, UsageRuntimeConfig, VideoTaskTruthSourceMode,
|
|
};
|
|
use crate::constants::*;
|
|
use crate::data::GatewayDataState;
|
|
use helpers::*;
|