ai-chat 0.5.6 → 0.5.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a8573f38afcb8919bc8a4c7612cd8d99ee54fa9271ee974fc1c266af3a6daaa5
-  data.tar.gz: 021e2c5e81597b2154142f65b4771932b615361b40c41d8e855675fb0db88a90
+  metadata.gz: bdaa537e8986ff0070250da8e7c6cc0a2b39fccab163aa4f2c858b3217bb6da3
+  data.tar.gz: 6e2077b0e87d86968d43c57b0f5bae1db7341b9fa16aafc1652712b87b67c438
 SHA512:
-  metadata.gz: 1ccf46c12086468abc44ecc967ed7ef3c52f19661ebe862f1516b79fd504e8b801300785cb39f7e8195443b55b8e101beb01955ca66bf43550f6f14d261e3fd6
-  data.tar.gz: 74dcc5634d3eaaa4d9f6b3e123e91b016385883e9c4a3bfb4d71904fde07651d79bad711dfe1f3e2735d31a153e8d0d9efc0a6593b887be2d728364c99160244
+  metadata.gz: 145f78aa4de9dd582d2772ab576c3bbe2c3a30623f6f8172eb2189edbdb198132778784855c498d7f28e0368f16ba8b5b1d7d258085b3c3645b833bd1fbaabd4
+  data.tar.gz: 48cfe57e2bffdcb039ce0177dd5961edba51f9df8e6c144bdcba8bea1dfc4902365fcfc43bb3cb6c76b11396c9e9e7383d0af327fe1ac82e3e08a45c15e84677

data/README.md

@@ -238,7 +238,7 @@ See [OpenAI's model documentation](https://platform.openai.com/docs/models) for
 
 ### API key
 
-The gem by default looks for an environment variable called `OPENAI_API_KEY` and uses that if it finds it.
+The gem by default looks for `AICHAT_API_KEY` first. If that is missing (or empty), it falls back to `OPENAI_API_KEY`.
 
 You can specify a different environment variable name:
 
@@ -404,7 +404,7 @@ AI::Chat.generate_schema!("A user profile with name (required), email (required)
 
 This method returns a String containing the JSON schema. The JSON schema also writes (or overwrites) to `schema.json` at the root of the project.
 
-Similar to generating messages with `AI::Chat` objects, this class method will assume that you have an API key called `OPENAI_API_KEY` defined. You can also pass the API key directly or choose a different environment variable key for it to use.
+Similar to generating messages with `AI::Chat` objects, this class method will look for `AICHAT_API_KEY` first, then fall back to `OPENAI_API_KEY` if needed. You can also pass the API key directly or choose a different environment variable key for it to use.
 
 ```rb
 # Passing the API key directly
@@ -748,7 +748,9 @@ This is particularly useful for background mode workflows. If you want to retrie
 ```ruby
 require "openai"
 
-client = OpenAI::Client.new(api_key: ENV.fetch("OPENAI_API_KEY"))
+api_key = ENV["AICHAT_API_KEY"]
+api_key = ENV.fetch("OPENAI_API_KEY") if api_key.nil? || api_key.empty?
+client = OpenAI::Client.new(api_key: api_key)
 
 response_id = "resp_abc123..." # e.g., load from your database
 response = client.responses.retrieve(response_id)

data/ai-chat.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |spec|
   spec.name = "ai-chat"
-  spec.version = "0.5.6"
+  spec.version = "0.5.7"
   spec.authors = ["Raghu Betina", "Jelani Woods"]
   spec.email = ["raghu@firstdraft.com", "jelani@firstdraft.com"]
   spec.homepage = "https://github.com/firstdraft/ai-chat"

data/lib/ai/chat.rb

@@ -23,8 +23,8 @@ module AI
 
     BASE_PROXY_URL = "https://prepend.me/api.openai.com/v1"
 
-    def initialize(api_key: nil, api_key_env_var: "OPENAI_API_KEY")
-      @api_key = api_key || ENV.fetch(api_key_env_var)
+    def initialize(api_key: nil, api_key_env_var: nil)
+      @api_key = self.class.resolve_api_key(api_key: api_key, api_key_env_var: api_key_env_var)
       @proxy = ENV["AICHAT_PROXY"] == "true"
       @messages = []
       @reasoning_effort = nil
@@ -39,8 +39,8 @@ module AI
       @verbosity = :medium
     end
 
-    def self.generate_schema!(description, location: "schema.json", api_key: nil, api_key_env_var: "OPENAI_API_KEY", proxy: nil)
-      api_key ||= ENV.fetch(api_key_env_var)
+    def self.generate_schema!(description, location: "schema.json", api_key: nil, api_key_env_var: nil, proxy: nil)
+      api_key = resolve_api_key(api_key: api_key, api_key_env_var: api_key_env_var)
       proxy = ENV["AICHAT_PROXY"] == "true" if proxy.nil?
       prompt_path = File.expand_path("../prompts/schema_generator.md", __dir__)
       system_prompt = File.read(prompt_path)
@@ -73,6 +73,16 @@ module AI
       content
     end
 
+    def self.resolve_api_key(api_key: nil, api_key_env_var: nil)
+      return api_key if api_key
+      return ENV.fetch(api_key_env_var) if api_key_env_var
+
+      aichat_api_key = ENV["AICHAT_API_KEY"]
+      return aichat_api_key if aichat_api_key && !aichat_api_key.empty?
+
+      ENV.fetch("OPENAI_API_KEY")
+    end
+
     # :reek:TooManyStatements
     # :reek:NilCheck
     def add(content, role: "user", response: nil, status: nil, image: nil, images: nil, file: nil, files: nil)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ai-chat
 version: !ruby/object:Gem::Version
-  version: 0.5.6
+  version: 0.5.7
 platform: ruby
 authors:
 - Raghu Betina