33 lines
1.1 KiB
Rust
33 lines
1.1 KiB
Rust
use platform_image::vector_engine::{
|
|
GPT_IMAGE_2_MODEL, VECTOR_ENGINE_PROVIDER, VectorEngineImageSettings,
|
|
build_vector_engine_image_request_body, vector_engine_images_edit_url,
|
|
vector_engine_images_generation_url,
|
|
};
|
|
|
|
#[test]
|
|
fn vector_engine_module_exposes_provider_protocol_helpers() {
|
|
let settings = VectorEngineImageSettings {
|
|
base_url: "https://vector.example/v1".to_string(),
|
|
api_key: "test-key".to_string(),
|
|
request_timeout_ms: 1_000,
|
|
};
|
|
|
|
let body =
|
|
build_vector_engine_image_request_body("雾海神殿", Some("文字，水印"), "16:9", 9, &[]);
|
|
|
|
assert_eq!(GPT_IMAGE_2_MODEL, "gpt-image-2");
|
|
assert_eq!(VECTOR_ENGINE_PROVIDER, "vector-engine");
|
|
assert_eq!(body["model"], GPT_IMAGE_2_MODEL);
|
|
assert_eq!(body["size"], "1536x1024");
|
|
assert_eq!(body["n"], 4);
|
|
assert_eq!(body["prompt"], "雾海神殿\n避免：文字，水印");
|
|
assert_eq!(
|
|
vector_engine_images_generation_url(&settings),
|
|
"https://vector.example/v1/images/generations"
|
|
);
|
|
assert_eq!(
|
|
vector_engine_images_edit_url(&settings),
|
|
"https://vector.example/v1/images/edits"
|
|
);
|
|
}
|