daniel543x/artcraft
1
0 Fork 0
mirror of https://github.com/storytold/artcraft.git synced 2026-11-04 09:28:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

I don't like this

Browse Source
This commit is contained in:
Brandon Thomas committed 2026-04-06 02:02:46 -04:00
1 parent aa4012f120
commit 5631ec15b4
2 files changed
+14 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/omni_gen_image_generate_handler.rs
+7 -1
View File
@@ -10,6 +10,7 @@ use artcraft_api_defs::omni_gen::generate_response::omni_gen_image_generate_resp
use enums::by_table::prompt_context_items::prompt_context_semantic_type::PromptContextSemanticType;
use enums::by_table::prompts::prompt_type::PromptType;
use enums::common::generation::common_generation_mode::CommonGenerationMode;
use enums::common::generation::common_model_type::CommonModelType;
use enums::common::generation_provider::GenerationProvider;
use enums::common::visibility::Visibility;
use http_server_common::request::get_request_ip::get_request_ip;
@@ -61,6 +62,11 @@ pub async fn omni_gen_image_generate_handler(
let mut generate_request = transform_request(&request)?;
let maybe_prompt_model_type: Option<CommonModelType> =
serde_json::to_string(&generate_request.model)
.ok()
.and_then(|json| serde_json::from_str(&json).ok());
// ==================== COST ==================== //
let cost_estimate = request_to_costs(&generate_request)?;
@@ -174,7 +180,7 @@ pub async fn omni_gen_image_generate_handler(
maybe_apriori_prompt_token: None,
prompt_type: PromptType::ArtcraftApp,
maybe_creator_user_token: Some(user_token),
maybe_model_type: None,
maybe_model_type: maybe_prompt_model_type,
maybe_generation_provider: Some(GenerationProvider::Artcraft),
maybe_positive_prompt: request.prompt.as_deref(),
maybe_negative_prompt: None,
crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/omni_gen_video_generate_handler.rs
+7 -1
View File
@@ -10,6 +10,7 @@ use artcraft_api_defs::omni_gen::generate_response::omni_gen_video_generate_resp
use enums::by_table::prompt_context_items::prompt_context_semantic_type::PromptContextSemanticType;
use enums::by_table::prompts::prompt_type::PromptType;
use enums::common::generation::common_generation_mode::CommonGenerationMode;
use enums::common::generation::common_model_type::CommonModelType;
use enums::common::generation_provider::GenerationProvider;
use enums::common::visibility::Visibility;
use http_server_common::request::get_request_ip::get_request_ip;
@@ -61,6 +62,11 @@ pub async fn omni_gen_video_generate_handler(
let mut generate_request = transform_request(&request)?;
let maybe_prompt_model_type: Option<CommonModelType> =
serde_json::to_string(&generate_request.model)
.ok()
.and_then(|json| serde_json::from_str(&json).ok());
// ==================== COST ==================== //
let cost_estimate = request_to_costs(&generate_request)?;
@@ -176,7 +182,7 @@ pub async fn omni_gen_video_generate_handler(
maybe_apriori_prompt_token: None,
prompt_type: PromptType::ArtcraftApp,
maybe_creator_user_token: Some(user_token),
maybe_model_type: None,
maybe_model_type: maybe_prompt_model_type,
maybe_generation_provider: Some(GenerationProvider::Artcraft),
maybe_positive_prompt: request.prompt.as_deref(),
maybe_negative_prompt: request.negative_prompt.as_deref(),