@drewsepsi/nextpi 0.1.2 → 0.1.3

package/lib/agent.ts

@@ -0,0 +1,105 @@
+import { 
+  createAgentSession, 
+  SessionManager, 
+  type AgentSession 
+} from "@mariozechner/pi-coding-agent";
+import { getModel, streamSimple } from "@mariozechner/pi-ai";
+import * as path from "path";
+import * as fs from "fs";
+import { getProjectRoot } from "./root-path";
+import { getEffectiveApiKey } from "./config";
+
+export interface ResearchAgentConfig {
+  sessionDir?: string;
+  sessionFile?: string;
+  modelId?: string;
+  thinkingLevel?: "off" | "minimal" | "low" | "medium" | "high" | "xhigh";
+}
+
+// Default OpenRouter model
+const DEFAULT_MODEL = "openrouter/free";
+
+// Research tools (web search, web fetch)
+const researchTools = [
+  {
+    name: "web_search",
+    description: "Search the web for information using Exa AI",
+    parameters: {
+      type: "object",
+      properties: {
+        query: { type: "string", description: "Search query" },
+        numResults: { type: "number", description: "Number of results (default: 5)" }
+      },
+      required: ["query"]
+    }
+  },
+  {
+    name: "web_fetch",
+    description: "Fetch content from a specific URL",
+    parameters: {
+      type: "object",
+      properties: {
+        url: { type: "string", description: "URL to fetch" }
+      },
+      required: ["url"]
+    }
+  }
+];
+
+export async function createResearchAgent(
+  config: ResearchAgentConfig = {}
+): Promise<AgentSession> {
+  const {
+    sessionDir = path.join(getProjectRoot(), ".sessions"),
+    sessionFile = path.join(sessionDir, "research-agent.jsonl"),
+    modelId = DEFAULT_MODEL,
+    thinkingLevel = "low",
+  } = config;
+
+  // Ensure session directory exists
+  fs.mkdirSync(sessionDir, { recursive: true });
+
+  // Use OpenRouter model
+  const model = getModel("openrouter", modelId as "openrouter/free");
+
+  // Create session with research tools and system prompt
+  const { session } = await createAgentSession({
+    model,
+    thinkingLevel,
+    systemPrompt: RESEARCH_SYSTEM_PROMPT,
+    sessionManager: SessionManager.open(sessionFile),
+    customTools: researchTools as any,
+  } as any);
+
+  // Wrap stream function with OpenRouter headers
+  const originalStreamFn = session.agent.streamFn;
+  session.agent.streamFn = (model, context, options) => {
+    return streamSimple(model, context, {
+      ...options,
+      headers: {
+        ...options?.headers,
+        "X-Title": "NextPi Agent",
+        "HTTP-Referer": "https://github.com/drewsepsi/nextpi",
+      },
+    });
+  };
+
+  return session;
+}
+
+// System prompt for research and coding
+export const RESEARCH_SYSTEM_PROMPT = `You are NextPi, an expert research and coding assistant. You have access to the web and the local file system.
+
+### GUIDELINES
+- ALWAYS start by explaining your plan to the user in the main message or thinking block.
+- BEFORE calling any tool that modifies the file system (write_file, edit_file, etc.), you MUST describe the specific changes you are about to make and why.
+- If you are editing a file, summarize the intended changes first.
+- If you are researching, explain what you hope to find.
+
+### CAPABILITIES
+1. **Web Search**: Search for docs, errors, news, etc.
+2. **Web Fetch**: Read specific URLs.
+3. **File Operations**: Read, write, list (ls), and edit files.
+4. **Shell**: Execute bash commands for testing or exploration.
+
+Always be surgical with edits and transparent with your process.`;

package/lib/config.js

@@ -0,0 +1,45 @@
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+
+const CONFIG_DIR = path.join(os.homedir(), '.nextpi');
+const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json');
+
+export function getConfig() {
+  try {
+    if (!fs.existsSync(CONFIG_FILE)) {
+      return {};
+    }
+    const data = fs.readFileSync(CONFIG_FILE, 'utf-8');
+    return JSON.parse(data);
+  } catch (err) {
+    console.error('Error reading config:', err);
+    return {};
+  }
+}
+
+export function saveConfig(newConfig) {
+  try {
+    if (!fs.existsSync(CONFIG_DIR)) {
+      fs.mkdirSync(CONFIG_DIR, { recursive: true });
+    }
+    const current = getConfig();
+    const updated = { ...current, ...newConfig };
+    fs.writeFileSync(CONFIG_FILE, JSON.stringify(updated, null, 2));
+    
+    // Also update current process env for immediate use
+    if (updated.openRouterApiKey) {
+      process.env.OPENROUTER_API_KEY = updated.openRouterApiKey;
+    }
+    
+    return updated;
+  } catch (err) {
+    console.error('Error saving config:', err);
+    throw err;
+  }
+}
+
+export function getEffectiveApiKey() {
+  // Priority: 1. Environment Variable, 2. Config File
+  return process.env.OPENROUTER_API_KEY || getConfig().openRouterApiKey;
+}

