ollama_chat 0.0.108 → 0.0.110

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 857b8cf92f68e5856238c1758cfebe463535ded52de3e4dc591cd83b501ccd63
-  data.tar.gz: fc90ce1e2145207dc49dfeb6512804dec0e0e3933c943ec0050221b5867bccd1
+  metadata.gz: 239476f6dca2ca4fca22dc0290b25bb048572cc89d12ef82c09d54a7ee429068
+  data.tar.gz: d7c51cf3d519c593666625bcedf487a1d40893099c251f211cd055a4610cb3db
 SHA512:
-  metadata.gz: 173caf67db6f004400f1f5199e07fcd16acd36c44a531a7d8b5618e5c2fb03f994aa63d03c5e4140d2d443022afd7fccc29d6c17d63f76e2d7085f41b9dd97e9
-  data.tar.gz: 5addb13b768f52b9ef3f5e3bd532aa632a37e5b5bf7b3c69c64be2d063b0f74ddc41640dd9e19a87d27e570c581e24f5456cf7b3a4d7c1351eba85e6d152e10b
+  metadata.gz: 24eadaa6eb35470ff3908fbfd7646ad0b685bded30694b7784839aef35a447b6f6926c9cac6a29f4eed55d723065cdc063f21d7c718f3c15dc4905892f860484
+  data.tar.gz: 98f4ab4370005f5a9523a7fc54ea62f56e64b411894d8ca40ae85518cae9962f8b9818086d5984f34aec392dde268377b10d33ef103e64ebc9b8abcd17702293

data/CHANGES.md

@@ -1,5 +1,113 @@
 # Changes
 
+## 2026-08-20 v0.0.110
+
+### Added
+
+*   **App State Support**: Introduced `009_add_app_states_table.rb` migration
+    and `app_state.rb` model.
+*   **Prompt Synchronization**: Added `/prompt sync` subcommand with unified
+    diffs and boot check capabilities.
+    *   Implemented `AppState` model with `get`/`set` helpers and a `seed` boot
+        check using per-prompt `Digest::SHA256` hashes XOR-folded into a
+        fingerprint.
+    *   Boot check prints `~`/`+`/`-` detail report on drift and prompts user
+        to acknowledge; declining retains old fingerprint for repeated notice
+        on next boot.
+    *   Added `OC::DIFF_COMMAND` setting (`diff -u --color=always`) to `oc.rb`
+        for unified diff display, decoded via `Shellwords.split`.
+    *   Implemented `prompt_sync` and `show_prompt_diff` methods in
+        `prompt_management.rb` to compare DB prompts against shipped defaults,
+        show unified diff, offer `OC::DIFF_TOOL` resolution, and clean up
+        orphaned default prompts.
+    *   Wired `/prompt sync` subcommand into `commands.rb` (regexp, completion,
+        dispatch).
+*   **Test Mode**: Added `OllamaChat.test_mode` flag in `ollama_chat.rb` to
+    suppress interactive prompts during specs; set in `spec_helper.rb`.
+*   **Documentation**: Added YARD `@!attribute` docs for `test_mode` and
+    `test_mode?`.
+
+### Changed
+
+*   **Web Searching**: Updated `web_searching.rb` to store `search_engine` in
+    local `engine` to avoid double method call.
+*   **Session Management**: Added call to `session_close` after
+    `clean_messages!` succeeds in `lib/ollama_chat/commands.rb`.
+*   **Error Messages**: Simplified stale context error message in `PatchFile`
+    when a checksum mismatch occurs.
+
+### Fixed
+
+*   **Spec Expectations**: Updated `web_searching_spec.rb` to replace `allow`
+    with `expect` for message expectations.
+    *   Changed `log` expectations to `.at_least(:once)` since `links.add` also
+        triggers `log` internally.
+    *   Removed unused `before { allow(chat).to receive(:log) }` from the
+        `#web` describe block.
+    *   Added `.at_least(:once)` to `document_policy.selected` as it is called
+        in both `if` and `elsif` branches.
+    *   Changed `fetch_source`, `summarize`, and `import` expectations to
+        `.twice` since they are called once per URL (2 URLs in test data).
+
+### Tests
+
+*   Added 7 new spec examples for `prompt_sync` and `show_prompt_diff` in
+    `prompt_management_spec.rb`.
+
+## 2026-08-18 v0.0.109
+
+### Added
+
+*   Support for exporting and importing model options via the `/model` command:
+    *   New `export_model_options` method in `model_handling.rb` to export
+        stored model profiles to a JSON file.
+    *   New `import_model_options` method in `model_handling.rb` to import
+        profiles from a JSON file, featuring selective model picking, diff
+        display, and override confirmation.
+    *   Extended `/model` command with `options export` and `options import`
+        subcommands.
+*   `-p` pattern flag for `/model options import` to narrow file search, with a
+    fallback to `**/*.json`.
+*   `-i` flag for `/paste` command to read input from stdin:
+    *   New `paste_from_stdin` method in `clipboard.rb`.
+    *   Updated `/paste` command regex and parsing to support both `-e` and
+        `-i` flags.
+*   `options` metadata to command definitions for `/copy`, `/paste`, `/toggle`,
+    and `/model`.
+*   Comprehensive test coverage for previously untested or under-tested
+    modules:
+    *   `spec/ollama_chat/conversation_spec.rb` (9 examples).
+    *   `spec/ollama_chat/favourites_management_spec.rb` (15 examples).
+    *   `spec/ollama_chat/rag_handling_spec.rb` covering
+        `OllamaChat::RAGHandling`.
+    *   `spec/ollama_chat/prompt_handling_spec.rb` (19 examples).
+    *   `spec/ollama_chat/prompt_management_spec.rb` (47 examples).
+    *   `spec/ollama_chat/system_prompt_management_spec.rb` (20 examples).
+    *   `spec/ollama_chat/config_handling_spec.rb` (15 examples).
+    *   `spec/ollama_chat/web_searching_spec.rb` (8 examples).
+    *   `spec/ollama_chat/information_spec.rb` (5 examples).
+
+### Changed
+
+*   Refactored session management to extract and reuse
+    `set_previous_session_on_change` as a shared private method in
+    `lib/ollama_chat/session_management.rb`.
+    *   `set_new_session`, `duplicate_session`, and `change_session` now
+        delegate to this method for consistent `@previous_session_id` handling.
+*   Migrated test stubs from `allow` to `expect` across multiple spec files to
+    eliminate false-green tests where stubs were never invoked.
+*   Normalized `context` and `name` arguments in `prompt_handling.rb` to handle
+    `Symbol` inputs safely in SQL queries.
+*   Updated `s.files` and `s.test_files` in `ollama_chat.gemspec` to include
+    new configuration, asset, and test files.
+
+### Fixed
+
+*   Infinite retry loop in `kramdown_ansi.rb` by using `retried ||= false` to
+    persist the guard across `retry` calls.
+*   Incorrect YARD `@return` documentation for `#info_prompt` in
+    `prompt_management.rb`.
+
 ## 2026-08-15 v0.0.108
 
 ### Added

