ollama_chat 0.0.124 → 0.0.125

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0d2664dce7c6742ff5c950e64330b84325bcbb573467edde9aceddc3fa209641
-  data.tar.gz: 87baa515acb398c9da1b7cf6c7fce173d946e4bd4d79ad0f1eb9dfbd2526235a
+  metadata.gz: 3de94c682281a4667e89feee589883bd33acf082165f37ce443072117a773167
+  data.tar.gz: 94cccbb92af48198a70879cf1028a0aebfe0cda49e25ccd82352cb35d9054bf7
 SHA512:
-  metadata.gz: a091addc87c893f700f405147386dc8fe8fcc5c3134ad12bb1ffda570ec8a0f64ac4acbaeaed70116a6c61d0765e14e2f085fa8d2c0b98496fb0881a56fdb0f8
-  data.tar.gz: 6d078a9c69b50719dda02e9c81a32d51929627ab6e50c7cae023630543e717dd2c1a86166495f339875178108e0ded60f9f451ebbec1275f16f06771c9b1b215
+  metadata.gz: d3bad30e34ad309fe6c2f2263802af2778eded9459affc6c6713c6383f58706fe54411e7a4b6f087c487718d0a6ea8e07e90dfed22ca1a64b563d207ce61287e
+  data.tar.gz: 8f9befaa4a6a519b8f7b4424f0d16380663b863e039f08142fc667b01ecb83aa989b9e635ea5373b22191c7ba8f568e97dd210d732d1d6d5099dd3e5cffb74ee

data/CHANGES.md

@@ -1,5 +1,92 @@
 # Changes
 
+## 2026-10-03 v0.0.125
+
+### Added
+
+* Session-scoped trigger injection mechanism:
+  * New migration `011_add_trigger_to_sessions` adding a `:text` JSON column to
+    `sessions`.
+  * `Session#trigger` attribute with `plugin :serialization, :json` and `nil`
+    default.
+  * `CollectionSearch#trigger_inject` dispatch logic including embedding
+    guards, collection switching, document finding, and reranking.
+  * New `/session trigger edit` command registered in `commands.rb`.
+  * `edit_session_trigger` method in `session_management.rb` supporting dual
+    chooser and prefill.
+  * Trigger configuration display (⚡ block) added to `info_rag` in
+    `information.rb`.
+  * `runtime_information_inject` extracted from `chat.rb` into
+    `information.rb`.
+  * New prompt contexts: `prompts.rerank` (`default`, `memory_rerank`,
+    `lorebook_rerank`), `prompts.prompt.snippets_retrieval`, and
+    `prompts.prompt.snippets_trigger`.
+  * `persona_name` line added to `prompts.system.persona`.
+  * `TagResult` changed from `Data.define` to `Struct.new` to allow mutable
+    fields.
+  * `∅` fallback for empty patterns in `collection_stats`.
+  * Info-level logging for embedding requests/responses and rerank survivors in
+    `collection_search.rb`.
+  * New spec file `collection_search_spec.rb` with 7 examples for
+    `#trigger_inject`.
+
+### Changed
+
+* `SessionLocking#lock?` now accepts a required `chat:` keyword argument to
+  decouple feedback delegation from the host object.
+* `search_knowledge.rb` updated to use `snippets_retrieval` DB prompt,
+  `collection` block, and `database_collection?` guard.
+* `@embedding` instance variable swept to `embedding` accessor in
+  `information.rb`, `source_fetching.rb`, and `web_searching.rb`.
+* `Tins::Duration#format` used for compaction voice timing in
+  `report_compaction`, replacing `%.1f seconds`.
+* `tins` dependency bumped from `~> **1.58**` to `~> **1.60**` to support
+  unpadded `%H`, `%M`, and `%u` directives.
+* `resolve_tag_spec.rb` updated to replace an unreachable rescue-path test with
+  a "tagged file does not exist" test.
+* `search_knowledge_spec.rb` updated for new prompt names, `collection` block,
+  and `tags_set` doubles.
+* `compaction_spec.rb` regex updated to match the new duration format.
+* `Tins::GO.go` flag string reordered in `bin/ollama_chat_log`.
+
+### Fixed
+
+* Spec expectations in `session_management_spec.rb`, `parsing_spec.rb`, and
+  `source_fetching_spec.rb` updated to use Ruby 3.1+ hash shorthand `chat:`.
+
+### Tests
+
+* Added specs for `CommandConcern` class methods:
+  * `help_message` via `OllamaChat::Commands`: table structure, category
+    headers, pattern filtering, subcommand expansion, and optional/mandatory
+    asterisks.
+  * `command_completions`: array shape, bare names, subcommand variants, and
+    sort order.
+  * `Command` class: `execute_if_match?` (match, captures, nil-regexp path,
+    no-context error), `completions` (product, optional bare, strict),
+    `command_names`, and `optional?`.
+  * Duplicate registration guard against the real `:copy` entry in
+    `OllamaChat::Commands`.
+* Added specs for remaining command dispatch gaps:
+  * `/paste -i` stdin path via `paste_from_stdin`.
+  * `/last -v` voice output via `speak_last`.
+  * `/regenerate` success path (drop + return content).
+  * `/prompt delete` routes to `choose_and_delete_prompt`.
+  * `/web` without explicit count passes `nil`.
+  * `/character` JSON success paths: `info` (pager), `load` (raw content),
+    `import` (delegation).
+  * `/character` PNG `load` with `PNGMetadataExtractor`.
+* Added specs for `SessionManagement#edit_session_trigger`:
+  * Setting trigger from `nil` to a configured hash with `enabled`,
+    `prompt_name`, and `text_count`.
+  * Updating config for an already-configured collection.
+  * Switching the trigger to a different collection name.
+* Added specs for session model options and trigger edit:
+  * `/session model options` delegates to `edit_session_model_options` and
+    returns `:next`.
+  * `/session trigger edit` delegates to `edit_session_trigger` and returns
+    `:next`.
+
 ## 2026-10-02 v0.0.124
 
 ### Added

