pplx-zero 1.1.8 → 2.0.0

package/package.json

@@ -1,80 +1,46 @@
 {
   "name": "pplx-zero",
-  "version": "1.1.8",
-  "description": "Fast Perplexity AI search CLI with multimodal support - minimal setup, maximal results",
+  "version": "2.0.0",
+  "description": "Minimal Perplexity AI CLI - search from terminal",
+  "author": "kenzo",
+  "license": "MIT",
   "type": "module",
-  "main": "dist/index.js",
+  "module": "src/index.ts",
   "bin": {
-    "pplx": "dist/index.js"
+    "pplx": "bin/pplx.js"
   },
   "scripts": {
-    "dev": "bun run src/cli/index.ts",
-    "dev:legacy": "bun run src/cli.ts",
-    "build": "rm -rf dist && bun build src/cli/index.ts --target node --outdir dist",
-    "build:legacy": "rm -rf dist && bun build src/cli.ts --target node --outdir dist",
-    "build:binary": "bun build --compile src/cli/index.ts --outfile=dist/pplx",
-    "build:binary:legacy": "bun build --compile src/cli.ts --outfile=dist/pplx-legacy",
-    "test": "bun test",
-    "test:watch": "bun test --watch",
-    "typecheck": "bun tsc --noEmit",
-    "lint": "bun run --bun eslint src/**/*.ts",
-    "lint:fix": "bun run --bun eslint src/**/*.ts --fix",
-    "clean": "rm -rf dist",
-    "dev:debug": "bun --inspect src/cli/index.ts",
-    "dev:debug:legacy": "bun --inspect src/cli.ts"
+    "build": "bun build src/index.ts --compile --outfile=pplx",
+    "dev": "bun run src/index.ts",
+    "test": "bun test"
   },
   "dependencies": {
-    "@perplexity-ai/perplexity_ai": "^0.11.0",
-    "abort-controller": "^3.0.0",
-    "commander": "^12.0.0",
-    "dotenv": "^16.3.1",
-    "zod": "^3.22.4"
+    "zod": "^4.0.0"
   },
   "devDependencies": {
-    "@types/node": "^20.0.0",
-    "typescript": "^5.0.0",
-    "bun-types": "latest",
-    "@typescript-eslint/eslint-plugin": "^6.0.0",
-    "@typescript-eslint/parser": "^6.0.0",
-    "eslint": "^8.0.0"
+    "@types/bun": "latest"
   },
-  "engines": {
-    "bun": ">=1.0.0"
+  "peerDependencies": {
+    "typescript": "^5"
   },
-  "keywords": [
-    "pplx-zero",
-    "perplexity",
-    "search",
-    "ai",
-    "cli",
-    "command-line",
-    "tool",
-    "typescript",
-    "bun",
-    "api",
-    "minimal",
-    "fast",
-    "productivity",
-    "zero-config",
-    "multimodal",
-    "attachments",
-    "images",
-    "documents",
-    "sonar",
-    "reasoning",
-    "research",
-    "async"
+  "files": [
+    "src",
+    "bin"
   ],
-  "author": "Kenzo",
-  "license": "MIT",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/codewithkenzo/pplx-zero.git"
   },
-  "files": [
-    "dist",
-    "README.md",
-    "LICENSE",
-    "package.json"
+  "bugs": {
+    "url": "https://github.com/codewithkenzo/pplx-zero/issues"
+  },
+  "homepage": "https://github.com/codewithkenzo/pplx-zero#readme",
+  "keywords": [
+    "perplexity",
+    "ai",
+    "search",
+    "cli",
+    "terminal",
+    "bun"
   ]
-}
+}

package/src/api.test.ts

@@ -0,0 +1,21 @@
+import { test, expect, describe } from 'bun:test';
+import { MODELS, type Model } from './api';
+
+describe('MODELS', () => {
+  test('includes all expected models', () => {
+    expect(MODELS).toContain('sonar');
+    expect(MODELS).toContain('sonar-pro');
+    expect(MODELS).toContain('sonar-reasoning');
+    expect(MODELS).toContain('sonar-reasoning-pro');
+    expect(MODELS).toContain('sonar-deep-research');
+  });
+
+  test('has exactly 5 models', () => {
+    expect(MODELS).toHaveLength(5);
+  });
+
+  test('Model type matches MODELS array', () => {
+    const model: Model = MODELS[0]!;
+    expect(MODELS.includes(model)).toBe(true);
+  });
+});

package/src/api.ts

@@ -0,0 +1,116 @@
+import { getEnv } from './env';
+import type { FileAttachment } from './files';
+
+const API_URL = 'https://api.perplexity.ai/chat/completions';
+
+export const MODELS = ['sonar', 'sonar-pro', 'sonar-reasoning', 'sonar-reasoning-pro', 'sonar-deep-research'] as const;
+export type Model = (typeof MODELS)[number];
+
+export interface SearchResult {
+  title: string;
+  url: string;
+  date?: string;
+}
+
+export interface StreamCallbacks {
+  onContent: (text: string) => void;
+  onDone: (citations: SearchResult[], usage: { prompt_tokens: number; completion_tokens: number }) => void;
+  onError: (error: Error) => void;
+}
+
+interface MessageContent {
+  type: 'text' | 'file_url';
+  text?: string;
+  file_url?: { url: string };
+  file_name?: string;
+}
+
+function buildMessages(query: string, file?: FileAttachment): { role: string; content: string | MessageContent[] }[] {
+  if (!file) {
+    return [{ role: 'user', content: query }];
+  }
+
+  const content: MessageContent[] = [
+    { type: 'text', text: query },
+    {
+      type: 'file_url',
+      file_url: { url: file.data },
+      file_name: file.filename,
+    },
+  ];
+
+  return [{ role: 'user', content }];
+}
+
+export async function search(
+  query: string,
+  model: Model,
+  callbacks: StreamCallbacks,
+  file?: FileAttachment
+): Promise<void> {
+  const body = JSON.stringify({
+    model,
+    messages: buildMessages(query, file),
+    stream: true,
+  });
+
+  const response = await fetch(API_URL, {
+    method: 'POST',
+    headers: {
+      'Authorization': `Bearer ${getEnv().PERPLEXITY_API_KEY}`,
+      'Content-Type': 'application/json',
+    },
+    body,
+  });
+
+  if (!response.ok) {
+    const text = await response.text();
+    callbacks.onError(new Error(`API error ${response.status}: ${text}`));
+    return;
+  }
+
+  if (!response.body) {
+    callbacks.onError(new Error('No response body'));
+    return;
+  }
+
+  const reader = response.body.getReader();
+  const decoder = new TextDecoder();
+  let buffer = '';
+  let citations: SearchResult[] = [];
+  let usage = { prompt_tokens: 0, completion_tokens: 0 };
+
+  while (true) {
+    const { done, value } = await reader.read();
+    if (done) break;
+
+    buffer += decoder.decode(value, { stream: true });
+    const lines = buffer.split('\n');
+    buffer = lines.pop() || '';
+
+    for (const line of lines) {
+      if (!line.startsWith('data: ')) continue;
+      const data = line.slice(6).trim();
+      if (data === '[DONE]') continue;
+
+      try {
+        const parsed = JSON.parse(data);
+        const delta = parsed.choices?.[0]?.delta?.content;
+        if (delta) {
+          callbacks.onContent(delta);
+        }
+
+        if (parsed.search_results) {
+          citations = parsed.search_results;
+        }
+        if (parsed.usage) {
+          usage = parsed.usage;
+        }
+      } catch {
+        continue;
+      }
+    }
+  }
+
+  callbacks.onDone(citations, usage);
+}

package/src/env.ts

@@ -0,0 +1,37 @@
+import { z } from 'zod';
+
+const c = {
+  reset: '\x1b[0m',
+  red: '\x1b[31m',
+  yellow: '\x1b[33m',
+  cyan: '\x1b[36m',
+  dim: '\x1b[2m',
+} as const;
+
+const envSchema = z.object({
+  PERPLEXITY_API_KEY: z.string().min(1),
+});
+
+let _env: z.infer<typeof envSchema> | null = null;
+
+export function getEnv() {
+  if (_env) return _env;
+
+  const key = process.env.PERPLEXITY_API_KEY || process.env.PERPLEXITY_AI_API_KEY;
+  
+  if (!key) {
+    console.error(`
+${c.red}✗ Missing API Key${c.reset}
+
+Set your Perplexity API key:
+
+  ${c.cyan}export PERPLEXITY_API_KEY="pplx-..."${c.reset}
+
+${c.dim}Get one at: https://perplexity.ai/settings/api${c.reset}
+`);
+    process.exit(2);
+  }
+
+  _env = { PERPLEXITY_API_KEY: key };
+  return _env;
+}

package/src/files.test.ts

@@ -0,0 +1,71 @@
+import { test, expect, describe } from 'bun:test';
+import { encodeFile, toDataUrl, type FileAttachment } from './files';
+import { writeFile, unlink } from 'node:fs/promises';
+import { join } from 'node:path';
+
+const TMP_DIR = '/tmp';
+
+describe('encodeFile', () => {
+  test('encodes text file correctly', async () => {
+    const testPath = join(TMP_DIR, 'test.txt');
+    await writeFile(testPath, 'hello world');
+    
+    const result = await encodeFile(testPath);
+    
+    expect(result.type).toBe('file');
+    expect(result.mimeType).toBe('text/plain');
+    expect(result.filename).toBe('test.txt');
+    expect(result.data).toBe(Buffer.from('hello world').toString('base64'));
+    
+    await unlink(testPath);
+  });
+
+  test('encodes PDF as file type', async () => {
+    const testPath = join(TMP_DIR, 'test.pdf');
+    await writeFile(testPath, '%PDF-1.4 test');
+    
+    const result = await encodeFile(testPath);
+    
+    expect(result.type).toBe('file');
+    expect(result.mimeType).toBe('application/pdf');
+    
+    await unlink(testPath);
+  });
+
+  test('encodes PNG as image type', async () => {
+    const testPath = join(TMP_DIR, 'test.png');
+    const pngHeader = Buffer.from([0x89, 0x50, 0x4E, 0x47, 0x0D, 0x0A, 0x1A, 0x0A]);
+    await writeFile(testPath, pngHeader);
+    
+    const result = await encodeFile(testPath);
+    
+    expect(result.type).toBe('image');
+    expect(result.mimeType).toBe('image/png');
+    
+    await unlink(testPath);
+  });
+
+  test('throws on unsupported file type', async () => {
+    const testPath = join(TMP_DIR, 'test.xyz');
+    await writeFile(testPath, 'test');
+    
+    await expect(encodeFile(testPath)).rejects.toThrow('Unsupported file type: .xyz');
+    
+    await unlink(testPath);
+  });
+});
+
+describe('toDataUrl', () => {
+  test('creates valid data URL', () => {
+    const attachment: FileAttachment = {
+      type: 'image',
+      data: 'aGVsbG8gd29ybGQ=',
+      mimeType: 'image/png',
+      filename: 'test.png',
+    };
+    
+    const result = toDataUrl(attachment);
+    
+    expect(result).toBe('data:image/png;base64,aGVsbG8gd29ybGQ=');
+  });
+});

package/src/files.ts

@@ -0,0 +1,44 @@
+import { readFile } from 'node:fs/promises';
+import { extname } from 'node:path';
+
+const MIME_TYPES: Record<string, string> = {
+  '.pdf': 'application/pdf',
+  '.txt': 'text/plain',
+  '.md': 'text/markdown',
+  '.png': 'image/png',
+  '.jpg': 'image/jpeg',
+  '.jpeg': 'image/jpeg',
+  '.gif': 'image/gif',
+  '.webp': 'image/webp',
+};
+
+export interface FileAttachment {
+  type: 'file' | 'image';
+  data: string;
+  mimeType: string;
+  filename: string;
+}
+
+export async function encodeFile(path: string): Promise<FileAttachment> {
+  const ext = extname(path).toLowerCase();
+  const mimeType = MIME_TYPES[ext];
+  
+  if (!mimeType) {
+    throw new Error(`Unsupported file type: ${ext}`);
+  }
+
+  const buffer = await readFile(path);
+  const data = buffer.toString('base64');
+  const isImage = mimeType.startsWith('image/');
+
+  return {
+    type: isImage ? 'image' : 'file',
+    data,
+    mimeType,
+    filename: path.split('/').pop() || 'file',
+  };
+}
+
+export function toDataUrl(attachment: FileAttachment): string {
+  return `data:${attachment.mimeType};base64,${attachment.data}`;
+}

package/src/index.ts

@@ -0,0 +1,96 @@
+#!/usr/bin/env bun
+import { parseArgs } from 'node:util';
+import { search, MODELS, type Model } from './api';
+import { encodeFile } from './files';
+import { getEnv } from './env';
+import { fmt, write, writeLn } from './output';
+
+getEnv();
+
+
+const { values, positionals } = parseArgs({
+  args: Bun.argv.slice(2),
+  options: {
+    model: { type: 'string', short: 'm', default: 'sonar' },
+    file: { type: 'string', short: 'f' },
+    image: { type: 'string', short: 'i' },
+    json: { type: 'boolean', default: false },
+    help: { type: 'boolean', short: 'h' },
+  },
+  allowPositionals: true,
+  strict: true,
+});
+
+if (values.help || positionals.length === 0) {
+  console.log(`
+pplx - Perplexity AI search from terminal
+
+Usage: pplx [options] <query>
+
+Options:
+  -m, --model <name>   Model: ${MODELS.join(', ')} (default: sonar)
+  -f, --file <path>    Attach a file (PDF, TXT, etc.)
+  -i, --image <path>   Attach an image (PNG, JPG, etc.)
+  --json               Output as JSON
+  -h, --help           Show this help
+
+Examples:
+  pplx "what is bun"
+  pplx -m sonar-pro "explain quantum computing"
+  pplx -f report.pdf "summarize this document"
+`);
+  process.exit(0);
+}
+
+const query = positionals.join(' ');
+const model = (MODELS.includes(values.model as Model) ? values.model : 'sonar') as Model;
+
+const filePath = values.file || values.image;
+const file = filePath ? await encodeFile(filePath) : undefined;
+
+const startTime = Date.now();
+let fullContent = '';
+
+if (!values.json) {
+  await write(fmt.model(model) + ' ');
+  await write(fmt.searching());
+}
+
+await search(query, model, {
+  onContent: async (text) => {
+    fullContent += text;
+    if (!values.json) {
+      await write(text);
+    }
+  },
+  onDone: async (citations, usage) => {
+    const elapsed = Date.now() - startTime;
+
+    if (values.json) {
+      const output = {
+        answer: fullContent,
+        citations: citations.map((c) => c.url),
+        model,
+        tokens: usage.prompt_tokens + usage.completion_tokens,
+        latency_ms: elapsed,
+      };
+      console.log(JSON.stringify(output, null, 2));
+    } else {
+      if (citations.length > 0) {
+        await writeLn(fmt.sources());
+        for (let i = 0; i < citations.length; i++) {
+          await writeLn(fmt.citation(i + 1, citations[i]!.url));
+        }
+      }
+      await write(fmt.stats(usage.prompt_tokens + usage.completion_tokens, elapsed));
+    }
+  },
+  onError: async (error) => {
+    if (values.json) {
+      console.error(JSON.stringify({ error: error.message }));
+    } else {
+      await write(fmt.error(error.message));
+    }
+    process.exit(1);
+  },
+}, file);

package/src/output.test.ts

@@ -0,0 +1,41 @@
+import { test, expect, describe } from 'bun:test';
+import { fmt } from './output';
+
+describe('fmt', () => {
+  test('model formats with cyan color', () => {
+    const result = fmt.model('sonar');
+    expect(result).toContain('[sonar]');
+    expect(result).toContain('\x1b[36m');
+    expect(result).toContain('\x1b[0m');
+  });
+
+  test('searching shows dim text', () => {
+    const result = fmt.searching();
+    expect(result).toContain('Searching...');
+    expect(result).toContain('\x1b[2m');
+  });
+
+  test('error formats with red color', () => {
+    const result = fmt.error('test error');
+    expect(result).toContain('Error: test error');
+    expect(result).toContain('\x1b[31m');
+  });
+
+  test('citation formats with number and URL', () => {
+    const result = fmt.citation(1, 'https://example.com');
+    expect(result).toContain('1.');
+    expect(result).toContain('https://example.com');
+  });
+
+  test('stats formats tokens and time', () => {
+    const result = fmt.stats(100, 1500);
+    expect(result).toContain('100 tokens');
+    expect(result).toContain('1.5s');
+  });
+
+  test('sources shows yellow header', () => {
+    const result = fmt.sources();
+    expect(result).toContain('Sources:');
+    expect(result).toContain('\x1b[33m');
+  });
+});

package/src/output.ts

@@ -0,0 +1,28 @@
+const c = {
+  reset: '\x1b[0m',
+  dim: '\x1b[2m',
+  bold: '\x1b[1m',
+  cyan: '\x1b[36m',
+  green: '\x1b[32m',
+  yellow: '\x1b[33m',
+  red: '\x1b[31m',
+  gray: '\x1b[90m',
+} as const;
+
+export const fmt = {
+  model: (name: string) => `${c.cyan}[${name}]${c.reset}`,
+  searching: () => `${c.dim}Searching...${c.reset}\n`,
+  error: (msg: string) => `${c.red}Error: ${msg}${c.reset}\n`,
+  citation: (i: number, url: string) => `${c.dim}  ${i}. ${url}${c.reset}`,
+  stats: (tokens: number, ms: number) => 
+    `\n${c.gray}[${tokens} tokens, ${(ms / 1000).toFixed(1)}s]${c.reset}\n`,
+  sources: () => `\n${c.yellow}Sources:${c.reset}`,
+};
+
+export async function write(text: string): Promise<void> {
+  await Bun.write(Bun.stdout, text);
+}
+
+export async function writeLn(text: string): Promise<void> {
+  await Bun.write(Bun.stdout, text + '\n');
+}