omniai 0.0.7 → 0.0.8
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/lib/omniai/chat/request.rb +7 -10
- data/lib/omniai/version.rb +1 -1
- metadata +1 -1
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: b478ff0decf827e07f67c3c85973ad576f546980417f82ff5f71949c99157e22
|
4
|
+
data.tar.gz: edcd0f60037d10fca5aec3ecc6cbd08c0c9b1c42aa9353506c232821cbd0fe0f
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: 69d8116cf9313da66fc89857332d8efb9e2005f007474dd377ec8c34ffb5c18e22155740390acb793f255447399f176605f0306e94c99c0e160346581fd1e597
|
7
|
+
data.tar.gz: 5c7975664071ff8158d5fd9e33fac37f7874b856d319edb0555deeefb78ac70e1dd7c13ffde015a5b0029c4b1ad26554d34ba0ca294bbf8acc81365adef1705c
|
data/lib/omniai/chat/request.rb
CHANGED
@@ -19,23 +19,20 @@ module OmniAI
|
|
19
19
|
# client.chat.completion(messages, model: "...", temperature: 0.0, format: :text)
|
20
20
|
class Request
|
21
21
|
# @param client [OmniAI::Client] the client
|
22
|
-
# @param messages [String]
|
23
|
-
# @param model [String]
|
24
|
-
# @param temperature [Float]
|
25
|
-
# @param
|
26
|
-
|
22
|
+
# @param messages [String] required
|
23
|
+
# @param model [String] required
|
24
|
+
# @param temperature [Float, nil] optional
|
25
|
+
# @param stream [Proc, nil] optional
|
26
|
+
# @param format [Symbol, nil] optional - :text or :json
|
27
|
+
def initialize(client:, messages:, model:, temperature: nil, stream: nil, format: nil)
|
27
28
|
@client = client
|
28
29
|
@messages = messages
|
29
30
|
@model = model
|
30
31
|
@temperature = temperature
|
32
|
+
@stream = stream
|
31
33
|
@format = format
|
32
34
|
end
|
33
35
|
|
34
|
-
# @param prompt [String, Message]
|
35
|
-
# @param model [String]
|
36
|
-
# @param format [Symbol] either :text or :json
|
37
|
-
# @param temperature [Float]
|
38
|
-
# @return [Hash]
|
39
36
|
# @raise [ExecutionError]
|
40
37
|
def process!
|
41
38
|
response = request!
|
data/lib/omniai/version.rb
CHANGED