data/Rakefile

@@ -49,7 +49,7 @@ GemHadar do
   dependency             'reverse_markdown',      '~> 3.0'
   dependency             'kramdown-ansi',         '~> 0.5'
   dependency             'complex_config',        '~> 0.22', '>= 0.22.2'
-  dependency             'tins',                  '~> 1.58'
+  dependency             'tins',                  '~> 1.60'
   dependency             'search_ui',             '>= 0.2'
   dependency             'amatch',                '~> 0.4'
   dependency             'pdf-reader',            '~> 2.0'

data/bin/ollama_chat_log

@@ -4,7 +4,7 @@ require 'ollama_chat'
 require 'file/tail'
 require 'tins/xt'
 
-opts     = Tins::GO.go 'fm:n:h:', defaults: { ?n => 10  }
+opts     = Tins::GO.go 'hfm:n:', defaults: { ?n => 10  }
 log_path = ARGV.shift || OC::OLLAMA::CHAT::LOG::CHAT
 
 if opts[?h]

data/lib/ollama_chat/chat.rb

@@ -314,10 +314,11 @@ class OllamaChat::Chat
   #   instance when a background thread was spawned (so callers can call
   #   `cancel_speaking`), or `nil` for inline playback or when voice is off
   def speak(text, background: true)
+    session or return
     voice.on? or return
-    voice = session.current_voice.full? or return
+    current_voice = session.current_voice.full? or return
     voice_handler.method_defined?(:speak) or return
-    voice_handler.new(chat: self, voice:).speak(text, background:)
+    voice_handler.new(chat: self, voice: current_voice).speak(text, background:)
   end
 
   # Returns a human-readable string representation of the Chat object,
@@ -553,17 +554,8 @@ class OllamaChat::Chat
           group_uuid:
         )
 
