From 32b1a9e0c82e4f8b7b3d07be641d02d18b18e438 Mon Sep 17 00:00:00 2001
From: jnjpng <jinjpeng@gmail.com>
Date: Mon, 23 Feb 2026 11:01:59 -0800
Subject: [PATCH] refactor: hoist resume model-refresh imports (#1093)

Co-authored-by: Letta <noreply@letta.com>
---
 src/headless.ts | 12 ++++++------
 src/index.ts    | 16 ++++++----------
 2 files changed, 12 insertions(+), 16 deletions(-)

diff --git a/src/headless.ts b/src/headless.ts
index 007b9e9..db20e57 100644
--- a/src/headless.ts
+++ b/src/headless.ts
@@ -25,7 +25,12 @@ import { createAgent } from "./agent/create";
 import { handleListMessages } from "./agent/listMessagesHandler";
 import { ISOLATED_BLOCK_LABELS } from "./agent/memory";
 import { getStreamToolContextId, sendMessageStream } from "./agent/message";
-import { getModelUpdateArgs } from "./agent/model";
+import {
+  getModelPresetUpdateForAgent,
+  getModelUpdateArgs,
+  resolveModel,
+} from "./agent/model";
+import { updateAgentLLMConfig, updateAgentSystemPrompt } from "./agent/modify";
 import { resolveSkillSourcesSelection } from "./agent/skillSources";
 import type { SkillSource } from "./agent/skills";
 import { SessionStats } from "./agent/stats";
@@ -898,10 +903,7 @@ export async function handleHeadlessCommand(
   // preset-derived fields in sync, then apply optional command-line
   // overrides (model/system prompt).
   if (isResumingAgent) {
-    const { updateAgentLLMConfig } = await import("./agent/modify");
-
     if (model) {
-      const { resolveModel } = await import("./agent/model");
       const modelHandle = resolveModel(model);
       if (typeof modelHandle !== "string") {
         console.error(`Error: Invalid model "${model}"`);
@@ -915,7 +917,6 @@ export async function handleHeadlessCommand(
       // Refresh agent state after model update
       agent = await client.agents.retrieve(agent.id);
     } else {
-      const { getModelPresetUpdateForAgent } = await import("./agent/model");
       const presetRefresh = getModelPresetUpdateForAgent(agent);
       if (presetRefresh) {
         // Resume preset refresh is intentionally scoped for now.
@@ -947,7 +948,6 @@ export async function handleHeadlessCommand(
     }
 
     if (systemPromptPreset) {
-      const { updateAgentSystemPrompt } = await import("./agent/modify");
       const result = await updateAgentSystemPrompt(
         agent.id,
         systemPromptPreset,
diff --git a/src/index.ts b/src/index.ts
index 09581e8..316de9f 100755
--- a/src/index.ts
+++ b/src/index.ts
@@ -12,6 +12,12 @@ import {
 import type { AgentProvenance } from "./agent/create";
 import { getLettaCodeHeaders } from "./agent/http-headers";
 import { ISOLATED_BLOCK_LABELS } from "./agent/memory";
+import {
+  getModelPresetUpdateForAgent,
+  getModelUpdateArgs,
+  resolveModel,
+} from "./agent/model";
+import { updateAgentLLMConfig, updateAgentSystemPrompt } from "./agent/modify";
 import { resolveSkillSourcesSelection } from "./agent/skillSources";
 import { LETTA_CLOUD_API_URL } from "./auth/oauth";
 import { ConversationSelector } from "./cli/components/ConversationSelector";
@@ -1561,7 +1567,6 @@ async function main(): Promise<void> {
 
         setLoadingState("initializing");
         const { createAgent } = await import("./agent/create");
-        const { getModelUpdateArgs } = await import("./agent/model");
 
         let agent: AgentState | null = null;
 
@@ -1793,12 +1798,7 @@ async function main(): Promise<void> {
         // preset-derived fields in sync, then apply optional command-line
         // overrides (model/system prompt).
         if (resuming) {
-          const { updateAgentLLMConfig } = await import("./agent/modify");
-
           if (model) {
-            const { resolveModel, getModelUpdateArgs } = await import(
-              "./agent/model"
-            );
             const modelHandle = resolveModel(model);
             if (!modelHandle) {
               console.error(`Error: Invalid model "${model}"`);
@@ -1812,9 +1812,6 @@ async function main(): Promise<void> {
             // Refresh agent state after model update
             agent = await client.agents.retrieve(agent.id);
           } else {
-            const { getModelPresetUpdateForAgent } = await import(
-              "./agent/model"
-            );
             const presetRefresh = getModelPresetUpdateForAgent(agent);
             if (presetRefresh) {
               // Resume preset refresh is intentionally scoped for now.
@@ -1851,7 +1848,6 @@ async function main(): Promise<void> {
           }
 
           if (systemPromptPreset) {
-            const { updateAgentSystemPrompt } = await import("./agent/modify");
             const result = await updateAgentSystemPrompt(
               agent.id,
               systemPromptPreset,