@vybestack/llxprt-code-core 0.1.14 → 0.1.16

package/dist/src/core/prompts-async.test.js

@@ -0,0 +1,115 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach, afterEach, beforeAll } from 'vitest';
+import { getCoreSystemPromptAsync, initializePromptSystem } from './prompts.js';
+import process from 'node:process';
+import fs from 'node:fs';
+import os from 'node:os';
+import path from 'node:path';
+describe('prompts async integration', () => {
+    let originalEnv;
+    let tempDir;
+    beforeAll(async () => {
+        // Create a temporary directory for test prompts
+        tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'llxprt-test-'));
+        process.env.LLXPRT_PROMPTS_DIR = tempDir;
+        // Initialize the prompt system once for all tests
+        await initializePromptSystem();
+    });
+    beforeEach(() => {
+        originalEnv = { ...process.env };
+        process.env.LLXPRT_PROMPTS_DIR = tempDir;
+    });
+    afterEach(() => {
+        // Restore original environment
+        Object.keys(process.env).forEach((key) => {
+            if (!(key in originalEnv)) {
+                delete process.env[key];
+            }
+        });
+        Object.assign(process.env, originalEnv);
+    });
+    describe('getCoreSystemPromptAsync', () => {
+        it('should return a valid prompt string', async () => {
+            const prompt = await getCoreSystemPromptAsync();
+            expect(prompt).toBeTruthy();
+            expect(typeof prompt).toBe('string');
+            expect(prompt.length).toBeGreaterThan(100);
+            // Should contain some key content from core.md
+            expect(prompt).toContain('interactive CLI agent');
+            expect(prompt).toContain('Core Mandates');
+        });
+        it('should include user memory when provided', async () => {
+            const userMemory = 'Remember: The user prefers concise responses.';
+            const prompt = await getCoreSystemPromptAsync(userMemory);
+            // Debug: log what we actually get
+            if (!prompt.includes(userMemory)) {
+                console.error('Prompt does not contain user memory. Last 500 chars:', prompt.slice(-500));
+            }
+            expect(prompt).toContain(userMemory);
+            // Should have both core content and user memory
+            expect(prompt).toContain('interactive CLI agent');
+            expect(prompt).toContain(userMemory);
+        });
+        it('should handle different models', async () => {
+            const prompt = await getCoreSystemPromptAsync(undefined, 'gemini-1.5-flash');
+            expect(prompt).toBeTruthy();
+            expect(typeof prompt).toBe('string');
+            // Flash models should have additional tool instructions
+            expect(prompt).toContain('IMPORTANT: You MUST use the provided tools when appropriate');
+        });
+        it('should handle custom tools list', async () => {
+            const tools = ['read_file', 'write_file', 'list_directory'];
+            const prompt = await getCoreSystemPromptAsync(undefined, undefined, tools);
+            expect(prompt).toBeTruthy();
+            expect(typeof prompt).toBe('string');
+            // Should still contain core content
+            expect(prompt).toContain('interactive CLI agent');
+        });
+        it('should handle git repository environment', async () => {
+            // Mock isGitRepository to return true
+            process.env.GIT_DIR = '.git';
+            const prompt = await getCoreSystemPromptAsync();
+            expect(prompt).toBeTruthy();
+            // Note: The git detection in buildPromptContext uses isGitRepository
+            // which checks the actual file system, not env vars
+        });
+        it('should handle sandbox environment', async () => {
+            process.env.SANDBOX = 'true';
+            const prompt = await getCoreSystemPromptAsync();
+            expect(prompt).toBeTruthy();
+            expect(prompt).toContain('Sandbox');
+        });
+        it('should handle different environments based on initial setup', async () => {
+            // This test just verifies the prompt system works
+            // We can't test environment changes without reset functionality
+            const prompt = await getCoreSystemPromptAsync();
+            expect(prompt).toBeTruthy();
+            expect(typeof prompt).toBe('string');
+            expect(prompt.length).toBeGreaterThan(100);
+        });
+    });
+    describe('prompt content validation', () => {
+        it('should include all required tool references', async () => {
+            const prompt = await getCoreSystemPromptAsync();
+            // Check for tool references (they should be replaced with actual tool names)
+            expect(prompt).toContain('Ls');
+            expect(prompt).toContain('Edit');
+            expect(prompt).toContain('Glob');
+            expect(prompt).toContain('Grep');
+            expect(prompt).toContain('ReadFile');
+            expect(prompt).toContain('WriteFile');
+            expect(prompt).toContain('Shell');
+        });
+        it('should properly format user memory with separator', async () => {
+            const userMemory = 'Custom user preferences here';
+            const prompt = await getCoreSystemPromptAsync(userMemory);
+            // Should have the separator before user memory
+            expect(prompt).toMatch(/---\s*Custom user preferences here/);
+        });
+    });
+});
+//# sourceMappingURL=prompts-async.test.js.map

