mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-03 09:50:21 +08:00
refactor ai serving modules and crates
This commit is contained in:
@@ -12,7 +12,7 @@ pub(crate) fn models_api_format(request_context: &GatewayPublicRequestContext) -
|
||||
.and_then(|decision| decision.auth_endpoint_signature.as_deref())
|
||||
.map(str::trim)
|
||||
.filter(|signature| !signature.is_empty())?;
|
||||
match crate::ai_pipeline::normalize_api_format_alias(signature).as_str() {
|
||||
match crate::ai_serving::normalize_api_format_alias(signature).as_str() {
|
||||
"openai:chat" => Some("openai:chat"),
|
||||
"openai:responses" => Some("openai:responses"),
|
||||
"openai:responses:compact" => Some("openai:responses:compact"),
|
||||
@@ -33,7 +33,7 @@ const MODELS_CROSS_FORMAT_QUERY_API_FORMATS: &[&str] = &[
|
||||
];
|
||||
|
||||
pub(super) fn models_query_api_formats(api_format: &str) -> &'static [&'static str] {
|
||||
match crate::ai_pipeline::normalize_api_format_alias(api_format).as_str() {
|
||||
match crate::ai_serving::normalize_api_format_alias(api_format).as_str() {
|
||||
"openai:chat"
|
||||
| "openai:responses"
|
||||
| "openai:responses:compact"
|
||||
|
||||
@@ -92,12 +92,12 @@ pub(super) async fn maybe_build_local_test_connection_route_response(
|
||||
let format_value = if endpoint.api_format.trim().is_empty() {
|
||||
"claude:messages".to_string()
|
||||
} else {
|
||||
crate::ai_pipeline::normalize_api_format_alias(&endpoint.api_format)
|
||||
crate::ai_serving::normalize_api_format_alias(&endpoint.api_format)
|
||||
};
|
||||
(endpoint, format_value)
|
||||
};
|
||||
|
||||
let format_value = crate::ai_pipeline::normalize_api_format_alias(&format_value);
|
||||
let format_value = crate::ai_serving::normalize_api_format_alias(&format_value);
|
||||
if !matches!(
|
||||
format_value.as_str(),
|
||||
"openai:chat" | "claude:messages" | "gemini:generate_content"
|
||||
|
||||
@@ -213,7 +213,7 @@ fn users_me_usage_api_format_defaults_to_non_stream(item: &StoredRequestUsageAud
|
||||
return false;
|
||||
};
|
||||
matches!(
|
||||
crate::ai_pipeline::normalize_api_format_alias(value).as_str(),
|
||||
crate::ai_serving::normalize_api_format_alias(value).as_str(),
|
||||
"openai:chat"
|
||||
| "openai:responses"
|
||||
| "openai:responses:compact"
|
||||
|
||||
Reference in New Issue
Block a user