store

Documentation

Overview

Package store persists user-selected LLM providers and credentials.

Index

• func ChangeDefault(Model types.LLMProvider) error
• type Config
  • func ListSavedModels() (*Config, error)
• type LLMProvider
• type StoreMethods
  • func KeyringInit() (*StoreMethods, error)
  • func NewStoreMethods() (*StoreMethods, error)
  • func (s *StoreMethods) CleanupCache() error
  • func (s *StoreMethods) ClearCache() error
  • func (s *StoreMethods) DefaultLLMKey() (*LLMProvider, error)
  • func (s *StoreMethods) DeleteModel(Model types.LLMProvider) error
  • func (s *StoreMethods) GetCacheHitRate() float64
  • func (s *StoreMethods) GetCacheManager() *cache.CacheManager
  • func (s *StoreMethods) GetCacheStats() *types.CacheStats
  • func (s *StoreMethods) GetCachedMessage(provider types.LLMProvider, diff string, opts *types.GenerationOptions) (*types.CacheEntry, bool)
  • func (s *StoreMethods) GetMostUsedProvider() (types.LLMProvider, int)
  • func (s *StoreMethods) GetOverallSuccessRate() float64
  • func (s *StoreMethods) GetProviderRanking() []types.LLMProvider
  • func (s *StoreMethods) GetUsageManager() *usage.StatsManager
  • func (s *StoreMethods) GetUsageStats() *types.UsageStats
  • func (s *StoreMethods) RecordGenerationEvent(event *types.GenerationEvent) error
  • func (s *StoreMethods) ResetUsageStats() error
  • func (s *StoreMethods) Save(LLMConfig LLMProvider) error
  • func (s *StoreMethods) SetCachedMessage(provider types.LLMProvider, diff string, opts *types.GenerationOptions, ...) error
  • func (s *StoreMethods) UpdateAPIKey(Model types.LLMProvider, APIKey string) error

Functions

func ChangeDefault

func ChangeDefault(Model types.LLMProvider) error

ChangeDefault updates the default LLM provider selection in the config.

Types

type Config

type Config struct {
	Default      types.LLMProvider   `json:"default"`
	LLMProviders []types.LLMProvider `json:"models"`
}

Config describes the on-disk structure for all saved LLM providers.

func ListSavedModels

func ListSavedModels() (*Config, error)

ListSavedModels loads all persisted LLM provider configurations.

type LLMProvider

type LLMProvider struct {
	LLM    types.LLMProvider `json:"model"`
	APIKey string            `json:"api_key"`
}

LLMProvider represents a single stored LLM provider and its credential.

type StoreMethods

type StoreMethods struct {
	// contains filtered or unexported fields
}

func KeyringInit

func KeyringInit() (*StoreMethods, error)

KeyringInit initializes a StoreMethods instance with both keyring and cache support. This function is kept for backward compatibility with main.go.

func NewStoreMethods

func NewStoreMethods() (*StoreMethods, error)

NewStoreMethods creates a new StoreMethods instance with cache support.

func (*StoreMethods) CleanupCache

func (s *StoreMethods) CleanupCache() error

CleanupCache removes old entries from the cache.

func (*StoreMethods) ClearCache

func (s *StoreMethods) ClearCache() error

ClearCache removes all entries from the cache.

func (*StoreMethods) DefaultLLMKey

func (s *StoreMethods) DefaultLLMKey() (*LLMProvider, error)

DefaultLLMKey returns the currently selected default LLM provider, if any.

func (*StoreMethods) DeleteModel

func (s *StoreMethods) DeleteModel(Model types.LLMProvider) error

DeleteModel removes the specified provider from the saved configuration.

func (*StoreMethods) GetCacheHitRate

func (s *StoreMethods) GetCacheHitRate() float64

GetCacheHitRate returns the cache hit rate as a percentage.

func (*StoreMethods) GetCacheManager

func (s *StoreMethods) GetCacheManager() *cache.CacheManager

GetCacheManager returns the cache manager instance.

func (*StoreMethods) GetCacheStats

func (s *StoreMethods) GetCacheStats() *types.CacheStats

GetCacheStats returns cache statistics.

func (*StoreMethods) GetCachedMessage

func (s *StoreMethods) GetCachedMessage(provider types.LLMProvider, diff string, opts *types.GenerationOptions) (*types.CacheEntry, bool)

GetCachedMessage retrieves a cached commit message if it exists.

func (*StoreMethods) GetMostUsedProvider

func (s *StoreMethods) GetMostUsedProvider() (types.LLMProvider, int)

GetMostUsedProvider returns the most frequently used LLM provider.

func (*StoreMethods) GetOverallSuccessRate

func (s *StoreMethods) GetOverallSuccessRate() float64

GetOverallSuccessRate returns the overall success rate as a percentage.

func (*StoreMethods) GetProviderRanking

func (s *StoreMethods) GetProviderRanking() []types.LLMProvider

GetProviderRanking returns providers ranked by usage frequency.

func (*StoreMethods) GetUsageManager

func (s *StoreMethods) GetUsageManager() *usage.StatsManager

GetUsageManager returns the usage statistics manager instance.

func (*StoreMethods) GetUsageStats

func (s *StoreMethods) GetUsageStats() *types.UsageStats

GetUsageStats returns comprehensive usage statistics.

func (*StoreMethods) RecordGenerationEvent

func (s *StoreMethods) RecordGenerationEvent(event *types.GenerationEvent) error

RecordGenerationEvent records a commit message generation event for statistics.

func (*StoreMethods) ResetUsageStats

func (s *StoreMethods) ResetUsageStats() error

ResetUsageStats clears all usage statistics.

func (*StoreMethods) Save

func (s *StoreMethods) Save(LLMConfig LLMProvider) error

Save persists or updates an LLM provider entry, marking it as the default.

func (*StoreMethods) SetCachedMessage

func (s *StoreMethods) SetCachedMessage(provider types.LLMProvider, diff string, opts *types.GenerationOptions, message string, cost float64, tokens *types.UsageInfo) error

SetCachedMessage stores a commit message in the cache.

func (*StoreMethods) UpdateAPIKey

func (s *StoreMethods) UpdateAPIKey(Model types.LLMProvider, APIKey string) error

UpdateAPIKey rotates the credential for an existing provider entry.