feat(tasks): wire decompose_and_store, list_subtasks_cmd, complete_subtask_cmd; add llm_engine to AppState
This commit is contained in:
@@ -4,10 +4,16 @@
|
||||
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
use uuid::Uuid;
|
||||
|
||||
use kon_storage::{
|
||||
complete_subtask_and_check_parent as db_complete_subtask,
|
||||
complete_task as db_complete_task,
|
||||
delete_task as db_delete_task,
|
||||
get_task_by_id as db_get_task,
|
||||
insert_subtask as db_insert_subtask,
|
||||
insert_task as db_insert_task,
|
||||
list_subtasks as db_list_subtasks,
|
||||
list_tasks as db_list_tasks,
|
||||
TaskRow,
|
||||
};
|
||||
@@ -115,3 +121,50 @@ pub async fn uncomplete_task_cmd(
|
||||
.map(|_| ())
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
pub async fn decompose_and_store(
|
||||
state: tauri::State<'_, AppState>,
|
||||
parent_task_id: String,
|
||||
) -> Result<Vec<TaskDto>, String> {
|
||||
let parent = db_get_task(&state.db, &parent_task_id)
|
||||
.await
|
||||
.map_err(|e| e.to_string())?
|
||||
.ok_or_else(|| format!("Task {parent_task_id} not found"))?;
|
||||
|
||||
let steps = state.llm_engine.decompose_task(&parent.text)?;
|
||||
|
||||
let mut created = Vec::new();
|
||||
for text in steps {
|
||||
let id = Uuid::new_v4().to_string();
|
||||
db_insert_subtask(&state.db, &id, &text, &parent_task_id)
|
||||
.await
|
||||
.map_err(|e| e.to_string())?;
|
||||
if let Some(row) = db_get_task(&state.db, &id).await.map_err(|e| e.to_string())? {
|
||||
created.push(TaskDto::from(row));
|
||||
}
|
||||
}
|
||||
|
||||
Ok(created)
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
pub async fn list_subtasks_cmd(
|
||||
state: tauri::State<'_, AppState>,
|
||||
parent_task_id: String,
|
||||
) -> Result<Vec<TaskDto>, String> {
|
||||
db_list_subtasks(&state.db, &parent_task_id)
|
||||
.await
|
||||
.map(|rows| rows.into_iter().map(TaskDto::from).collect())
|
||||
.map_err(|e| e.to_string())
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
pub async fn complete_subtask_cmd(
|
||||
state: tauri::State<'_, AppState>,
|
||||
subtask_id: String,
|
||||
) -> Result<(), String> {
|
||||
db_complete_subtask(&state.db, &subtask_id)
|
||||
.await
|
||||
.map_err(|e| e.to_string())
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user