feat: lancedb vector tool selection (#2654)

Co-authored-by: Wendy Tang <wendytang@squareup.com> Co-authored-by: Alice Hau <ahau@squareup.com>
2025-05-28 23:23:02 -04:00
parent cf7bb08ee1
commit bf1c0d51e4
23 changed files with 3661 additions and 301 deletions
@@ -268,12 +268,16 @@ async fn remove_extension(
        .get_agent()
        .await
        .map_err(|_| StatusCode::PRECONDITION_FAILED)?;
-    agent.remove_extension(&name).await;
-
-    Ok(Json(ExtensionResponse {
-        error: false,
-        message: None,
-    }))
+    match agent.remove_extension(&name).await {
+        Ok(_) => Ok(Json(ExtensionResponse {
+            error: false,
+            message: None,
+        })),
+        Err(e) => Ok(Json(ExtensionResponse {
+            error: true,
+            message: Some(format!("Failed to remove extension: {:?}", e)),
+        })),
+    }
 }

 /// Registers the extension management routes with the Axum router.