From 03e8f246d11205a6f5844e9248e307a067627d84 Mon Sep 17 00:00:00 2001
From: Agent Zero <agent@ingwaz.work>
Date: Thu, 19 Mar 2026 15:59:34 +0000
Subject: [PATCH] Implement batch_store endpoint for Issue #12

---
 src/db.rs                |  4 +++-
 src/tools/batch_store.rs |  2 +-
 src/transport.rs         | 15 ---------------
 3 files changed, 4 insertions(+), 17 deletions(-)

diff --git a/src/db.rs b/src/db.rs
index 748ee4e..8e810bf 100644
--- a/src/db.rs
+++ b/src/db.rs
@@ -9,6 +9,8 @@ use tokio_postgres::NoTls;
 use tracing::info;
 use uuid::Uuid;
 
+use serde::Serialize;
+use serde_json::Value;
 use crate::config::DatabaseConfig;
 
 /// Database wrapper with connection pool
@@ -177,7 +179,7 @@ impl Database {
 
 
 /// Result for a single batch entry
-#[derive(Debug, Clone)]
+#[derive(Debug, Clone, Serialize)]
 pub struct BatchStoreResult {
     pub id: String,
     pub status: String,
diff --git a/src/tools/batch_store.rs b/src/tools/batch_store.rs
index 1397387..cb190b5 100644
--- a/src/tools/batch_store.rs
+++ b/src/tools/batch_store.rs
@@ -6,7 +6,7 @@
 use anyhow::{Context, Result, anyhow};
 use serde_json::Value;
 use std::sync::Arc;
-use tracing::{info, warn};
+use tracing::info;
 
 use crate::embedding::extract_keywords;
 use crate::AppState;
diff --git a/src/transport.rs b/src/transport.rs
index a88b7f0..12b2414 100644
--- a/src/transport.rs
+++ b/src/transport.rs
@@ -188,21 +188,6 @@ async fn sse_handler(
             .event("endpoint")
             .data(endpoint));
 
-        // Send initial tools list so Agent Zero knows what's available
-        let tools_response = JsonRpcResponse {
-            jsonrpc: "2.0".to_string(),
-            id: serde_json::json!("initial-tools"),
-            result: Some(serde_json::json!({
-                "tools": tools::get_tool_definitions()
-            })),
-            error: None,
-        };
-
-        yield Ok(Event::default()
-            .event("message")
-            .json_data(&tools_response)
-            .unwrap());
-
         loop {
             tokio::select! {
                 maybe_message = session_rx.recv() => {