package/lib/config.ts

@@ -0,0 +1,50 @@
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+
+const CONFIG_DIR = path.join(os.homedir(), '.nextpi');
+const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json');
+
+export interface NextPiConfig {
+  openRouterApiKey?: string;
+  defaultModel?: string;
+}
+
+export function getConfig(): NextPiConfig {
+  try {
+    if (!fs.existsSync(CONFIG_FILE)) {
+      return {};
+    }
+    const data = fs.readFileSync(CONFIG_FILE, 'utf-8');
+    return JSON.parse(data);
+  } catch (err) {
+    console.error('Error reading config:', err);
+    return {};
+  }
+}
+
+export function saveConfig(newConfig: Partial<NextPiConfig>) {
+  try {
+    if (!fs.existsSync(CONFIG_DIR)) {
+      fs.mkdirSync(CONFIG_DIR, { recursive: true });
+    }
+    const current = getConfig();
+    const updated = { ...current, ...newConfig };
+    fs.writeFileSync(CONFIG_FILE, JSON.stringify(updated, null, 2));
+    
+    // Also update current process env for immediate use
+    if (updated.openRouterApiKey) {
+      process.env.OPENROUTER_API_KEY = updated.openRouterApiKey;
+    }
+    
+    return updated;
+  } catch (err) {
+    console.error('Error saving config:', err);
+    throw err;
+  }
+}
+
+export function getEffectiveApiKey(): string | undefined {
+  // Priority: 1. Environment Variable, 2. Config File
+  return process.env.OPENROUTER_API_KEY || getConfig().openRouterApiKey;
+}

package/lib/root-path.ts

@@ -0,0 +1,15 @@
+import path from 'path';
+
+export function getProjectRoot() {
+  // PI_ROOT can be set via environment variable to override the root directory
+  const envRoot = process.env.PI_ROOT;
+  
+  if (envRoot) {
+    // Resolve the path to handle relative paths from where the process was started
+    return path.resolve(envRoot);
+  }
+
+  // If no PI_ROOT is set, fallback to process.cwd()
+  // Note: in a Next.js environment during dev/build, process.cwd() is the project root.
+  return process.cwd();
+}

package/lib/session-singleton.ts

@@ -0,0 +1,48 @@
+import { createResearchAgent } from '@/lib/agent';
+
+type AgentSession = Awaited<ReturnType<typeof createResearchAgent>>;
+
+/**
+ * In Next.js development mode, modules are frequently re-evaluated due to HMR.
+ * Attaching the session to `globalThis` ensures that all API routes (which may
+ * be running in different contexts or re-evaluated modules) share exactly
+ * the same AgentSession instance.
+ */
+const globalForAgent = globalThis as unknown as {
+  _sharedSession?: AgentSession;
+  _sharedSessionPromise?: Promise<AgentSession>;
+};
+
+let currentModelId: string | undefined = undefined;
+
+export async function getSharedSession(): Promise<AgentSession> {
+  // Return the existing session if already initialized
+  if (globalForAgent._sharedSession) {
+    return globalForAgent._sharedSession;
+  }
+
+  // If a session is being created, wait for it
+  if (globalForAgent._sharedSessionPromise) {
+    return globalForAgent._sharedSessionPromise;
+  }
+
+  // Otherwise, create a new session and store the promise globally
+  console.log(`[Agent] Initializing new shared session singleton (Model: ${currentModelId || 'default'})...`);
+  globalForAgent._sharedSessionPromise = createResearchAgent({ modelId: currentModelId }).then((session) => {
+    globalForAgent._sharedSession = session;
+    console.log('[Agent] Shared session initialized');
+    return session;
+  });
+
+  return globalForAgent._sharedSessionPromise;
+}
+
+export function resetSharedSession(modelId?: string): void {
+  console.log(`[Agent] Resetting shared session${modelId ? ` to model ${modelId}` : ''}...`);
+  if (modelId) {
+    currentModelId = modelId;
+  }
+  globalForAgent._sharedSession = undefined;
+  globalForAgent._sharedSessionPromise = undefined;
+}
+

package/lib/utils.ts

@@ -0,0 +1,6 @@
+import { clsx, type ClassValue } from "clsx"
+import { twMerge } from "tailwind-merge"
+
+export function cn(...inputs: ClassValue[]) {
+  return twMerge(clsx(inputs))
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@drewsepsi/nextpi",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "The Agentic IDE - An AI-powered research and coding assistant workspace.",
   "type": "module",
   "bin": {
@@ -8,6 +8,7 @@
   },
   "files": [
     "bin",
+    "lib",
     ".next/standalone",
     ".next/static",
     "public",