@elyracode/coding-agent 0.1.0 → 0.3.0

package/dist/core/agent-session.js

@@ -26,9 +26,12 @@ 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 { detectStack } from "./stack-detector.js";
 import { buildSystemPrompt } from "./system-prompt.js";
 import { createLocalBashOperations } from "./tools/bash.js";
 import { createAllToolDefinitions } from "./tools/index.js";
@@ -130,6 +133,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,
@@ -506,7 +510,7 @@ export class AgentSession {
      * Call this when completely done with the session.
      */
     dispose() {
-        this._extensionRunner.invalidate("This extension ctx is stale after session replacement or reload. Do not use a captured pi or command ctx after ctx.newSession(), ctx.fork(), ctx.switchSession(), or ctx.reload(). For newSession, fork, and switchSession, move post-replacement work into withSession and use the ctx passed to withSession. For reload, do not use the old ctx after await ctx.reload().");
+        this._extensionRunner.invalidate("This extension ctx is stale after session replacement or reload. Do not use a captured elyra or command ctx after ctx.newSession(), ctx.fork(), ctx.switchSession(), or ctx.reload(). For newSession, fork, and switchSession, move post-replacement work into withSession and use the ctx passed to withSession. For reload, do not use the old ctx after await ctx.reload().");
         this._disconnectFromAgent();
         this._eventListeners = [];
         cleanupSessionResources(this.sessionId);
@@ -663,6 +667,11 @@ 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;
+        // Detect project technology stack
+        const stackResult = detectStack(this._cwd);
+        const stackInfo = stackResult.summary || undefined;
         this._baseSystemPromptOptions = {
             cwd: this._cwd,
             skills: loadedSkills,
@@ -672,6 +681,8 @@ export class AgentSession {
             selectedTools: validToolNames,
             toolSnippets,
             promptGuidelines,
+            memoryContent,
+            stackInfo,
         };
         return buildSystemPrompt(this._baseSystemPromptOptions);
     }
@@ -680,7 +691,7 @@ export class AgentSession {
     // =========================================================================
     /**
      * Send a prompt to the agent.
-     * - Handles extension commands (registered via pi.registerCommand) immediately, even during streaming
+     * - Handles extension commands (registered via elyra.registerCommand) immediately, even during streaming
      * - Expands file-based prompt templates by default
      * - During streaming, queues via steer() or followUp() based on streamingBehavior option
      * - Validates model and API key before sending (when not streaming)
@@ -693,7 +704,7 @@ export class AgentSession {
         let messages;
         try {
             // Handle extension commands first (execute immediately, even during streaming)
-            // Extension commands manage their own LLM interaction via pi.sendMessage()
+            // Extension commands manage their own LLM interaction via elyra.sendMessage()
             if (expandPromptTemplates && text.startsWith("/")) {
                 const handled = await this._tryExecuteExtensionCommand(text);
                 if (handled) {
@@ -1857,6 +1868,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();