internal/app/memory_components.go

Rama main · réplica de lectura del repositorio original.

package app

import (
	"log/slog"
	"time"

	"github.com/yeomyeonggeori/bluememo"

	"github.com/yeomyeonggeori/blueclaw/internal/config"
	"github.com/yeomyeonggeori/blueclaw/internal/identity"
	"github.com/yeomyeonggeori/blueclaw/internal/llm"
	"github.com/yeomyeonggeori/blueclaw/internal/memory"
)

const (
	memoryMaintenanceInterval      = time.Hour
	memoryMaintenanceRetryInterval = time.Minute
)

type memoryComponents struct {
	stores *memory.Stores
}

func newMemoryComponents(runtimeConfiguration config.RuntimeConfiguration, kernel agentKernel, services taskServices, identityService *identity.IdentityService, logger *slog.Logger) memoryComponents {
	logger.Info("application.initializing", "stage", "memory")
	embeddingModelName := llm.ConfiguredEmbeddingModelName(runtimeConfiguration)
	embeddingDimensions := llm.ConfiguredEmbeddingDimensions(runtimeConfiguration)
	configuration := bluememo.Configuration{
		Embedder:      kernel.embeddingClient,
		Model:         memory.LanguageModel{Provider: kernel.taskTierLanguageModels.Low},
		RecallSources: true,
		Logger:        logger,
	}
	if kernel.decisionModel != nil {
		configuration.Chooser = memory.Chooser{DecisionModel: kernel.decisionModel}
	}
	workspaceRootPath := firstNonEmptyString(runtimeConfiguration.Terminal.WorkspaceRootPath, "/workspace")
	stores := memory.NewStores(workspaceRootPath, configuration, kernel.terminalService.WorkspaceActorFactory())
	if !runtimeConfiguration.Memory.ExtractionDisabled {
		services.taskRunService.RegisterTaskRunTransitionObserver(memory.TaskRunTransitionObserver{
			Stores:   stores,
			TaskRuns: services.taskRunService,
			Steps:    services.taskStepService,
			Access:   identityService,
			Logger:   logger,
		}.Observe)
	}
	logger.Info("application.memory.store_configured",
		"workspaceRoot", workspaceRootPath,
		"embeddingModel", embeddingModelName,
		"embeddingDimensions", embeddingDimensions,
		"extractionDisabled", runtimeConfiguration.Memory.ExtractionDisabled)
	return memoryComponents{stores: stores}
}