@oh-my-pi/pi-ai 6.7.67 → 6.8.0

package/README.md

@@ -10,38 +10,38 @@ Unified LLM API with automatic model discovery, provider configuration, token an
 - [Installation](#installation)
 - [Quick Start](#quick-start)
 - [Tools](#tools)
-  - [Defining Tools](#defining-tools)
-  - [Handling Tool Calls](#handling-tool-calls)
-  - [Streaming Tool Calls with Partial JSON](#streaming-tool-calls-with-partial-json)
-  - [Validating Tool Arguments](#validating-tool-arguments)
-  - [Complete Event Reference](#complete-event-reference)
+   - [Defining Tools](#defining-tools)
+   - [Handling Tool Calls](#handling-tool-calls)
+   - [Streaming Tool Calls with Partial JSON](#streaming-tool-calls-with-partial-json)
+   - [Validating Tool Arguments](#validating-tool-arguments)
+   - [Complete Event Reference](#complete-event-reference)
 - [Image Input](#image-input)
 - [Thinking/Reasoning](#thinkingreasoning)
-  - [Unified Interface](#unified-interface-streamsimplecompletesimple)
-  - [Provider-Specific Options](#provider-specific-options-streamcomplete)
-  - [Streaming Thinking Content](#streaming-thinking-content)
+   - [Unified Interface](#unified-interface-streamsimplecompletesimple)
+   - [Provider-Specific Options](#provider-specific-options-streamcomplete)
+   - [Streaming Thinking Content](#streaming-thinking-content)
 - [Stop Reasons](#stop-reasons)
 - [Error Handling](#error-handling)
-  - [Aborting Requests](#aborting-requests)
-  - [Continuing After Abort](#continuing-after-abort)
+   - [Aborting Requests](#aborting-requests)
+   - [Continuing After Abort](#continuing-after-abort)
 - [APIs, Models, and Providers](#apis-models-and-providers)
-  - [Providers and Models](#providers-and-models)
-  - [Querying Providers and Models](#querying-providers-and-models)
-  - [Custom Models](#custom-models)
-  - [OpenAI Compatibility Settings](#openai-compatibility-settings)
-  - [Type Safety](#type-safety)
+   - [Providers and Models](#providers-and-models)
+   - [Querying Providers and Models](#querying-providers-and-models)
+   - [Custom Models](#custom-models)
+   - [OpenAI Compatibility Settings](#openai-compatibility-settings)
+   - [Type Safety](#type-safety)
 - [Cross-Provider Handoffs](#cross-provider-handoffs)
 - [Context Serialization](#context-serialization)
 - [Browser Usage](#browser-usage)
-  - [Environment Variables](#environment-variables-nodejs-only)
-  - [Checking Environment Variables](#checking-environment-variables)
+   - [Environment Variables](#environment-variables-nodejs-only)
+   - [Checking Environment Variables](#checking-environment-variables)
 - [OAuth Providers](#oauth-providers)
-  - [Vertex AI (ADC)](#vertex-ai-adc)
-  - [CLI Login](#cli-login)
-  - [Programmatic OAuth](#programmatic-oauth)
-  - [Login Flow Example](#login-flow-example)
-  - [Using OAuth Tokens](#using-oauth-tokens)
-  - [Provider Notes](#provider-notes)
+   - [Vertex AI (ADC)](#vertex-ai-adc)
+   - [CLI Login](#cli-login)
+   - [Programmatic OAuth](#programmatic-oauth)
+   - [Login Flow Example](#login-flow-example)
+   - [Using OAuth Tokens](#using-oauth-tokens)
+   - [Provider Notes](#provider-notes)
 - [License](#license)
 
 ## Supported Providers
@@ -156,7 +156,7 @@ for (const call of toolCalls) {
 					timeZone: call.arguments.timezone || "UTC",
 					dateStyle: "full",
 					timeStyle: "long",
-			  })
+				})
 			: "Unknown tool";
 
 	// Add tool result to context (supports text and images)
@@ -443,7 +443,7 @@ const response = await completeSimple(
 	},
 	{
 		reasoning: "medium", // 'minimal' | 'low' | 'medium' | 'high' | 'xhigh' (xhigh maps to high on non-OpenAI providers)
-	}
+	},
 );
 
 // Access thinking and text blocks
@@ -563,7 +563,7 @@ const s = stream(
 	},
 	{
 		signal: controller.signal,
-	}
+	},
 );
 
 for await (const event of s) {
@@ -856,7 +856,7 @@ const response = await complete(
 	},
 	{
 		apiKey: "your-api-key",
-	}
+	},
 );
 ```
 
@@ -957,9 +957,9 @@ Official docs: [Application Default Credentials](https://cloud.google.com/docs/a
 The quickest way to authenticate:
 
 ```bash
-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 available providers
+bunx @oh-my-pi/pi-ai login              # interactive provider selection
+bunx @oh-my-pi/pi-ai login anthropic    # login to specific provider
+bunx @oh-my-pi/pi-ai list               # list available providers
 ```
 
 Credentials are saved to `auth.json` in the current directory.
@@ -1035,7 +1035,7 @@ const response = await complete(
 	{
 		messages: [{ role: "user", content: "Hello!" }],
 	},
-	{ apiKey: result.apiKey }
+	{ apiKey: result.apiKey },
 );
 ```
 

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@oh-my-pi/pi-ai",
-	"version": "6.7.67",
+	"version": "6.8.0",
 	"description": "Unified LLM API with automatic model discovery and provider configuration",
 	"type": "module",
 	"main": "./src/index.ts",
@@ -17,6 +17,7 @@
 		"test": "bun test"
 	},
 	"dependencies": {
+		"@oh-my-pi/pi-utils": "6.8.0",
 		"@anthropic-ai/sdk": "0.71.2",
 		"@aws-sdk/client-bedrock-runtime": "^3.968.0",
 		"@bufbuild/protobuf": "^2.10.2",

package/src/cli.ts

@@ -1,7 +1,7 @@
 #!/usr/bin/env bun
-import "./utils/migrate-env";
-import { existsSync, readFileSync, writeFileSync } from "node:fs";
 import { createInterface } from "readline";
+import { CliAuthStorage } from "./storage";
+import "./utils/migrate-env";
 import { loginAnthropic } from "./utils/oauth/anthropic";
 import { loginGitHubCopilot } from "./utils/oauth/github-copilot";
 import { loginAntigravity } from "./utils/oauth/google-antigravity";
@@ -10,92 +10,81 @@ import { getOAuthProviders } from "./utils/oauth/index";
 import { loginOpenAICodex } from "./utils/oauth/openai-codex";
 import type { OAuthCredentials, OAuthProvider } from "./utils/oauth/types";
 
-const AUTH_FILE = "auth.json";
 const PROVIDERS = getOAuthProviders();
 
 function prompt(rl: ReturnType<typeof createInterface>, question: string): Promise<string> {
-	return new Promise((resolve) => rl.question(question, resolve));
-}
-
-function loadAuth(): Record<string, { type: "oauth" } & OAuthCredentials> {
-	if (!existsSync(AUTH_FILE)) return {};
-	try {
-		return JSON.parse(readFileSync(AUTH_FILE, "utf-8"));
-	} catch {
-		return {};
-	}
-}
-
-function saveAuth(auth: Record<string, { type: "oauth" } & OAuthCredentials>): void {
-	writeFileSync(AUTH_FILE, JSON.stringify(auth, null, 2), "utf-8");
+	const { promise, resolve } = Promise.withResolvers<string>();
+	rl.question(question, resolve);
+	return promise;
 }
 
 async function login(provider: OAuthProvider): Promise<void> {
 	const rl = createInterface({ input: process.stdin, output: process.stdout });
 
 	const promptFn = (msg: string) => prompt(rl, `${msg} `);
+	const storage = new CliAuthStorage();
 
 	try {
 		let credentials: OAuthCredentials;
 
 		switch (provider) {
 			case "anthropic":
-				credentials = await loginAnthropic(
-					(url) => {
+				credentials = await loginAnthropic({
+					onAuth(info) {
+						const { url } = info;
 						console.log(`\nOpen this URL in your browser:\n${url}\n`);
 					},
-					async () => {
-						return await promptFn("Paste the authorization code:");
+					onProgress(message) {
+						console.log(message);
 					},
-				);
+				});
 				break;
 
 			case "github-copilot":
 				credentials = await loginGitHubCopilot({
-					onAuth: (url, instructions) => {
+					onAuth(url, instructions) {
 						console.log(`\nOpen this URL in your browser:\n${url}`);
 						if (instructions) console.log(instructions);
 						console.log();
 					},
-					onPrompt: async (p) => {
+					async onPrompt(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);
+				credentials = await loginGeminiCli({
+					onAuth(info) {
+						const { url, instructions } = info;
+						console.log(`\nOpen this URL in your browser:\n${url}`);
+						if (instructions) console.log(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);
+				credentials = await loginAntigravity({
+					onAuth(info) {
+						const { url, instructions } = info;
+						console.log(`\nOpen this URL in your browser:\n${url}`);
+						if (instructions) console.log(instructions);
 						console.log();
 					},
-					(msg) => console.log(msg),
-				);
+				});
 				break;
 			case "openai-codex":
 				credentials = await loginOpenAICodex({
-					onAuth: (info) => {
-						console.log(`\nOpen this URL in your browser:\n${info.url}`);
-						if (info.instructions) console.log(info.instructions);
+					onAuth(info) {
+						const { url, instructions } = info;
+						console.log(`\nOpen this URL in your browser:\n${url}`);
+						if (instructions) console.log(instructions);
 						console.log();
 					},
-					onPrompt: async (p) => {
+					async onPrompt(p) {
 						return await promptFn(`${p.message}${p.placeholder ? ` (${p.placeholder})` : ""}:`);
 					},
-					onProgress: (msg) => console.log(msg),
 				});
 				break;
 
@@ -103,12 +92,11 @@ async function login(provider: OAuthProvider): Promise<void> {
 				throw new Error(`Unknown provider: ${provider}`);
 		}
 
-		const auth = loadAuth();
-		auth[provider] = { type: "oauth", ...credentials };
-		saveAuth(auth);
+		storage.saveOAuth(provider, credentials);
 
-		console.log(`\nCredentials saved to ${AUTH_FILE}`);
+		console.log(`\nCredentials saved to ~/.omp/agent/agent.db`);
 	} finally {
+		storage.close();
 		rl.close();
 	}
 }
@@ -118,10 +106,12 @@ async function main(): Promise<void> {
 	const command = args[0];
 
 	if (!command || command === "help" || command === "--help" || command === "-h") {
-		console.log(`Usage: npx @oh-my-pi/pi-ai <command> [provider]
+		console.log(`Usage: bunx @oh-my-pi/pi-ai <command> [provider]
 
 Commands:
   login [provider]  Login to an OAuth provider
+  logout [provider] Logout from an OAuth provider
+  status            Show logged-in providers
   list              List available providers
 
 Providers:
@@ -130,15 +120,43 @@ Providers:
   google-gemini-cli Google Gemini CLI
   google-antigravity Antigravity (Gemini 3, Claude, GPT-OSS)
   openai-codex      OpenAI Codex (ChatGPT Plus/Pro)
+  cursor            Cursor (Claude, GPT, etc.)
 
 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
+  bunx @oh-my-pi/pi-ai login              # interactive provider selection
+  bunx @oh-my-pi/pi-ai login anthropic    # login to specific provider
+  bunx @oh-my-pi/pi-ai logout anthropic   # logout from specific provider
+  bunx @oh-my-pi/pi-ai status             # show logged-in providers
+  bunx @oh-my-pi/pi-ai list               # list providers
 `);
 		return;
 	}
 
+	if (command === "status") {
+		const storage = new CliAuthStorage();
+		try {
+			const providers = storage.listProviders();
+			if (providers.length === 0) {
+				console.log("No OAuth credentials stored.");
+				console.log(`Use 'bunx @oh-my-pi/pi-ai login' to authenticate.`);
+			} else {
+				console.log("Logged-in providers:\n");
+				for (const provider of providers) {
+					const oauth = storage.getOAuth(provider);
+					if (oauth) {
+						const expires = new Date(oauth.expires);
+						const expired = Date.now() >= oauth.expires;
+						const status = expired ? "(expired)" : `(expires ${expires.toLocaleString()})`;
+						console.log(`  ${provider.padEnd(20)} ${status}`);
+					}
+				}
+			}
+		} finally {
+			storage.close();
+		}
+		return;
+	}
+
 	if (command === "list") {
 		console.log("Available OAuth providers:\n");
 		for (const p of PROVIDERS) {
@@ -147,6 +165,50 @@ Examples:
 		return;
 	}
 
+	if (command === "logout") {
+		let provider = args[1] as OAuthProvider | undefined;
+		const storage = new CliAuthStorage();
+
+		try {
+			if (!provider) {
+				const providers = storage.listProviders();
+				if (providers.length === 0) {
+					console.log("No OAuth credentials stored.");
+					return;
+				}
+
+				const rl = createInterface({ input: process.stdin, output: process.stdout });
+				console.log("Select a provider to logout:\n");
+				for (let i = 0; i < providers.length; i++) {
+					console.log(`  ${i + 1}. ${providers[i]}`);
+				}
+				console.log();
+
+				const choice = await prompt(rl, `Enter number (1-${providers.length}): `);
+				rl.close();
+
+				const index = parseInt(choice, 10) - 1;
+				if (index < 0 || index >= providers.length) {
+					console.error("Invalid selection");
+					process.exit(1);
+				}
+				provider = providers[index] as OAuthProvider;
+			}
+
+			const oauth = storage.getOAuth(provider);
+			if (!oauth) {
+				console.error(`Not logged in to ${provider}`);
+				process.exit(1);
+			}
+
+			storage.deleteProvider(provider);
+			console.log(`Logged out from ${provider}`);
+		} finally {
+			storage.close();
+		}
+		return;
+	}
+
 	if (command === "login") {
 		let provider = args[1] as OAuthProvider | undefined;
 
@@ -171,7 +233,7 @@ Examples:
 
 		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`);
+			console.error(`Use 'bunx @oh-my-pi/pi-ai list' to see available providers`);
 			process.exit(1);
 		}
 
@@ -181,7 +243,7 @@ Examples:
 	}
 
 	console.error(`Unknown command: ${command}`);
-	console.error(`Use 'npx @oh-my-pi/pi-ai --help' for usage`);
+	console.error(`Use 'bunx @oh-my-pi/pi-ai --help' for usage`);
 	process.exit(1);
 }