replace some function

tests/test_deepseek_ocr.rs

@@ -24,7 +24,7 @@ fn deepseek_ocr_generate() -> Result<()> {
                     },              
                     {
                         "type": "text", 
-                        "text": "<image>\n<|grounding|>Convert the document to markdown. "
+                        "text": "<image>\nConvert the document to markdown. "
                     }
                 ]
             }

tests/test_hunyuan_ocr.rs

@@ -24,7 +24,7 @@ fn hunyuan_ocr_generate() -> Result<()> {
                     },              
                     {
                         "type": "text", 
-                        "text": "检测并识别图片中的文字，将文本坐标格式化输出。"
+                        "text": "识别图片中的文字"
                     }
                 ]
             }

tests/test_qwen3vl.rs

@@ -7,7 +7,7 @@ use rocket::futures::StreamExt;
 
 #[test]
 fn qwen3vl_generate() -> Result<()> {
-    // test with cuda: RUST_BACKTRACE=1 cargo test -F cuda qwen3vl_generate -r -- --nocapture
+    // test with cuda: RUST_BACKTRACE=1 cargo test -F cuda,ffmpeg qwen3vl_generate -r -- --nocapture
 
     let model_path = "/home/jhq/huggingface_model/Qwen/Qwen3-VL-2B-Instruct/";
 

tests/test_robo_brain.rs

@@ -0,0 +1,42 @@
+use std::time::Instant;
+
+use aha::models::{GenerateModel, qwen2_5vl::generate::Qwen2_5VLGenerateModel};
+use aha_openai_dive::v1::resources::chat::ChatCompletionParameters;
+use anyhow::Result;
+
+#[test]
+fn robo_brain_generate() -> Result<()> {
+    // test with cuda: RUST_BACKTRACE=1 cargo test -F cuda robo_brain_generate -r -- --nocapture
+
+    let model_path = "/home/jhq/huggingface_model/BAAI/RoboBrain2.0-3B/";
+
+    let message = r#"
+    {
+        "model": "qwen2.5vl",
+        "messages": [
+            {
+                "role": "user",
+                "content": [           
+                    {
+                        "type": "text", 
+                        "text": "hello RoboBrain"
+                    }
+                ]
+            }
+        ]
+    }
+    "#;
+    let mes: ChatCompletionParameters = serde_json::from_str(message)?;
+    let i_start = Instant::now();
+    let mut model = Qwen2_5VLGenerateModel::init(model_path, None, None)?;
+    let i_duration = i_start.elapsed();
+    println!("Time elapsed in load model is: {:?}", i_duration);
+
+    let i_start = Instant::now();
+    let result = model.generate(mes)?;
+    println!("generate: \n {:?}", result);
+    let i_duration = i_start.elapsed();
+    println!("Time elapsed in generate is: {:?}", i_duration);
+
+    Ok(())
+}