pi-coralbricks-provider 1.0.0

package/models.json

@@ -0,0 +1,133 @@
+[
+  {
+    "id": "glm-5.2-fp4",
+    "name": "GLM 5.2 FP4",
+    "reasoning": true,
+    "input": [
+      "text"
+    ],
+    "cost": {
+      "input": 1.12,
+      "output": 4.4,
+      "cacheRead": 0,
+      "cacheWrite": 0
+    },
+    "contextWindow": 1048576,
+    "maxTokens": 131072,
+    "thinkingLevelMap": {
+      "off": "none",
+      "minimal": null,
+      "low": null,
+      "medium": null,
+      "high": "high",
+      "xhigh": null,
+      "max": "max"
+    },
+    "compat": {
+      "supportsStore": false,
+      "supportsDeveloperRole": false,
+      "supportsReasoningEffort": true,
+      "maxTokensField": "max_tokens",
+      "thinkingFormat": "zai"
+    }
+  },
+  {
+    "id": "glm-5.3-fp4",
+    "name": "GLM 5.3 FP4",
+    "reasoning": true,
+    "input": [
+      "text"
+    ],
+    "cost": {
+      "input": 1.12,
+      "output": 4.4,
+      "cacheRead": 0,
+      "cacheWrite": 0
+    },
+    "contextWindow": 1048576,
+    "maxTokens": 131072,
+    "thinkingLevelMap": {
+      "off": "none",
+      "minimal": null,
+      "low": "low",
+      "medium": null,
+      "high": "high",
+      "xhigh": null,
+      "max": "max"
+    },
+    "compat": {
+      "supportsStore": false,
+      "supportsDeveloperRole": false,
+      "supportsReasoningEffort": true,
+      "maxTokensField": "max_tokens",
+      "thinkingFormat": "zai"
+    }
+  },
+  {
+    "id": "gpt-oss-120b",
+    "name": "GPT-OSS 120B",
+    "reasoning": true,
+    "input": [
+      "text"
+    ],
+    "cost": {
+      "input": 0.12,
+      "output": 0.6,
+      "cacheRead": 0,
+      "cacheWrite": 0
+    },
+    "contextWindow": 131072,
+    "maxTokens": 40960,
+    "thinkingLevelMap": {
+      "off": null,
+      "minimal": null,
+      "low": "low",
+      "medium": "medium",
+      "high": "high",
+      "xhigh": null,
+      "max": null
+    },
+    "compat": {
+      "supportsStore": false,
+      "supportsDeveloperRole": false,
+      "supportsReasoningEffort": true,
+      "maxTokensField": "max_tokens",
+      "thinkingFormat": "openai"
+    }
+  },
+  {
+    "id": "kimi-k3",
+    "name": "Kimi K3",
+    "reasoning": true,
+    "input": [
+      "text",
+      "image"
+    ],
+    "cost": {
+      "input": 3,
+      "output": 15,
+      "cacheRead": 0,
+      "cacheWrite": 0
+    },
+    "contextWindow": 1048576,
+    "maxTokens": 131072,
+    "thinkingLevelMap": {
+      "off": null,
+      "minimal": null,
+      "low": "low",
+      "medium": null,
+      "high": "high",
+      "xhigh": null,
+      "max": "max"
+    },
+    "compat": {
+      "supportsStore": false,
+      "supportsDeveloperRole": false,
+      "supportsReasoningEffort": true,
+      "maxTokensField": "max_tokens",
+      "supportsStrictMode": false,
+      "thinkingFormat": "openai",
+      "requiresReasoningContentOnAssistantMessages": true
+    }
+  }
+]

package/package.json

