Fix PR 434 check failures

This commit is contained in:
fawney19
2026-05-14 01:51:23 +08:00
parent e251a63cb3
commit 1f8ff7f6d2
5 changed files with 11 additions and 10 deletions

View File

@@ -29,9 +29,9 @@ pub(super) fn provider_query_standard_test_client_api_format(
provider_api_format: &str,
) -> &'static str {
let normalized_api_format = crate::ai_serving::normalize_api_format_alias(provider_api_format);
if aether_ai_formats::is_embedding_api_format(&normalized_api_format) {
if crate::ai_serving::is_embedding_api_format(&normalized_api_format) {
"openai:embedding"
} else if aether_ai_formats::is_rerank_api_format(&normalized_api_format) {
} else if crate::ai_serving::is_rerank_api_format(&normalized_api_format) {
"openai:rerank"
} else {
"openai:chat"

View File

@@ -287,7 +287,7 @@ fn provider_query_parse_provider_model_mapping_object_lenient(
.map(|formats| {
formats
.into_iter()
.map(|value| aether_ai_formats::normalize_api_format_alias(&value))
.map(|value| crate::ai_serving::normalize_api_format_alias(&value))
.collect()
});
let endpoint_ids = provider_query_parse_mapping_string_list(