naiad-cli 0.1.0

package/dist/sync/session-upload.js

@@ -0,0 +1,69 @@
+import { readFileSync, readdirSync, statSync } from "fs";
+import { join } from "path";
+import { createHash } from "crypto";
+const CHUNK_SIZE = 4 * 1024 * 1024; // 4 MiB
+export class SessionUploader {
+    client;
+    sessionId;
+    sessionDir;
+    pollTimer = null;
+    constructor(client, sessionId, sessionDir) {
+        this.client = client;
+        this.sessionId = sessionId;
+        this.sessionDir = sessionDir;
+    }
+    start() {
+        this.pollTimer = setInterval(() => this.checkAndUpload(), 10_000);
+    }
+    async stop() {
+        if (this.pollTimer) {
+            clearInterval(this.pollTimer);
+            this.pollTimer = null;
+        }
+        await this.checkAndUpload();
+    }
+    async checkAndUpload() {
+        try {
+            const files = readdirSync(this.sessionDir).filter((f) => f.endsWith(".jsonl"));
+            for (const file of files) {
+                await this.uploadFile(join(this.sessionDir, file));
+            }
+        }
+        catch {
+            // Session dir may not exist yet
+        }
+    }
+    async uploadFile(filePath) {
+        const stat = statSync(filePath);
+        if (stat.size === 0)
+            return;
+        const data = readFileSync(filePath);
+        const totalBytes = data.length;
+        const chunkCount = Math.ceil(totalBytes / CHUNK_SIZE);
+        const fileHash = createHash("sha256").update(data).digest("hex");
+        // Build chunks
+        const chunks = [];
+        for (let i = 0; i < chunkCount; i++) {
+            const start = i * CHUNK_SIZE;
+            const end = Math.min(start + CHUNK_SIZE, totalBytes);
+            const chunkData = data.subarray(start, end);
+            const chunkHash = createHash("sha256").update(chunkData).digest("hex");
+            chunks.push({ start, end, sha256_hex: chunkHash, data: chunkData });
+        }
+        // Get presigned URLs
+        const { urls } = await this.client.getUploadUrls(this.sessionId, chunks.map((c) => ({ start: c.start, end: c.end, sha256_hex: c.sha256_hex })));
+        // Upload chunks
+        for (let i = 0; i < urls.length; i++) {
+            const res = await fetch(urls[i].url, {
+                method: "PUT",
+                body: new Uint8Array(chunks[i].data),
+            });
+            if (!res.ok) {
+                throw new Error(`Chunk upload failed: ${res.status}`);
+            }
+        }
+        // Finalize
+        await this.client.finalizeSession(this.sessionId, totalBytes, chunkCount, fileHash);
+    }
+}
+//# sourceMappingURL=session-upload.js.map

package/dist/sync/session-upload.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-upload.js","sourceRoot":"","sources":["../../src/sync/session-upload.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,IAAI,CAAC;AACzD,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAGpC,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,QAAQ;AAE5C,MAAM,OAAO,eAAe;IAClB,MAAM,CAAY;IAClB,SAAS,CAAS;IAClB,UAAU,CAAS;IACnB,SAAS,GAA0C,IAAI,CAAC;IAEhE,YAAY,MAAiB,EAAE,SAAiB,EAAE,UAAkB;QAClE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAED,KAAK;QACH,IAAI,CAAC,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,MAAM,CAAC,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC9B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACxB,CAAC;QACD,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;IAC9B,CAAC;IAEO,KAAK,CAAC,cAAc;QAC1B,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;YAC/E,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC;YACrD,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,gCAAgC;QAClC,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,QAAgB;QACvC,MAAM,IAAI,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAChC,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC;YAAE,OAAO;QAE5B,MAAM,IAAI,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC;QAC/B,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC,CAAC;QACtD,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAEjE,eAAe;QACf,MAAM,MAAM,GAA4E,EAAE,CAAC;QAC3F,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG,CAAC,GAAG,UAAU,CAAC;YAC7B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,UAAU,EAAE,UAAU,CAAC,CAAC;YACrD,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;YAC5C,MAAM,SAAS,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACvE,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;QACtE,CAAC;QAED,qBAAqB;QACrB,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,aAAa,CAC9C,IAAI,CAAC,SAAS,EACd,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,GAAG,EAAE,UAAU,EAAE,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC,CAC9E,CAAC;QAEF,gBAAgB;QAChB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE;gBACnC,MAAM,EAAE,KAAK;gBACb,IAAI,EAAE,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;aACrC,CAAC,CAAC;YACH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,wBAAwB,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;QAED,WAAW;QACX,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;IACtF,CAAC;CACF"}

