tiny_pair 1.0.0 → 3.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a8277eb9531bc7cac4382beef85c9005df5052ff947cec691c3017bf00f276e5
-  data.tar.gz: df2adc92464f9b22fccb83f593a0e9ea047217c5545329f30f43e09adebee20c
+  metadata.gz: 2977ae760f8e81ef8aa7dda582cd99efeb159f58183c704fbe3b68e6f6f0eb98
+  data.tar.gz: 870bd35db707a58b45aa6ac7031d6361bf01f0e5f58a223ddf9f87e3f729ced7
 SHA512:
-  metadata.gz: 83f3d378f2d080da1031a80813c39011beba51d4e7255e6515055618fee687f32be091671ae202780b0d77ac2c265d13b3ee855409367173ef34a89f106c766f
-  data.tar.gz: 28b4e2b443c0deae01f215d76c1aa7faf4c02ad19002b2fffc472c30c361a6eaef453510d1d3b98a7228a5b5c8ed0658a7383e2de9e35b2cd78b6a70d9072b4e
+  metadata.gz: 282da0f95ba8017b5e65e6d09207b400be09111788a09038ad9d04646d594b4060bf83633b2dafea77878d780033527ec1046479c90dbce78524c204aabcc106
+  data.tar.gz: 8d77e1d6fb175755307d4386cc55ea84072d3ad860ddabc46acb7859d1f60c5430e8fd36f0db8f2fe06888c87b65fed19965e7c791e0d58a9d451d215550a875

data/lib/tiny_pair.rb

@@ -1,4 +1,4 @@
-require 'tiny_gemini'
+require 'gemini-ai'
 
 # if you can write the specification for some code, shouldn't the computer be
 # able to write the implementation?
@@ -6,18 +6,22 @@ require 'tiny_gemini'
 # get your Google Gemini API key ready, and then run the following code to see
 # what happens:
 #
-#   tp = TinyPair.new
-#   puts tp.implement(tests: TinyPair::TESTS)
+#   tp = TinyPair.new(instructions: TinyPair::TEST_MODEL_INSTRUCTIONS)
+#   puts tp.prompt(TinyPair::TEST_TEXT)
+#
+# override model instructions if you like:
+#   tp = TinyPair.new(instructions: <your custom instructions>)
+#
 class TinyPair
-  MODEL_INSTRUCTIONS = <<~INSTR
+  TEST_MODEL_INSTRUCTIONS = <<~INSTR
     You will be given some minitest tests in Ruby, and I want you to return the matching Ruby implementation.
 
-    You should use the latest version of Ruby you know of, and try to keep the output code to an 80-column width
+    You should use the latest version of Ruby you know of, and try to keep the output code to an 80-column width.
 
-    Do not explain the code or any surrounding documentation: only provide the implementation code in plaintext and nothing else.
+    Only reply with a code snippet; no documentation, and no explanation.
   INSTR
 
-  TESTS = <<~TEST_TEXT
+  TEST_TEXT = <<~TEST_TEXT
     class TestCalc < Minitest::Test
       def test_add_two_and_two
         assert_equal 4, Calc.add(2, 2)
@@ -37,19 +41,46 @@ class TinyPair
     end
   TEST_TEXT
 
-  def initialize(instructions: MODEL_INSTRUCTIONS)
-    @client = TinyGemini.new(
-      model: 'gemini-1.5-flash',
-      system_instruction: MODEL_INSTRUCTIONS,
-      api_key: ENV['GEMINI_KEY']
+  attr_reader :msgs
+
+  def initialize(api_key: ENV['TINY_PAIR_GEMINI_API_KEY'], model: ENV['TINY_PAIR_GEMINI_MODEL'], instructions: nil)
+    @gemini = Gemini.new(
+      credentials: {
+        service: 'generative-language-api',
+        api_key: api_key,
+      },
+      options: { model: model, server_sent_events: false }
     )
+
+    @msgs = []
+
+    instructions&.strip!
+    @msgs << {
+      role: 'user',
+      parts: [{ text: instructions.strip }]
+    } if instructions && !instructions.empty?
   end
 
-  # tests: the raw text of a series of minitest tests
-  def implement(tests:)
-    @client.prompt({
-      parts: { text: tests },
-      role: 'user'
-    })
+  # starts or continues a conversation with the msg
+  def prompt(msg)
+    if @msgs.length == 1
+      @msgs.last[:parts] << { text: msg }
+    else
+      @msgs << { role: 'user', parts: [{ text: msg }] }
+    end
+
+    request_body = { contents: @msgs }
+    response = @gemini.generate_content(request_body)
+    candidate = response['candidates']&.first
+    model_content = candidate&.dig('content')
+
+    if model_content
+      @msgs << model_content
+      model_content.dig('parts', 0, 'text') || ""
+    else
+      error_message = "Failed to get a valid response from the API. Response: #{response.to_json}"
+      @msgs.pop
+      raise error_message
+    end
   end
 end

metadata

@@ -1,17 +1,17 @@
 --- !ruby/object:Gem::Specification
 name: tiny_pair
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 3.0.0
 platform: ruby
 authors:
 - Jeff Lunt
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-08-01 00:00:00.000000000 Z
+date: 2025-06-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: tiny_gemini
+  name: gemini-ai
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -24,7 +24,7 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: a tiny gem pair programming with a computer
+description: a tiny pair programming gem that uses an LLM
 email: jefflunt@gmail.com
 executables: []
 extensions: []
@@ -50,8 +50,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.19
+rubygems_version: 3.5.23
 signing_key:
 specification_version: 4
-summary: a tiny gem pair programming with a computer
+summary: a tiny pair programming gem that uses an LLM
 test_files: []