lex-prompt 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: '0790ff654b0cbe3cd763f78988e7392301a4e51532a45f94ab31f13911547803'
+  data.tar.gz: 118bf80427c47630a7a551634fe963fb10fe4bd2ed5363ead4d3d334bcb7b299
+SHA512:
+  metadata.gz: a037896893b6dbbb6832328029c941c69972a367f7fbbdf0ebba0b3f29a1118c6345e77db43f0e47490e646275040bde14c3884380785465f5a95cef1f079799
+  data.tar.gz: ed3e3334cb08cf1422538b7b848a376d508411763bd41036f43212608c39d5fdb40a98170792d14aaaa1a66ea8668d42f368ca53bef223a5eff37c412be03748

data/lib/legion/extensions/prompt/client.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module Prompt
+      class Client
+        include Runners::Prompt
+
+        def initialize(db: nil, **opts)
+          @db   = db
+          @opts = opts
+        end
+      end
+    end
+  end
+end

data/lib/legion/extensions/prompt/helpers/template_renderer.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+require 'openssl'
+
+module Legion
+  module Extensions
+    module Prompt
+      module Helpers
+        module TemplateRenderer
+          module_function
+
+          def render(template, variables: {})
+            result = template.dup
+            variables.each { |key, val| result.gsub!("{{#{key}}}", val.to_s) }
+            result
+          end
+
+          def extract_variables(template)
+            template.scan(/\{\{(\w+)\}\}/).flatten.uniq
+          end
+
+          def content_hash(template, model_params = {})
+            payload = "#{template}|#{model_params.sort}"
+            OpenSSL::Digest.new('SHA256').hexdigest(payload)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/legion/extensions/prompt/runners/prompt.rb

@@ -0,0 +1,111 @@
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module Prompt
+      module Runners
+        module Prompt
+          def create_prompt(name:, template:, description: nil, model_params: {}, **)
+            now = Time.now.utc
+            prompt_id = db[:prompts].insert(name: name, description: description, created_at: now, updated_at: now)
+            hash = Helpers::TemplateRenderer.content_hash(template, model_params)
+            db[:prompt_versions].insert(
+              prompt_id: prompt_id, version: 1, template: template,
+              model_params: serialize_params(model_params), content_hash: hash, created_at: now
+            )
+            { created: true, name: name, version: 1, prompt_id: prompt_id }
+          end
+
+          def update_prompt(name:, template:, model_params: {}, **)
+            prompt = db[:prompts].where(name: name).first
+            return { error: 'not_found' } unless prompt
+
+            hash = Helpers::TemplateRenderer.content_hash(template, model_params)
+            latest = db[:prompt_versions].where(prompt_id: prompt[:id]).order(Sequel.desc(:version)).first
+            return { updated: false, reason: 'no_change' } if latest && latest[:content_hash] == hash
+
+            new_version = (latest ? latest[:version] : 0) + 1
+            db[:prompt_versions].insert(
+              prompt_id: prompt[:id], version: new_version, template: template,
+              model_params: serialize_params(model_params), content_hash: hash, created_at: Time.now.utc
+            )
+            db[:prompts].where(id: prompt[:id]).update(updated_at: Time.now.utc)
+            { updated: true, name: name, version: new_version }
+          end
+
+          def get_prompt(name:, version: nil, tag: nil, **)
+            prompt = db[:prompts].where(name: name).first
+            return { error: 'not_found' } unless prompt
+
+            pv = resolve_version(prompt[:id], version: version, tag: tag)
+            return { error: 'version_not_found' } unless pv
+
+            { name: name, version: pv[:version], template: pv[:template],
+              model_params: deserialize_params(pv[:model_params]),
+              content_hash: pv[:content_hash], created_at: pv[:created_at] }
+          end
+
+          def list_prompts(**)
+            db[:prompts].all.map do |prompt|
+              latest = db[:prompt_versions].where(prompt_id: prompt[:id]).order(Sequel.desc(:version)).first
+              { name: prompt[:name], description: prompt[:description],
+                latest_version: latest ? latest[:version] : nil, updated_at: prompt[:updated_at] }
+            end
+          end
+
+          def tag_prompt(name:, tag:, version: nil, **)
+            prompt = db[:prompts].where(name: name).first
+            return { error: 'not_found' } unless prompt
+
+            pv = resolve_version(prompt[:id], version: version)
+            return { error: 'version_not_found' } unless pv
+
+            db[:prompt_tags].insert_conflict(target: %i[prompt_id tag], update: { version_id: pv[:id] })
+                            .insert(prompt_id: prompt[:id], tag: tag, version_id: pv[:id], created_at: Time.now.utc)
+            { tagged: true, name: name, tag: tag, version: pv[:version] }
+          end
+
+          def render_prompt(name:, variables: {}, version: nil, tag: nil, **)
+            pv = get_prompt(name: name, version: version, tag: tag)
+            return pv if pv[:error]
+
+            rendered = Helpers::TemplateRenderer.render(pv[:template], variables: variables)
+            { rendered: rendered, prompt_version: pv[:version] }
+          end
+
+          private
+
+          def resolve_version(prompt_id, version: nil, tag: nil)
+            if tag
+              tag_row = db[:prompt_tags].where(prompt_id: prompt_id, tag: tag).first
+              tag_row ? db[:prompt_versions].where(id: tag_row[:version_id]).first : nil
+            elsif version
+              db[:prompt_versions].where(prompt_id: prompt_id, version: version).first
+            else
+              db[:prompt_versions].where(prompt_id: prompt_id).order(Sequel.desc(:version)).first
+            end
+          end
+
+          def serialize_params(params)
+            return '{}' if params.nil? || params.empty?
+
+            ::JSON.dump(params)
+          end
+
+          def deserialize_params(raw)
+            return {} if raw.nil? || raw.empty? || raw == '{}'
+
+            result = ::JSON.parse(raw)
+            result.is_a?(Hash) ? result.transform_keys(&:to_sym) : {}
+          rescue ::JSON::ParserError
+            {}
+          end
+
+          def db
+            @db
+          end
+        end
+      end
+    end
+  end
+end

data/lib/legion/extensions/prompt/version.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module Prompt
+      VERSION = '0.1.0'
+    end
+  end
+end

data/lib/legion/extensions/prompt.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+require_relative 'prompt/version'
+require_relative 'prompt/helpers/template_renderer'
+require_relative 'prompt/runners/prompt'
+require_relative 'prompt/client'
+
+module Legion
+  module Extensions
+    module Prompt
+      extend Legion::Extensions::Core if defined?(Legion::Extensions::Core)
+    end
+  end
+end

metadata

@@ -0,0 +1,47 @@
+--- !ruby/object:Gem::Specification
+name: lex-prompt
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Matthew Iverson
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies: []
+description: Provides versioned prompt storage with immutable versions, tagged releases,
+  template rendering, and a playground for testing prompts.
+email:
+- matt@iverson.io
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/legion/extensions/prompt.rb
+- lib/legion/extensions/prompt/client.rb
+- lib/legion/extensions/prompt/helpers/template_renderer.rb
+- lib/legion/extensions/prompt/runners/prompt.rb
+- lib/legion/extensions/prompt/version.rb
+homepage: https://github.com/LegionIO/lex-prompt
+licenses:
+- MIT
+metadata:
+  rubygems_mfa_required: 'true'
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '3.4'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.9
+specification_version: 4
+summary: Versioned prompt management for LegionIO
+test_files: []