promptspec 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 022ef82dfab3a71fccb829c8358442d8ef7f77026d637bfad15586adef3dfce5
-  data.tar.gz: 990779a5f40d7baffe57dbc186198541ea96bd316328741e352b823f6e3b587c
+  metadata.gz: b8344f339aa0ef8a57c14422dc2f50046a203076ec9c2daee525c4432f57f6bf
+  data.tar.gz: 96014c0bc00da57d4b0289df8c39fa4a3c8505dc6a0b6814a717a31f735ad591
 SHA512:
-  metadata.gz: 391028f8084149662b27a3bca42bf6a7e8d1ff06e0ad043e9c5b82390186d6804416d153be55160d096b4e9fafa0239cd74d1777d74a46ad48276b6d045c23e4
-  data.tar.gz: 5fb14570770bf048e695ee101a5a8043d5eec74dcb245f1ac369573a4c1f10cb796d2c5d4c35f8ac4aa97505e4c798978e48b86590bba05e8fbc278d16a2f88f
+  metadata.gz: 1516fbdb9607f0724faa3c10081269b3d99c243145b1f31a19588faec6a5b7602120d536acba56fd1629b7afc1b6bd9f2d25f7cd18e4491687bceb66964c0534
+  data.tar.gz: 611b943f18480666c2d9da0a157eb186f0a4392ee876897af67d4c026829391a701882abcf19c1e9343eca7e8d35aee2131de888887537fbbf0fd96e075caf37

data/lib/promptspec.rb

@@ -6,12 +6,6 @@ require 'net/http'
 require 'uri'
 
 class PromptSpec
-  class Error < StandardError; end
-  class FileNotFoundError < Error; end
-  class ParseError < Error; end
-  class RequiredParameterError < Error; end
-  class EndpointError < Error; end
-
   attr_reader :file_path, :validate_required_params
 
   def initialize(file_path, validate_required_params: true)
@@ -21,7 +15,7 @@ class PromptSpec
   end
 
   def call(**parameters)
-    validate_required_inputs!(parameters) if validate_required_params
+    validate_required_inputs!(parameters) if @validate_required_params
     parse_prompt_messages(parameters)
     construct_endpoint_request
   end
@@ -75,9 +69,7 @@ class PromptSpec
   def construct_default_endpoint
     @model = @yaml_content['prompt']['model']
     case @model
-    when 'gpt-4'
-      'https://api.openai.com/v1/chat/completions'
-    when 'gpt-3.5-turbo'
+    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'
     else
       raise EndpointError, "Unknown model: #{model}"
@@ -85,14 +77,18 @@ class PromptSpec
   end
 
   def construct_headers
-    # Placeholder logic for constructing headers
     headers = @yaml_content['headers'] || {}
-    if @model == 'gpt-4' && headers.empty?
-      api_key = ENV['OPENAI_API_KEY']
-      raise EndpointError, "Missing OpenAI API Key" unless api_key
-      headers['Authorization'] = "Bearer #{api_key}"
+
+    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
-    headers['Content-Type'] = 'application/json'
+
+    headers['Content-Type'] = 'application/json' unless headers.key?('Content-Type')
     headers
   end
 end

data/promptspec.gemspec

@@ -1,7 +1,7 @@
 # promptspec.gemspec
 Gem::Specification.new do |spec|
   spec.name          = "promptspec"
-  spec.version       = "0.0.4"
+  spec.version       = "0.0.5"
   spec.authors       = ["Hyperaide, John Paul, Daniel Paul"]
   spec.email         = ["jp@hyperaide.com"]
 
@@ -14,7 +14,6 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
 
   spec.add_runtime_dependency "yaml", "~> 0.1.0"
-  spec.add_runtime_dependency "erb", "~> 2.2.0"
 
   spec.add_development_dependency "bundler", "~> 2.0"
   spec.add_development_dependency "rake", "~> 13.0"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: promptspec
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Hyperaide, John Paul, Daniel Paul
@@ -24,20 +24,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.1.0
-- !ruby/object:Gem::Dependency
-  name: erb
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 2.2.0
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 2.2.0
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement