@pedrohnas/opencode-telegram 1.2.1 → 1.3.0

package/src/session-manager.test.ts

@@ -196,3 +196,49 @@ describe("SessionManager", () => {
     expect(sm.get("12345")?.sessionId).toBe("existing")
   })
 })
+
+// --- Phase 5: model/agent override fields ---
+
+describe("SessionEntry overrides", () => {
+  let sm: SessionManager
+
+  beforeEach(() => {
+    sm = new SessionManager({ maxEntries: 10, ttlMs: 60000 })
+  })
+
+  test("set/get preserves modelOverride", () => {
+    sm.set("123", {
+      sessionId: "s1",
+      directory: "/tmp",
+      modelOverride: { providerID: "anthropic", modelID: "claude-sonnet-4-5-20250929" },
+    })
+    const entry = sm.get("123")
+    expect(entry?.modelOverride).toEqual({
+      providerID: "anthropic",
+      modelID: "claude-sonnet-4-5-20250929",
+    })
+  })
+
+  test("set/get preserves agentOverride", () => {
+    sm.set("123", {
+      sessionId: "s1",
+      directory: "/tmp",
+      agentOverride: "code",
+    })
+    const entry = sm.get("123")
+    expect(entry?.agentOverride).toBe("code")
+  })
+
+  test("getOrCreate returns entry without overrides by default", async () => {
+    const sdk = {
+      session: {
+        create: mock(async () => ({
+          data: { id: "new-session", title: "", directory: "/tmp" },
+        })),
+      },
+    }
+    const entry = await sm.getOrCreate("456", sdk as any)
+    expect(entry.modelOverride).toBeUndefined()
+    expect(entry.agentOverride).toBeUndefined()
+  })
+})

package/src/session-manager.ts

@@ -15,6 +15,8 @@ export type SessionEntry = {
   directory: string
   createdAt: number
   lastAccessAt: number
+  modelOverride?: { providerID: string; modelID: string }
+  agentOverride?: string
 }
 
 export type SessionManagerOptions = {
@@ -79,7 +81,12 @@ export class SessionManager {
 
   set(
     chatKey: string,
-    init: { sessionId: string; directory: string },
+    init: {
+      sessionId: string
+      directory: string
+      modelOverride?: { providerID: string; modelID: string }
+      agentOverride?: string
+    },
   ): void {
     // Clean up old binding if exists
     const old = this.map.get(chatKey)
@@ -92,6 +99,8 @@ export class SessionManager {
       directory: init.directory,
       createdAt: Date.now(),
       lastAccessAt: Date.now(),
+      modelOverride: init.modelOverride,
+      agentOverride: init.agentOverride,
     }
 
     this.insert(chatKey, entry)