luo 0.2.6 → 0.2.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8d99e88006e5445420b8659fd680122ca1b047e15d9e97e7b62b6fd2144930b9
-  data.tar.gz: e7b6e686e73e1daa3bc3d345e8a1523267220e768bdd474d345dc263f2e9ab6b
+  metadata.gz: 4569a6fd6ae87797597f59bc73f8c1b6ac8d805f3410ecc716ffc7e1ddf428c1
+  data.tar.gz: 9695305bfe27cec9708fbc6fea8d1b000ff8450b17befdaa1a0c822da815f157
 SHA512:
-  metadata.gz: 20eafa8f4fafc7483e3acf28f532729eb6a993dad888d93f5c28d7bd7608643909d7fe771db4772c0e896658d9c3a6587760c9e85c115e81e7d29ea4813afc7a
-  data.tar.gz: 0b6cf883f769c6cbad3a178faa16ae6cd02dc7f240335e2b17b30f096b1d4f2b126a07cde18ecf8b5dacf4d8ead7d7a0add0f308abbeb4dd406f818c9778fe68
+  metadata.gz: 281fca6621c8aaccc3c21559c4a0e8ef80738fdf648bb5db68fd639b3692bb4a5f5890c5ad20b94dab9159ae27a5569b99c1a3cf1408603af5894be76f315586
+  data.tar.gz: 80a2b4894e507b1874e21ba0928bc8b5c16cc88fda9870e5ee2c8b333dd1dcc0e2d80c3d5f954ce6abf8f35b56f93127d1aad1ebca34278c7d7427241429c904

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    luo (0.2.6)
+    luo (0.2.7)
       dotenv (~> 2.8, >= 2.8.1)
       dry-configurable (~> 1.0, >= 1.0.1)
       dry-schema (~> 1.13, >= 1.13.1)

data/lib/luo/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Luo
-  VERSION = "0.2.6"
+  VERSION = "0.2.7"
 end

data/lib/luo/xinghuo.rb

@@ -26,12 +26,31 @@ module Luo
       optional(:stream).maybe(:bool)
     end
 
+    EMBEDDING_PARAMS = Dry::Schema.Params do
+      required(:input).filled(:string)
+    end
+
     # header uid max length is 32 todo
 
     def request_chat(params, &block)
       client.post('/v1/spark/completions', params.to_h, &block)
     end
 
+    def embedding(params)
+      client.post('/v1/embedding', params.to_h)
+    end
+
+    def create_embedding(text, model: 'text-embedding-ada-002')
+      params = EMBEDDING_PARAMS.call(input: text)
+      return params.errors unless params.success?
+      response = embedding(params)
+      if response.success?
+        response.body.dig("data")
+      else
+        raise "create_embeddings failed: #{response.body}"
+      end
+    end
+
     def chat(messages, random_threshold: nil, &block)
       if messages.is_a?(Messages)
         messages = messages.to_a

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: luo
 version: !ruby/object:Gem::Version
-  version: 0.2.6
+  version: 0.2.7
 platform: ruby
 authors:
 - MJ