@oh-my-pi/pi-ai 0.1.0

package/package.json

@@ -0,0 +1,60 @@
+{
+	"name": "@oh-my-pi/pi-ai",
+	"version": "0.1.0",
+	"description": "Unified LLM API with automatic model discovery and provider configuration",
+	"type": "module",
+	"main": "./src/index.ts",
+	"types": "./src/index.ts",
+	"exports": {
+		".": "./src/index.ts",
+		"./utils/json-parse": "./src/utils/json-parse.ts"
+	},
+	"bin": {
+		"pi-ai": "./src/cli.ts"
+	},
+	"files": [
+		"src",
+		"README.md"
+	],
+	"scripts": {
+		"generate-models": "bun scripts/generate-models.ts",
+		"test": "vitest --run",
+		"prepublishOnly": "bun run generate-models"
+	},
+	"dependencies": {
+		"@anthropic-ai/sdk": "0.71.2",
+		"@google/genai": "1.34.0",
+		"@mistralai/mistralai": "1.10.0",
+		"@sinclair/typebox": "^0.34.41",
+		"ajv": "^8.17.1",
+		"ajv-formats": "^3.0.1",
+		"chalk": "^5.6.2",
+		"openai": "6.10.0",
+		"partial-json": "^0.1.7",
+		"zod-to-json-schema": "^3.24.6"
+	},
+	"keywords": [
+		"ai",
+		"llm",
+		"openai",
+		"anthropic",
+		"gemini",
+		"unified",
+		"api"
+	],
+	"author": "Mario Zechner",
+	"license": "MIT",
+	"repository": {
+		"type": "git",
+		"url": "git+https://github.com/can1357/oh-my-pi.git",
+		"directory": "packages/ai"
+	},
+	"engines": {
+		"bun": ">=1.0.0"
+	},
+	"devDependencies": {
+		"@types/node": "^24.3.0",
+		"canvas": "^3.2.0",
+		"vitest": "^3.2.4"
+	}
+}

package/src/cli.ts

