neoctl 0.1.12 → 0.1.14

package/dist/repl/index.js

@@ -20,6 +20,7 @@ import { listDirectoryTool, readFileTool } from "../tools/builtins/filesystem-to
 import { grepTool } from "../tools/builtins/grep-tool.js";
 import { searchTool } from "../tools/builtins/search-tool.js";
 import { planTool } from "../tools/builtins/plan-tool.js";
+import { createOpenAIImageGenerationTool } from "../tools/builtins/image-generation-tool.js";
 import { createAgentTool, resumeAgentTask } from "../agents/agent-tool.js";
 import { createTaskTools } from "../tasks/task-tools.js";
 import { TaskStore } from "../tasks/task-store.js";
@@ -156,6 +157,8 @@ async function createRuntime() {
     tools.register(readFileTool);
     tools.register(grepTool);
     tools.register(searchTool);
+    if (modelConfig?.provider === "openai")
+        tools.register(createOpenAIImageGenerationTool());
     tools.register(planTool);
     const agentRuntime = { modelGateway, tools, taskStore };
     tools.register(createAgentTool(agentRuntime));
@@ -199,12 +202,18 @@ async function createRuntime() {
         agentRuntime,
         usage: new SessionUsageTracker(),
         taskStore,
+        tools,
         initialMetrics,
         defaultReasoning: modelConfig?.defaultReasoning,
         envPath: process.env.NEO_ENV_FILE?.trim() ? path.resolve(process.env.NEO_ENV_FILE.trim()) : envLoad.userDotEnvPath,
         envNotice: envLoad.createdUserDotEnv ? formatCreatedEnvNotice(envLoad.userDotEnvPath) : undefined,
     };
 }
+function syncImageGenerationTool(runtime, provider) {
+    runtime.tools.unregister("image2");
+    if (provider === "openai")
+        runtime.tools.register(createOpenAIImageGenerationTool());
+}
 function formatCreatedEnvNotice(path) {
     return `Created default config file: ${path}\nSet MODEL_PROVIDER and the matching provider section (for example OPENAI_API_KEY or KIMI_API_KEY), then restart neo.`;
 }
@@ -1979,6 +1988,7 @@ async function handleModelCommand(command, runtime) {
                         fallbackModel: config.fallbackModel,
                         reasoning: config.defaultReasoning,
                     });
+                    syncImageGenerationTool(runtime, config.provider);
                     runtime.defaultReasoning = config.defaultReasoning;
                 }
             }
@@ -2605,6 +2615,7 @@ async function submitLoginForm(state, runtime, append, setLoginFormState, setSta
             fallbackModel: config.fallbackModel,
             reasoning: config.defaultReasoning,
         });
+        syncImageGenerationTool(runtime, config.provider);
         runtime.defaultReasoning = config.defaultReasoning;
         const metrics = await runtime.engine.contextMetrics();
         setStatus((current) => ({