promptspec 0.0.5 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b8344f339aa0ef8a57c14422dc2f50046a203076ec9c2daee525c4432f57f6bf
-  data.tar.gz: 96014c0bc00da57d4b0289df8c39fa4a3c8505dc6a0b6814a717a31f735ad591
+  metadata.gz: ff9862c476f10363aff3327d9f365b0dce6a3d09967569d6336bdd78b7897571
+  data.tar.gz: 33b50174079b344a29d8d230ba844bc363fd43cb8fb2d63f9fdf70c11ad48821
 SHA512:
-  metadata.gz: 1516fbdb9607f0724faa3c10081269b3d99c243145b1f31a19588faec6a5b7602120d536acba56fd1629b7afc1b6bd9f2d25f7cd18e4491687bceb66964c0534
-  data.tar.gz: 611b943f18480666c2d9da0a157eb186f0a4392ee876897af67d4c026829391a701882abcf19c1e9343eca7e8d35aee2131de888887537fbbf0fd96e075caf37
+  metadata.gz: cebe769d93847ca3e7383283a6cddd37d0235f53de7d6595b4c9ce642f4626b92d5e46dd25dd200d9a75ee6025e9e4172c8b932f3e47cdd38af75457eaee4c7a
+  data.tar.gz: ca1d578071ceeefcaffe5af466d974327fe99faa75289aa3590df3f6065aec45c1b60734ca7ada1db979ab2c18e0c49db15a5c8e96dedb8d4f9c8e42b274dedd

data/lib/promptspec.rb

@@ -6,7 +6,7 @@ require 'net/http'
 require 'uri'
 
 class PromptSpec
-  attr_reader :file_path, :validate_required_params
+  attr_reader :file_path, :validate_required_params, :parameters, :yaml_content
 
   def initialize(file_path, validate_required_params: true)
     @file_path = file_path
@@ -15,8 +15,9 @@ class PromptSpec
   end
 
   def call(**parameters)
-    validate_required_inputs!(parameters) if @validate_required_params
-    parse_prompt_messages(parameters)
+    @parameters = parameters
+    validate_required_inputs! if @validate_required_params
+    parse_prompt_messages
     construct_endpoint_request
   end
 
@@ -30,21 +31,20 @@ class PromptSpec
     raise ParseError, "YAML parsing error: #{e.message}"
   end
 
-  def validate_required_inputs!(parameters)
+  def validate_required_inputs!
     required_params = @yaml_content.dig('parameters', 'required') || []
-    missing_params = required_params - parameters.keys.map(&:to_s)
+    missing_params = required_params - @parameters.keys.map(&:to_s)
     raise RequiredParameterError, "Missing required parameters: #{missing_params.join(', ')}" unless missing_params.empty?
   end
 
-  def parse_prompt_messages(parameters)
-    messages = @yaml_content['prompt']['messages'].map do |message|
+  def parse_prompt_messages
+    @yaml_content['prompt']['messages'].each do |message|
       content = message['content']
-      parameters.each do |key, value|
-        content.gsub!("{#{key}}", value.to_s)
+      @parameters.each do |key, value|
+        content.gsub!("{#{key}}", value.to_s) if content.include?("{#{key}}")
       end
-      { role: message['role'], content: content }
+      message['content'] = content
     end
-    @yaml_content['prompt']['messages'] = messages
   end
 
   def construct_endpoint_request
@@ -67,7 +67,7 @@ class PromptSpec
   end
 
   def construct_default_endpoint
-    @model = @yaml_content['prompt']['model']
+    @model ||= @yaml_content['prompt']['model']
     case @model
     when 'gpt-4', 'gpt-4-0613', 'gpt-4-32k', 'gpt-4-32k-0613', 'gpt-3.5-turbo', 'gpt-3.5-turbo-16k', 'gpt-3.5-turbo-0613', 'gpt-3.5-turbo-16k-0613'
       'https://api.openai.com/v1/chat/completions'
@@ -77,18 +77,20 @@ class PromptSpec
   end
 
   def construct_headers
-    headers = @yaml_content['headers'] || {}
+    @headers ||= begin
+      headers = @yaml_content.fetch('headers', {})
 
-    if headers.empty?
-      case @model
-      when 'gpt-4', 'gpt-4-0613', 'gpt-4-32k', 'gpt-4-32k-0613', 'gpt-3.5-turbo', 'gpt-3.5-turbo-16k', 'gpt-3.5-turbo-0613', 'gpt-3.5-turbo-16k-0613'
-        api_key = ENV['OPENAI_API_KEY']
-        headers['Authorization'] = "Bearer #{api_key}" if api_key
-      # Add more cases here for other providers
+      if headers.empty?
+        case @model
+        when 'gpt-4', 'gpt-4-0613', 'gpt-4-32k', 'gpt-4-32k-0613', 'gpt-3.5-turbo', 'gpt-3.5-turbo-16k', 'gpt-3.5-turbo-0613', 'gpt-3.5-turbo-16k-0613'
+          api_key = ENV['OPENAI_API_KEY']
+          headers['Authorization'] = "Bearer #{api_key}" if api_key
+        # Add more cases here for other providers
+        end
       end
-    end
 
-    headers['Content-Type'] = 'application/json' unless headers.key?('Content-Type')
-    headers
+      headers['Content-Type'] = 'application/json' unless headers.key?('Content-Type')
+      headers
+    end
   end
-end
+end

data/promptspec.gemspec

@@ -1,7 +1,7 @@
 # promptspec.gemspec
 Gem::Specification.new do |spec|
   spec.name          = "promptspec"
-  spec.version       = "0.0.5"
+  spec.version       = "0.0.6"
   spec.authors       = ["Hyperaide, John Paul, Daniel Paul"]
   spec.email         = ["jp@hyperaide.com"]
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: promptspec
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors:
 - Hyperaide, John Paul, Daniel Paul
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-11-03 00:00:00.000000000 Z
+date: 2023-11-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: yaml