Add layered memory capture APIs and demos

DemoApp/AssistantRuntimeDemoApp.xcodeproj/project.pbxproj

@@ -22,6 +22,9 @@
 		1A2B3C4D5E6F70000000000D /* DemoUIComponents.swift in Sources */ = {isa = PBXBuildFile; fileRef = 2A2B3C4D5E6F70000000000D /* DemoUIComponents.swift */; };
 		1A2B3C4D5E6F70000000000E /* StructuredOutputDemoView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 2A2B3C4D5E6F70000000000E /* StructuredOutputDemoView.swift */; };
 		1A2B3C4D5E6F70000000000F /* ThreadDetailView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 2A2B3C4D5E6F70000000000F /* ThreadDetailView.swift */; };
+		1A2B3C4D5E6F700000000010 /* DemoMemoryExamples.swift in Sources */ = {isa = PBXBuildFile; fileRef = 2A2B3C4D5E6F700000000010 /* DemoMemoryExamples.swift */; };
+		1A2B3C4D5E6F700000000011 /* AgentDemoViewModel+Memory.swift in Sources */ = {isa = PBXBuildFile; fileRef = 2A2B3C4D5E6F700000000011 /* AgentDemoViewModel+Memory.swift */; };
+		1A2B3C4D5E6F700000000012 /* MemoryDemoView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 2A2B3C4D5E6F700000000012 /* MemoryDemoView.swift */; };
 		7482123BC63AC10F104DE092 /* AssistantRuntimeDemoApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 5A6999E6475919476E726E8C /* AssistantRuntimeDemoApp.swift */; };
 		84726927B752451499D9257F /* Foundation.framework in Frameworks */ = {isa = PBXBuildFile; fileRef = 906A95007C8ECB92CFC2CE15 /* Foundation.framework */; };
 		B060448C6464C41789B56EED /* AgentDemoView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 3CA22585116A120BA97F76B8 /* AgentDemoView.swift */; };
@@ -47,6 +50,9 @@
 		2A2B3C4D5E6F70000000000D /* DemoUIComponents.swift */ = {isa = PBXFileReference; includeInIndex = 1; lastKnownFileType = sourcecode.swift; path = DemoUIComponents.swift; sourceTree = "<group>"; };
 		2A2B3C4D5E6F70000000000E /* StructuredOutputDemoView.swift */ = {isa = PBXFileReference; includeInIndex = 1; lastKnownFileType = sourcecode.swift; path = StructuredOutputDemoView.swift; sourceTree = "<group>"; };
 		2A2B3C4D5E6F70000000000F /* ThreadDetailView.swift */ = {isa = PBXFileReference; includeInIndex = 1; lastKnownFileType = sourcecode.swift; path = ThreadDetailView.swift; sourceTree = "<group>"; };
+		2A2B3C4D5E6F700000000010 /* DemoMemoryExamples.swift */ = {isa = PBXFileReference; includeInIndex = 1; lastKnownFileType = sourcecode.swift; path = DemoMemoryExamples.swift; sourceTree = "<group>"; };
+		2A2B3C4D5E6F700000000011 /* AgentDemoViewModel+Memory.swift */ = {isa = PBXFileReference; includeInIndex = 1; lastKnownFileType = sourcecode.swift; path = "AgentDemoViewModel+Memory.swift"; sourceTree = "<group>"; };
+		2A2B3C4D5E6F700000000012 /* MemoryDemoView.swift */ = {isa = PBXFileReference; includeInIndex = 1; lastKnownFileType = sourcecode.swift; path = MemoryDemoView.swift; sourceTree = "<group>"; };
 		2481147A958D00EB4A70C928 /* AgentDemoViewModel.swift */ = {isa = PBXFileReference; includeInIndex = 1; lastKnownFileType = sourcecode.swift; path = AgentDemoViewModel.swift; sourceTree = "<group>"; };
 		3CA22585116A120BA97F76B8 /* AgentDemoView.swift */ = {isa = PBXFileReference; includeInIndex = 1; lastKnownFileType = sourcecode.swift; path = AgentDemoView.swift; sourceTree = "<group>"; };
 		5A6999E6475919476E726E8C /* AssistantRuntimeDemoApp.swift */ = {isa = PBXFileReference; includeInIndex = 1; lastKnownFileType = sourcecode.swift; path = AssistantRuntimeDemoApp.swift; sourceTree = "<group>"; };
