llmemory 0.1.13 → 0.1.15

data/lib/llmemory/vector_store/openai_embeddings.rb

@@ -2,6 +2,7 @@
 
 require "faraday"
 require "json"
+require "digest"
 require_relative "base"
 
 module Llmemory
@@ -13,10 +14,46 @@ module Llmemory
       def initialize(api_key: nil, model: nil)
         @api_key = api_key || Llmemory.configuration.llm_api_key
         @model = model || DEFAULT_MODEL
+        @cache = {}
+        @cache_order = []
       end
 
       def embed(text)
         return Array.new(DEFAULT_DIMS, 0.0) if text.to_s.strip.empty?
+
+        if Llmemory.configuration.embedding_cache_enabled
+          key = cache_key(text)
+          return @cache[key].dup if @cache.key?(key)
+        end
+
+        result = fetch_embedding(text)
+
+        if Llmemory.configuration.embedding_cache_enabled
+          evict_if_needed
+          @cache[cache_key(text)] = result.dup
+          @cache_order << cache_key(text)
+        end
+
+        result
+      end
+
+      private
+
+      def cache_key(text)
+        Digest::SHA256.hexdigest("#{@model}:#{text.to_s.strip}")
+      end
+
+      def evict_if_needed
+        max = Llmemory.configuration.embedding_cache_max_entries.to_i
+        return if max <= 0 || @cache.size < max
+
+        while @cache_order.any? && @cache.size >= max
+          k = @cache_order.shift
+          @cache.delete(k)
+        end
+      end
+
+      def fetch_embedding(text)
         response = connection.post("embeddings") do |req|
           req.headers["Authorization"] = "Bearer #{@api_key}"
           req.headers["Content-Type"] = "application/json"
@@ -27,16 +64,6 @@ module Llmemory
         body.dig("data", 0, "embedding")&.map(&:to_f) || Array.new(DEFAULT_DIMS, 0.0)
       end
 
-      def store(id:, embedding:, metadata: {})
-        raise NotImplementedError, "OpenAIEmbeddings does not store; use a VectorStore backend (e.g. MemoryStore)"
-      end
-
-      def search(query_embedding, top_k: 10)
-        raise NotImplementedError, "OpenAIEmbeddings does not search; use a VectorStore backend"
-      end
-
-      private
-
       def connection
         @connection ||= Faraday.new(url: "https://api.openai.com/v1") do |f|
           f.request :json
@@ -44,6 +71,14 @@ module Llmemory
           f.adapter Faraday.default_adapter
         end
       end
+
+      def store(id:, embedding:, metadata: {})
+        raise NotImplementedError, "OpenAIEmbeddings does not store; use a VectorStore backend (e.g. MemoryStore)"
+      end
+
+      def search(query_embedding, top_k: 10)
+        raise NotImplementedError, "OpenAIEmbeddings does not search; use a VectorStore backend"
+      end
     end
   end
 end

data/lib/llmemory/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Llmemory
-  VERSION = "0.1.13"
+  VERSION = "0.1.15"
 end

data/lib/tasks/release.rake

