feat(memory): auto-wire memoryProvider on direct agent.stream() + drop dead MEMORY_TIMEOUT_MS

Author: jddunn

src/api/agent.ts

@@ -290,9 +290,6 @@ async function loadRecordedAgentOSUsage(
   return getRecordedAgentOSUsage(options);
 }
 
-/** Timeout for memory operations to prevent blocking generation. */
-const MEMORY_TIMEOUT_MS = 5000;
-
 /**
  * Convert HEXACO trait values (0-1) into behavioral descriptions the LLM can act on.
  *
@@ -473,13 +470,18 @@ export function agent(opts: AgentOptions): Agent {
     },
 
     stream(prompt: MessageContent, extra?: Partial<GenerateTextOptions>): StreamTextResult {
-      const streamOpts: Partial<GenerateTextOptions> = {
-        ...baseOpts,
-        ...extra,
-        usageLedger: mergeUsageLedgerOptions(baseOpts.usageLedger, extra?.usageLedger, {
-          source: extra?.usageLedger?.source ?? 'agent.stream',
-        }),
-      };
+      const userText = typeof prompt === 'string' ? prompt : extractTextFromContent(prompt);
+      const streamOpts: Partial<GenerateTextOptions> = applyMemoryProvider(
+        {
+          ...baseOpts,
+          ...extra,
+          usageLedger: mergeUsageLedgerOptions(baseOpts.usageLedger, extra?.usageLedger, {
+            source: extra?.usageLedger?.source ?? 'agent.stream',
+          }),
+        },
+        opts.memoryProvider,
+        userText,
+      );
       if (typeof prompt === 'string') {
         streamOpts.prompt = prompt;
       } else {

src/api/runtime/__tests__/agentPromptEngine.test.ts

@@ -219,6 +219,38 @@ describe('agent() PromptEngine/Memory/Skills integration', () => {
       const combined = systemMsgs.map((m: any) => m.content).join('\n');
       expect(combined).toContain('Memory: user likes hiking');
     });
+
+    it('calls getContext before direct agent.stream() (new in 0.2.0)', async () => {
+      const memory = createMockMemory();
+      const a = agent({ instructions: 'test', memoryProvider: memory });
+
+      const streamResult = a.stream('hello from stream');
+      // Drain the stream to ensure generation completes
+      for await (const _chunk of streamResult.textStream) {
+        // consume
+      }
+      await streamResult.text;
+
+      expect(memory.getContext).toHaveBeenCalledWith(
+        'hello from stream',
+        expect.objectContaining({ tokenBudget: expect.any(Number) }),
+      );
+    });
+
+    it('calls observe after direct agent.stream() completes (new in 0.2.0)', async () => {
+      const memory = createMockMemory();
+      const a = agent({ instructions: 'test', memoryProvider: memory });
+
+      const streamResult = a.stream('hello from stream');
+      for await (const _chunk of streamResult.textStream) {
+        // consume
+      }
+      await streamResult.text;
+      await new Promise((resolve) => setImmediate(resolve));
+
+      expect(memory.observe).toHaveBeenCalledWith('user', 'hello from stream');
+      expect(memory.observe).toHaveBeenCalledWith('assistant', 'streamed');
+    });
   });
 
   describe('all three compose together', () => {