promptspec 0.0.5 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b8344f339aa0ef8a57c14422dc2f50046a203076ec9c2daee525c4432f57f6bf
-  data.tar.gz: 96014c0bc00da57d4b0289df8c39fa4a3c8505dc6a0b6814a717a31f735ad591
+  metadata.gz: 02cda944eeaf09e58bd5ce6735e6c3f04346624d40be692d6de61b584717846c
+  data.tar.gz: acac7f476102409d6b1b0f9f277a8ccd00de474ffdd2044f4469960fb221f0f8
 SHA512:
-  metadata.gz: 1516fbdb9607f0724faa3c10081269b3d99c243145b1f31a19588faec6a5b7602120d536acba56fd1629b7afc1b6bd9f2d25f7cd18e4491687bceb66964c0534
-  data.tar.gz: 611b943f18480666c2d9da0a157eb186f0a4392ee876897af67d4c026829391a701882abcf19c1e9343eca7e8d35aee2131de888887537fbbf0fd96e075caf37
+  metadata.gz: 6a665777ac2a33223b7104dce51e23dd0420c37d72973bbd4fe3b32126cecba11c4e614ce3981a0b107ceb3df6a79ff3825f1bf6fea9a2ee50921d906555be2c
+  data.tar.gz: f8573b02222b519c0b79ee3c62385f4fcd49a1ef90d034e3c3ad32edf345a1dabb4efb1af1c24987ad978196c774427114d656f65fd82bf20b7e9a3e84d88b62

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,25 +31,24 @@ 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
-    endpoint = @yaml_content['endpoint'] || construct_default_endpoint
+    endpoint = @yaml_content['url'] || construct_default_endpoint
     headers = construct_headers
     payload = @yaml_content['prompt']
   
@@ -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,22 @@ class PromptSpec
   end
 
   def construct_headers
-    headers = @yaml_content['headers'] || {}
-
-    if headers.empty?
+    @headers ||= begin
+      yaml_headers = @yaml_content.fetch('headers', {})
+      headers = {}
+  
       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
+      else
+        headers.merge!(yaml_headers)
+      end
+  
+      headers['Content-Type'] = 'application/json' unless headers.key?('Content-Type')
+      headers
     end
-
-    headers['Content-Type'] = 'application/json' unless headers.key?('Content-Type')
-    headers
   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.7"
   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.7
 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-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: yaml