package/dist/src/core/prompts-async.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompts-async.test.js","sourceRoot":"","sources":["../../../src/core/prompts-async.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAChF,OAAO,EAAE,wBAAwB,EAAE,sBAAsB,EAAE,MAAM,cAAc,CAAC;AAChF,OAAO,OAAO,MAAM,cAAc,CAAC;AACnC,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACzC,IAAI,WAA8B,CAAC;IACnC,IAAI,OAAe,CAAC;IAEpB,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,gDAAgD;QAChD,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,cAAc,CAAC,CAAC,CAAC;QACjE,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,OAAO,CAAC;QAEzC,kDAAkD;QAClD,MAAM,sBAAsB,EAAE,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,UAAU,CAAC,GAAG,EAAE;QACd,WAAW,GAAG,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;QACjC,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,OAAO,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,+BAA+B;QAC/B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YACvC,IAAI,CAAC,CAAC,GAAG,IAAI,WAAW,CAAC,EAAE,CAAC;gBAC1B,OAAO,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,MAAM,GAAG,MAAM,wBAAwB,EAAE,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,CAAC;YAC5B,MAAM,CAAC,OAAO,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YAC3C,+CAA+C;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,UAAU,GAAG,+CAA+C,CAAC;YACnE,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,UAAU,CAAC,CAAC;YAE1D,kCAAkC;YAClC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;gBACjC,OAAO,CAAC,KAAK,CACX,sDAAsD,EACtD,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CACnB,CAAC;YACJ,CAAC;YAED,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YACrC,gDAAgD;YAChD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAC3C,SAAS,EACT,kBAAkB,CACnB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,CAAC;YAC5B,MAAM,CAAC,OAAO,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACrC,wDAAwD;YACxD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CACtB,6DAA6D,CAC9D,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,KAAK,GAAG,CAAC,WAAW,EAAE,YAAY,EAAE,gBAAgB,CAAC,CAAC;YAC5D,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAC3C,SAAS,EACT,SAAS,EACT,KAAK,CACN,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,CAAC;YAC5B,MAAM,CAAC,OAAO,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACrC,oCAAoC;YACpC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,sCAAsC;YACtC,OAAO,CAAC,GAAG,CAAC,OAAO,GAAG,MAAM,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,wBAAwB,EAAE,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,CAAC;YAC5B,qEAAqE;YACrE,oDAAoD;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,OAAO,CAAC,GAAG,CAAC,OAAO,GAAG,MAAM,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,wBAAwB,EAAE,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,CAAC;YAC5B,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,kDAAkD;YAClD,gEAAgE;YAChE,MAAM,MAAM,GAAG,MAAM,wBAAwB,EAAE,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,CAAC;YAC5B,MAAM,CAAC,OAAO,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,MAAM,GAAG,MAAM,wBAAwB,EAAE,CAAC;YAEhD,6EAA6E;YAC7E,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACjC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACjC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACjC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,UAAU,GAAG,8BAA8B,CAAC;YAClD,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,UAAU,CAAC,CAAC;YAE1D,+CAA+C;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,oCAAoC,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/core/prompts.d.ts

@@ -3,7 +3,14 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-export declare function getCoreSystemPrompt(userMemory?: string, model?: string): string;
+/**
+ * Async version of getCoreSystemPrompt that uses the new PromptService
+ */
+export declare function getCoreSystemPromptAsync(userMemory?: string, model?: string, tools?: string[]): Promise<string>;
+/**
+ * Initialize the prompt system - call this early in application startup
+ */
+export declare function initializePromptSystem(): Promise<void>;
 /**
  * Provides the system prompt for the history compression process.
  * This prompt instructs the model to act as a specialized state manager,