@elyracode/coding-agent 0.1.0 → 0.2.0

package/dist/core/agent-session.js

@@ -26,8 +26,10 @@ import { exportSessionToHtml } from "./export-html/index.js";
 import { createToolHtmlRenderer } from "./export-html/tool-renderer.js";
 import { ExtensionRunner, wrapRegisteredTools, } from "./extensions/index.js";
 import { emitSessionShutdownEvent } from "./extensions/runner.js";
+import { loadMemory } from "./memory.js";
 import { expandPromptTemplate } from "./prompt-templates.js";
 import { CURRENT_SESSION_VERSION, getLatestCompactionEntry } from "./session-manager.js";
+import { routeNextTurn } from "./smart-router.js";
 import { createSyntheticSourceInfo } from "./source-info.js";
 import { buildSystemPrompt } from "./system-prompt.js";
 import { createLocalBashOperations } from "./tools/bash.js";
@@ -130,6 +132,7 @@ export class AgentSession {
         // (session persistence, extensions, auto-compaction, retry logic)
         this._unsubscribeAgent = this.agent.subscribe(this._handleAgentEvent);
         this._installAgentToolHooks();
+        this._installSmartRouting();
         this._buildRuntime({
             activeToolNames: this._initialActiveToolNames,
             includeAllExtensionTools: true,
@@ -663,6 +666,8 @@ export class AgentSession {
         const appendSystemPrompt = loaderAppendSystemPrompt.length > 0 ? loaderAppendSystemPrompt.join("\n\n") : undefined;
         const loadedSkills = this._resourceLoader.getSkills().skills;
         const loadedContextFiles = this._resourceLoader.getAgentsFiles().agentsFiles;
+        // Load codebase memory if enabled
+        const memoryContent = this.settingsManager.getCodebaseMemory() ? loadMemory(this._cwd) : undefined;
         this._baseSystemPromptOptions = {
             cwd: this._cwd,
             skills: loadedSkills,
@@ -672,6 +677,7 @@ export class AgentSession {
             selectedTools: validToolNames,
             toolSnippets,
             promptGuidelines,
+            memoryContent,
         };
         return buildSystemPrompt(this._baseSystemPromptOptions);
     }
@@ -1857,6 +1863,47 @@ export class AgentSession {
         }
         this.setActiveToolsByName([...new Set(nextActiveToolNames)]);
     }
+    /**
+     * Install smart model routing via the prepareNextTurn hook.
+     * When enabled, the router analyzes turn complexity and swaps to a
+     * cheaper or more powerful model automatically.
+     */
+    _installSmartRouting() {
+        const existingHook = this.agent.prepareNextTurn;
+        this.agent.prepareNextTurn = async (signal) => {
+            // Run existing hook first (if any)
+            const existing = existingHook ? await existingHook(signal) : undefined;
+            if (!this.settingsManager.getSmartRouting()) {
+                return existing;
+            }
+            // Build available models list from scoped models or all available
+            let availableModels;
+            if (this._scopedModels.length > 0) {
+                availableModels = this._scopedModels
+                    .filter((sm) => this._modelRegistry.hasConfiguredAuth(sm.model))
+                    .map((sm) => sm.model);
+            }
+            else {
+                availableModels = await this._modelRegistry.getAvailable();
+            }
+            if (availableModels.length <= 1) {
+                return existing;
+            }
+            const config = {
+                enabled: true,
+                availableModels,
+                preferredModel: this.agent.state.model,
+            };
+            const result = routeNextTurn({ messages: this.agent.state.messages, thinkingLevel: this.agent.state.thinkingLevel }, config);
+            if (!result) {
+                return existing;
+            }
+            return {
+                ...existing,
+                model: result.model,
+            };
+        };
+    }
     _buildRuntime(options) {
         const autoResizeImages = this.settingsManager.getImageAutoResize();
         const shellCommandPrefix = this.settingsManager.getShellCommandPrefix();