@@ -0,0 +1,100 @@
+# frozen_string_literal: true
+
+namespace :release do
+  desc "Bump version (patch|minor|major). Checks: branch=main, no uncommitted changes, tests pass. Then: Gemfile.lock, CHANGELOG, commit, push, tag"
+  task :bump, [:bump_type] => [] do |_t, args|
+    require_relative "../llmemory/version"
+
+    # Pre-flight checks
+    current_branch = `git rev-parse --abbrev-ref HEAD`.strip
+    abort "Current branch must be main (got: #{current_branch})" unless current_branch == "main"
+
+    # Allow only release-related files to be modified (we'll commit them)
+    release_files = %w[lib/llmemory/version.rb Gemfile.lock CHANGELOG.txt]
+    status_lines = `git status --porcelain`.strip.lines
+    other_changes = status_lines.reject do |line|
+      path = line.sub(/\A..\s+/, "").strip
+      release_files.include?(path)
+    end
+    abort "Working tree has uncommitted changes outside release files. Commit or stash them first." unless other_changes.empty?
+
+    puts "Running tests..."
+    sh "bundle exec rspec"
+    puts "Tests passed.\n\n"
+
+    bump_type = (args[:bump_type] || "patch").to_s.downcase
+    unless %w[patch minor major].include?(bump_type)
+      abort "Bump type must be: patch, minor, or major"
+    end
+
+    seg = Gem::Version.new(Llmemory::VERSION).segments
+    new_version = case bump_type
+    when "patch" then Gem::Version.new("#{seg[0]}.#{seg[1] || 0}.#{(seg[2] || 0) + 1}")
+    when "minor" then Gem::Version.new("#{seg[0]}.#{(seg[1] || 0) + 1}.0")
+    when "major" then Gem::Version.new("#{(seg[0] || 0) + 1}.0.0")
+    end
+    new_version_s = new_version.to_s
+
+    puts "Bumping #{Llmemory::VERSION} -> #{new_version_s} (#{bump_type})"
+
+    # 1. Update version.rb
+    version_file = File.expand_path("../llmemory/version.rb", __dir__)
+    content = File.read(version_file)
+    content = content.sub(/VERSION = "[^"]+"/, %(VERSION = "#{new_version_s}"))
+    File.write(version_file, content)
+    puts "  Updated lib/llmemory/version.rb"
+
+    # 2. bundle install
+    sh "bundle install"
+    puts "  Updated Gemfile.lock"
+
+    # 3. Update CHANGELOG.txt
+    changelog_path = File.expand_path("../../CHANGELOG.txt", __dir__)
+    changelog_content = if File.exist?(changelog_path)
+      File.read(changelog_path)
+    else
+      ""
+    end
+
+    today = Time.now.strftime("%Y-%m-%d")
+    last_tag = `git describe --tags --abbrev=0 2>/dev/null`.strip
+    commits = if last_tag.empty?
+      `git log --oneline`.strip
+    else
+      `git log #{last_tag}..HEAD --oneline`.strip
+    end
+
+    new_entry = <<~CHANGELOG
+
+      ## [#{new_version_s}] - #{today}
+
+      ### Changes
+      #{commits.lines.map { |l| "- #{l.strip}" }.join("\n")}
+    CHANGELOG
+
+    header = "# Changelog\n\n"
+    if changelog_content.empty?
+      changelog_content = header + new_entry
+    else
+      changelog_content = header + changelog_content unless changelog_content.start_with?(header)
+      changelog_content = changelog_content.sub(/(# Changelog\n\n)/m, "\\1#{new_entry.lstrip}")
+    end
+    File.write(changelog_path, changelog_content)
+    puts "  Updated CHANGELOG.txt"
+
+    # 4. Commit
+    sh "git add lib/llmemory/version.rb Gemfile.lock CHANGELOG.txt"
+    sh "git commit -m 'Release v#{new_version_s}'"
+
+    # 5. Push
+    sh "git push"
+
+    # 6. Tag
+    sh "git tag v#{new_version_s}"
+
+    # 7. Push tag
+    sh "git push origin v#{new_version_s}"
+
+    puts "\nDone. Released v#{new_version_s}"
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llmemory
 version: !ruby/object:Gem::Version
-  version: 0.1.13
+  version: 0.1.15
 platform: ruby
 authors:
 - llmemory
@@ -37,6 +37,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.6'
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.22'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.22'
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement
@@ -180,12 +194,18 @@ files:
 - lib/llmemory/mcp/tools/memory_timeline.rb
 - lib/llmemory/mcp/tools/memory_timeline_context.rb
 - lib/llmemory/memory.rb
+- lib/llmemory/noise_filter.rb
 - lib/llmemory/retrieval.rb
+- lib/llmemory/retrieval/bm25_scorer.rb
 - lib/llmemory/retrieval/context_assembler.rb
 - lib/llmemory/retrieval/engine.rb
+- lib/llmemory/retrieval/mmr_reranker.rb
 - lib/llmemory/retrieval/temporal_ranker.rb
 - lib/llmemory/short_term.rb
 - lib/llmemory/short_term/checkpoint.rb
+- lib/llmemory/short_term/message_sanitizer.rb
+- lib/llmemory/short_term/pruner.rb
+- lib/llmemory/short_term/session_lifecycle.rb
 - lib/llmemory/short_term/stores/active_record_checkpoint.rb
 - lib/llmemory/short_term/stores/active_record_store.rb
 - lib/llmemory/short_term/stores/base.rb
@@ -199,6 +219,7 @@ files:
 - lib/llmemory/vector_store/memory_store.rb
 - lib/llmemory/vector_store/openai_embeddings.rb
 - lib/llmemory/version.rb
+- lib/tasks/release.rake
 homepage: https://github.com/entaina/llmemory
 licenses:
 - MIT