llmed 0.1.16 → 0.1.17

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '09c50ad00371154217526840bd23d9905bdd41db9603e677a94508f27518c09f'
-  data.tar.gz: fdea6c57beeb69aa45a5324f4533431ca732da087538130e4c483c22b6ceef0a
+  metadata.gz: 66603fc5fdc0f63f5c066ca085a80dee2a259e4a1d30f315f52e018f5bfec0de
+  data.tar.gz: 244a8d4938c85cd395a541eb18491e007b8e116d4dbf997d4f71a05050d5b5a9
 SHA512:
-  metadata.gz: 25adbc23488893e2c29ba284e3da0ab45fc466d5258bc6f54753a87bf2eb7273d0eafbfe95eda1b77d1deaad58bc897d47d3d6cdc7e83eac4f3cb43a712bf0ba
-  data.tar.gz: f1fe49760f451316f9ff3155c482155c3fb6653139f386602175dab09dc95b578d90cc9147218ede345246cfb2fdbbebf44b5e7e271d01c7ab30e04aeafe138b
+  metadata.gz: 2f9c7c6e0f9949b7a2876051912a645e786bd65eaba09e23bf837f1e52ae1cb8c0d645194028ac5f8b65c5e0a0f8d5d6197d6ac1155753dfa38f887ac57a6fc9
+  data.tar.gz: 308e33e710860f4ed3b63b007875aa4c339edac3bbb9f8ab6c13452cb48919408e610282991f53fefb38c27a009fb3e9acf0e4035944f40867a9f833825b2e49

data/lib/llm.rb

@@ -16,7 +16,7 @@ class LLMed
       end
     end
 
-    Response = Struct.new(:provider, :model, :source_code, :total_tokens, keyword_init: true)
+    Response = Struct.new(:provider, :model, :source_code, :duration_seconds, :total_tokens, keyword_init: true)
 
     class OpenAI
       def initialize(**args)
@@ -33,9 +33,12 @@ class LLMed
           end
         end
 
+        start = Time.now
         llm_response = @llm.chat(messages: messages)
+        stop = Time.now
         Response.new({ provider: :openai,
                        model: @llm.chat_parameters[:model],
+                       duration_seconds: stop.to_i - start.to_i,
                        source_code: source_code(llm_response.chat_completion),
                        total_tokens: llm_response.total_tokens })
       end
@@ -57,6 +60,7 @@ class LLMed
 
         Response.new({ provider: :test,
                        model: 'test',
+                       duration_seconds: 0,
                        source_code: @output,
                        total_tokens: 0 })
       end

data/lib/llmed.rb

@@ -189,7 +189,8 @@ You must only modify the following source code:
           provider: response.provider,
           model: response.model,
           release: @release,
-          total_tokens: response.total_tokens
+          total_tokens: response.total_tokens,
+          duration_seconds: response.duration_seconds
         }
         file.puts stat.to_json
       end
@@ -226,11 +227,7 @@ You must only modify the following source code:
 
   def compile(output_dir:, release_dir: nil)
     @applications.each do |app|
-      app.notify('COMPILE START')
-      _, elapsed_seconds = measure do
-        compile_application(app, output_dir, release_dir)
-      end
-      app.notify("COMPILE DONE #{elapsed_seconds}")
+      compile_application(app, output_dir, release_dir)
     end
   end
 
@@ -239,6 +236,7 @@ You must only modify the following source code:
   def compile_application(app, output_dir, release_dir)
     release_dir ||= output_dir
 
+    app.notify('COMPILE START')
     @logger.info("APPLICATION #{app.name} COMPILING")
 
     llm = @configuration.llm
@@ -258,6 +256,7 @@ You must only modify the following source code:
     @logger.info("APPLICATION #{app.name} TOTAL TOKENS #{llm_response.total_tokens}")
     write_output(app, output_dir, llm_response.source_code)
     write_statistics(app, release_dir, llm_response)
+    app.notify("COMPILE DONE #{llm_response.duration_seconds}")
   end
 
   def write_statistics(app, release_dir, response)
@@ -269,13 +268,6 @@ You must only modify the following source code:
       file.write(output)
     end
   end
-
-  def measure
-    start = Time.now
-    result = yield
-    stop = Time.now
-    [result, stop - start]
-  end
 end
 
 require_relative 'llm'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llmed
 version: !ruby/object:Gem::Version
-  version: 0.1.16
+  version: 0.1.17
 platform: ruby
 authors:
 - Jovany Leandro G.C