data/lib/ollama_chat/clipboard.rb

@@ -143,4 +143,15 @@ module OllamaChat::Clipboard
     log(:error, e, data: { method: __method__ })
     STDERR.puts e.message
   end
+
+  # Read the full text from STDIN and return it.
+  #
+  # @param edit [Boolean] If true, the text is passed through
+  #   {#edit_text} before being returned.
+  # @return [String] The raw (or edited) content read from STDIN.
+  def paste_from_stdin(edit: false)
+    text = STDIN.read
+    edit and text = edit_text(text)
+    text
+  end
 end

data/lib/ollama_chat/commands.rb

@@ -27,6 +27,7 @@ module OllamaChat::Commands
   command(
     name: :copy,
     regexp: %r(^/copy(\s+-e)?\s*$),
+    options: '[-e]',
     help: <<~EOT
       📋 Copy the last response to the clipboard.
          Options: -e to edit before copying.
@@ -39,15 +40,22 @@ module OllamaChat::Commands
 
   command(
     name: :paste,
-    regexp: %r(^/paste(\s+-e)?\s*$),
+    regexp: %r(^/paste(\s+-[ie])*\s*$),
+    options: '[-e|-i]',
     help: <<~EOT
-      📋 Paste content from the clipboard.
-         Options: -e to edit after pasting.
+      📋 Paste content from the clipboard or stdin.
+         Options:
+           -e  Edit after pasting.
+           -i  Read from stdin instead of clipboard.
     EOT
   ) do |opts|
     disable_content_parsing
-    opts = go_command('e', opts)
-    paste_from_clipboard(edit: opts[?e])
+    opts = go_command('ie', opts)
+    if opts[?i]
+      paste_from_stdin(edit: opts[?e])
+    else
+      paste_from_clipboard(edit: opts[?e])
+    end
   end
 
   ## Settings
@@ -88,6 +96,7 @@ module OllamaChat::Commands
     name: :toggle,
     regexp: %r(^/toggle(?:\s+(markdown|stream|location|runtime_info|voice|think_loud|think_strip|embedding)(?:\s+(-[yn]))?)?$),
     complete: [ 'toggle', %w[ markdown stream location runtime_info voice think_loud think_strip embedding ] ],
+    options: '[-y|-n]',
     help: <<~EOT
       🎛️ Toggle feature switches
       (markdown, stream, location, runtime_info,
@@ -139,17 +148,21 @@ module OllamaChat::Commands
 
   command(
     name: :model,
-    regexp: %r(^/model(?:\s+(change|options(?: copy| delete)?|options from session|options to session))?((?:\s+(?:-m))*)$),
-    complete: [ 'model', %w[ change options options\ copy options\ delete options\ from\ session options\ to\ session ] ],
+    regexp: %r(^/model(?:\s+(change|options(?: copy| delete| export| import)?|options from session|options to session))?((?:\s+(?:-m|-p\s+[\S]+))*)$),
+    complete: [ 'model', %w[ change options options\ copy options\ delete options\ export options\ import options\ from\ session options\ to\ session ] ],
+    options: '[-m|-p pattern]',
     help: <<~EOT
       🤖 Manage AI models & profiles:
          - change: Switch active model
          - options: Edit saved profile config
          - options copy: Copy profile from another model
          - options delete: Delete a saved profile
+         - options export: Export ALL model profiles to JSON
+         - options import: Selectively import profiles from JSON
          - options from session: Save live → Saved
          - options to session: Apply Saved → Live
          -m interactively choose a model
+         -p PATTERN narrow file search for import
     EOT
   ) do |subcommand, opts|
     if subcommand == 'change'
@@ -163,7 +176,7 @@ module OllamaChat::Commands
       end
       next :next
     end
-    opts    = go_command('m', opts)
+    opts    = go_command('mp:', opts)
     model   = opts[?m] ? choose_model('', @model) : @model
     case subcommand
     when 'options'
@@ -179,6 +192,12 @@ module OllamaChat::Commands
     when 'options to session'
       profile = choose_profile_for_model(model) || 'default'
       copy_model_options_to_session(model, profile:)
+    when 'options export'
+      export_model_options
+    when 'options import'
+      pattern  = opts[?p] || '**/*.json'
+      filename = choose_filename(pattern) or next :next
+      import_model_options(filename)
     end
     :next
   end
@@ -465,14 +484,15 @@ module OllamaChat::Commands
 
   command(
     name: :prompt,
-    regexp: %r(^/prompt(?:\s+(edit|info|add|delete|list|duplicate|import|export|reset|rename|-e))?(\s+(?:-[ef]|-c\s+(?:\w+|\?)))?(?:\s+([^-].*))?$),
-    complete: [ 'prompt', %w[ edit info add delete list duplicate import export reset rename ] ],
+    regexp: %r(^/prompt(?:\s+(edit|info|add|delete|list|duplicate|import|export|reset|rename|sync|-e))?(\s+(?:-[ef]|-c\s+(?:\w+|\?)))?(?:\s+([^-].*))?$),
+    complete: [ 'prompt', %w[ edit info add delete list duplicate import export reset rename sync ] ],
     optional: true,
     options: '[-c CONTEXT|-e|-f]',
     help: <<~EOT,
       📝 Manage prompt templates:
-         Subcommands: edit, info, add, delete, list,
-         duplicate, import, export, reset, rename.
+          Subcommands: edit, info, add, delete, list,
+          duplicate, import, export, reset, rename,
+          sync.
          Options: -c [context]
                      (? for interactive in /prompt),
                   -e (edit next)
@@ -525,6 +545,8 @@ module OllamaChat::Commands
           STDOUT.puts "No default value found for prompt #{bold{prompt.name}}."
         end
       end
+    when 'sync'
+      prompt_sync(context:)
     when nil, '-e'
       if prompt = choose_prompt(
           prompt: 'Which template shall guide the next response? %s',
@@ -600,6 +622,7 @@ module OllamaChat::Commands
         )
       then
         messages.clean_messages!
+        session_close
         STDOUT.puts "Conversation cleaned."
       else
         STDOUT.puts 'Cancelled.'

data/lib/ollama_chat/database/migrations/009_add_app_states_table.rb

@@ -0,0 +1,11 @@
+Sequel.migration do
+  change do
+    create_table :app_states do
+      primary_key :id
+      String :key, null: false, unique: true
+      Text :value # JSON-encoded payload
+      DateTime :created_at, default: Sequel::CURRENT_TIMESTAMP, null: false
+      DateTime :updated_at, default: Sequel::CURRENT_TIMESTAMP, null: false
+    end
+  end
+end

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

@@ -0,0 +1,122 @@
+require 'digest'
+
+# Represents a key/value pair of persistent application state stored in
+# the database.
+#
+# This model provides a generic, JSON-backed store for small bits of
+# runtime state that need to survive across sessions—such as the
+# XOR-SHA256 fingerprint of the shipped default prompts used for boot
+# drift detection.
+class OllamaChat::Database::Models::AppState < Sequel::Model(OllamaChat::DB)
+  plugin :timestamps, update_on_create: true
+  plugin :serialization, :json, :value
+  plugin :validation_helpers
+
+  # Validates the application state entry.
+  #
+  # Ensures that the `key` is present and unique.
+  def validate
+    super
+    validates_presence :key
+    validates_unique :key
+  end
+
+  # Retrieves the JSON payload stored under the given key.
+  #
+  # @param key [String, Symbol] the state key to look up
+  # @return [Hash, nil] the deserialized JSON payload, or nil if no
+  #   record exists for the key
+  def self.get(key)
+    where(key: key.to_s).first&.value
+  end
+
+  # Creates or updates the JSON payload stored under the given key.
+  #
+  # @param key [String, Symbol] the state key to write
+  # @param value [Hash, nil] the JSON-serializable payload to store
+  # @return [OllamaChat::Database::Models::AppState] the saved record
+  def self.set(key, value)
+    if found = where(key: key.to_s).first
+      found.update(value:)
+      found
+    else
+      create(key: key.to_s, value:)
+    end
+  end
+
+  # Computes per-prompt SHA256 hashes and an XOR fingerprint of all shipped
+  # default prompts.
+  #
+  # At boot, if the stored fingerprint differs from the current one, a detail
+  # report (modified / added / removed) is printed to STDERR and the user is
+  # prompted to accept the new fingerprint. If declined, the old fingerprint
+  # is retained and the notice will reappear on the next boot.
+  #
+  # This method is invoked automatically by the model seeding loop in
+  # `OllamaChat::Database.setup_models`.
+  #
+  # @param chat [OllamaChat::Chat] the chat instance providing the config
+  #   and `confirm?`
+  def self.seed(chat)
+    stored = get(:prompt_defaults_fingerprint)
+
+    stored_fingerprint = stored&.dig('fingerprint')
+
+    hashes = {}
+    chat.config.prompts.to_h.each do |context, prompts|
+      prompts.each do |name, content|
+        hashes["#{context}/#{name}"] =
+          Digest::SHA256.hexdigest(content.to_s)
+      end
+    end
+
+    fingerprint = hashes.values.reduce(0) do |sum, hex|
+      sum ^ hex.to_i(16)
+    end.to_s(16)
+
+    return true if fingerprint == stored_fingerprint
+
+    if stored&.dig('hashes')
+      old     = stored['hashes']
+      added   = hashes.keys - old.keys
+      removed = old.keys - hashes.keys
+      changed = hashes.select { |k, v| old[k] && old[k] != v }.keys
+
+      STDERR.puts "⚠️  Shipped prompts changed since last boot:"
+      changed.each { |k| STDERR.puts "  ~ #{k} (modified)" }
+      added.each   { |k| STDERR.puts "  + #{k} (new)" }
+      removed.each { |k| STDERR.puts "  - #{k} (removed)" }
+    elsif !OllamaChat.test_mode?
+      STDERR.puts '⚠️  First run — storing prompt fingerprints.'
+      set(:prompt_defaults_fingerprint,
+          { fingerprint:, hashes:, computed_at: Time.now.iso8601 })
+      return true
+    end
+
+    prompt = <<~EOT.chomp << ' '
+      Keep local prompts (stop nagging)?
+      Consider adopting via /prompt sync or /prompt reset. (y/n) %s
+    EOT
+    if OllamaChat.test_mode? || chat.confirm?(prompt:, yes: /\Ay/i, timeout: 5)
+      set(:prompt_defaults_fingerprint,
+          { fingerprint:, hashes:, computed_at: Time.now.iso8601 })
+    end
+
+    true
+  end
+
+  # @!attribute [v] id
+  #   @return [Integer] The primary key for the state entry.
+  #
+  # @!attribute [v] key
+  #   @return [String] The unique state key.
+  #
+  # @!attribute [v] value
+  #   @return [Hash, nil] The JSON-serialized state payload.
+  #
+  # @!attribute [v] created_at
+  #   @return [Time, nil] The timestamp when the state was created.
+  #
+  # @!attribute [v] updated_at
+  #   @return [Time, nil] The timestamp of the last update to the state.
+end

data/lib/ollama_chat/kramdown_ansi.rb

@@ -39,7 +39,7 @@ module OllamaChat::KramdownANSI
   # @return [ String ] the content formatted with ANSI escape sequences
   #   according to the configured styles
   def kramdown_ansi_parse(content)
-    retried = false
+    retried ||= false
     content.nil? and return ''
     Kramdown::ANSI.parse(content, ansi_styles: @kramdown_ansi_styles)
   rescue Encoding::UndefinedConversionError

data/lib/ollama_chat/model_handling.rb

@@ -268,6 +268,122 @@ module OllamaChat::ModelHandling
     end
   end
 
+  # Exports all stored model options for every model to a JSON file.
+  #
+  # The resulting JSON structure is:
+  #   [
+  #     { "model_name": "…", "profiles": [ { "profile": "…", "options": {…} } ] },
+  #     …
+  #   ]
+  #
+  # @return [Pathname, nil] the path of the written file, or nil if cancelled
+  def export_model_options
+    model_names = models::ModelOptions.distinct.map(&:model_name).sort
+    unless model_names.any?
+      STDERR.puts "❌ No model options stored yet!"
+      return
+    end
+
+    payload = model_names.map do |name|
+      { model_name: name,
+        profiles: models::ModelOptions.where(model_name: name)
+          .order(:profile).map { |m|
+            { profile: m.profile, options: m.options.to_h }
+          } }
+    end
+
+    filename = determine_valid_output_filename('to export model options to') or return
+    filename.write(JSON.pretty_generate(payload))
+    total = model_names.sum { |n|
+      models::ModelOptions.where(model_name: n).count
+    }
+    log(:info, "Model options exported",
+        data: { models: model_names, dest: filename.to_s })
+    STDOUT.puts "✅ #{model_names.size} model(s), #{total} profile(s) "\
+      "exported to #{filename.to_path.inspect}."
+    filename
+  end
+
+  # Imports model options from a JSON file into the database.
+  #
+  # The JSON file must be an array of entries:
+  #   [ { "model_name": "…", "profiles": [ … ] }, … ]
+  #
+  # If multiple models are present, the user selects which to import.
+  # For each profile, if an existing record has different options, both
+  # are displayed side-by-side and the user decides per-profile.
+  #
+  # @param filename [String, Pathname] the path to the JSON file to import
+  # @return [Boolean, nil] true on success, nil if cancelled
+  def import_model_options(filename)
+    filename = Pathname.new(filename)
+    data     = JSON.parse(filename.read)
+    unless data.is_a?(Array) && data.all? { |d| d['model_name'] }
+      STDERR.puts "❌ Invalid format: expected array of { 'model_name', 'profiles' }!"
+      return
+    end
+
+    # Let user pick models if more than one
+    if data.size > 1
+      options = (['[ALL]'] + data.map { |d| d['model_name'] } + ['[EXIT]'])
+      chosen  = choose_entry(options, prompt: 'Which model(s) to import? %s')
+      case chosen
+      when '[EXIT]', nil
+        STDOUT.puts "Cancelled."
+        return
+      when '[ALL]'
+        selected = data
+      else
+        selected = data.select { |d| d['model_name'] == chosen }
+      end
+    else
+      selected = data
+    end
+
+    imported = 0
+    selected.each do |entry|
+      model_name = entry['model_name']
+      profiles   = Array(entry['profiles'])
+      STDOUT.puts "\n📦 #{bold{model_name}} (#{profiles.size} profile(s))"
+
+      profiles.each do |profile_data|
+        profile = profile_data['profile'] || 'default'
+        options = profile_data['options'] || {}
+
+        existing = stored_model_options_exist?(model_name, profile:)
+        if existing
+          current = existing.options.to_h.symbolize_keys_recursive
+          incoming = options.to_h.symbolize_keys_recursive
+          if current == incoming
+            STDOUT.puts "   • #{italic{profile}}: identical, skipping."
+            next
+          end
+          STDOUT.puts "   • #{italic{profile}}: differs!"
+          STDOUT.puts "     📤 Current:"
+          STDOUT.puts "       " + JSON.pretty_generate(current).sub(/^/m, '     ')
+          STDOUT.puts "     📥 Incoming:"
+          STDOUT.puts "       " + JSON.pretty_generate(incoming).sub(/^/m, '     ')
+          unless confirm?(prompt: "     ⚠️ Overwrite? (y/n) ", yes: /\Ay/i)
+            STDOUT.puts "     Skipped."
+            next
+          end
+        end
+
+        store_model_options(model_name, options, profile:)
+        imported += 1
+        STDOUT.puts "   • #{italic{profile}}: ✅"
+      end
+    end
+
+    log(:info, "Model options imported",
+        data: { models: selected.map { |e| e['model_name'] },
+                imported:, source: filename.to_s })
+    STDOUT.puts "\n✅ Imported #{imported} profile(s) "\
+      "from #{filename.to_path.inspect}."
+    true
+  end
+
+
   # The model_present? method checks if the specified Ollama model is
   # available.
   #

data/lib/ollama_chat/oc.rb

@@ -1,5 +1,6 @@
 require 'const_conf'
 require 'pathname'
+require 'shellwords'
 
 # Environment configuration module for OllamaChat
 #
@@ -64,6 +65,14 @@ module OC
     end
   end
 
+  DIFF_COMMAND = set do
+    description 'Command (string) to generate a unified diff with'
+
+    required true
+    default { 'diff -u --color=always' }
+    decode { Shellwords.split(_1) if _1.present? }
+  end
+
   DIFF_TOOL = set do
     description 'Diff tool to apply changes with'
 

data/lib/ollama_chat/ollama_chat_config/default_config.yml

@@ -240,7 +240,6 @@ prompts:
       How does this implementation handle malformed JSON or missing keys in the
       middle of a 2GB file without crashing the process?
 
-
       What is the time complexity of the current insert logic, and can we implement
       bulk inserts to reduce database round trips?
 

data/lib/ollama_chat/prompt_handling.rb

@@ -10,7 +10,7 @@ module OllamaChat::PromptHandling
   # @return [OllamaChat::Database::Models::Prompt, nil] the prompt model
   #   instance or nil if not found
   def prompt(name, context: nil)
-    context ||= 'prompt'
+    context = (context || 'prompt').to_s
     models::Prompt.where(context:, name: name.to_s).first
   end
 
@@ -19,7 +19,7 @@ module OllamaChat::PromptHandling
   # @yield [prompt] yields each prompt model instance
   # @return [Enumerator] an enumerator if no block is given
   def each_prompt(context: nil, default: nil, &block)
-    context ||= 'prompt'
+    context = (context || 'prompt').to_s
     block or return enum_for(__method__, context:, default:)
     prompts = models::Prompt.where(context:)
     case default
@@ -39,7 +39,7 @@ module OllamaChat::PromptHandling
   # @param name [String, Symbol] the name of the prompt to delete
   # @return [Boolean] true if deleted, false otherwise
   def delete_prompt(name, context: nil)
-    context ||= 'prompt'
+    context = (context || 'prompt').to_s
     if found = prompt(name, context:) and !found.metadata['default']
       found.destroy
       return true
@@ -54,25 +54,25 @@ module OllamaChat::PromptHandling
   # @return [OllamaChat::Database::Models::Prompt] the saved prompt model
   #   instance
   def store_prompt(name, content, context: nil)
-    context ||= 'prompt'
+    context = (context || 'prompt').to_s
     write_prompt(name, content, context:)
   end
 
   # Creates or updates a prompt in the specified context.
   #
-  # @param context [String] the context (e.g., 'prompt' or 'system')
-  # @param name [String] the name of the prompt
+  # @param context [String, Symbol] the context (e.g., 'prompt' or 'system')
+  # @param name [String, Symbol] the name of the prompt
   # @param content [String] the content of the prompt
   # @return [OllamaChat::Database::Models::Prompt] the created or updated
   #   prompt model instance
   def write_prompt(name, content, context: nil)
-    context ||= 'prompt'
+    context = (context || 'prompt').to_s
     obj = nil
-    if found = models::Prompt.where(context:, name:).first
+    if found = models::Prompt.where(context:, name: name.to_s).first
       found.metadata['content'] = content
       obj = found
     else
-      obj = models::Prompt.create(name:, context:)
+      obj = models::Prompt.create(name: name.to_s, context:)
       obj.metadata = { default: false, content: }.stringify_keys_recursive
     end
     obj.tap(&:save)