@@ -137,6 +143,9 @@
 				2A2B3C4D5E6F70000000000D /* DemoUIComponents.swift */,
 				2A2B3C4D5E6F70000000000E /* StructuredOutputDemoView.swift */,
 				2A2B3C4D5E6F70000000000F /* ThreadDetailView.swift */,
+				2A2B3C4D5E6F700000000010 /* DemoMemoryExamples.swift */,
+				2A2B3C4D5E6F700000000011 /* AgentDemoViewModel+Memory.swift */,
+				2A2B3C4D5E6F700000000012 /* MemoryDemoView.swift */,
 			);
 			name = Shared;
 			path = Shared;
@@ -228,6 +237,9 @@
 				1A2B3C4D5E6F70000000000D /* DemoUIComponents.swift in Sources */,
 				1A2B3C4D5E6F70000000000E /* StructuredOutputDemoView.swift in Sources */,
 				1A2B3C4D5E6F70000000000F /* ThreadDetailView.swift in Sources */,
+				1A2B3C4D5E6F700000000010 /* DemoMemoryExamples.swift in Sources */,
+				1A2B3C4D5E6F700000000011 /* AgentDemoViewModel+Memory.swift in Sources */,
+				1A2B3C4D5E6F700000000012 /* MemoryDemoView.swift in Sources */,
 				7482123BC63AC10F104DE092 /* AssistantRuntimeDemoApp.swift in Sources */,
 				BB4F38E64D1EBBB3821AC4E3 /* AgentDemoRuntimeFactory.swift in Sources */,
 				B060448C6464C41789B56EED /* AgentDemoView.swift in Sources */,

DemoApp/AssistantRuntimeDemoApp/AssistantRuntimeDemoApp.swift

@@ -3,6 +3,7 @@ import SwiftUI
 enum DemoTab: Hashable {
     case assistant
     case structuredOutput
+    case memory
     case healthCoach
 }
 
@@ -45,6 +46,19 @@ struct AssistantRuntimeDemoApp: App {
                     Label("Structured", systemImage: "square.stack.3d.up")
                 }
 
+                NavigationStack {
+                    MemoryDemoView(
+                        viewModel: viewModel,
+                        selectedTab: $selectedTab
+                    )
+                    .navigationTitle("Memory")
+                    .navigationBarTitleDisplayMode(.inline)
+                }
+                .tag(DemoTab.memory)
+                .tabItem {
+                    Label("Memory", systemImage: "brain")
+                }
+
                 NavigationStack {
                     HealthCoachView(viewModel: viewModel)
                         .navigationTitle("Health Coach")

DemoApp/AssistantRuntimeDemoApp/Shared/AgentDemoRuntimeFactory.swift

@@ -102,7 +102,21 @@ enum AgentDemoRuntimeFactory {
                 )
             ),
             approvalPresenter: approvalInbox,
-            stateStore: FileRuntimeStateStore(url: stateURL ?? defaultStateURL())
+            stateStore: FileRuntimeStateStore(url: stateURL ?? defaultStateURL()),
+            memory: .init(
+                store: try! SQLiteMemoryStore(url: defaultMemoryURL()),
+                automaticCapturePolicy: .init(
+                    source: .lastTurn,
+                    options: .init(
+                        defaults: .init(
+                            namespace: DemoMemoryExamples.namespace,
+                            kind: "preference",
+                            tags: ["demo", "auto-capture"]
+                        ),
+                        maxMemories: 2
+                    )
+                )
+            )
         ))
     }
     #endif
@@ -129,7 +143,21 @@ enum AgentDemoRuntimeFactory {
                 )
             ),
             approvalPresenter: NonInteractiveApprovalPresenter(),