-        if runtime_info.on?
-          tool_name = 'runtime_information'
-          messages << OllamaChat::Message.new(
-            role:        'user',
-            tool_name:   ,
-            sender_name: tool_name,
-            content:     dynamic_runtime_information,
-            images:      @images.dup,
-            group_uuid:
-          )
-        end
+        runtime_information_inject(group_uuid:)
+        trigger_inject(content, group_uuid:)
       end
       @images.clear
       handler = OllamaChat::FollowChat.new(

data/lib/ollama_chat/collection_search.rb

@@ -24,14 +24,37 @@ module OllamaChat::CollectionSearch
     tags = Documentrix::Utils::Tags.new(
       tags, valid_tag: /\A#*([-\w.\]\[]+)/
     )
-    @documents.find_where(
-      query.first(config.embedding.model.context_length),
+    effective_query  = query.first(
+      config.embedding.model.context_length
+    )
+    embedding_prompt = config.embedding.model.prompt?
+    log(
+      :info, 'Embedding search request',
+      data: {
+        model:      config.embedding.model.to_h,
+        collection: collection,
+        query:      effective_query,
+        prompt:     embedding_prompt,
+      }
+    )
+    records = @documents.find_where(
+      effective_query,
       tags:,
-      prompt:         config.embedding.model.prompt?,
+      prompt:         embedding_prompt,
       text_size:,
       text_count:,
       min_similarity:
     )
+    log(
+      :info, 'Embedding search response',
+      data: {
+        collection:   collection,
+        query:        effective_query,
+        result_count: records.size,
+        top:          records.first&.similarity&.round(4),
+      }
+    )
+    records
   end
 
   # Reranks a set of records using the LLM with the `rerank` prompt.
@@ -43,11 +66,11 @@ module OllamaChat::CollectionSearch
   # @return [Array<Documentrix::Utils::TagResult>] filtered records
   def rerank_records(query, records, prompt_name: nil)
     candidates = records.each_with_index.map { |r, i|
-      "[#{i}] #{truncate(r.text.strip, length: 300)}"
+      "[#{i}] #{r.tags_set.to_s(link: false)} #{truncate(r.text.squeeze, length: 300)}"
     }.join("\n")
 
-    prompt_name ||= 'rerank'
-    rerank_prompt = prompt(prompt_name) or
+    prompt_name ||= 'default'
+    rerank_prompt = prompt(prompt_name, context: 'rerank') or
       raise 'missing prompt %s' % prompt_name.inspect
     rerank_prompt = rerank_prompt.to_s
       .named_placeholders_interpolate({ query:, candidates: })
@@ -63,4 +86,77 @@ module OllamaChat::CollectionSearch
     end
     records
   end
+
+  # Injects relevant snippets from the session's trigger collection
+  # into the conversation as a tool message.
+  #
+  # Reads the first entry of the session's `trigger` hash, performs a
+  # vector search in the configured collection, reranks the candidates
+  # via the LLM, and appends a `trigger_inject` tool message containing
+  # the surviving snippets (text, similarity, tags). The message is
+  # grouped with the current user message via `group_uuid`.
+  #
+  # @param content [String] the parsed user message text used as the
+  #   search query
+  # @param group_uuid [String, nil] the group UUID for message grouping
+  #   (shared with the user message and runtime-info message)
+  #
+  # @return [self, nil] self if snippets were injected, nil if no
+  #   trigger is configured, the trigger or embedding is disabled, the
+  #   collection or prompt is missing, or no snippets survived reranking
+  def trigger_inject(content, group_uuid:)
+    embedding.on? or return
+    trigger = session.trigger&.first or return
+    collection, config = trigger
+    unless config['enabled']
+      log(:info, 'Trigger for collection %s disabled.' % collection)
+      return
+    end
+    unless col = database_collection?(collection)
+      log(:error, 'Unknown collection named %s' % collection)
+      return
+    end
+    prompt_name = config['prompt_name']
+    unless rerank_prompt = prompt(prompt_name, context: 'rerank').full?(:to_s)
+      log(
+        :error, 'Unknown rerank prompt named %s' % prompt_name,
+        data: { config: }
+      )
+      return
+    end
+    records = []
+    switch_collection(collection) do
+      records = find_document_records(content, text_count: config['text_count'])
+      records.empty? and return
+      pre_rerank = records.size
+      records    = rerank_records(content, records, prompt_name:)
+      log(:info, 'Trigger: %d/%d passed rerank for %s' % [ records.size, pre_rerank, collection ],
+          data: { collection:, prompt_name: })
+      records.empty? and return
+    end
+    snippets = records.map { |record|
+      {
+        text:       record.text,
+        similarity: record.similarity.to_f,
+        tags:       record.tags_set.to_s(link: false),
+      }
+    }
+    message_content = {
+      prompt: prompt('snippets_trigger').to_s,
+      collection: {
+        name:        collection,
+        description: col.description&.to_s,
+      },
+      snippets:
+    }.to_json
+    tool_name       = 'trigger_inject'
+    messages << OllamaChat::Message.new(
+      role:        'user',
+      tool_name:   ,
+      sender_name: tool_name,
+      content:     message_content,
+      group_uuid:
+    )
+    self
+  end
 end

data/lib/ollama_chat/commands.rb

@@ -113,15 +113,16 @@ module OllamaChat::Commands
 
   command(
     name: :session,
-    regexp: %r(^/session(?:\s+(change|previous|list|new|duplicate|rename|delete|model options change|model options))?((?:\s+-(?:p\s*\w+))*)(?:\s+([^-].*))?$),
-    complete: [ 'session', %w[ change previous list new duplicate rename delete model\ options\ change model\ options ] ],
+    regexp: %r(^/session(?:\s+(change|previous|list|new|duplicate|rename|delete|model options change|model options|trigger edit))?((?:\s+-(?:p\s*\w+))*)(?:\s+([^-].*))?$),
+    complete: [ 'session', %w[ change previous list new duplicate rename delete model\ options\ change model\ options trigger\ edit ] ],
     optional: true,
     options: "[-p profile]\n[name]",
     help: <<~EOT
       💬 Manage sessions:
          - list/new/delete/rename/duplicate
          - change [name]/previous
-         - model options/change
+          - model options/change
+          - trigger edit
     EOT
   ) do |subcommand, opts, name|
     case subcommand
@@ -152,6 +153,8 @@ module OllamaChat::Commands
       else
         feedback("No previous session defined.", type: :info)
       end
+    when 'trigger edit'
+      edit_session_trigger
     end
     :next
   end

data/lib/ollama_chat/compaction.rb

@@ -295,11 +295,15 @@ module OllamaChat::Compaction
       Context:      #{result.context_before} → #{result.context_after}
       Stored total: #{result.stored_total}
     EOT
+    duration        = Tins::Duration.new(duration)
+    duration_string = duration.format(
+      duration.minutes? ? '%M minutes and %u seconds' : '%u seconds'
+    )
     speak(
-      'Compressed conversation context from %.f to %.f in %.1f seconds.' % [
+      'Compressed conversation context from %.f to %.f in %s.' % [
         100 * result.context_filled_before,
         100 * result.context_filled_after,
-        duration,
+        duration_string,
       ]
     )
   end

data/lib/ollama_chat/database/migrations/011_add_trigger_to_sessions.rb

@@ -0,0 +1,7 @@
+Sequel.migration do
+  change do
+    alter_table :sessions do
+      add_column :trigger, :text
+    end
+  end
+end

data/lib/ollama_chat/database/models/session.rb

@@ -12,6 +12,7 @@ class OllamaChat::Database::Models::Session < Sequel::Model(OllamaChat::DB)
   plugin :touch
   plugin :serialization, :json, :tools_default_enabled
   plugin :serialization, :json, :model_options
+  plugin :serialization, :json, :trigger
   plugin :validation_helpers
 
   # Validates the session instance.
@@ -133,6 +134,15 @@ class OllamaChat::Database::Models::Session < Sequel::Model(OllamaChat::DB)
   # @!attribute [v] model_options
   #   @return [String, nil] A JSON-serialized string containing model-specific options.
   #
+  # @!attribute [v] trigger
+  #   @return [Hash, nil] Maps collection names to search/rerank
+  #     overrides, e.g.
+  #     `{ 'memory-miyu' => { 'prompt_name' => 'memory_rerank',
+  #       'text_count' => 5 } }`.
+  #     Recognized keys: `prompt_name`, `text_count`, `text_size`,
+  #     `min_similarity`.
+  #     `nil` when the trigger mechanism is disabled for this session.
+  #
   # @!attribute [v] messages
   #   @return [String] The full conversation history, stored in JSONL format.
   #
@@ -184,6 +194,7 @@ class OllamaChat::Database::Models::Session < Sequel::Model(OllamaChat::DB)
       context_format:        chat.config.context.format,
       working_directory:     Dir.pwd,
       messages:              '',
+      trigger:               nil,
     }
     new(attributes)
   end

data/lib/ollama_chat/database/session_locking.rb

@@ -25,10 +25,11 @@ module OllamaChat::Database::SessionLocking
 
   # Attempts to acquire a lock on the session.
   #
+  # @param chat [OllamaChat] the chat instance that receives feedback
   # @return [Boolean] true if the lock was successfully acquired, false if already locked.
-  def lock?
+  def lock?(chat:)
     if running_pid = locked?
-      feedback("session #{id} locked by running process #{running_pid}", type: :warn)
+      chat.feedback("session #{id} locked by running process #{running_pid}", type: :warn)
       false
     else
       lock

data/lib/ollama_chat/information.rb

@@ -81,7 +81,7 @@ module OllamaChat::Information
       Current Collection
         Name: #{bold{collection}}
         Status: #{col&.enabled ? '✅' : '⛔'}
-        Patterns: #{italic{col&.patterns&.join(' ')}}
+        Patterns: #{italic{col&.patterns.full?(:join, ' ') || '∅' }}
         #Embeddings: #{@documents.size}
         #Tags: #{@documents.tags.size}
         Tags:
@@ -190,15 +190,23 @@ module OllamaChat::Information
   #
   # @param output [IO] the output stream to write the information to, defaults to STDOUT
   def info_rag(output: STDOUT)
-    if @embedding.on?
+    if embedding.on?
       output.puts "🗄️ Current RAG model is #{bold{@embedding_model}}"
       if @embedding_model_options.present?
         output.puts "  Options: #{JSON.pretty_generate(@embedding_model_options).gsub(/(?<!\A)^/, '  ')}"
       end
       output.puts "Text splitter is #{bold{config.embedding.splitter.name}}."
       collection_stats(output:)
+      if session.trigger.full?
+        session.trigger.each do |col, cfg|
+          state = cfg['enabled'] ? '✅' : '⛔'
+          output.puts "\u26A1 Trigger: #{state} #{bold{col}}"
+          output.puts "     Prompt: #{bold{cfg['prompt_name']}}"
+          output.puts "     Text count: #{cfg['text_count'] || 'unlimited'}"
+        end
+      end
     end
-    @embedding.show(output:)
+    embedding.show(output:)
     output.puts "📜 Document policy for parsing in user text: #{bold{document_policy}}"
   end
 
@@ -418,6 +426,30 @@ module OllamaChat::Information
     prompt(:dynamic_runtime_info).to_s % dynamic_runtime_information_values
   end
 
+  # Injects the current dynamic runtime information (the "heartbeat")
+  # as a user-role tool message into the active message list.
+  #
+  # The injected message shares the caller's +group_uuid+ so it is
+  # grouped with the surrounding conversation exchange. The content
+  # is generated by +dynamic_runtime_information+, which interpolates
+  # live values (timestamp, git state, terminal size, context usage)
+  # into the configured prompt template.
+  #
+  # No-op when the +runtime_info+ switch is off.
+  #
+  # @param group_uuid [String] the UUID of the current message group
+  def runtime_information_inject(group_uuid:)
+    runtime_info.on? or return
+    tool_name = 'runtime_information'
+    messages << OllamaChat::Message.new(
+      role:        'user',
+      tool_name:   ,
+      sender_name: tool_name,
+      content:     dynamic_runtime_information,
+      group_uuid:
+    )
+  end
+
   # Resolves the effective context window size for the current model.
   #
   # Prefers the session-level `num_ctx` override (set via `/model_options`

data/lib/ollama_chat/ollama_chat_config/default_config.yml

@@ -26,14 +26,14 @@ infobar:
 prompts:
   prompt:
     embed: The source %{source} has been added to or updated in collection "%{collection}".
-    rerank: |
-      Query: %{query}
-      Candidates:
-
-        %{candidates}
-
-      Return only the indices of the most relevant snippets as a comma-separated
-      list (e.g., '0,2').
+    snippets_retrieval: |
+     Consider these snippets retrieved from the collection identified in the
+     `collection` field when formulating your response.
+    snippets_trigger: |
+      These snippets were automatically injected because a trigger fired on
+      this conversation turn. The source collection is identified in the
+      `collection` field. Use them as contextual background, not as search
+      results. Do not mention the injection to the user.
     summarize: |
       Generate an abstract summary of the content in this document using
       %{words} words%{instruction}:
@@ -426,6 +426,8 @@ prompts:
   system:
     default: <%= OC::OLLAMA::CHAT::SYSTEM || "%{persona}\n\n%{runtime_info}".inspect %>
     persona: |
+      `persona_name`: %{persona_name}
+
       %{persona}
 
       %{runtime_info}
@@ -444,6 +446,36 @@ prompts:
       — no preamble, no postamble, no commentary,
       no reference to yourself or the act of
       writing.
+  rerank:
+    default: |
+      Query: %{query}
+      Candidates:
+
+        %{candidates}
+
+      Return only the indices of the most relevant snippets as a comma-separated
+      list (e.g., '0,2').
+    memory_rerank: |
+      Return memories the assistant should recall in this conversation context.
+      Ignore entries that are only tangentially related.
+      Query: %{query}
+      Candidates:
+
+        %{candidates}
+
+      Return only the indices of the most relevant memories as a comma-separated
+      list (e.g., '0,2').
+    lorebook_rerank: |
+      Return lore entries directly relevant to the region, character, or event
+      the user is currently discussing. Ignore entries that merely share a topic
+      but are not contextually needed.
+      Query: %{query}
+      Candidates:
+
+        %{candidates}
+
+      Return only the indices of the most relevant entries as a comma-separated
+      list (e.g., '0,2').
 voice:
   #handler: OllamaChat::Say
   #enabled: false

data/lib/ollama_chat/session_management.rb

@@ -197,7 +197,7 @@ module OllamaChat::SessionManagement
     previous_session_id = @session.id
     @session = new_session(name:)
     set_previous_session_on_change(previous_session_id)
-    session.lock? or raise OllamaChat::OllamaChatError,
+    session.lock?(chat: self) or raise OllamaChat::OllamaChatError,
       "Could not lock session #{session.id} #{session.errors.full?(:inspect)}"
     session_apply
     messages.clear
@@ -236,7 +236,9 @@ module OllamaChat::SessionManagement
       use_model(_1)
       copy_model_options_to_session
     }
-    log(:info, "Session duplicated", data: { session_id: session.id, name: session.name, old_session_id: old_session.id })
+    log(:info, "Session duplicated", data: {
+      session_id: session.id, name: session.name, old_session_id: old_session.id
+    })
     nil
   end
 
@@ -253,7 +255,7 @@ module OllamaChat::SessionManagement
                  preferred_session
                end
     session or abort "No session named #{bold{session_name.inspect}} found."
-    if session.lock?
+    if session.lock?(chat: self)
       messages.read_conversation_jsonl(session.messages.to_s)
       session_apply
     else
@@ -483,6 +485,80 @@ module OllamaChat::SessionManagement
     end
   end
 
+  # Interactively edits the trigger configuration for a collection.
+  #
+  # Presents a chooser for available rerank prompt names, then prompts
+  # for the collection, text_count, and enabled state. Existing values
+  # are prefilled when re-editing a known trigger so the user can simply
+  # accept defaults or toggle enabled off.
+  def edit_session_trigger
+    prompts = models::Prompt.where(context: 'rerank').map(&:name).sort
+    return feedback('No rerank prompts available.', type: :warn) \
+      unless prompts.any?
+
+    collections = all_collections.map(&:name).sort
+    return feedback('No collections available.', type: :warn) \
+      unless collections.any?
+
+    existing = session.trigger.to_h
+
+    collection = nil
+    choose_with_state do
+      # Prefill with the first configured collection as a starting
+      # point. Fine for the single-trigger case; with multiple
+      # configured, it's just an arbitrary default the user can
+      # override in the chooser.
+      if selected = existing.first&.first&.dup
+        self.current_search_state = SearchUI::Search::State.new(selected, 0)
+      end
+      collection = choose_entry(collections, prompt: 'Collection? %s')
+      return feedback('Exiting chooser.', type: :info) unless collection
+    end
+
+    prompt_name = nil
+    choose_with_state do
+      if selected = existing.dig(collection, 'prompt_name')
+        self.current_search_state = SearchUI::Search::State.new(selected, 0)
+      end
+      prompt_name = choose_entry(prompts, prompt: 'Rerank prompt? %s')
+      return feedback('Exiting chooser.', type: :info) unless prompt_name
+    end
+
+    prefill_tc      = ''
+    prefill_enabled = 'y'
+    if cfg = existing[collection]
+      prefill_tc      = cfg['text_count'].to_s
+      prefill_enabled = cfg['enabled'] ? 'y' : 'n'
+    end
+
+    tc_raw = ask?(
+      prompt:  'text_count (empty for no limit): ',
+      prefill: prefill_tc
+    )
+    return feedback('Cancelled.', type: :cancel) if tc_raw.nil?
+    text_count = tc_raw.strip.empty? ? nil : tc_raw.strip.to_i
+
+    enabled_raw = ask?(
+      prompt: 'Enabled? (y/n): ',
+      prefill: prefill_enabled
+    )
+    return feedback('Cancelled.', type: :cancel) if enabled_raw.nil?
+    enabled = enabled_raw.strip.match?(/\Ay/i)
+
+    cfg = {
+      'enabled'     => enabled,
+      'prompt_name' => prompt_name,
+    }
+    cfg['text_count'] = text_count if text_count
+
+    session.update(trigger: { collection => cfg })
+
+    feedback(
+      "Trigger for #{collection.inspect} → #{cfg.inspect}.",
+      type: :info
+    )
+  end
+
   # Derives a title for the session based on its content.
   #
   # @param length [Integer] the maximum length of the title (default: 128)
@@ -551,7 +627,7 @@ module OllamaChat::SessionManagement
         session_close
         previous_session = session
         @session = chosen_session
-        if session.lock?
+        if session.lock?(chat: self)
           messages.read_conversation_jsonl(session.messages.to_s)
           if current_collection = session.current_collection.full? and
             database_collection?(current_collection)

data/lib/ollama_chat/source_fetching.rb

@@ -194,7 +194,7 @@ module OllamaChat::SourceFetching
   # @return [Array, String, nil] The embedded chunks or processed content, or
   #   nil if embedding is disabled or fails
   def embed_source(source_io, source, tags: [], count: nil)
-    @embedding.on? or return parse_source(source_io)
+    embedding.on? or return parse_source(source_io)
     unless @documents.source_modified?(source)
       infobar.puts "Source #{source.to_s.inspect} already up-to-date. => Skipping."
       log(:info, "Source up-to-date", data: { source: source.to_s })
@@ -272,7 +272,7 @@ module OllamaChat::SourceFetching
   #   nil if the operation fails
   def embed(source, tags: [], prompt: prompt(:embed))
     prompt = prompt.to_s
-    @embedding.on? or return
+    embedding.on? or return
     fetch_source(source) do |source_io|
       content = parse_source(source_io)
       content.present? or return