tiny_pair 1.0.0 → 2.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a8277eb9531bc7cac4382beef85c9005df5052ff947cec691c3017bf00f276e5
-  data.tar.gz: df2adc92464f9b22fccb83f593a0e9ea047217c5545329f30f43e09adebee20c
+  metadata.gz: 4acbf237fbd17f3191100d48e5549fee2ba87c760d8c6b5ff92fcea828aa3efd
+  data.tar.gz: 72001b4be9760b31e800842d994d4c98216c37bf738fdf7b42fc3c8e0976b140
 SHA512:
-  metadata.gz: 83f3d378f2d080da1031a80813c39011beba51d4e7255e6515055618fee687f32be091671ae202780b0d77ac2c265d13b3ee855409367173ef34a89f106c766f
-  data.tar.gz: 28b4e2b443c0deae01f215d76c1aa7faf4c02ad19002b2fffc472c30c361a6eaef453510d1d3b98a7228a5b5c8ed0658a7383e2de9e35b2cd78b6a70d9072b4e
+  metadata.gz: 61ebbb83b85b7bee5facdcacb02f5f3b4c6ec325dbc5ebe7efe9bca57d1b8849e1d2d22689e03ec7acf7f56553c48058b8be6e3f7ae0980a1c66988ad2556944
+  data.tar.gz: 131394bfe45e73929477de3080cb66513da319f8eb30f98cf31d42a8a0d8d63967abd3079c20c7c58231711c5812909ad6ae334d5dda1e6c7d274a3d94d60e3b

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?
@@ -7,17 +7,21 @@ require 'tiny_gemini'
 # what happens:
 #
 #   tp = TinyPair.new
-#   puts tp.implement(tests: TinyPair::TESTS)
+#   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,38 @@ 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']
+  def initialize(api_key: ENV['TINY_PAIR_GEMINI_API_KEY'], model: ENV['TINY_PAIR_GEMINI_MODEL'])
+    @client = Gemini.new(
+      credentials: {
+        service: 'generative-language-api',
+        api_key: api_key,
+      },
+      options: { model: model, server_sent_events: false }
     )
   end
 
   # tests: the raw text of a series of minitest tests
-  def implement(tests:)
-    @client.prompt({
-      parts: { text: tests },
-      role: 'user'
-    })
+  def llm(prompt:, instructions:)
+    instructions = instructions.lines.map(&:strip).select{|l| !l.empty? }.join(' ')
+    request_body = {
+# NOTE: this doesn't seem to work, and I'm not sure why, so I put it in the
+# prompt
+#      system_instruction: {
+#        role: 'user',
+#        parts: { text: instructions }
+#      },
+      contents: {
+        role: 'user',
+        parts: [
+          { text: instructions },
+          { text: prompt }
+        ]
+      }
+    }
+
+    @client
+      .generate_content(request_body)['candidates']
+      .first['content']['parts']
+      .first['text']
   end
 end

metadata

@@ -1,17 +1,17 @@
 --- !ruby/object:Gem::Specification
 name: tiny_pair
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 2.0.0
 platform: ruby
 authors:
 - Jeff Lunt
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-08-01 00:00:00.000000000 Z
+date: 2025-06-15 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: []