llm_lib 0.1.2 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a7fe9391bf23bcbe47c61ac8e8449536eb3052a19f8150813bf125861492d167
-  data.tar.gz: c2107e7106b73200502730c128fe0220a2a374d8f1ba6cfe5e6db48c8e78b95d
+  metadata.gz: 4285431acf89a0a82b7c4decb0d981f80616bd92bdd915e917375b8d33408dac
+  data.tar.gz: e54cd6b5a882426fd77bee80ae2f1ca20d91e918dea02edcf93563e624e44367
 SHA512:
-  metadata.gz: cc56b149e4e2e127857528a6c1754cf4e472eab9a09a4e34932b9a63d5d200eba80718b9c87dfa0f2d6d7901e9c95526b28dd348e870cc171a601b245c465830
-  data.tar.gz: d456b665985410472ba8248a517ab0bcc9126532cf11df46bf2e4134e8aafbb3903cdf9215164b280ff8f7b1ce55920f5af41792bb70268b54d8c71ca202300b
+  metadata.gz: 89c9ce0b97eb8a94e0eec825e7213dfaf535b513e7e2e1a7226dba32a5d9e5ef3904848d5ec3edaede23611e0af682c3c01f09da0db49bc04a16e6d59a164ea6
+  data.tar.gz: 3f784c7baac2ed5a08a67df24215b0cf5352f801c10d90feb3d6407876b6278bb782534a3ac55c38a686fcb60ec796d8c2b7226ac2862b9666e348859b59e91e

data/lib/llm_lib.rb

@@ -75,18 +75,33 @@ module LlmLib
 
         def hugging_falcon_call(query, model = "tiiuae/falcon-40b-instruct")
             response = HuggingFace.send(@api_key,
-                query, 
-                model)
+                                    model, 
+                                    query)
             response
         end
 
         def hugging_llama2_call(query, model = "meta-llama/Llama-2-70b-chat-hf")
             response = HuggingFace.send(@api_key,
-                query, 
-                model)
+                                    model, 
+                                    query)
             response
         end
 
+        def hugging_dolly2_call(query, model = "databricks/dolly-v2-12b")
+            response = HuggingFace.send(@api_key,
+                                    model,
+                                    query 
+                                    )
+            response
+        end
+
+        def hugging_flant5_call(query, model = "google/flan-t5-xl")
+            response = HuggingFace.send(@api_key,
+                                    model,
+                                    query 
+                                    )
+            response
+        end
     end
 
 end    

metadata

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