Add Raindrop observability integration

extensions/cli/package.json

@@ -45,6 +45,8 @@
   },
   "homepage": "https://continue.dev",
   "dependencies": {
+    "@raindrop-ai/ai-sdk": "^0.0.15",
+    "ai": "^6.0.86",
     "@sentry/profiling-node": "^9.47.1",
     "fdir": "^6.4.2",
     "find-up": "^8.0.0",

extensions/cli/src/services/index.ts

@@ -332,6 +332,10 @@ export async function initializeServices(initOptions: ServiceInitOptions = {}) {
     [], // No dependencies
   );
 
+  // Initialize Raindrop observability before services create LLM API instances
+  const raindropService = await import("../telemetry/raindropService.js");
+  await raindropService.initialize();
+
   // Eagerly initialize all services to ensure they're ready when needed
   // This avoids race conditions and "service not ready" errors
   await serviceContainer.initializeAll();

extensions/cli/src/smoke-api/headless-all-tools.test.ts

@@ -8,7 +8,7 @@ import {
   createSmokeContext,
   cleanupSmokeContext,
   writeAnthropicConfig,
-  runHeadless,
+  runHeadlessWithRetry,
   type SmokeTestContext,
 } from "./smoke-api-helpers.js";
 
@@ -62,7 +62,7 @@ describe.skipIf(!ANTHROPIC_API_KEY)(
 After all tools complete, say exactly "ALL_TOOLS_DONE" as your final message.`;
 
     it("should exercise 10 built-in tools and complete successfully", async () => {
-      const result = await runHeadless(
+      const result = await runHeadlessWithRetry(
         ctx,
         ["-p", "--auto", "--config", ctx.configPath, PROMPT],
         { timeout: 120000 },

extensions/cli/src/smoke-api/headless-anthropic.test.ts

@@ -4,7 +4,7 @@ import {
   createSmokeContext,
   cleanupSmokeContext,
   writeAnthropicConfig,
-  runHeadless,
+  runHeadlessWithRetry,
   type SmokeTestContext,
 } from "./smoke-api-helpers.js";
 
@@ -25,7 +25,7 @@ describe.skipIf(!ANTHROPIC_API_KEY)(
     });
 
     it("should complete a round-trip and return a response", async () => {
-      const result = await runHeadless(ctx, [
+      const result = await runHeadlessWithRetry(ctx, [
         "-p",
         "--config",
         ctx.configPath,

extensions/cli/src/smoke-api/headless-continue-proxy.test.ts

@@ -4,7 +4,7 @@ import {
   createSmokeContext,
   cleanupSmokeContext,
   writeContinueProxyConfig,
-  runHeadless,
+  runHeadlessWithRetry,
   type SmokeTestContext,
 } from "./smoke-api-helpers.js";
 
@@ -29,7 +29,7 @@ describe.skipIf(!CONTINUE_API_KEY || !SMOKE_PROXY_MODEL)(
     });
 
     it("should complete a round-trip and return a response", async () => {
-      const result = await runHeadless(ctx, [
+      const result = await runHeadlessWithRetry(ctx, [
         "-p",
         "--config",
         ctx.configPath,

extensions/cli/src/smoke-api/serve-anthropic.test.ts

@@ -7,7 +7,8 @@ import {
   writeAnthropicConfig,
   spawnServe,
   waitForPattern,
-  pollUntilIdle,
+  sendMessageAndWait,
+  getLastAssistantContent,
   shutdownServe,
   type SmokeTestContext,
 } from "./smoke-api-helpers.js";
@@ -45,37 +46,14 @@ describe.skipIf(!ANTHROPIC_API_KEY)(
       // Wait for the server to start
       await waitForPattern(proc, "Server started", 30000);
 
-      // Send a message
-      const msgRes = await fetch(`${baseUrl}/message`, {
-        method: "POST",
-        headers: { "Content-Type": "application/json" },
-        body: JSON.stringify({
-          message: "Reply with exactly the word 'hello' and nothing else.",
-        }),
-      });
-      expect(msgRes.ok).toBe(true);
-
-      // Poll until the agent finishes processing
-      const state = await pollUntilIdle(baseUrl, 60000);
-
-      // State shape: { session: { history: ChatHistoryItem[] }, ... }
-      // Each ChatHistoryItem has { message: { role, content }, ... }
-      const history: any[] = state.session?.history ?? [];
-      const assistantItems = history.filter(
-        (item: any) => item.message?.role === "assistant",
+      // Send a message (retries with backoff on rate-limit errors)
+      const state = await sendMessageAndWait(
+        baseUrl,
+        "Reply with exactly the word 'hello' and nothing else.",
       );
-      expect(assistantItems.length).toBeGreaterThan(0);
-
-      const lastMsg = assistantItems[assistantItems.length - 1].message;
-      const content =
-        typeof lastMsg.content === "string"
-          ? lastMsg.content
-          : lastMsg.content
-              ?.filter((p: any) => p.type === "text")
-              .map((p: any) => p.text)
-              .join("");
 
-      expect(content?.toLowerCase()).toContain("hello");
+      const content = getLastAssistantContent(state);
+      expect(content).toContain("hello");
 
       // Graceful exit
       const exitRes = await fetch(`${baseUrl}/exit`, { method: "POST" });