@@ -0,0 +1,59 @@
+{
+  "name": "pi-coralbricks-provider",
+  "version": "1.0.0",
+  "description": "CoralBricks provider extension for pi - Access GLM 5.2/5.3, Kimi K3, and GPT-OSS 120B through the Coral Inference API",
+  "author": "monotykamary",
+  "homepage": "https://github.com/monotykamary/pi-coralbricks-provider#readme",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/monotykamary/pi-coralbricks-provider.git"
+  },
+  "bugs": {
+    "url": "https://github.com/monotykamary/pi-coralbricks-provider/issues"
+  },
+  "type": "module",
+  "main": "index.ts",
+  "scripts": {
+    "clean": "echo 'nothing to clean'",
+    "build": "echo 'nothing to build'",
+    "check": "tsc --noEmit && vitest run",
+    "test": "vitest run",
+    "update-models": "node scripts/update-models.js"
+  },
+  "keywords": [
+    "pi",
+    "extension",
+    "provider",
+    "coralbricks",
+    "coral",
+    "ai",
+    "llm",
+    "glm",
+    "kimi",
+    "gpt-oss"
+  ],
+  "license": "MIT",
+  "files": [
+    "index.ts",
+    "models.json",
+    "deprecated-models.json",
+    "custom-models.json",
+    "patch.json",
+    "scripts/update-models.js",
+    "scripts/probe-stream-fix.ts",
+    "README.md",
+    "LICENSE"
+  ],
+  "devDependencies": {
+    "@earendil-works/pi-ai": "0.84.4",
+    "@earendil-works/pi-coding-agent": "0.84.4",
+    "@types/node": "^22.20.0",
+    "typescript": "6.0.3",
+    "vitest": "4.1.7"
+  },
+  "pi": {
+    "extensions": [
+      "./index.ts"
+    ]
+  }
+}

package/patch.json

@@ -0,0 +1 @@
+{}

package/scripts/probe-stream-fix.ts

@@ -0,0 +1,62 @@
+/**
+ * Live probe: stream gpt-oss-120b through streamCoral (the extension's
+ * streamSimple path) and verify the tool-call index bug is repaired — the
+ * accumulated message must contain exactly one complete tool call.
+ */
+import { streamCoral } from "../index.ts";
+import modelsData from "../models.json" with { type: "json" };
+
+const apiKey = process.env.CORALBRICKS_API_KEY;
+if (!apiKey) throw new Error("CORALBRICKS_API_KEY not set");
+
+const modelId = process.argv[2] ?? "gpt-oss-120b";
+const model = {
+  ...(modelsData as any[]).find((m) => m.id === modelId)!,
+  provider: "coralbricks",
+  baseUrl: "https://inference.coralbricks.ai/v1",
+  api: "openai-completions",
+};
+
+const context = {
+  systemPrompt: "You are a helpful assistant.",
+  messages: [
+    { role: "user", content: [{ type: "text", text: "What is the weather in Paris? Use the tool." }] },
+  ],
+  tools: [
+    {
+      name: "get_weather",
+      description: "Get current weather for a location",
+      parameters: {
+        type: "object",
+        properties: { location: { type: "string", description: "City name" } },
+        required: ["location"],
+      } as any,
+    },
+  ],
+};
+
+const events = streamCoral(model, context, { apiKey, reasoning: "high" } as any);
+let finalMessage: any = null;
+for await (const event of events) {
+  if (event.type === "done") finalMessage = event.message;
+  if (event.type === "error") finalMessage = event.error;
+}
+
+if (!finalMessage) throw new Error("no final message");
+if (finalMessage.stopReason === "error") {
+  throw new Error("stream error: " + finalMessage.errorMessage);
+}
+
+const calls = (finalMessage.content ?? []).filter((c: any) => c.type === "toolCall");
+console.log("stopReason:", finalMessage.stopReason);
+console.log("toolCalls:", calls.length);
+for (const call of calls) {
+  console.log("  -", call.name, JSON.stringify(call.arguments));
+}
+
+if (calls.length !== 1) throw new Error(`expected exactly 1 tool call, got ${calls.length}`);
+const args = typeof calls[0].arguments === "string" ? JSON.parse(calls[0].arguments) : calls[0].arguments;
+if (!args.location || !/paris/i.test(String(args.location))) {
+  throw new Error("arguments malformed: " + JSON.stringify(calls[0].arguments));
+}
+console.log("PASS: streamCoral produced exactly one complete tool call");