1

server-rs/crates/api-server/src/custom_world_agent_entities.rs

@@ -1,6 +1,8 @@
 use platform_llm::{LlmClient, LlmMessage, LlmTextRequest};
 use serde_json::{Map as JsonMap, Value as JsonValue};
 use shared_contracts::runtime::ExecuteCustomWorldAgentActionRequest;
+
+use crate::llm_model_routing::CREATION_TEMPLATE_LLM_MODEL;
 use spacetime_client::CustomWorldAgentSessionRecord;
 
 const CUSTOM_WORLD_AGENT_CHARACTER_EXPANSION_SYSTEM_PROMPT: &str =
@@ -92,10 +94,15 @@ pub async fn generate_custom_world_agent_entities(
     };
 
     let response = llm_client
-        .request_text(LlmTextRequest::new(vec![
-            LlmMessage::system(system_prompt),
-            LlmMessage::user(user_prompt),
-        ]))
+        .request_text(
+            LlmTextRequest::new(vec![
+                LlmMessage::system(system_prompt),
+                LlmMessage::user(user_prompt),
+            ])
+            .with_model(CREATION_TEMPLATE_LLM_MODEL)
+            .with_responses_api()
+            .with_web_search(true),
+        )
         .await
         .map_err(|error| format!("{action} LLM 请求失败：{error}"))?;
     let generated_entities = parse_json_array_response(response.content.as_str())