-            stateStore: FileRuntimeStateStore(url: defaultStateURL())
+            stateStore: FileRuntimeStateStore(url: defaultStateURL()),
+            memory: .init(
+                store: try! SQLiteMemoryStore(url: defaultMemoryURL()),
+                automaticCapturePolicy: .init(
+                    source: .lastTurn,
+                    options: .init(
+                        defaults: .init(
+                            namespace: DemoMemoryExamples.namespace,
+                            kind: "preference",
+                            tags: ["demo", "auto-capture"]
+                        ),
+                        maxMemories: 2
+                    )
+                )
+            )
         ))
     }
 
@@ -143,6 +171,17 @@ enum AgentDemoRuntimeFactory {
             .appendingPathComponent("AssistantRuntimeDemoApp", isDirectory: true)
             .appendingPathComponent("runtime-state.json")
     }
+
+    static func defaultMemoryURL() -> URL {
+        let baseDirectory = FileManager.default.urls(
+            for: .applicationSupportDirectory,
+            in: .userDomainMask
+        ).first ?? URL(fileURLWithPath: NSTemporaryDirectory())
+
+        return baseDirectory
+            .appendingPathComponent("AssistantRuntimeDemoApp", isDirectory: true)
+            .appendingPathComponent("memory.sqlite")
+    }
 }
 
 private struct NonInteractiveApprovalPresenter: ApprovalPresenting {

DemoApp/AssistantRuntimeDemoApp/Shared/AgentDemoViewModel+Memory.swift

@@ -0,0 +1,205 @@
+import CodexKit
+import Foundation
+
+@MainActor
+extension AgentDemoViewModel {
+    func runAutomaticPolicyMemoryDemo() async {
+        guard session != nil else {
+            lastError = "Sign in before running policy-based memory capture."
+            return
+        }
+        guard !isRunningMemoryDemo else {
+            return
+        }
+
+        isRunningMemoryDemo = true
+        lastError = nil
+        defer {
+            isRunningMemoryDemo = false
+        }
+
+        do {
+            let thread = try await runtime.createThread(
+                title: "Memory Demo: Automatic Policy",
+                memoryContext: AgentMemoryContext(
+                    namespace: DemoMemoryExamples.namespace,
+                    scopes: [DemoMemoryExamples.healthCoachScope],
+                    kinds: ["preference"]
+                )
+            )
+            _ = try await runtime.sendMessage(
+                UserMessageRequest(text: DemoMemoryExamples.automaticPolicyPrompt),
+                in: thread.id
+            )
+
+            let store = try SQLiteMemoryStore(url: AgentDemoRuntimeFactory.defaultMemoryURL())
+            let result = try await store.query(
+                MemoryQuery(
+                    namespace: DemoMemoryExamples.namespace,
+                    scopes: [DemoMemoryExamples.healthCoachScope],
+                    text: "direct blunt steps",
+                    limit: 4,
+                    maxCharacters: 800
+                )
+            )
+
+            automaticPolicyMemoryResult = AutomaticPolicyMemoryDemoResult(
+                threadID: thread.id,
+                threadTitle: thread.title ?? "Memory Demo: Automatic Policy",
+                prompt: DemoMemoryExamples.automaticPolicyPrompt,
+                records: result.matches.map(\.record)
+            )
+            threads = await runtime.threads()
+        } catch {
+            lastError = error.localizedDescription
+        }
+    }
+
+    func runAutomaticMemoryDemo() async {
+        guard session != nil else {
+            lastError = "Sign in before running automatic memory capture."
+            return
+        }
+        guard !isRunningMemoryDemo else {
+            return
+        }
+
+        isRunningMemoryDemo = true
+        lastError = nil
+        defer {
+            isRunningMemoryDemo = false
+        }
+
+        do {
+            let thread = try await runtime.createThread(
+                title: "Memory Demo: Automatic Capture",
+                memoryContext: DemoMemoryExamples.previewContext
+            )
+            let capture = try await runtime.captureMemories(
+                from: .text(DemoMemoryExamples.automaticCaptureTranscript),
+                for: thread.id,
+                options: .init(
+                    defaults: DemoMemoryExamples.guidedDefaults,
+                    maxMemories: 3
+                )
+            )
+            automaticMemoryResult = AutomaticMemoryDemoResult(
+                threadID: thread.id,
+                threadTitle: thread.title ?? "Memory Demo: Automatic Capture",
+                capture: capture
+            )
+            threads = await runtime.threads()
+        } catch {
+            lastError = error.localizedDescription
+        }
+    }
+
+    func runGuidedMemoryDemo() async {
+        guard !isRunningMemoryDemo else {
+            return
+        }
+
+        isRunningMemoryDemo = true
+        lastError = nil
+        defer {
+            isRunningMemoryDemo = false
+        }
+
+        do {
+            let writer = try await runtime.memoryWriter(defaults: DemoMemoryExamples.guidedDefaults)
+            let record = try await writer.upsert(DemoMemoryExamples.guidedDraft)
+            let diagnostics = try await writer.diagnostics()
+
+            guidedMemoryResult = GuidedMemoryDemoResult(
+                record: record,
+                diagnostics: diagnostics
+            )
+        } catch {
+            lastError = error.localizedDescription
+        }
+    }
+
+    func runRawMemoryDemo() async {
+        guard !isRunningMemoryDemo else {
+            return
+        }
+
+        isRunningMemoryDemo = true
+        lastError = nil
+        defer {
+            isRunningMemoryDemo = false
+        }
+
+        do {
+            let store = try SQLiteMemoryStore(url: AgentDemoRuntimeFactory.defaultMemoryURL())
+            try await store.upsert(
+                DemoMemoryExamples.rawRecord,
+                dedupeKey: DemoMemoryExamples.rawRecord.dedupeKey ?? DemoMemoryExamples.rawRecord.id
+            )
+            let diagnostics = try await store.diagnostics(namespace: DemoMemoryExamples.namespace)
+
+            rawMemoryResult = RawMemoryDemoResult(
+                record: DemoMemoryExamples.rawRecord,
+                diagnostics: diagnostics
+            )
+        } catch {
+            lastError = error.localizedDescription
+        }
+    }
+
+    func runMemoryPreviewDemo() async {
+        guard !isRunningMemoryDemo else {
+            return
+        }
+
+        isRunningMemoryDemo = true
+        lastError = nil
+        defer {
+            isRunningMemoryDemo = false
+        }
+
+        do {
+            let result: MemoryQueryResult
+            var previewThreadID: String?
+            var previewThreadTitle: String?
+
+            if session != nil {
+                let thread = try await runtime.createThread(
+                    title: "Memory Demo: Prompt Injection",
+                    memoryContext: DemoMemoryExamples.previewContext
+                )
+                previewThreadID = thread.id
+                previewThreadTitle = thread.title
+                result = try await runtime.memoryQueryPreview(
+                    for: thread.id,
+                    request: UserMessageRequest(text: DemoMemoryExamples.previewRequestText)
+                ) ?? MemoryQueryResult(matches: [], truncated: false)
+                threads = await runtime.threads()
+            } else {
+                let store = try SQLiteMemoryStore(url: AgentDemoRuntimeFactory.defaultMemoryURL())
+                result = try await store.query(
+                    MemoryQuery(
+                        namespace: DemoMemoryExamples.namespace,
+                        scopes: DemoMemoryExamples.previewContext.scopes,
+                        text: DemoMemoryExamples.previewRequestText,
+                        limit: DemoMemoryExamples.previewBudget.maxItems,
+                        maxCharacters: DemoMemoryExamples.previewBudget.maxCharacters
+                    )
+                )
+            }
+
+            memoryPreviewResult = MemoryPreviewDemoResult(
+                threadID: previewThreadID,
+                threadTitle: previewThreadTitle,
+                requestText: DemoMemoryExamples.previewRequestText,
+                result: result,
+                renderedPrompt: DefaultMemoryPromptRenderer().render(
+                    result: result,
+                    budget: DemoMemoryExamples.previewBudget
+                )
+            )
+        } catch {
+            lastError = error.localizedDescription
+        }
+    }
+}