@riotprompt/riotprompt 0.0.8 → 0.0.10

package/FIXES-APPLIED.md

@@ -0,0 +1,437 @@
+# Code Review Fixes Applied
+
+All issues identified in the comprehensive code review have been fixed.
+
+## Summary
+
+- **Total Issues Fixed:** 19
+- **Critical Issues:** 4
+- **High Priority Issues:** 6
+- **Medium Priority Issues:** 9
+
+---
+
+## 🔴 Critical Issues Fixed
+
+### 1. **Memory Leak in TokenBudgetManager** ✅
+**File:** `src/token-budget.ts`
+
+**Problem:** Creating temporary `TokenBudgetManager` instances without disposing encoders.
+
+**Fix:** Modified `compressAdaptive()` to temporarily modify config instead of creating new instance.
+
+**Before:**
+```typescript
+const tempManager = new TokenBudgetManager(modifiedConfig, 'gpt-4o', this.logger);
+return tempManager.compressFIFO(messages, targetTokens);
+```
+
+**After:**
+```typescript
+const originalPreserveRecent = this.config.preserveRecent;
+this.config.preserveRecent = 5;
+const result = this.compressFIFO(messages, targetTokens);
+this.config.preserveRecent = originalPreserveRecent;
+return result;
+```
+
+---
+
+### 2. **Context Injection Position Tracking Bug** ✅
+**File:** `src/conversation.ts`
+
+**Problem:** Multiple items injected at same position had incorrect tracking.
+
+**Fix:** Track each item at `position + index` to maintain correct positions.
+
+**Before:**
+```typescript
+this.state.messages.splice(position, 0, contextMessage);
+this.state.contextManager.track(item, position);
+```
+
+**After:**
+```typescript
+const actualPosition = position + i;
+this.state.messages.splice(actualPosition, 0, contextMessage);
+this.state.contextManager.track(item, actualPosition);
+```
+
+---
+
+### 3. **Shallow Copy Causing Shared State** ✅
+**File:** `src/conversation.ts`
+
+**Problem:** Cloning conversations with shallow copy causes `tool_calls` arrays to be shared.
+
+**Fix:** Deep copy messages including nested `tool_calls` arrays.
+
+**Before:**
+```typescript
+cloned.state.messages = this.state.messages.map(msg => ({ ...msg }));
+```
+
+**After:**
+```typescript
+cloned.state.messages = this.state.messages.map(msg => ({
+    ...msg,
+    tool_calls: msg.tool_calls ? msg.tool_calls.map(tc => ({
+        ...tc,
+        function: { ...tc.function }
+    })) : undefined
+}));
+```
+
+---
+
+### 4. **Uncaught JSON Parse Error** ✅
+**File:** `src/iteration-strategy.ts`
+
+**Problem:** No error handling for `JSON.parse()` on tool arguments.
+
+**Fix:** Wrapped JSON.parse in try-catch with descriptive error message.
+
+**After:**
+```typescript
+let toolArgs: any;
+try {
+    toolArgs = JSON.parse(toolCall.function.arguments);
+} catch (parseError) {
+    throw new Error(`Invalid JSON in tool arguments: ${parseError.message}`);
+}
+```
+
+---
+
+## 🟡 High Priority Issues Fixed
+
+### 5. **Hardcoded Model in Recipe ExecuteWith** ✅
+**File:** `src/recipes.ts`
+
+**Problem:** `executeWith()` hardcoded `'gpt-4o'` model.
+
+**Fix:** Added `model` parameter with default value.
+
+**After:**
+```typescript
+executeWith: async (
+    llm: LLMClient,
+    strategy: IterationStrategy,
+    model: Model = 'gpt-4o',
+    tokenBudget?: TokenBudgetConfig
+)
+```
+
+---
+
+### 6. **Section Array Append Missing Options** ✅
+**File:** `src/items/section.ts`
+
+**Problem:** Options not propagated when appending arrays.
+
+**Fix:** Pass `options` parameter in recursive calls.
+
+**After:**
+```typescript
+if (Array.isArray(item)) {
+    item.forEach((item) => {
+        append(item, options);  // Now propagates options
+    });
+}
+```
+
+---
+
+### 7. **JSONL File Race Condition** ✅
+**File:** `src/conversation-logger.ts`
+
+**Problem:** Fire-and-forget async writes could corrupt JSONL files.
+
+**Fix:** Added write queue and cached output path.
+
+**After:**
+```typescript
+private cachedOutputPath?: string;
+private writeQueue: Promise<void> = Promise.resolve();
+
+// Queue writes
+this.writeQueue = this.writeQueue
+    .then(() => this.appendToJSONL(loggedMessage))
+    .catch(this.config.onError);
+```
+
+---
+
+### 8. **Hardcoded Model Detection System** ✅
+**Files:** New `src/model-config.ts`, `src/chat.ts`, `src/message-builder.ts`, `src/token-budget.ts`
+
+**Problem:** Model names hardcoded throughout codebase, inflexible for new models.
+
+**Fix:** Created flexible `ModelRegistry` system with pattern-based configuration.
+
+**Key Features:**
+- Pattern-based model matching (regex)
+- Configurable role mapping (system vs developer)
+- Tokenizer encoding configuration
+- User-extensible via `configureModel()`
+
+**Example Usage:**
+```typescript
+import { configureModel } from 'riotprompt';
+
+// Add support for new model family
+configureModel({
+    pattern: /^gemini/i,
+    personaRole: 'system',
+    encoding: 'cl100k_base',
+    family: 'gemini'
+});
+```
+
+---
+
+### 9. **Conversation Replayer JSON Parse Error** ✅
+**File:** `src/conversation-logger.ts`
+
+**Problem:** No error handling when parsing tool call arguments from logs.
+
+**Fix:** Added try-catch with fallback object containing raw arguments.
+
+---
+
+### 10. **Iteration Counter Misplaced** ✅
+**File:** `src/iteration-strategy.ts`
+
+**Problem:** Counter incremented per phase instead of per iteration.
+
+**Fix:** Moved `incrementIteration()` inside the iteration loop.
+
+---
+
+## 🟠 Medium Priority Issues Fixed
+
+### 11. **FIFO Compression O(n*m) Complexity** ✅
+**File:** `src/token-budget.ts`
+
+**Problem:** Using `includes()` on array is O(n*m) for large conversations.
+
+**Fix:** Use `Set` for O(1) lookups, reducing to O(n).
+
+---
+
+### 12. **Overly Aggressive Similar Content Detection** ✅
+**File:** `src/context-manager.ts`
+
+**Problem:** "Hello" and "Hello world" considered duplicates.
+
+**Fix:** Added similarity threshold (default 90%) for substring matching.
+
+**After:**
+```typescript
+hasSimilarContent(content: string, similarityThreshold: number = 0.9): boolean {
+    const lengthRatio = shorter.length / longer.length;
+    if (lengthRatio >= similarityThreshold) {
+        if (longer.includes(shorter)) {
+            return true;
+        }
+    }
+}
+```
+
+---
+
+### 13. **Duplicate Context Without IDs** ✅
+**File:** `src/context-manager.ts`
+
+**Problem:** Same content tracked multiple times if no ID provided.
+
+**Fix:** Check content hash before tracking items without IDs.
+
+---
+
+### 14. **Tool Category Filter Edge Case** ✅
+**File:** `src/recipes.ts`
+
+**Problem:** Tools without category could match empty string in filter array.
+
+**Fix:** Added explicit check for truthy `tool.category`.
+
+---
+
+### 15. **Circuit Breaker for Failing Tools** ✅
+**File:** `src/iteration-strategy.ts`
+
+**Problem:** No protection against repeatedly calling broken tools.
+
+**Fix:** Added circuit breaker with configurable threshold (default: 3 consecutive failures).
+
+**Features:**
+- Tracks consecutive failures per tool
+- Configurable via `maxConsecutiveToolFailures` in phase config
+- Resets counter on successful execution
+- Logs circuit breaker triggers
+
+---
+
+### 16. **Generic Parser Error Message** ✅
+**File:** `src/parser.ts`
+
+**Problem:** Error said "instructions" but parser is used for all content types.
+
+**Fix:** Changed to generic "content" in error message.
+
+---
+
+### 17. **Override Array Mutation** ✅
+**File:** `src/override.ts`
+
+**Problem:** Mutating `appends` array with `.reverse()`.
+
+**Fix:** Create copy before reversing: `[...appends].reverse()`.
+
+---
+
+### 18. **Type Assertion Hack in Loader** ✅
+**File:** `src/loader.ts`
+
+**Problem:** Double cast `as unknown as T` circumventing type system.
+
+**Fix:** Removed unnecessary cast - `Section.add()` correctly accepts `Section<T>`.
+
+---
+
+### 19. **Ignore Pattern Only Tests Filename** ✅
+**File:** `src/loader.ts`
+
+**Problem:** Regex patterns only tested against filename, not full path.
+
+**Fix:** Test against both filename and full path for flexibility.
+
+---
+
+## New Features Added
+
+### Model Configuration System
+
+Created a comprehensive, user-configurable model system:
+
+**File:** `src/model-config.ts`
+
+**Exports:**
+- `ModelRegistry` - Main registry class
+- `getModelRegistry()` - Get global instance
+- `configureModel()` - Register custom models
+- `getPersonaRole()` - Get role for model
+- `getEncoding()` - Get tokenizer encoding
+- `supportsToolCalls()` - Check tool support
+- `getModelFamily()` - Get model family
+
+**Default Configurations:**
+- GPT-4 family (uses 'system' role)
+- O-series models (uses 'developer' role)
+- Claude family (uses 'system' role)
+- Default fallback for unknown models
+
+**Benefits:**
+- No more hardcoded model names
+- Easy to add new models without code changes
+- Pattern-based matching (e.g., `/^o\d+/` matches o1, o2, o3, etc.)
+- User-extensible at runtime
+
+---
+
+## Testing Recommendations
+
+1. **Memory Leak Test:** Run long conversation with many compressions, monitor memory
+2. **Context Injection Test:** Inject multiple items, verify position tracking
+3. **Clone Test:** Clone conversation, modify tool_calls, verify no cross-contamination
+4. **Invalid JSON Test:** Send malformed tool arguments, verify graceful error handling
+5. **Circuit Breaker Test:** Configure tool to fail repeatedly, verify circuit breaker triggers
+6. **Model Registry Test:** Register custom model, verify correct role/encoding selection
+7. **JSONL Race Test:** Add many messages rapidly, verify file integrity
+8. **Performance Test:** Compress large conversations (1000+ messages), verify Set optimization
+
+---
+
+## Breaking Changes
+
+⚠️ **Minor Breaking Change in Recipe API:**
+
+```typescript
+// OLD
+builder.executeWith(llm, strategy, tokenBudget)
+
+// NEW
+builder.executeWith(llm, strategy, model, tokenBudget)
+// OR use default
+builder.executeWith(llm, strategy) // defaults to 'gpt-4o'
+```
+
+---
+
+## Migration Guide
+
+### For Custom Model Support
+
+**Before:** Had to modify source code to add new models
+
+**After:** Configure at runtime:
+
+```typescript
+import { configureModel } from 'riotprompt';
+
+// Add new model family
+configureModel({
+    pattern: /^my-model/i,
+    personaRole: 'system',
+    encoding: 'gpt-4o',
+    supportsToolCalls: true,
+    family: 'my-family'
+});
+
+// Add specific override
+configureModel({
+    exactMatch: 'my-model-v2-special',
+    personaRole: 'developer',
+    encoding: 'cl100k_base'
+});
+```
+
+---
+
+## Files Modified
+
+1. `src/model-config.ts` (NEW)
+2. `src/chat.ts`
+3. `src/conversation.ts`
+4. `src/token-budget.ts`
+5. `src/iteration-strategy.ts`
+6. `src/recipes.ts`
+7. `src/conversation-logger.ts`
+8. `src/context-manager.ts`
+9. `src/items/section.ts`
+10. `src/parser.ts`
+11. `src/loader.ts`
+12. `src/override.ts`
+13. `src/message-builder.ts`
+14. `src/riotprompt.ts`
+
+---
+
+## Next Steps
+
+1. ✅ All critical issues fixed
+2. ✅ All high priority issues fixed
+3. ✅ All medium priority issues fixed
+4. ✅ No linting errors
+5. ⏭️ Run test suite to verify fixes
+6. ⏭️ Update documentation for model configuration system
+7. ⏭️ Consider adding unit tests for circuit breaker
+8. ⏭️ Performance test compression optimization
+
+---
+
+**Review completed:** December 27, 2025
+**Total fixes applied:** 19
+**Status:** ✅ All issues resolved
+