@@ -0,0 +1,173 @@
+#!/usr/bin/env bun
+
+import { logger } from "@oh-my-pi/pi-coding-agent";
+import { loginAnthropic } from "./utils/oauth/anthropic";
+import { loginGitHubCopilot } from "./utils/oauth/github-copilot";
+import { loginAntigravity } from "./utils/oauth/google-antigravity";
+import { loginGeminiCli } from "./utils/oauth/google-gemini-cli";
+import { getOAuthProviders } from "./utils/oauth/index";
+import type { OAuthCredentials, OAuthProvider } from "./utils/oauth/types";
+
+const AUTH_FILE = "auth.json";
+const PROVIDERS = getOAuthProviders();
+
+function prompt(question: string): Promise<string> {
+	return new Promise((resolve) => {
+		process.stdout.write(question);
+		process.stdin.resume();
+		process.stdin.once("data", (data) => {
+			process.stdin.pause();
+			resolve(data.toString().trim());
+		});
+	});
+}
+
+async function loadAuth(): Promise<Record<string, { type: "oauth" } & OAuthCredentials>> {
+	const file = Bun.file(AUTH_FILE);
+	if (!(await file.exists())) return {};
+	try {
+		return await file.json();
+	} catch (err) {
+		logger.debug("Failed to parse config file", { error: String(err) });
+		return {};
+	}
+}
+
+async function saveAuth(auth: Record<string, { type: "oauth" } & OAuthCredentials>): Promise<void> {
+	await Bun.write(AUTH_FILE, JSON.stringify(auth, null, 2));
+}
+
+async function login(provider: OAuthProvider): Promise<void> {
+	const promptFn = (msg: string) => prompt(`${msg} `);
+
+	let credentials: OAuthCredentials;
+
+	switch (provider) {
+		case "anthropic":
+			credentials = await loginAnthropic(
+				(url) => {
+					console.log(`\nOpen this URL in your browser:\n${url}\n`);
+				},
+				async () => {
+					return await promptFn("Paste the authorization code:");
+				},
+			);
+			break;
+
+		case "github-copilot":
+			credentials = await loginGitHubCopilot({
+				onAuth: (url, instructions) => {
+					console.log(`\nOpen this URL in your browser:\n${url}`);
+					if (instructions) console.log(instructions);
+					console.log();
+				},
+				onPrompt: async (p) => {
+					return await promptFn(`${p.message}${p.placeholder ? ` (${p.placeholder})` : ""}:`);
+				},
+				onProgress: (msg) => console.log(msg),
+			});
+			break;
+
+		case "google-gemini-cli":
+			credentials = await loginGeminiCli(
+				(info) => {
+					console.log(`\nOpen this URL in your browser:\n${info.url}`);
+					if (info.instructions) console.log(info.instructions);
+					console.log();
+				},
+				(msg) => console.log(msg),
+			);
+			break;
+
+		case "google-antigravity":
+			credentials = await loginAntigravity(
+				(info) => {
+					console.log(`\nOpen this URL in your browser:\n${info.url}`);
+					if (info.instructions) console.log(info.instructions);
+					console.log();
+				},
+				(msg) => console.log(msg),
+			);
+			break;
+	}
+
+	const auth = await loadAuth();
+	auth[provider] = { type: "oauth", ...credentials };
+	await saveAuth(auth);
+
+	console.log(`\nCredentials saved to ${AUTH_FILE}`);
+}
+
+async function main(): Promise<void> {
+	const args = process.argv.slice(2);
+	const command = args[0];
+
+	if (!command || command === "help" || command === "--help" || command === "-h") {
+		console.log(`Usage: npx @oh-my-pi/pi-ai <command> [provider]
+
+Commands:
+  login [provider]  Login to an OAuth provider
+  list              List available providers
+
+Providers:
+  anthropic         Anthropic (Claude Pro/Max)
+  github-copilot    GitHub Copilot
+  google-gemini-cli Google Gemini CLI
+  google-antigravity Antigravity (Gemini 3, Claude, GPT-OSS)
+
+Examples:
+  npx @oh-my-pi/pi-ai login              # interactive provider selection
+  npx @oh-my-pi/pi-ai login anthropic    # login to specific provider
+  npx @oh-my-pi/pi-ai list               # list providers
+`);
+		return;
+	}
+
+	if (command === "list") {
+		console.log("Available OAuth providers:\n");
+		for (const p of PROVIDERS) {
+			console.log(`  ${p.id.padEnd(20)} ${p.name}`);
+		}
+		return;
+	}
+
+	if (command === "login") {
+		let provider = args[1] as OAuthProvider | undefined;
+
+		if (!provider) {
+			console.log("Select a provider:\n");
+			for (let i = 0; i < PROVIDERS.length; i++) {
+				console.log(`  ${i + 1}. ${PROVIDERS[i].name}`);
+			}
+			console.log();
+
+			const choice = await prompt("Enter number (1-4): ");
+
+			const index = parseInt(choice, 10) - 1;
+			if (index < 0 || index >= PROVIDERS.length) {
+				console.error("Invalid selection");
+				process.exit(1);
+			}
+			provider = PROVIDERS[index].id;
+		}
+
+		if (!PROVIDERS.some((p) => p.id === provider)) {
+			console.error(`Unknown provider: ${provider}`);
+			console.error(`Use 'npx @oh-my-pi/pi-ai list' to see available providers`);
+			process.exit(1);
+		}
+
+		console.log(`Logging in to ${provider}...`);
+		await login(provider);
+		return;
+	}
+
+	console.error(`Unknown command: ${command}`);
+	console.error(`Use 'npx @oh-my-pi/pi-ai --help' for usage`);
+	process.exit(1);
+}
+
+main().catch((err) => {
+	console.error("Error:", err.message);
+	process.exit(1);
+});

package/src/index.ts

@@ -0,0 +1,13 @@
+export * from "./models";
+export * from "./providers/anthropic";
+export * from "./providers/google";
+export * from "./providers/google-gemini-cli";
+export * from "./providers/openai-completions";
+export * from "./providers/openai-responses";
+export * from "./stream";
+export * from "./types";
+export * from "./utils/event-stream";
+export * from "./utils/oauth/index";
+export * from "./utils/overflow";
+export * from "./utils/typebox-helpers";
+export * from "./utils/validation";