pi-omlx-picker 0.2.3 → 0.2.4

package/README.md

@@ -2,7 +2,8 @@
 
 > Seamlessly integrate your local [oMLX](https://github.com/jundot/omlx) models into Pi.
 
-This extension discovers models from a local OMLX server and registers them as native Pi providers. Switch between your local and remote models effortlessly using Pi's built-in `/model` command.
+This extension discovers models from a local OMLX server and registers them as native Pi providers.
+Switch between your local and remote models effortlessly using Pi's built-in `/model` command.
 
 ![Pi OMLX Picker Demo](./.assets/demo.gif)
 
@@ -14,17 +15,17 @@ This extension discovers models from a local OMLX server and registers them as n
 
 ## 📦 Installation
 
-```sh
+```bash
 pi install npm:pi-omlx-picker
 ```
 
 ## 🛠️ Development
 
-- `npm install`
-- `npm run check`
-- `npm run format`
-- `npm test`
-- `npm run test:watch`
+* `npm install`
+* `npm run check`
+* `npm run format`
+* `npm test`
+* `npm run test:watch`
 
 ## 🚀 Quick Start
 
@@ -35,6 +36,7 @@ pi install npm:pi-omlx-picker
 
 The default base URL is `http://127.0.0.1:8000/v1`. Set `OMLX_BASE_URL` before starting Pi if your OMLX server uses a different URL.
 
+
 ## ⚙️ Configuration
 
 Env-var overrides, model metadata overlay, and stream timeout knobs are documented in [docs/CONFIGURATION.md](./docs/CONFIGURATION.md).

package/index.ts

@@ -6,6 +6,9 @@ import {
 } from "@earendil-works/pi-ai";
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { PROVIDER_KEY } from "./src/auth-storage.ts";
+import { loadDotenvFromExtensionDir } from "./src/dotenv.ts";
+
+loadDotenvFromExtensionDir(import.meta.url);
 import {
 	fetchModels,
 	type OmlxModel,

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "pi-omlx-picker",
-	"version": "0.2.3",
+	"version": "0.2.4",
 	"type": "module",
 	"description": "Pi extension that discovers models from a local OMLX server and registers them as a native Pi provider.",
 	"license": "MIT",

package/src/dotenv.ts

@@ -0,0 +1,49 @@
+import { existsSync, readFileSync } from "node:fs";
+import { dirname, join } from "node:path";
+import { fileURLToPath } from "node:url";
+
+export function parseDotenv(content: string): Record<string, string> {
+	const result: Record<string, string> = {};
+	for (const raw of content.split(/\r?\n/)) {
+		const line = raw.trim();
+		if (!line || line.startsWith("#")) continue;
+		const stripped = line.startsWith("export ") ? line.slice(7).trim() : line;
+		const eq = stripped.indexOf("=");
+		if (eq === -1) continue;
+		const key = stripped.slice(0, eq).trim();
+		if (!key) continue;
+		let val = stripped.slice(eq + 1).trim();
+		if (
+			(val.startsWith('"') && val.endsWith('"')) ||
+			(val.startsWith("'") && val.endsWith("'"))
+		) {
+			val = val.slice(1, -1);
+		}
+		result[key] = val;
+	}
+	return result;
+}
+
+export function mergeDotenv(
+	parsed: Record<string, string>,
+	env: NodeJS.ProcessEnv = process.env,
+): void {
+	for (const [key, val] of Object.entries(parsed)) {
+		if (env[key] === undefined || env[key] === "") env[key] = val;
+	}
+}
+
+export function loadDotenvFromExtensionDir(
+	importMetaUrl: string,
+	env: NodeJS.ProcessEnv = process.env,
+): void {
+	try {
+		const dir = dirname(fileURLToPath(importMetaUrl));
+		const dotenvPath = join(dir, ".env");
+		if (!existsSync(dotenvPath)) return;
+		const content = readFileSync(dotenvPath, "utf8");
+		mergeDotenv(parseDotenv(content), env);
+	} catch {
+		// non-fatal
+	}
+}