package/README.md

@@ -75,11 +75,11 @@ RiotPrompt is designed to be completely generic and unopinionated. Unlike other
 
 ## 🤝 Contributing
 
-Contributions are welcome! Please see our [Contributing Guide](CONTRIBUTING.md) for details. Test.
+Contributions are welcome! Please see our [Contributing Guide](CONTRIBUTING.md) for details.
 
 ## 📄 License
 
-MIT License - see [LICENSE](LICENSE) for details.
+Apache-2.0 License - see [LICENSE](LICENSE) for details.
 
 ---
 

package/dist/builder.js

@@ -10,6 +10,9 @@ import { create as create$1 } from './parser.js';
 import { create as create$3 } from './loader.js';
 import { create as create$2 } from './override.js';
 import './recipes.js';
+import './conversation.js';
+import 'tiktoken';
+import './tools.js';
 
 const OptionSchema = z.object({
     logger: z.any().optional().default(DEFAULT_LOGGER),

package/dist/builder.js.map

@@ -1 +1 @@
-{"version":3,"file":"builder.js","sources":["../src/builder.ts"],"sourcesContent":["import path from \"path\";\nimport { z } from \"zod\";\nimport { ParametersSchema } from \"./items/parameters\";\nimport { SectionOptions, SectionOptionsSchema } from \"./items/section\";\nimport { DEFAULT_LOGGER, wrapLogger } from \"./logger\";\nimport { Content, Context, createPrompt, createSection, Instruction, Loader, Override, Parser, Prompt, Section, Weighted } from \"./riotprompt\";\n\nconst OptionSchema = z.object({\n    logger: z.any().optional().default(DEFAULT_LOGGER),\n    basePath: z.string(),\n    overridePaths: z.array(z.string()).optional().default([\"./\"]),\n    overrides: z.boolean().optional().default(false),\n    parameters: ParametersSchema.optional().default({}),\n});\n\nexport type Options = z.infer<typeof OptionSchema>;\n\nexport type OptionsParam = Required<Pick<Options, 'basePath'>> & Partial<Omit<Options, 'basePath'>>;\n\nexport interface Instance {\n    addPersonaPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContextPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addInstructionPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContentPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContent(content: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContext(context: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    loadContext(contextDirectories: string[], sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    loadContent(contentDirectories: string[], sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    build(): Promise<Prompt>;\n}\n\nexport const create = (builderOptions: OptionsParam): Instance => {\n    const options: Required<Options> = OptionSchema.parse(builderOptions) as Required<Options>;\n\n    const logger = wrapLogger(options.logger, 'Builder');\n    const parser = Parser.create({ logger });\n    const override = Override.create({\n        logger, configDirs: options.overridePaths || [\"./\"],\n        overrides: options.overrides || false\n    });\n    const loader = Loader.create({ logger });\n\n    const personaSection: Section<Instruction> = createSection({ title: \"Persona\" });\n    const contextSection: Section<Context> = createSection({ title: \"Context\" });\n    const instructionSection: Section<Instruction> = createSection({ title: \"Instruction\" });\n    const contentSection: Section<Content> = createSection({ title: \"Content\" });\n    const parameters = options.parameters;\n\n\n    const instance: Partial<Instance> = {}\n\n    const loadOptions = (sectionOptions: Partial<SectionOptions> = {}): SectionOptions => {\n        const currentOptions = SectionOptionsSchema.parse(sectionOptions);\n        return {\n            ...currentOptions,\n            parameters: {\n                ...parameters,\n                ...currentOptions.parameters\n            }\n        }\n    }\n\n    const loadDirectories = async <T extends Weighted>(\n        directories: string[],\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Section<T>[]> => {\n        const currentOptions = loadOptions(sectionOptions);\n        logger.debug(\"Loading directories\", directories);\n        const sections: Section<T>[] = await loader.load<T>(directories, currentOptions);\n        return sections;\n    }\n\n    const loadContext = async (\n        contextDirectories: string[],\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        const currentOptions = loadOptions(sectionOptions);\n        logger.debug('Loading context', contextDirectories);\n        const context: Section<Context>[] = await loadDirectories<Context>(contextDirectories, currentOptions);\n        contextSection.add(context);\n        return instance as Instance;\n    }\n    instance.loadContext = loadContext;\n\n    const loadContent = async (\n        contentDirectories: string[],\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        const currentOptions = loadOptions(sectionOptions);\n        const content: Section<Content>[] = await loadDirectories<Content>(contentDirectories, currentOptions);\n        contentSection.add(content);\n        return instance as Instance;\n    }\n    instance.loadContent = loadContent;\n\n    const loadPath = async <T extends Weighted>(\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Section<T>> => {\n        const currentOptions = loadOptions(sectionOptions);\n        const defaultPath = path.join(options.basePath as string, contentPath);\n        const section: Section<T> = await parser.parseFile<T>(defaultPath, currentOptions);\n        const overrideSection = await override.customize<T>(contentPath, section, currentOptions);\n        return overrideSection;\n    }\n\n    const addPersonaPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        const currentOptions = loadOptions(sectionOptions);\n        const persona: Section<Instruction> = await loadPath<Instruction>(contentPath, currentOptions);\n        personaSection.add(persona);\n        return instance as Instance;\n    }\n    instance.addPersonaPath = addPersonaPath;\n\n    const addContextPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding context path\", contentPath);\n        const currentOptions = loadOptions(sectionOptions);\n        const context: Section<Context> = await loadPath<Context>(contentPath, currentOptions);\n        contextSection.add(context);\n        return instance as Instance;\n    }\n    instance.addContextPath = addContextPath;\n\n    const addInstructionPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding instruction path\", contentPath);\n        const currentOptions = loadOptions(sectionOptions);\n        const instruction: Section<Instruction> = await loadPath<Instruction>(contentPath, currentOptions);\n        instructionSection.add(instruction);\n        return instance as Instance;\n    }\n    instance.addInstructionPath = addInstructionPath;\n\n    const addContentPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding content path\", contentPath);\n        const currentOptions = loadOptions(sectionOptions);\n        const content: Section<Content> = await loadPath<Content>(contentPath, currentOptions);\n        contentSection.add(content);\n        return instance as Instance;\n    }\n    instance.addContentPath = addContentPath;\n\n    const addContent = async (\n        content: string | Buffer,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding content\", typeof content);\n        const currentOptions = loadOptions(sectionOptions);\n        const parsedContentSection: Section<Content> = await parser.parse<Content>(content, currentOptions);\n        contentSection.add(parsedContentSection);\n        return instance as Instance;\n    }\n    instance.addContent = addContent;\n\n    const addContext = async (\n        context: string | Buffer,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding context\", typeof context);\n        const currentOptions = loadOptions(sectionOptions);\n        const parsedContextSection: Section<Context> = await parser.parse<Context>(context, currentOptions);\n        contextSection.add(parsedContextSection);\n        return instance as Instance;\n    }\n    instance.addContext = addContext;\n\n    const build = async () => {\n        logger.debug(\"Building prompt\", {});\n        const prompt = createPrompt({ persona: personaSection, contexts: contextSection, instructions: instructionSection, contents: contentSection });\n        return prompt;\n    }\n    instance.build = build;\n\n    return instance as Instance;\n}\n"],"names":["OptionSchema","z","object","logger","any","optional","default","DEFAULT_LOGGER","basePath","string","overridePaths","array","overrides","boolean","parameters","ParametersSchema","create","builderOptions","options","parse","wrapLogger","parser","Parser","override","Override","configDirs","loader","Loader","personaSection","createSection","title","contextSection","instructionSection","contentSection","instance","loadOptions","sectionOptions","currentOptions","SectionOptionsSchema","loadDirectories","directories","debug","sections","load","loadContext","contextDirectories","context","add","loadContent","contentDirectories","content","loadPath","contentPath","defaultPath","path","join","section","parseFile","overrideSection","customize","addPersonaPath","persona","addContextPath","addInstructionPath","instruction","addContentPath","addContent","parsedContentSection","addContext","parsedContextSection","build","prompt","createPrompt","contexts","instructions","contents"],"mappings":";;;;;;;;;;;;;AAOA,MAAMA,YAAAA,GAAeC,CAAAA,CAAEC,MAAM,CAAC;AAC1BC,IAAAA,MAAAA,EAAQF,EAAEG,GAAG,EAAA,CAAGC,QAAQ,EAAA,CAAGC,OAAO,CAACC,cAAAA,CAAAA;AACnCC,IAAAA,QAAAA,EAAUP,EAAEQ,MAAM,EAAA;IAClBC,aAAAA,EAAeT,CAAAA,CAAEU,KAAK,CAACV,CAAAA,CAAEQ,MAAM,EAAA,CAAA,CAAIJ,QAAQ,EAAA,CAAGC,OAAO,CAAC;AAAC,QAAA;AAAK,KAAA,CAAA;AAC5DM,IAAAA,SAAAA,EAAWX,EAAEY,OAAO,EAAA,CAAGR,QAAQ,EAAA,CAAGC,OAAO,CAAC,KAAA,CAAA;AAC1CQ,IAAAA,UAAAA,EAAYC,gBAAAA,CAAiBV,QAAQ,EAAA,CAAGC,OAAO,CAAC,EAAC;AACrD,CAAA,CAAA;AAkBO,MAAMU,SAAS,CAACC,cAAAA,GAAAA;IACnB,MAAMC,OAAAA,GAA6BlB,YAAAA,CAAamB,KAAK,CAACF,cAAAA,CAAAA;AAEtD,IAAA,MAAMd,MAAAA,GAASiB,UAAAA,CAAWF,OAAAA,CAAQf,MAAM,EAAE,SAAA,CAAA;IAC1C,MAAMkB,QAAAA,GAASC,QAAa,CAAC;AAAEnB,QAAAA;AAAO,KAAA,CAAA;IACtC,MAAMoB,UAAAA,GAAWC,QAAe,CAAC;AAC7BrB,QAAAA,MAAAA;QAAQsB,UAAAA,EAAYP,OAAAA,CAAQR,aAAa,IAAI;AAAC,YAAA;AAAK,SAAA;QACnDE,SAAAA,EAAWM,OAAAA,CAAQN,SAAS,IAAI;AACpC,KAAA,CAAA;IACA,MAAMc,QAAAA,GAASC,QAAa,CAAC;AAAExB,QAAAA;AAAO,KAAA,CAAA;AAEtC,IAAA,MAAMyB,iBAAuCC,QAAAA,CAAc;QAAEC,KAAAA,EAAO;AAAU,KAAA,CAAA;AAC9E,IAAA,MAAMC,iBAAmCF,QAAAA,CAAc;QAAEC,KAAAA,EAAO;AAAU,KAAA,CAAA;AAC1E,IAAA,MAAME,qBAA2CH,QAAAA,CAAc;QAAEC,KAAAA,EAAO;AAAc,KAAA,CAAA;AACtF,IAAA,MAAMG,iBAAmCJ,QAAAA,CAAc;QAAEC,KAAAA,EAAO;AAAU,KAAA,CAAA;IAC1E,MAAMhB,UAAAA,GAAaI,QAAQJ,UAAU;AAGrC,IAAA,MAAMoB,WAA8B,EAAC;AAErC,IAAA,MAAMC,WAAAA,GAAc,CAACC,cAAAA,GAA0C,EAAE,GAAA;QAC7D,MAAMC,cAAAA,GAAiBC,oBAAAA,CAAqBnB,KAAK,CAACiB,cAAAA,CAAAA;QAClD,OAAO;AACH,YAAA,GAAGC,cAAc;YACjBvB,UAAAA,EAAY;AACR,gBAAA,GAAGA,UAAU;AACb,gBAAA,GAAGuB,eAAevB;AACtB;AACJ,SAAA;AACJ,IAAA,CAAA;AAEA,IAAA,MAAMyB,eAAAA,GAAkB,OACpBC,WAAAA,EACAJ,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;QACnCjC,MAAAA,CAAOsC,KAAK,CAAC,qBAAA,EAAuBD,WAAAA,CAAAA;AACpC,QAAA,MAAME,QAAAA,GAAyB,MAAMhB,QAAAA,CAAOiB,IAAI,CAAIH,WAAAA,EAAaH,cAAAA,CAAAA;QACjE,OAAOK,QAAAA;AACX,IAAA,CAAA;AAEA,IAAA,MAAME,WAAAA,GAAc,OAChBC,kBAAAA,EACAT,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;QACnCjC,MAAAA,CAAOsC,KAAK,CAAC,iBAAA,EAAmBI,kBAAAA,CAAAA;QAChC,MAAMC,OAAAA,GAA8B,MAAMP,eAAAA,CAAyBM,kBAAAA,EAAoBR,cAAAA,CAAAA;AACvFN,QAAAA,cAAAA,CAAegB,GAAG,CAACD,OAAAA,CAAAA;QACnB,OAAOZ,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAASU,WAAW,GAAGA,WAAAA;AAEvB,IAAA,MAAMI,WAAAA,GAAc,OAChBC,kBAAAA,EACAb,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;QACnC,MAAMc,OAAAA,GAA8B,MAAMX,eAAAA,CAAyBU,kBAAAA,EAAoBZ,cAAAA,CAAAA;AACvFJ,QAAAA,cAAAA,CAAec,GAAG,CAACG,OAAAA,CAAAA;QACnB,OAAOhB,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAASc,WAAW,GAAGA,WAAAA;AAEvB,IAAA,MAAMG,QAAAA,GAAW,OACbC,WAAAA,EACAhB,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;AACnC,QAAA,MAAMiB,cAAcC,aAAAA,CAAKC,IAAI,CAACrC,OAAAA,CAAQV,QAAQ,EAAY4C,WAAAA,CAAAA;AAC1D,QAAA,MAAMI,OAAAA,GAAsB,MAAMnC,QAAAA,CAAOoC,SAAS,CAAIJ,WAAAA,EAAahB,cAAAA,CAAAA;AACnE,QAAA,MAAMqB,kBAAkB,MAAMnC,UAAAA,CAASoC,SAAS,CAAIP,aAAaI,OAAAA,EAASnB,cAAAA,CAAAA;QAC1E,OAAOqB,eAAAA;AACX,IAAA,CAAA;AAEA,IAAA,MAAME,cAAAA,GAAiB,OACnBR,WAAAA,EACAhB,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;QACnC,MAAMyB,OAAAA,GAAgC,MAAMV,QAAAA,CAAsBC,WAAAA,EAAaf,cAAAA,CAAAA;AAC/ET,QAAAA,cAAAA,CAAemB,GAAG,CAACc,OAAAA,CAAAA;QACnB,OAAO3B,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAAS0B,cAAc,GAAGA,cAAAA;AAE1B,IAAA,MAAME,cAAAA,GAAiB,OACnBV,WAAAA,EACAhB,cAAAA,GAA0C,EAAE,GAAA;QAE5CjC,MAAAA,CAAOsC,KAAK,CAAC,qBAAA,EAAuBW,WAAAA,CAAAA;AACpC,QAAA,MAAMf,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;QACnC,MAAMU,OAAAA,GAA4B,MAAMK,QAAAA,CAAkBC,WAAAA,EAAaf,cAAAA,CAAAA;AACvEN,QAAAA,cAAAA,CAAegB,GAAG,CAACD,OAAAA,CAAAA;QACnB,OAAOZ,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAAS4B,cAAc,GAAGA,cAAAA;AAE1B,IAAA,MAAMC,kBAAAA,GAAqB,OACvBX,WAAAA,EACAhB,cAAAA,GAA0C,EAAE,GAAA;QAE5CjC,MAAAA,CAAOsC,KAAK,CAAC,yBAAA,EAA2BW,WAAAA,CAAAA;AACxC,QAAA,MAAMf,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;QACnC,MAAM4B,WAAAA,GAAoC,MAAMb,QAAAA,CAAsBC,WAAAA,EAAaf,cAAAA,CAAAA;AACnFL,QAAAA,kBAAAA,CAAmBe,GAAG,CAACiB,WAAAA,CAAAA;QACvB,OAAO9B,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAAS6B,kBAAkB,GAAGA,kBAAAA;AAE9B,IAAA,MAAME,cAAAA,GAAiB,OACnBb,WAAAA,EACAhB,cAAAA,GAA0C,EAAE,GAAA;QAE5CjC,MAAAA,CAAOsC,KAAK,CAAC,qBAAA,EAAuBW,WAAAA,CAAAA;AACpC,QAAA,MAAMf,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;QACnC,MAAMc,OAAAA,GAA4B,MAAMC,QAAAA,CAAkBC,WAAAA,EAAaf,cAAAA,CAAAA;AACvEJ,QAAAA,cAAAA,CAAec,GAAG,CAACG,OAAAA,CAAAA;QACnB,OAAOhB,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAAS+B,cAAc,GAAGA,cAAAA;AAE1B,IAAA,MAAMC,UAAAA,GAAa,OACfhB,OAAAA,EACAd,cAAAA,GAA0C,EAAE,GAAA;QAE5CjC,MAAAA,CAAOsC,KAAK,CAAC,gBAAA,EAAkB,OAAOS,OAAAA,CAAAA;AACtC,QAAA,MAAMb,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;AACnC,QAAA,MAAM+B,oBAAAA,GAAyC,MAAM9C,QAAAA,CAAOF,KAAK,CAAU+B,OAAAA,EAASb,cAAAA,CAAAA;AACpFJ,QAAAA,cAAAA,CAAec,GAAG,CAACoB,oBAAAA,CAAAA;QACnB,OAAOjC,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAASgC,UAAU,GAAGA,UAAAA;AAEtB,IAAA,MAAME,UAAAA,GAAa,OACftB,OAAAA,EACAV,cAAAA,GAA0C,EAAE,GAAA;QAE5CjC,MAAAA,CAAOsC,KAAK,CAAC,gBAAA,EAAkB,OAAOK,OAAAA,CAAAA;AACtC,QAAA,MAAMT,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;AACnC,QAAA,MAAMiC,oBAAAA,GAAyC,MAAMhD,QAAAA,CAAOF,KAAK,CAAU2B,OAAAA,EAAST,cAAAA,CAAAA;AACpFN,QAAAA,cAAAA,CAAegB,GAAG,CAACsB,oBAAAA,CAAAA;QACnB,OAAOnC,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAASkC,UAAU,GAAGA,UAAAA;AAEtB,IAAA,MAAME,KAAAA,GAAQ,UAAA;QACVnE,MAAAA,CAAOsC,KAAK,CAAC,iBAAA,EAAmB,EAAC,CAAA;AACjC,QAAA,MAAM8B,SAASC,QAAAA,CAAa;YAAEX,OAAAA,EAASjC,cAAAA;YAAgB6C,QAAAA,EAAU1C,cAAAA;YAAgB2C,YAAAA,EAAc1C,kBAAAA;YAAoB2C,QAAAA,EAAU1C;AAAe,SAAA,CAAA;QAC5I,OAAOsC,MAAAA;AACX,IAAA,CAAA;AACArC,IAAAA,QAAAA,CAASoC,KAAK,GAAGA,KAAAA;IAEjB,OAAOpC,QAAAA;AACX;;;;"}
+{"version":3,"file":"builder.js","sources":["../src/builder.ts"],"sourcesContent":["import path from \"path\";\nimport { z } from \"zod\";\nimport { ParametersSchema } from \"./items/parameters\";\nimport { SectionOptions, SectionOptionsSchema } from \"./items/section\";\nimport { DEFAULT_LOGGER, wrapLogger } from \"./logger\";\nimport { Content, Context, createPrompt, createSection, Instruction, Loader, Override, Parser, Prompt, Section, Weighted } from \"./riotprompt\";\n\nconst OptionSchema = z.object({\n    logger: z.any().optional().default(DEFAULT_LOGGER),\n    basePath: z.string(),\n    overridePaths: z.array(z.string()).optional().default([\"./\"]),\n    overrides: z.boolean().optional().default(false),\n    parameters: ParametersSchema.optional().default({}),\n});\n\nexport type Options = z.infer<typeof OptionSchema>;\n\nexport type OptionsParam = Required<Pick<Options, 'basePath'>> & Partial<Omit<Options, 'basePath'>>;\n\nexport interface Instance {\n    addPersonaPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContextPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addInstructionPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContentPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContent(content: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContext(context: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    loadContext(contextDirectories: string[], sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    loadContent(contentDirectories: string[], sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    build(): Promise<Prompt>;\n}\n\nexport const create = (builderOptions: OptionsParam): Instance => {\n    const options: Required<Options> = OptionSchema.parse(builderOptions) as Required<Options>;\n\n    const logger = wrapLogger(options.logger, 'Builder');\n    const parser = Parser.create({ logger });\n    const override = Override.create({\n        logger, configDirs: options.overridePaths || [\"./\"],\n        overrides: options.overrides || false\n    });\n    const loader = Loader.create({ logger });\n\n    const personaSection: Section<Instruction> = createSection({ title: \"Persona\" });\n    const contextSection: Section<Context> = createSection({ title: \"Context\" });\n    const instructionSection: Section<Instruction> = createSection({ title: \"Instruction\" });\n    const contentSection: Section<Content> = createSection({ title: \"Content\" });\n    const parameters = options.parameters;\n\n\n    const instance: Partial<Instance> = {}\n\n    const loadOptions = (sectionOptions: Partial<SectionOptions> = {}): SectionOptions => {\n        const currentOptions = SectionOptionsSchema.parse(sectionOptions);\n        return {\n            ...currentOptions,\n            parameters: {\n                ...parameters,\n                ...currentOptions.parameters\n            }\n        }\n    }\n\n    const loadDirectories = async <T extends Weighted>(\n        directories: string[],\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Section<T>[]> => {\n        const currentOptions = loadOptions(sectionOptions);\n        logger.debug(\"Loading directories\", directories);\n        const sections: Section<T>[] = await loader.load<T>(directories, currentOptions);\n        return sections;\n    }\n\n    const loadContext = async (\n        contextDirectories: string[],\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        const currentOptions = loadOptions(sectionOptions);\n        logger.debug('Loading context', contextDirectories);\n        const context: Section<Context>[] = await loadDirectories<Context>(contextDirectories, currentOptions);\n        contextSection.add(context);\n        return instance as Instance;\n    }\n    instance.loadContext = loadContext;\n\n    const loadContent = async (\n        contentDirectories: string[],\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        const currentOptions = loadOptions(sectionOptions);\n        const content: Section<Content>[] = await loadDirectories<Content>(contentDirectories, currentOptions);\n        contentSection.add(content);\n        return instance as Instance;\n    }\n    instance.loadContent = loadContent;\n\n    const loadPath = async <T extends Weighted>(\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Section<T>> => {\n        const currentOptions = loadOptions(sectionOptions);\n        const defaultPath = path.join(options.basePath as string, contentPath);\n        const section: Section<T> = await parser.parseFile<T>(defaultPath, currentOptions);\n        const overrideSection = await override.customize<T>(contentPath, section, currentOptions);\n        return overrideSection;\n    }\n\n    const addPersonaPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        const currentOptions = loadOptions(sectionOptions);\n        const persona: Section<Instruction> = await loadPath<Instruction>(contentPath, currentOptions);\n        personaSection.add(persona);\n        return instance as Instance;\n    }\n    instance.addPersonaPath = addPersonaPath;\n\n    const addContextPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding context path\", contentPath);\n        const currentOptions = loadOptions(sectionOptions);\n        const context: Section<Context> = await loadPath<Context>(contentPath, currentOptions);\n        contextSection.add(context);\n        return instance as Instance;\n    }\n    instance.addContextPath = addContextPath;\n\n    const addInstructionPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding instruction path\", contentPath);\n        const currentOptions = loadOptions(sectionOptions);\n        const instruction: Section<Instruction> = await loadPath<Instruction>(contentPath, currentOptions);\n        instructionSection.add(instruction);\n        return instance as Instance;\n    }\n    instance.addInstructionPath = addInstructionPath;\n\n    const addContentPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding content path\", contentPath);\n        const currentOptions = loadOptions(sectionOptions);\n        const content: Section<Content> = await loadPath<Content>(contentPath, currentOptions);\n        contentSection.add(content);\n        return instance as Instance;\n    }\n    instance.addContentPath = addContentPath;\n\n    const addContent = async (\n        content: string | Buffer,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding content\", typeof content);\n        const currentOptions = loadOptions(sectionOptions);\n        const parsedContentSection: Section<Content> = await parser.parse<Content>(content, currentOptions);\n        contentSection.add(parsedContentSection);\n        return instance as Instance;\n    }\n    instance.addContent = addContent;\n\n    const addContext = async (\n        context: string | Buffer,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding context\", typeof context);\n        const currentOptions = loadOptions(sectionOptions);\n        const parsedContextSection: Section<Context> = await parser.parse<Context>(context, currentOptions);\n        contextSection.add(parsedContextSection);\n        return instance as Instance;\n    }\n    instance.addContext = addContext;\n\n    const build = async () => {\n        logger.debug(\"Building prompt\", {});\n        const prompt = createPrompt({ persona: personaSection, contexts: contextSection, instructions: instructionSection, contents: contentSection });\n        return prompt;\n    }\n    instance.build = build;\n\n    return instance as Instance;\n}\n"],"names":["OptionSchema","z","object","logger","any","optional","default","DEFAULT_LOGGER","basePath","string","overridePaths","array","overrides","boolean","parameters","ParametersSchema","create","builderOptions","options","parse","wrapLogger","parser","Parser","override","Override","configDirs","loader","Loader","personaSection","createSection","title","contextSection","instructionSection","contentSection","instance","loadOptions","sectionOptions","currentOptions","SectionOptionsSchema","loadDirectories","directories","debug","sections","load","loadContext","contextDirectories","context","add","loadContent","contentDirectories","content","loadPath","contentPath","defaultPath","path","join","section","parseFile","overrideSection","customize","addPersonaPath","persona","addContextPath","addInstructionPath","instruction","addContentPath","addContent","parsedContentSection","addContext","parsedContextSection","build","prompt","createPrompt","contexts","instructions","contents"],"mappings":";;;;;;;;;;;;;;;;AAOA,MAAMA,YAAAA,GAAeC,CAAAA,CAAEC,MAAM,CAAC;AAC1BC,IAAAA,MAAAA,EAAQF,EAAEG,GAAG,EAAA,CAAGC,QAAQ,EAAA,CAAGC,OAAO,CAACC,cAAAA,CAAAA;AACnCC,IAAAA,QAAAA,EAAUP,EAAEQ,MAAM,EAAA;IAClBC,aAAAA,EAAeT,CAAAA,CAAEU,KAAK,CAACV,CAAAA,CAAEQ,MAAM,EAAA,CAAA,CAAIJ,QAAQ,EAAA,CAAGC,OAAO,CAAC;AAAC,QAAA;AAAK,KAAA,CAAA;AAC5DM,IAAAA,SAAAA,EAAWX,EAAEY,OAAO,EAAA,CAAGR,QAAQ,EAAA,CAAGC,OAAO,CAAC,KAAA,CAAA;AAC1CQ,IAAAA,UAAAA,EAAYC,gBAAAA,CAAiBV,QAAQ,EAAA,CAAGC,OAAO,CAAC,EAAC;AACrD,CAAA,CAAA;AAkBO,MAAMU,SAAS,CAACC,cAAAA,GAAAA;IACnB,MAAMC,OAAAA,GAA6BlB,YAAAA,CAAamB,KAAK,CAACF,cAAAA,CAAAA;AAEtD,IAAA,MAAMd,MAAAA,GAASiB,UAAAA,CAAWF,OAAAA,CAAQf,MAAM,EAAE,SAAA,CAAA;IAC1C,MAAMkB,QAAAA,GAASC,QAAa,CAAC;AAAEnB,QAAAA;AAAO,KAAA,CAAA;IACtC,MAAMoB,UAAAA,GAAWC,QAAe,CAAC;AAC7BrB,QAAAA,MAAAA;QAAQsB,UAAAA,EAAYP,OAAAA,CAAQR,aAAa,IAAI;AAAC,YAAA;AAAK,SAAA;QACnDE,SAAAA,EAAWM,OAAAA,CAAQN,SAAS,IAAI;AACpC,KAAA,CAAA;IACA,MAAMc,QAAAA,GAASC,QAAa,CAAC;AAAExB,QAAAA;AAAO,KAAA,CAAA;AAEtC,IAAA,MAAMyB,iBAAuCC,QAAAA,CAAc;QAAEC,KAAAA,EAAO;AAAU,KAAA,CAAA;AAC9E,IAAA,MAAMC,iBAAmCF,QAAAA,CAAc;QAAEC,KAAAA,EAAO;AAAU,KAAA,CAAA;AAC1E,IAAA,MAAME,qBAA2CH,QAAAA,CAAc;QAAEC,KAAAA,EAAO;AAAc,KAAA,CAAA;AACtF,IAAA,MAAMG,iBAAmCJ,QAAAA,CAAc;QAAEC,KAAAA,EAAO;AAAU,KAAA,CAAA;IAC1E,MAAMhB,UAAAA,GAAaI,QAAQJ,UAAU;AAGrC,IAAA,MAAMoB,WAA8B,EAAC;AAErC,IAAA,MAAMC,WAAAA,GAAc,CAACC,cAAAA,GAA0C,EAAE,GAAA;QAC7D,MAAMC,cAAAA,GAAiBC,oBAAAA,CAAqBnB,KAAK,CAACiB,cAAAA,CAAAA;QAClD,OAAO;AACH,YAAA,GAAGC,cAAc;YACjBvB,UAAAA,EAAY;AACR,gBAAA,GAAGA,UAAU;AACb,gBAAA,GAAGuB,eAAevB;AACtB;AACJ,SAAA;AACJ,IAAA,CAAA;AAEA,IAAA,MAAMyB,eAAAA,GAAkB,OACpBC,WAAAA,EACAJ,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;QACnCjC,MAAAA,CAAOsC,KAAK,CAAC,qBAAA,EAAuBD,WAAAA,CAAAA;AACpC,QAAA,MAAME,QAAAA,GAAyB,MAAMhB,QAAAA,CAAOiB,IAAI,CAAIH,WAAAA,EAAaH,cAAAA,CAAAA;QACjE,OAAOK,QAAAA;AACX,IAAA,CAAA;AAEA,IAAA,MAAME,WAAAA,GAAc,OAChBC,kBAAAA,EACAT,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;QACnCjC,MAAAA,CAAOsC,KAAK,CAAC,iBAAA,EAAmBI,kBAAAA,CAAAA;QAChC,MAAMC,OAAAA,GAA8B,MAAMP,eAAAA,CAAyBM,kBAAAA,EAAoBR,cAAAA,CAAAA;AACvFN,QAAAA,cAAAA,CAAegB,GAAG,CAACD,OAAAA,CAAAA;QACnB,OAAOZ,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAASU,WAAW,GAAGA,WAAAA;AAEvB,IAAA,MAAMI,WAAAA,GAAc,OAChBC,kBAAAA,EACAb,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;QACnC,MAAMc,OAAAA,GAA8B,MAAMX,eAAAA,CAAyBU,kBAAAA,EAAoBZ,cAAAA,CAAAA;AACvFJ,QAAAA,cAAAA,CAAec,GAAG,CAACG,OAAAA,CAAAA;QACnB,OAAOhB,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAASc,WAAW,GAAGA,WAAAA;AAEvB,IAAA,MAAMG,QAAAA,GAAW,OACbC,WAAAA,EACAhB,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;AACnC,QAAA,MAAMiB,cAAcC,aAAAA,CAAKC,IAAI,CAACrC,OAAAA,CAAQV,QAAQ,EAAY4C,WAAAA,CAAAA;AAC1D,QAAA,MAAMI,OAAAA,GAAsB,MAAMnC,QAAAA,CAAOoC,SAAS,CAAIJ,WAAAA,EAAahB,cAAAA,CAAAA;AACnE,QAAA,MAAMqB,kBAAkB,MAAMnC,UAAAA,CAASoC,SAAS,CAAIP,aAAaI,OAAAA,EAASnB,cAAAA,CAAAA;QAC1E,OAAOqB,eAAAA;AACX,IAAA,CAAA;AAEA,IAAA,MAAME,cAAAA,GAAiB,OACnBR,WAAAA,EACAhB,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;QACnC,MAAMyB,OAAAA,GAAgC,MAAMV,QAAAA,CAAsBC,WAAAA,EAAaf,cAAAA,CAAAA;AAC/ET,QAAAA,cAAAA,CAAemB,GAAG,CAACc,OAAAA,CAAAA;QACnB,OAAO3B,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAAS0B,cAAc,GAAGA,cAAAA;AAE1B,IAAA,MAAME,cAAAA,GAAiB,OACnBV,WAAAA,EACAhB,cAAAA,GAA0C,EAAE,GAAA;QAE5CjC,MAAAA,CAAOsC,KAAK,CAAC,qBAAA,EAAuBW,WAAAA,CAAAA;AACpC,QAAA,MAAMf,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;QACnC,MAAMU,OAAAA,GAA4B,MAAMK,QAAAA,CAAkBC,WAAAA,EAAaf,cAAAA,CAAAA;AACvEN,QAAAA,cAAAA,CAAegB,GAAG,CAACD,OAAAA,CAAAA;QACnB,OAAOZ,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAAS4B,cAAc,GAAGA,cAAAA;AAE1B,IAAA,MAAMC,kBAAAA,GAAqB,OACvBX,WAAAA,EACAhB,cAAAA,GAA0C,EAAE,GAAA;QAE5CjC,MAAAA,CAAOsC,KAAK,CAAC,yBAAA,EAA2BW,WAAAA,CAAAA;AACxC,QAAA,MAAMf,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;QACnC,MAAM4B,WAAAA,GAAoC,MAAMb,QAAAA,CAAsBC,WAAAA,EAAaf,cAAAA,CAAAA;AACnFL,QAAAA,kBAAAA,CAAmBe,GAAG,CAACiB,WAAAA,CAAAA;QACvB,OAAO9B,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAAS6B,kBAAkB,GAAGA,kBAAAA;AAE9B,IAAA,MAAME,cAAAA,GAAiB,OACnBb,WAAAA,EACAhB,cAAAA,GAA0C,EAAE,GAAA;QAE5CjC,MAAAA,CAAOsC,KAAK,CAAC,qBAAA,EAAuBW,WAAAA,CAAAA;AACpC,QAAA,MAAMf,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;QACnC,MAAMc,OAAAA,GAA4B,MAAMC,QAAAA,CAAkBC,WAAAA,EAAaf,cAAAA,CAAAA;AACvEJ,QAAAA,cAAAA,CAAec,GAAG,CAACG,OAAAA,CAAAA;QACnB,OAAOhB,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAAS+B,cAAc,GAAGA,cAAAA;AAE1B,IAAA,MAAMC,UAAAA,GAAa,OACfhB,OAAAA,EACAd,cAAAA,GAA0C,EAAE,GAAA;QAE5CjC,MAAAA,CAAOsC,KAAK,CAAC,gBAAA,EAAkB,OAAOS,OAAAA,CAAAA;AACtC,QAAA,MAAMb,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;AACnC,QAAA,MAAM+B,oBAAAA,GAAyC,MAAM9C,QAAAA,CAAOF,KAAK,CAAU+B,OAAAA,EAASb,cAAAA,CAAAA;AACpFJ,QAAAA,cAAAA,CAAec,GAAG,CAACoB,oBAAAA,CAAAA;QACnB,OAAOjC,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAASgC,UAAU,GAAGA,UAAAA;AAEtB,IAAA,MAAME,UAAAA,GAAa,OACftB,OAAAA,EACAV,cAAAA,GAA0C,EAAE,GAAA;QAE5CjC,MAAAA,CAAOsC,KAAK,CAAC,gBAAA,EAAkB,OAAOK,OAAAA,CAAAA;AACtC,QAAA,MAAMT,iBAAiBF,WAAAA,CAAYC,cAAAA,CAAAA;AACnC,QAAA,MAAMiC,oBAAAA,GAAyC,MAAMhD,QAAAA,CAAOF,KAAK,CAAU2B,OAAAA,EAAST,cAAAA,CAAAA;AACpFN,QAAAA,cAAAA,CAAegB,GAAG,CAACsB,oBAAAA,CAAAA;QACnB,OAAOnC,QAAAA;AACX,IAAA,CAAA;AACAA,IAAAA,QAAAA,CAASkC,UAAU,GAAGA,UAAAA;AAEtB,IAAA,MAAME,KAAAA,GAAQ,UAAA;QACVnE,MAAAA,CAAOsC,KAAK,CAAC,iBAAA,EAAmB,EAAC,CAAA;AACjC,QAAA,MAAM8B,SAASC,QAAAA,CAAa;YAAEX,OAAAA,EAASjC,cAAAA;YAAgB6C,QAAAA,EAAU1C,cAAAA;YAAgB2C,YAAAA,EAAc1C,kBAAAA;YAAoB2C,QAAAA,EAAU1C;AAAe,SAAA,CAAA;QAC5I,OAAOsC,MAAAA;AACX,IAAA,CAAA;AACArC,IAAAA,QAAAA,CAASoC,KAAK,GAAGA,KAAAA;IAEjB,OAAOpC,QAAAA;AACX;;;;"}

package/dist/chat.d.ts

@@ -1,5 +1,5 @@
 export type Role = "user" | "assistant" | "system" | "developer";
-export type Model = "gpt-4o" | "gpt-4o-mini" | "o1-preview" | "o1-mini" | "o1" | "o3-mini" | "o1-pro";
+export type Model = string;
 export interface Message {
     role: Role;
     content: string | string[];

package/dist/chat.js

@@ -1,10 +1,7 @@
-import { DEFAULT_PERSONA_ROLE } from './constants.js';
+import { getPersonaRole as getPersonaRole$1 } from './model-config.js';
 
 const getPersonaRole = (model)=>{
-    if (model === "gpt-4o" || model === "gpt-4o-mini") {
-        return "system";
-    }
-    return DEFAULT_PERSONA_ROLE;
+    return getPersonaRole$1(model);
 };
 const createRequest = (model)=>{
     const messages = [];

package/dist/chat.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat.js","sources":["../src/chat.ts"],"sourcesContent":["import { DEFAULT_PERSONA_ROLE } from \"./constants\";\n\nexport type Role = \"user\" | \"assistant\" | \"system\" | \"developer\";\n\nexport type Model = \"gpt-4o\" | \"gpt-4o-mini\" | \"o1-preview\" | \"o1-mini\" | \"o1\" | \"o3-mini\" | \"o1-pro\";\n\nexport interface Message {\n    role: Role;\n    content: string | string[];\n    name?: string;\n}\n\nexport interface Request {\n    messages: Message[];\n    model: Model;\n\n    addMessage(message: Message): void;\n}\n\nexport const getPersonaRole = (model: Model): Role => {\n    if (model === \"gpt-4o\" || model === \"gpt-4o-mini\") {\n        return \"system\";\n    }\n    return DEFAULT_PERSONA_ROLE;\n}\n\nexport const createRequest = (model: Model): Request => {\n    const messages: Message[] = [];\n\n    return {\n        model,\n        messages,\n        addMessage: (message: Message) => {\n            messages.push(message);\n        }\n    }\n}\n"],"names":["getPersonaRole","model","DEFAULT_PERSONA_ROLE","createRequest","messages","addMessage","message","push"],"mappings":";;AAmBO,MAAMA,iBAAiB,CAACC,KAAAA,GAAAA;IAC3B,IAAIA,KAAAA,KAAU,QAAA,IAAYA,KAAAA,KAAU,aAAA,EAAe;QAC/C,OAAO,QAAA;AACX,IAAA;IACA,OAAOC,oBAAAA;AACX;AAEO,MAAMC,gBAAgB,CAACF,KAAAA,GAAAA;AAC1B,IAAA,MAAMG,WAAsB,EAAE;IAE9B,OAAO;AACHH,QAAAA,KAAAA;AACAG,QAAAA,QAAAA;AACAC,QAAAA,UAAAA,EAAY,CAACC,OAAAA,GAAAA;AACTF,YAAAA,QAAAA,CAASG,IAAI,CAACD,OAAAA,CAAAA;AAClB,QAAA;AACJ,KAAA;AACJ;;;;"}
+{"version":3,"file":"chat.js","sources":["../src/chat.ts"],"sourcesContent":["import { getPersonaRole as getPersonaRoleFromRegistry } from \"./model-config\";\n\nexport type Role = \"user\" | \"assistant\" | \"system\" | \"developer\";\n\n// Model is now a flexible string type\nexport type Model = string;\n\nexport interface Message {\n    role: Role;\n    content: string | string[];\n    name?: string;\n}\n\nexport interface Request {\n    messages: Message[];\n    model: Model;\n\n    addMessage(message: Message): void;\n}\n\nexport const getPersonaRole = (model: Model): Role => {\n    return getPersonaRoleFromRegistry(model);\n}\n\nexport const createRequest = (model: Model): Request => {\n    const messages: Message[] = [];\n\n    return {\n        model,\n        messages,\n        addMessage: (message: Message) => {\n            messages.push(message);\n        }\n    }\n}\n"],"names":["getPersonaRole","model","getPersonaRoleFromRegistry","createRequest","messages","addMessage","message","push"],"mappings":";;AAoBO,MAAMA,iBAAiB,CAACC,KAAAA,GAAAA;AAC3B,IAAA,OAAOC,gBAAAA,CAA2BD,KAAAA,CAAAA;AACtC;AAEO,MAAME,gBAAgB,CAACF,KAAAA,GAAAA;AAC1B,IAAA,MAAMG,WAAsB,EAAE;IAE9B,OAAO;AACHH,QAAAA,KAAAA;AACAG,QAAAA,QAAAA;AACAC,QAAAA,UAAAA,EAAY,CAACC,OAAAA,GAAAA;AACTF,YAAAA,QAAAA,CAASG,IAAI,CAACD,OAAAA,CAAAA;AAClB,QAAA;AACJ,KAAA;AACJ;;;;"}

package/dist/constants.js

@@ -1,6 +1,5 @@
 const DEFAULT_CHARACTER_ENCODING = "utf8";
 const LIBRARY_NAME = "riotprompt";
-const DEFAULT_PERSONA_ROLE = "developer";
 const DEFAULT_IGNORE_PATTERNS = [
     "^\\..*",
     "\\.(jpg|jpeg|png|gif|bmp|svg|webp|ico)$",
@@ -19,5 +18,5 @@ const DEFAULT_FORMAT_OPTIONS = {
     sectionDepth: 0
 };
 
-export { DEFAULT_CHARACTER_ENCODING, DEFAULT_FORMAT_OPTIONS, DEFAULT_IGNORE_PATTERNS, DEFAULT_PERSONA_ROLE, DEFAULT_SECTION_INDENTATION, DEFAULT_SECTION_SEPARATOR, DEFAULT_SECTION_TITLE_PROPERTY, LIBRARY_NAME };
+export { DEFAULT_CHARACTER_ENCODING, DEFAULT_FORMAT_OPTIONS, DEFAULT_IGNORE_PATTERNS, DEFAULT_SECTION_INDENTATION, DEFAULT_SECTION_SEPARATOR, DEFAULT_SECTION_TITLE_PROPERTY, LIBRARY_NAME };
 //# sourceMappingURL=constants.js.map

package/dist/constants.js.map

@@ -1 +1 @@
-{"version":3,"file":"constants.js","sources":["../src/constants.ts"],"sourcesContent":["import { SectionSeparator } from \"formatter\";\n\nimport { FormatOptions } from \"formatter\";\n\nexport const DEFAULT_CHARACTER_ENCODING = \"utf8\";\nexport const LIBRARY_NAME = \"riotprompt\";\n\nexport const DEFAULT_PERSONA_ROLE = \"developer\";\n\nexport const DEFAULT_INSTRUCTIONS_AREA_TITLE = \"Instructions\";\nexport const DEFAULT_CONTENTS_AREA_TITLE = \"Contents\";\nexport const DEFAULT_CONTEXT_AREA_TITLE = \"Context\";\n\nexport const DEFAULT_IGNORE_PATTERNS: string[] = [\n    \"^\\\\..*\", // Hidden files (e.g., .git, .DS_Store)\n    \"\\\\.(jpg|jpeg|png|gif|bmp|svg|webp|ico)$\", // Image files\n    \"\\\\.(mp3|wav|ogg|aac|flac)$\", // Audio files\n    \"\\\\.(mp4|mov|avi|mkv|webm)$\", // Video files\n    \"\\\\.(pdf|doc|docx|xls|xlsx|ppt|pptx)$\", // Document files\n    \"\\\\.(zip|tar|gz|rar|7z)$\" // Compressed files\n];\n\nexport const DEFAULT_SECTION_SEPARATOR: SectionSeparator = \"tag\";\nexport const DEFAULT_SECTION_INDENTATION = true;\nexport const DEFAULT_SECTION_TAG = \"section\";\nexport const DEFAULT_SECTION_TITLE_PROPERTY = \"title\";\n\nexport const DEFAULT_FORMAT_OPTIONS: FormatOptions = {\n    sectionSeparator: DEFAULT_SECTION_SEPARATOR,\n    sectionIndentation: DEFAULT_SECTION_INDENTATION,\n    sectionTitleProperty: DEFAULT_SECTION_TITLE_PROPERTY,\n    sectionDepth: 0,\n}\n"],"names":["DEFAULT_CHARACTER_ENCODING","LIBRARY_NAME","DEFAULT_PERSONA_ROLE","DEFAULT_IGNORE_PATTERNS","DEFAULT_SECTION_SEPARATOR","DEFAULT_SECTION_INDENTATION","DEFAULT_SECTION_TITLE_PROPERTY","DEFAULT_FORMAT_OPTIONS","sectionSeparator","sectionIndentation","sectionTitleProperty","sectionDepth"],"mappings":"AAIO,MAAMA,6BAA6B;AACnC,MAAMC,eAAe;AAErB,MAAMC,uBAAuB;MAMvBC,uBAAAA,GAAoC;AAC7C,IAAA,QAAA;AACA,IAAA,yCAAA;AACA,IAAA,4BAAA;AACA,IAAA,4BAAA;AACA,IAAA,sCAAA;AACA,IAAA,yBAAA;;AAGG,MAAMC,4BAA8C;AACpD,MAAMC,8BAA8B;AAEpC,MAAMC,iCAAiC;MAEjCC,sBAAAA,GAAwC;IACjDC,gBAAAA,EAAkBJ,yBAAAA;IAClBK,kBAAAA,EAAoBJ,2BAAAA;IACpBK,oBAAAA,EAAsBJ,8BAAAA;IACtBK,YAAAA,EAAc;AAClB;;;;"}
+{"version":3,"file":"constants.js","sources":["../src/constants.ts"],"sourcesContent":["import { SectionSeparator } from \"formatter\";\n\nimport { FormatOptions } from \"formatter\";\n\nexport const DEFAULT_CHARACTER_ENCODING = \"utf8\";\nexport const LIBRARY_NAME = \"riotprompt\";\n\nexport const DEFAULT_PERSONA_ROLE = \"developer\";\n\nexport const DEFAULT_INSTRUCTIONS_AREA_TITLE = \"Instructions\";\nexport const DEFAULT_CONTENTS_AREA_TITLE = \"Contents\";\nexport const DEFAULT_CONTEXT_AREA_TITLE = \"Context\";\n\nexport const DEFAULT_IGNORE_PATTERNS: string[] = [\n    \"^\\\\..*\", // Hidden files (e.g., .git, .DS_Store)\n    \"\\\\.(jpg|jpeg|png|gif|bmp|svg|webp|ico)$\", // Image files\n    \"\\\\.(mp3|wav|ogg|aac|flac)$\", // Audio files\n    \"\\\\.(mp4|mov|avi|mkv|webm)$\", // Video files\n    \"\\\\.(pdf|doc|docx|xls|xlsx|ppt|pptx)$\", // Document files\n    \"\\\\.(zip|tar|gz|rar|7z)$\" // Compressed files\n];\n\nexport const DEFAULT_SECTION_SEPARATOR: SectionSeparator = \"tag\";\nexport const DEFAULT_SECTION_INDENTATION = true;\nexport const DEFAULT_SECTION_TAG = \"section\";\nexport const DEFAULT_SECTION_TITLE_PROPERTY = \"title\";\n\nexport const DEFAULT_FORMAT_OPTIONS: FormatOptions = {\n    sectionSeparator: DEFAULT_SECTION_SEPARATOR,\n    sectionIndentation: DEFAULT_SECTION_INDENTATION,\n    sectionTitleProperty: DEFAULT_SECTION_TITLE_PROPERTY,\n    sectionDepth: 0,\n}\n"],"names":["DEFAULT_CHARACTER_ENCODING","LIBRARY_NAME","DEFAULT_IGNORE_PATTERNS","DEFAULT_SECTION_SEPARATOR","DEFAULT_SECTION_INDENTATION","DEFAULT_SECTION_TITLE_PROPERTY","DEFAULT_FORMAT_OPTIONS","sectionSeparator","sectionIndentation","sectionTitleProperty","sectionDepth"],"mappings":"AAIO,MAAMA,6BAA6B;AACnC,MAAMC,eAAe;MAQfC,uBAAAA,GAAoC;AAC7C,IAAA,QAAA;AACA,IAAA,yCAAA;AACA,IAAA,4BAAA;AACA,IAAA,4BAAA;AACA,IAAA,sCAAA;AACA,IAAA,yBAAA;;AAGG,MAAMC,4BAA8C;AACpD,MAAMC,8BAA8B;AAEpC,MAAMC,iCAAiC;MAEjCC,sBAAAA,GAAwC;IACjDC,gBAAAA,EAAkBJ,yBAAAA;IAClBK,kBAAAA,EAAoBJ,2BAAAA;IACpBK,oBAAAA,EAAsBJ,8BAAAA;IACtBK,YAAAA,EAAc;AAClB;;;;"}