@midpath/gptmitra-pi 0.0.0-stage → 1.4.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Midpath Software Pvt Ltd
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -1,3 +1,51 @@
-# Temporary Holding Version
+# GPT Mitra for Pi
 
-This version is a temporary placeholder for this package. An operational version to replace this has been submitted for review and is awaiting a staged release.
+Use GPT Mitra's OpenAI-compatible API in Pi with per-call supplier pricing and persisted session costs.
+
+## Install
+
+```sh
+pi install npm:@midpath/gptmitra-pi
+export GPTMITRA_API_KEY='your-consumer-key'
+```
+
+Restart Pi or run `/reload`. Add your model to `~/.pi/agent/models.json`:
+
+```json
+{
+  "providers": {
+    "gptmitra": {
+      "api": "gptmitra",
+      "baseUrl": "https://app.gptmitra.in/v1",
+      "models": [{
+        "id": "your-model-id",
+        "name": "Your model",
+        "input": ["text"],
+        "reasoning": false,
+        "contextWindow": 32768,
+        "maxTokens": 4096
+      }]
+    }
+  }
+}
+```
+
+Select the model under the `gptmitra` provider. Replace the example ID and capabilities with those supported by your supplier. A consumer key needs an enabled owned or shared supplier serving that model. See [GPT Mitra setup](https://gptmitra.in/docs.html).
+
+Keep `api: gptmitra`; using `openai-completions` bypasses cost handling. For another deployment, change `baseUrl`, retaining `/v1`.
+
+## Costs
+
+Uses the server's final cost, or complete token usage with the response's supplier rates. Missing usage is marked unavailable, not treated as free; Pi's numeric total excludes unavailable costs and the extension warns that the total is incomplete. Costs persist in Pi sessions. No local prices or additional requests are needed.
+
+Targets Pi's `@earendil-works` provider API; tested with Pi 1.4.0. This is not a budget or payment system.
+
+## Publish
+
+From the repository, after `npm login`:
+
+```sh
+./pi-extension/scripts/publish.sh
+```
+
+Runs tests and publishes publicly. The npm prepack hook syncs the version from root `VERSION.txt`; no Go build is required. Bump that file before subsequent releases. To preview without publishing, pass `--dry-run`.

package/index.ts

@@ -0,0 +1,85 @@
+import { createAssistantMessageEventStream, openAICompletionsApi, type Api, type Model, type SimpleStreamOptions, type TranscriptContext } from '@earendil-works/pi-ai/compat';
+import type { ExtensionAPI } from '@earendil-works/pi-coding-agent';
+
+const decimal = /^(0|[1-9]\d*)(\.\d{1,12})?$/;
+function money(value: unknown): number | undefined {
+  if (typeof value !== 'string' || !decimal.test(value)) return;
+  const n = Number(value);
+  return Number.isFinite(n) && n >= 0 ? n : undefined;
+}
+function tokens(n: unknown): n is number { return Number.isSafeInteger(n) && (n as number) >= 0; }
+
+// Every invocation owns its metadata. Never mutate model-wide prices or fetch.
+export function streamGPTMitra(model: Model<Api>, context: TranscriptContext, options?: SimpleStreamOptions) {
+  const stream = createAssistantMessageEventStream();
+  let inputRate: number | undefined, outputRate: number | undefined;
+  let counts: {input: number; output: number} | undefined;
+  let reported: {input: number; output: number; cacheRead: number; cacheWrite: number; total: number} | undefined;
+  const innerModel = {...model, api: 'openai-completions' as const, cost: {input: 0, output: 0, cacheRead: 0, cacheWrite: 0}};
+  const inner = openAICompletionsApi().streamSimple(innerModel, context, {
+    ...options,
+    onResponse: async (response, m) => {
+      const headers = new Headers(response.headers);
+      if (headers.get('X-GPTMitra-Currency') === 'USD') {
+        inputRate = money(headers.get('X-GPTMitra-Input-Per-Million'));
+        outputRate = money(headers.get('X-GPTMitra-Output-Per-Million'));
+      }
+      await options?.onResponse?.(response, m);
+    },
+    onProviderStreamEvent: async (event, m) => {
+      const u = (event as any)?.usage;
+      if (u) { counts = undefined; reported = undefined; }
+      if (tokens(u?.prompt_tokens) && tokens(u?.completion_tokens)) {
+        counts = {input: u.prompt_tokens, output: u.completion_tokens};
+        const c = u.gptmitra_cost;
+        if (c?.currency === 'USD') {
+          const input = money(c.input_cost), output = money(c.output_cost), total = money(c.total);
+          const close = (a: number, b: number) => Math.abs(a - b) <= Math.max(1e-12, Math.abs(b) * Number.EPSILON * 4);
+          if (input !== undefined && output !== undefined && total !== undefined && inputRate !== undefined && outputRate !== undefined
+              && close(input, counts.input * inputRate / 1e6) && close(output, counts.output * outputRate / 1e6) && close(input + output, total)) {
+            reported = {input, output, cacheRead: 0, cacheWrite: 0, total};
+          }
+        }
+      }
+      await options?.onProviderStreamEvent?.(event, m);
+    },
+  });
+  void (async () => {
+    for await (const event of inner) {
+      if (event.type === 'done' || event.type === 'error') {
+        const message = event.type === 'done' ? event.message : event.error;
+        const fallback = counts && inputRate !== undefined && outputRate !== undefined
+          ? {input: counts.input * inputRate / 1e6, output: counts.output * outputRate / 1e6, cacheRead: 0, cacheWrite: 0, total: (counts.input * inputRate + counts.output * outputRate) / 1e6}
+          : undefined;
+        message.usage.cost = reported ?? fallback ?? {input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0};
+        // NOTE: Pi requires numeric costs. Unknown calls are excluded, with a
+        // persisted status and explicit warning rather than claiming a free call.
+        (message.usage as any).gptmitraCostStatus = reported ? 'reported' : fallback ? 'calculated' : 'unavailable';
+        message.api = model.api;
+      }
+      stream.push(event);
+    }
+    stream.end();
+  })();
+  return stream;
+}
+
+export default function (pi: ExtensionAPI) {
+  pi.registerProvider('gptmitra', {
+    baseUrl: 'https://app.gptmitra.in/v1',
+    apiKey: '$GPTMITRA_API_KEY',
+    api: 'gptmitra',
+    streamSimple: streamGPTMitra,
+  });
+  // NOTE: O(n) scan per message keeps warnings consistent with native session
+  // totals across branches; use an incremental counter if huge sessions need it.
+  const update = (event: any, ctx: any) => {
+    const messages = ctx.sessionManager.getEntries().filter((e: any) => e.type === 'message').map((e: any) => e.message);
+    // message_end runs before the new message is appended to session storage.
+    if (event.message && !messages.includes(event.message)) messages.push(event.message);
+    const missing = messages.filter((m: any) => m.role === 'assistant' && m.usage?.gptmitraCostStatus === 'unavailable').length;
+    ctx.ui.setStatus('gptmitra', missing ? `GPT Mitra: ${missing} call(s) have unavailable cost; USD total is incomplete` : undefined);
+  };
+  pi.on('session_start', update);
+  pi.on('message_end', update);
+}

package/package.json

@@ -1,6 +1,29 @@
 {
   "name": "@midpath/gptmitra-pi",
-  "version": "0.0.0-stage",
-  "stub": true,
-  "description": "Temporary package placeholder for staged publishing"
-}
+  "version": "1.4.0",
+  "description": "GPT Mitra provider for Pi with per-call supplier pricing and persisted session costs",
+  "license": "MIT",
+  "author": "Midpath Software Pvt Ltd",
+  "publishConfig": {
+    "access": "public"
+  },
+  "files": [
+    "index.ts"
+  ],
+  "scripts": {
+    "test": "node test.cjs",
+    "prepack": "node sync-version.cjs"
+  },
+  "keywords": [
+    "pi-package"
+  ],
+  "pi": {
+    "extensions": [
+      "./index.ts"
+    ]
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-ai": "*",
+    "@earendil-works/pi-coding-agent": "*"
+  }
+}