package/dist/utils/package-root.d.ts

@@ -0,0 +1 @@
+export declare function findPackageRoot(): string;

package/dist/utils/package-root.js

@@ -0,0 +1,13 @@
+import { dirname, join } from "path";
+import { fileURLToPath } from "url";
+import { existsSync } from "fs";
+export function findPackageRoot() {
+    let dir = dirname(fileURLToPath(import.meta.url));
+    while (dir !== dirname(dir)) {
+        if (existsSync(join(dir, "package.json")))
+            return dir;
+        dir = dirname(dir);
+    }
+    throw new Error("Cannot find package root");
+}
+//# sourceMappingURL=package-root.js.map

package/dist/utils/package-root.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"package-root.js","sourceRoot":"","sources":["../../src/utils/package-root.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAEhC,MAAM,UAAU,eAAe;IAC7B,IAAI,GAAG,GAAG,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAClD,OAAO,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QAC5B,IAAI,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;YAAE,OAAO,GAAG,CAAC;QACtD,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;IACrB,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;AAC9C,CAAC"}

package/extensions/naiad-extension.ts

@@ -0,0 +1,330 @@
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import {
+  type AssistantMessage,
+  type AssistantMessageEventStream,
+  type Context,
+  type Model,
+  type SimpleStreamOptions,
+  type ToolCall,
+  createAssistantMessageEventStream,
+} from "@mariozechner/pi-ai";
+
+export default function (pi: ExtensionAPI) {
+  const inferenceUrl = process.env.NAIAD_INFERENCE_URL;
+  const apiKey = process.env.NAIAD_API_KEY ?? "";
+  const model = process.env.NAIAD_MODEL ?? "llama-3.1-8b";
+  const threadId = process.env.NAIAD_THREAD_ID ?? "";
+  const sessionId = process.env.NAIAD_SESSION_ID ?? "";
+
+  if (!inferenceUrl) {
+    console.error("[naiad-extension] NAIAD_INFERENCE_URL not set, skipping provider registration");
+    return;
+  }
+
+  // Event forwarding via HTTP callback (for interactive mode)
+  const callbackUrl = process.env.NAIAD_CALLBACK_URL;
+  if (callbackUrl && typeof (pi as any).on === "function") {
+    const postEvent = (type: string, data: unknown) => {
+      fetch(callbackUrl, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ type, data }),
+      }).catch(() => {
+        // Best effort — don't crash the agent if callback fails
+      });
+    };
+
+    (pi as any).on("agent_start", (event: unknown) => postEvent("agent_start", event));
+    (pi as any).on("agent_end", (event: unknown) => postEvent("agent_end", event));
+    (pi as any).on("turn_start", (event: unknown) => postEvent("turn_start", event));
+    (pi as any).on("turn_end", (event: unknown) => postEvent("turn_end", event));
+    (pi as any).on("message_start", (event: unknown) => postEvent("message_start", event));
+    (pi as any).on("message_end", (event: unknown) => postEvent("message_end", event));
+    (pi as any).on("message_update", (event: unknown) => postEvent("message_update", event));
+    (pi as any).on("tool_execution_start", (event: unknown) => postEvent("tool_execution_start", event));
+    (pi as any).on("tool_execution_end", (event: unknown) => postEvent("tool_execution_end", event));
+  }
+
+  pi.registerProvider("naiad-proxy", {
+    baseUrl: inferenceUrl,
+    apiKey: "NAIAD_API_KEY",
+    api: "naiad-openai",
+    models: [
+      {
+        id: model,
+        name: `Naiad Model (${model})`,
+        input: ["text"],
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+        contextWindow: 131072,
+        maxTokens: 8192,
+      },
+    ],
+    streamSimple: streamNaiad,
+  });
+
+  function convertMessages(context: Context): any[] {
+    const messages: any[] = [];
+
+    if (context.systemPrompt) {
+      messages.push({ role: "system", content: context.systemPrompt });
+    }
+
+    for (const msg of context.messages) {
+      if (msg.role === "user") {
+        const content =
+          typeof msg.content === "string"
+            ? msg.content
+            : msg.content
+                .filter((c): c is { type: "text"; text: string } => c.type === "text")
+                .map((c) => c.text)
+                .join("\n");
+        messages.push({ role: "user" as const, content });
+      } else if (msg.role === "assistant") {
+        const assistantMsg: any = { role: "assistant", content: null };
+
+        const textBlocks = msg.content.filter((b): b is { type: "text"; text: string } => b.type === "text");
+        if (textBlocks.length > 0) {
+          assistantMsg.content = textBlocks.map((b) => b.text).join("");
+        }
+
+        const toolCalls = msg.content.filter((b): b is ToolCall => b.type === "toolCall");
+        if (toolCalls.length > 0) {
+          assistantMsg.tool_calls = toolCalls.map((tc) => ({
+            id: tc.id,
+            type: "function",
+            function: {
+              name: tc.name,
+              arguments: JSON.stringify(tc.arguments),
+            },
+          }));
+        }
+
+        // Skip empty assistant messages
+        if (assistantMsg.content === null && !assistantMsg.tool_calls) continue;
+        messages.push(assistantMsg);
+      } else if (msg.role === "toolResult") {
+        const textResult = msg.content
+          .filter((c): c is { type: "text"; text: string } => c.type === "text")
+          .map((c) => c.text)
+          .join("\n");
+        messages.push({
+          role: "tool",
+          tool_call_id: msg.toolCallId,
+          content: textResult || "(no output)",
+        });
+      }
+    }
+
+    return messages;
+  }
+
+  function convertTools(context: Context): any[] | undefined {
+    if (!context.tools || context.tools.length === 0) return undefined;
+    return context.tools.map((tool) => ({
+      type: "function",
+      function: {
+        name: tool.name,
+        description: tool.description,
+        parameters: tool.parameters,
+      },
+    }));
+  }
+
+  function streamNaiad(
+    mdl: Model,
+    context: Context,
+    options?: SimpleStreamOptions,
+  ): AssistantMessageEventStream {
+    const stream = createAssistantMessageEventStream();
+
+    (async () => {
+      const output: AssistantMessage = {
+        role: "assistant",
+        content: [],
+        api: mdl.api,
+        provider: mdl.provider,
+        model: mdl.id,
+        usage: {
+          input: 0,
+          output: 0,
+          cacheRead: 0,
+          cacheWrite: 0,
+          totalTokens: 0,
+          cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+        },
+        stopReason: "stop",
+        timestamp: Date.now(),
+      };
+
+      try {
+        const messages = convertMessages(context);
+        const tools = convertTools(context);
+
+        const body: any = {
+          model: mdl.id,
+          messages,
+          stream: true,
+          stream_options: { include_usage: true },
+        };
+        if (tools) {
+          body.tools = tools;
+        }
+
+        const res = await fetch(`${inferenceUrl}/v1/chat/completions`, {
+          method: "POST",
+          headers: {
+            "Content-Type": "application/json",
+            "Authorization": `Bearer ${apiKey}`,
+            "X-Naiad-Thread-Id": threadId,
+            "X-Naiad-Session-Id": sessionId,
+          },
+          body: JSON.stringify(body),
+          signal: options?.signal,
+        });
+
+        if (!res.ok) {
+          throw new Error(`Inference proxy error: ${res.status} ${await res.text()}`);
+        }
+
+        stream.push({ type: "start", partial: output });
+
+        const reader = res.body!.getReader();
+        const decoder = new TextDecoder();
+        let buffer = "";
+
+        // Track current content block for proper event sequencing
+        let currentBlock: any = null;
+        const blocks = output.content;
+        const blockIndex = () => blocks.length - 1;
+
+        const finishCurrentBlock = (block: any) => {
+          if (!block) return;
+          if (block.type === "text") {
+            stream.push({
+              type: "text_end",
+              contentIndex: blockIndex(),
+              content: block.text,
+              partial: output,
+            });
+          } else if (block.type === "toolCall") {
+            try {
+              block.arguments = JSON.parse(block.partialArgs || "{}");
+            } catch {
+              block.arguments = {};
+            }
+            delete block.partialArgs;
+            stream.push({
+              type: "toolcall_end",
+              contentIndex: blockIndex(),
+              toolCall: block as ToolCall,
+              partial: output,
+            });
+          }
+        };
+
+        while (true) {
+          const { done, value } = await reader.read();
+          if (done) break;
+          buffer += decoder.decode(value, { stream: true });
+
+          const lines = buffer.split("\n");
+          buffer = lines.pop() ?? "";
+
+          for (const line of lines) {
+            if (!line.startsWith("data: ") || line === "data: [DONE]") continue;
+            try {
+              const chunk = JSON.parse(line.slice(6));
+
+              if (chunk.usage) {
+                output.usage.input = chunk.usage.prompt_tokens ?? 0;
+                output.usage.output = chunk.usage.completion_tokens ?? 0;
+                output.usage.totalTokens = chunk.usage.total_tokens ?? 0;
+              }
+
+              const choice = chunk.choices?.[0];
+              if (!choice) continue;
+
+              if (choice.finish_reason) {
+                if (choice.finish_reason === "tool_calls" || choice.finish_reason === "function_call") {
+                  output.stopReason = "toolUse";
+                } else if (choice.finish_reason === "length") {
+                  output.stopReason = "length";
+                } else {
+                  output.stopReason = "stop";
+                }
+              }
+
+              // Handle text content delta
+              const delta = choice.delta?.content;
+              if (delta) {
+                if (!currentBlock || currentBlock.type !== "text") {
+                  finishCurrentBlock(currentBlock);
+                  currentBlock = { type: "text", text: "" };
+                  output.content.push(currentBlock);
+                  stream.push({ type: "text_start", contentIndex: blockIndex(), partial: output });
+                }
+                currentBlock.text += delta;
+                stream.push({ type: "text_delta", contentIndex: blockIndex(), delta, partial: output });
+              }
+
+              // Handle tool call deltas
+              if (choice.delta?.tool_calls) {
+                for (const toolCall of choice.delta.tool_calls) {
+                  if (
+                    !currentBlock ||
+                    currentBlock.type !== "toolCall" ||
+                    (toolCall.id && currentBlock.id !== toolCall.id)
+                  ) {
+                    finishCurrentBlock(currentBlock);
+                    currentBlock = {
+                      type: "toolCall",
+                      id: toolCall.id || "",
+                      name: toolCall.function?.name || "",
+                      arguments: {},
+                      partialArgs: "",
+                    };
+                    output.content.push(currentBlock);
+                    stream.push({ type: "toolcall_start", contentIndex: blockIndex(), partial: output });
+                  }
+                  if (currentBlock.type === "toolCall") {
+                    if (toolCall.id) currentBlock.id = toolCall.id;
+                    if (toolCall.function?.name) currentBlock.name = toolCall.function.name;
+                    let argDelta = "";
+                    if (toolCall.function?.arguments) {
+                      argDelta = toolCall.function.arguments;
+                      currentBlock.partialArgs += argDelta;
+                      try {
+                        currentBlock.arguments = JSON.parse(currentBlock.partialArgs);
+                      } catch {
+                        // partial JSON, keep accumulating
+                      }
+                    }
+                    stream.push({
+                      type: "toolcall_delta",
+                      contentIndex: blockIndex(),
+                      delta: argDelta,
+                      partial: output,
+                    });
+                  }
+                }
+              }
+            } catch {
+              // skip unparseable lines
+            }
+          }
+        }
+
+        finishCurrentBlock(currentBlock);
+        stream.push({ type: "done", reason: output.stopReason as "stop" | "length" | "toolUse", message: output });
+        stream.end();
+      } catch (error) {
+        output.stopReason = options?.signal?.aborted ? "aborted" : "error";
+        output.errorMessage = error instanceof Error ? error.message : String(error);
+        stream.push({ type: "error", reason: output.stopReason, error: output });
+        stream.end();
+      }
+    })();
+
+    return stream;
+  }
+}

package/package.json

@@ -0,0 +1,31 @@
+{
+  "name": "naiad-cli",
+  "version": "0.1.0",
+  "type": "module",
+  "bin": {
+    "naiad": "./dist/index.js"
+  },
+  "files": [
+    "dist/",
+    "extensions/"
+  ],
+  "engines": {
+    "node": ">=20"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsx src/index.ts",
+    "test": "tsx --test src/callback/server.test.ts"
+  },
+  "dependencies": {
+    "@mariozechner/pi-coding-agent": "^0.55.1"
+  },
+  "devDependencies": {
+    "@types/node": "^22.0.0",
+    "tsx": "^4.0.0",
+    "typescript": "^5.7.0"
+  }
+}