dorian-commit 0.4.4 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: eb42f850e1f1489f8ec86917d590be0c9c44d79c77da416e3f163f78179654f7
-  data.tar.gz: d7827274748c78005fba5c43ebd75cc8a029dd5e38c6fac0e481daffee590ff1
+  metadata.gz: 1cfd254ac72b0cb783a5617779fb203ed8fe84cec8685702a9f52ab18ac6de0b
+  data.tar.gz: 8158f5df736374fa3c9fc3e203755344dcdd93610d42227c127da25f14d577e1
 SHA512:
-  metadata.gz: 69d4bd2e6c98bb3f201eb0fa7631c4595a52a08f8e79e607b5108c34a8d726be9c1d1e8cfa6b556ab94e86ae7c75bb9c957ff5ade8668b684a43b045312fac53
-  data.tar.gz: '09655ba89547e1e5c4b0045eaddc0ede62d850bf7d5b098a4efa54237fd16dcbe2a9bcdee3c18ad2b88b7798d9b6b69d2ca84976e6d0cde200ba698eaa124f94'
+  metadata.gz: a328add85459b999de4436e92db6ea24cfd03912d3c0672fabe13823d24a88cfbe247db4bb2b0a2c43b1be596e3144ca2cbb5d2e67015bb25cdbcce7b480d1bb
+  data.tar.gz: 9dc85ba07932fc0be0c5781b4c5d6d94bbe5de98d686c61dc8cd234f9982608f350bd3b8436735ca48c483d2466f784c50d09994ac44325b4bf9b30a17adae91

data/bin/commit

@@ -5,6 +5,7 @@ require "dorian/arguments"
 require "git"
 require "json"
 require "net/http"
+require "tiktoken_ruby"
 require "uri"
 
 TOKEN_FILE = File.join(Dir.home, ".commit")
@@ -23,7 +24,7 @@ PROMPT_2 = "for the following diff:"
 PROMPT_3 = "for the following git status:"
 PROMPT_4 = "for the following comment:"
 
-parsed = Dorian::Arguments.parse(version: { alias: :v }, help: { alias: :h })
+parsed = Dorian::Arguments.parse(tokens: { type: :integer, alias: :t, default: 1000 }, version: { alias: :v }, help: { alias: :h })
 
 abort parsed.help if parsed.options.help
 
@@ -38,6 +39,30 @@ content_4 += parsed.files.map { |file| File.read(file) }.join("\n")
 
 abort "no staged files" if content_2.empty?
 
+encoder = Tiktoken.encoding_for_model("gpt-4o")
+
+content_2_encoded = encoder.encode(content_2)
+
+if content_2_encoded.length > parsed.options.tokens
+  messages_2 = []
+else
+  messages_2 = [
+    { role: :system, content: PROMPT_2 },
+    { role: :user, content: content_2 }
+  ]
+end
+
+content_3_encoded = encoder.encode(content_3)
+
+if content_3_encoded.length > parsed.options.tokens
+  messages_3 = []
+else
+  messages_3 = [
+    { role: :system, content: PROMPT_3 },
+    { role: :user, content: content_3 },
+  ]
+end
+
 if content_4.empty?
   messages_4 = []
 else
@@ -65,10 +90,8 @@ request.body = {
   model: "gpt-4o",
   messages: [
     { role: :system, content: PROMPT_1 },
-    { role: :system, content: PROMPT_2 },
-    { role: :user, content: content_2 },
-    { role: :system, content: PROMPT_3 },
-    { role: :user, content: content_3 },
+    *messages_2,
+    *messages_3,
     *messages_4
   ]
 }.to_json

metadata

@@ -1,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: dorian-commit
 version: !ruby/object:Gem::Version
-  version: 0.4.4
+  version: 1.1.0
 platform: ruby
 authors:
 - Dorian Marié
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-08-22 00:00:00.000000000 Z
+date: 2024-08-25 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: dorian-arguments
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: git
   requirement: !ruby/object:Gem::Requirement
@@ -25,7 +39,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: dorian-arguments
+  name: tiktoken_ruby
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="