koodilabran v0.1

network-poc/node/src/lib.rs

@@ -9,6 +9,7 @@ use burn::backend::{Wgpu, NdArray};
 pub mod storage;
 pub mod smollm;
 pub mod qwen;
+pub mod qwen_coder;
 pub mod phi3;
 
 #[macro_export]
@@ -159,7 +160,7 @@ pub async fn start_agent_node(hub_url: String, has_webgpu: bool, device_info_jso
     HAS_WEBGPU.store(has_webgpu, Ordering::SeqCst);
     SELECTED_TASK.store(task_id, Ordering::SeqCst);
     let backend_name = if has_webgpu { "WebGPU" } else { "CPU (NdArray)" };
-    let task_names = ["tokenize", "smollm-135m", "qwen-05b", "phi3-mini"];
+    let task_names = ["tokenize", "smollm-135m", "qwen-05b", "phi3-mini", "qwen-coder-05b", "qwen-coder-3b"];
     let task_name = task_names.get(task_id as usize).unwrap_or(&"tokenize");
     console_log!("Kipinä Agent Node käynnistyy — backend: {} | tehtävä: {}", backend_name, task_name);
 
@@ -248,6 +249,21 @@ pub async fn start_agent_node(hub_url: String, has_webgpu: bool, device_info_jso
                         });
                     }
                 }
+            } else if msg.contains("llm_prompt") && (current_task == 4 || current_task == 5) {
+                // Qwen2.5-Coder: 4 = 0.5B, 5 = 3B
+                if LLM_BUSY.load(Ordering::SeqCst) {
+                } else if let Ok(task) = serde_json::from_str::<serde_json::Value>(&msg) {
+                    let prompt = task.get("prompt").and_then(|v| v.as_str()).unwrap_or("").to_string();
+                    if !prompt.is_empty() {
+                        let use_3b = current_task == 5;
+                        LLM_BUSY.store(true, Ordering::SeqCst);
+                        let ws_for_async = ws_clone.clone();
+                        wasm_bindgen_futures::spawn_local(async move {
+                            qwen_coder::run_coder_inference(prompt, ws_for_async, use_3b).await;
+                            LLM_BUSY.store(false, Ordering::SeqCst);
+                        });
+                    }
+                }
             } else if msg.contains("ai_task") {
                 console_log!("Hub task vastaanotettu, ajetaan GPU:lla...");
                 let ws_for_async = ws_clone.clone();