llm_lib 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 85f973672c9b0eb31e43f3fbeb04c583a2b92bd5886c3caa320db5f972d614a2
-  data.tar.gz: f2fa7bb2bdb6d06a688a0987017c22f9fe55eb4ad6861abe1efd205773df35fc
+  metadata.gz: a7fe9391bf23bcbe47c61ac8e8449536eb3052a19f8150813bf125861492d167
+  data.tar.gz: c2107e7106b73200502730c128fe0220a2a374d8f1ba6cfe5e6db48c8e78b95d
 SHA512:
-  metadata.gz: ab1ea2cbdd89d7af8229195addb8edeb03bacb93b99879de60f78ee4924c410a0b350f5d513bc91282d9ba0a1655c8ea2abde72cf64daec495a1b014750a42f6
-  data.tar.gz: 2916daaa07a84273eeffd20c500ba3d27be918f40490b91cd5099eae793311a8c3d08ef4526b3fe731a38099bf6e6b60d4b7b84d8607884a76b5c01250847675
+  metadata.gz: cc56b149e4e2e127857528a6c1754cf4e472eab9a09a4e34932b9a63d5d200eba80718b9c87dfa0f2d6d7901e9c95526b28dd348e870cc171a601b245c465830
+  data.tar.gz: d456b665985410472ba8248a517ab0bcc9126532cf11df46bf2e4134e8aafbb3903cdf9215164b280ff8f7b1ce55920f5af41792bb70268b54d8c71ca202300b

data/lib/llm_lib/restclient.rb

@@ -1,6 +1,5 @@
 class LlmLib::Restclient
 
-# def self.post(url:, body:, headers: {})
 def self.post(url:, body:, apikey:)
 
     url = URI(url)
@@ -13,10 +12,7 @@ def self.post(url:, body:, apikey:)
     request = Net::HTTP::Post.new(url.path)
     request['Content-Type'] = 'application/json'
     request["cache-control"] = 'no-cache'
-    # request["apikey"] = apikey
-    # headers.each { |key, value| request[key] = value }
     request['Authorization'] = "Bearer #{apikey}" 
-    # request.set_form_data(body)
     request.body = JSON.generate(body)
 
     begin

data/lib/llm_lib.rb

@@ -11,43 +11,13 @@ module LlmLib
         # response = client.chat_gpt_call(prompt, max_tokens)
         # puts response
 
-        # remove
-        # attr_reader :apikey
-        # def def initialize(apikey, query)
-        #   @apikey, @query = apikey, query
-        # end
-
         def initialize(apikey)
             @apikey = apikey
         end
 
-        # def chat_gpt_call(prompt, max_tokens, temperature = 0, top_p = 1, n = 1, stream = false, logprobs = nil, stop = "\n")
-        #     model = "text-davinci-003"
-        #     response = self.class.send(@api_key, {
-        #       "model" => model,
-        #       "prompt" => prompt,
-        #       "max_tokens" => max_tokens,
-        #       "temperature" => temperature,
-        #       "top_p" => top_p,
-        #       "n" => n,
-        #       "stream" => stream,
-        #       "logprobs" => logprobs,
-        #       "stop" => stop
-        #     })
-        #     response
-        #   end
-
-        #   def self.send(apikey, body)
-        #     LlmLib::Restclient.post(
-        #       body: body,
-        #       url: "https://api.openai.com/v1/completions",
-        #       apikey: apikey
-        #     )
-        #   end
-
         def chat_gpt_call(prompt, max_tokens, temperature = 0, top_p = 1, n = 1, stream = false, stop = "\n")
             model = "gpt-3.5-turbo"
-            response = OpenAI.send(@api_key,
+            response = OpenAI.send(@apikey,
               model,
               prompt,
               max_tokens,
@@ -62,7 +32,7 @@ module LlmLib
 
         def gpt4_call(prompt, max_tokens, temperature = 0, top_p = 1, n = 1, stream = false, stop = "\n")
             model = "gpt-4"
-            response = OpenAI.send(@api_key,
+            response = OpenAI.send(@apikey,
                 model,
                 prompt,
                 max_tokens,
@@ -97,8 +67,9 @@ module LlmLib
 
         def hugging_bloom_call(query, model = "bigscience/bloom")
             response = HuggingFace.send(@api_key,
-                                    query, 
-                                    model)
+                                    model,
+                                    query 
+                                    )
             response
         end
 
@@ -116,13 +87,6 @@ module LlmLib
             response
         end
 
-        # def self.send(apikey, body, model)
-        #     LlmLib::Restclient.post(
-        #     body: body,
-        #     url: "https://api-inference.huggingface.co/models/#{model}",
-        #     apikey: apikey
-        #     )
-        # end
     end
 
 end    

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: llm_lib
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Chamath Attanayaka
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-07-28 00:00:00.000000000 Z
+date: 2023-07-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: uri