Kaede Akino
3a59d89c52
Review every 10 turns or 15 non-memory tool calls and save explicit corrections promptly. Invalidate outdated learning after lifecycle and history changes, and prevent older correction retries from overwriting newer corrections. Replace capacity bars with entry and character counts, retain per-entry and model-input budgets, and migrate away from frozen snapshots. Upgrade the AgentKit package and submodule to signed release 0.15.0.
50 lines
2.3 KiB
Swift
50 lines
2.3 KiB
Swift
import AgentKit
|
|
import Foundation
|
|
|
|
/// Only successful history changes schedule indexing; disabled memory performs no additional I/O.
|
|
@MainActor public final class MemoryIndexedAgentRepository: AgentRunPersisting {
|
|
private let base: any AgentRunPersisting
|
|
private let memory: MemoryModel
|
|
public init(base: any AgentRunPersisting, memory: MemoryModel) {
|
|
self.base = base
|
|
self.memory = memory
|
|
memory.attachHistory(base)
|
|
}
|
|
public func save(_ snapshot: AgentRunSnapshot) async throws {
|
|
try await base.save(snapshot)
|
|
if snapshot.state.isTerminal { memory.historyChanged() }
|
|
}
|
|
public func record(_ event: AgentEvent, runID: UUID) async throws { try await base.record(event, runID: runID) }
|
|
public func persistCompaction(_ record: AgentCompactionRecord, conversationID: UUID) async throws {
|
|
try await base.persistCompaction(record, conversationID: conversationID)
|
|
}
|
|
public func loadConversations() async throws -> [StoredAgentConversation] { try await base.loadConversations() }
|
|
public func deleteConversation(_ id: UUID) async throws {
|
|
try await base.deleteConversation(id)
|
|
memory.invalidateLearning(for: id)
|
|
memory.historyChanged()
|
|
}
|
|
public func setTitle(_ title: String, conversationID: UUID) async throws {
|
|
try await base.setTitle(title, conversationID: conversationID)
|
|
memory.historyChanged()
|
|
}
|
|
public func setConversationMetadata(_ metadata: StoredAgentConversationMetadata, conversationID: UUID) async throws
|
|
{
|
|
try await base.setConversationMetadata(metadata, conversationID: conversationID)
|
|
}
|
|
public func reorderConversations(_ updates: [StoredAgentConversationOrder]) async throws {
|
|
try await base.reorderConversations(updates)
|
|
}
|
|
public func truncateConversation(_ id: UUID, fromUserMessageID: UUID, replacingWith: AgentTranscriptMessage)
|
|
async throws
|
|
{
|
|
try await base.truncateConversation(id, fromUserMessageID: fromUserMessageID, replacingWith: replacingWith)
|
|
memory.invalidateLearning(for: id)
|
|
memory.historyChanged()
|
|
}
|
|
public func createConversation(_ conversation: StoredAgentConversation) async throws {
|
|
try await base.createConversation(conversation)
|
|
memory.historyChanged()
|
|
}
|
|
}
|