Function process_model_queue_item

src/worker.rs:169–234

Process a single model-type queue item: build prompt, call LLM, update row.

(item: &QueueItem)

/// Process a single model-type queue item: build prompt, call LLM, update row.
fn process_model_queue_item(item: &QueueItem) -> Result<(), String> {
    let api_key = crate::GUC_API_KEY
        .get()
        .ok_or("pgclaw.api_key not set")?
        .to_str()
        .map_err(|_| "pgclaw.api_key is not valid UTF-8")?
        .to_string();

    let provider = crate::get_provider();
    let api_url = crate::get_api_url();

    // Resolve agent config
    let (model, system_prompt) = resolve_agent_config(item)?;

    // Load conversation history
    let history = load_row_history(item);

    // Build user message with row context
    let user_content = format!(
        "Event: {} on table (oid={}), row pk={}.\n\nCurrent row state:\n{}\n\nRespond with a JSON object of columns to update, e.g. {{\"column\": \"value\"}}.\nOr {{}} if no changes needed.\nTo update agent memory, include {{\"__memory\": {{...}}}}.",
        item.event, item.table_oid.as_u32(), item.pk_value, item.row_data
    );

    // Build messages array
    let mut messages: Vec<http::Message> = Vec::new();
    for h in &history {
        messages.push(h.clone());
    }
    messages.push(http::Message {
        role: "user".to_string(),
        content: user_content.clone(),
    });

    // Call LLM
    let response = http::call_llm(&provider, &api_key, api_url.as_deref(), &model, &system_prompt, &messages, 1024)?;

    // Record in history
    BackgroundWorker::transaction(|| {
        Spi::run(&format!(
            "INSERT INTO claw.history (table_oid, pk_value, claw_col, role, content)
             VALUES ({}, '{}', '{}', 'user', '{}')",
            item.table_oid.as_u32(),
            item.pk_value.replace('\'', "''"),
            item.claw_col.replace('\'', "''"),
            user_content.replace('\'', "''")
        ))
        .expect("SPI failed");

        Spi::run(&format!(
            "INSERT INTO claw.history (table_oid, pk_value, claw_col, role, content)
             VALUES ({}, '{}', '{}', 'assistant', '{}')",
            item.table_oid.as_u32(),
            item.pk_value.replace('\'', "''"),
            item.claw_col.replace('\'', "''"),
            response.replace('\'', "''")
        ))
        .expect("SPI failed");
    });