omniai-mistral 1.9.3 → 1.9.6
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/Gemfile +13 -10
- data/lib/omniai/mistral/chat.rb +24 -22
- data/lib/omniai/mistral/client.rb +1 -1
- data/lib/omniai/mistral/config.rb +3 -3
- data/lib/omniai/mistral/embed.rb +2 -2
- data/lib/omniai/mistral/version.rb +1 -1
- data/lib/omniai/mistral.rb +3 -3
- metadata +7 -10
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: fc1a094aecb12eaac9663fa11782de2b6aef77ae22cb32090cebdb27c4f66cda
|
4
|
+
data.tar.gz: 2232c97c97ea560a5c57c7f509985d972a963b192b94d9785557e8ca9f4a1fb4
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: 48ad9c34c199cd6b1e0248b5ce8ad761ec9a2ede91f3acc137dfff68cbe30b07b9f0989db88c61f9f073d6b3247eec6d029dcff15af2753ad8dd9a0872314c1e
|
7
|
+
data.tar.gz: a4f68fee9f21920365207ff11e74eab60c0c3a81dc6c006934b30868e011857fd6cf877b00ce3103010f6da4577be7180e1892bf9a8e0476b73c17b3d0b06ad1
|
data/Gemfile
CHANGED
@@ -1,15 +1,18 @@
|
|
1
1
|
# frozen_string_literal: true
|
2
2
|
|
3
|
-
source
|
3
|
+
source "https://rubygems.org"
|
4
4
|
|
5
5
|
gemspec
|
6
6
|
|
7
|
-
gem
|
8
|
-
gem
|
9
|
-
gem
|
10
|
-
gem
|
11
|
-
gem
|
12
|
-
gem
|
13
|
-
gem
|
14
|
-
gem
|
15
|
-
gem
|
7
|
+
gem "irb"
|
8
|
+
gem "logger"
|
9
|
+
gem "rake"
|
10
|
+
gem "rspec"
|
11
|
+
gem "rspec_junit_formatter"
|
12
|
+
gem "rubocop"
|
13
|
+
gem "rubocop-basic"
|
14
|
+
gem "rubocop-rake"
|
15
|
+
gem "rubocop-rspec"
|
16
|
+
gem "simplecov"
|
17
|
+
gem "webmock"
|
18
|
+
gem "yard"
|
data/lib/omniai/mistral/chat.rb
CHANGED
@@ -13,23 +13,25 @@ module OmniAI
|
|
13
13
|
# chat.completion([{ role: 'system', content: 'Tell me a joke.' }])
|
14
14
|
class Chat < OmniAI::Chat
|
15
15
|
module Model
|
16
|
-
|
17
|
-
|
18
|
-
|
19
|
-
|
20
|
-
|
21
|
-
|
22
|
-
|
23
|
-
|
24
|
-
|
25
|
-
|
26
|
-
|
27
|
-
|
28
|
-
|
29
|
-
|
30
|
-
|
31
|
-
|
32
|
-
|
16
|
+
MISTRAL_SMALL_2402 = "mistral-small-2402" # LEGACY
|
17
|
+
MISTRAL_SMALL_2409 = "mistral-small-2409" # LEGACY
|
18
|
+
MISTRAL_SMALL_2501 = "mistral-small-2501"
|
19
|
+
MISTRAL_SMALL_LATEST = "mistral-small-latest"
|
20
|
+
MISTRAL_MEDIUM_2312 = "mistral-medium-2312" # LEGACY
|
21
|
+
MISTRAL_MEDIUM_LATEST = "mistral-medium-latest" # LEGACY
|
22
|
+
MISTRAL_LARGE_2411 = "mistral-large-2411"
|
23
|
+
MISTRAL_LARGE_LATEST = "mistral-large-latest"
|
24
|
+
CODESTRAL_2405 = "codestral-2405"
|
25
|
+
CODESTRAL_LATEST = "codestral-latest"
|
26
|
+
MINISTRAL_3B_2410 = "ministral-3b-2410"
|
27
|
+
MINISTRAL_3B_LATEST = "ministral-3b-latest"
|
28
|
+
MINISTRAL_8B_2410 = "ministral-8b-2410"
|
29
|
+
MINISTRAL_8B_LATEST = "ministral-8b-latest"
|
30
|
+
MISTRAL_MODERATION_2411 = "mistral-moderation-2411"
|
31
|
+
MISTRAL_MODERATION_LATEST = "mistral-moderation-latest"
|
32
|
+
MISTRAL_EMBED = "mistral-embed"
|
33
|
+
PIXTRAL_LARGE_2411 = "pixtral-large-2411"
|
34
|
+
PIXTRAL_LARGE_LATEST = "pixtral-large-latest"
|
33
35
|
SMALL = MISTRAL_SMALL_LATEST
|
34
36
|
MEDIUM = MISTRAL_MEDIUM_LATEST # LEGACY
|
35
37
|
LARGE = MISTRAL_LARGE_LATEST
|
@@ -40,14 +42,14 @@ module OmniAI
|
|
40
42
|
DEFAULT_MODEL = Model::PIXTRAL_LARGE_LATEST
|
41
43
|
|
42
44
|
module Role
|
43
|
-
ASSISTANT =
|
44
|
-
USER =
|
45
|
-
SYSTEM =
|
45
|
+
ASSISTANT = "assistant"
|
46
|
+
USER = "user"
|
47
|
+
SYSTEM = "system"
|
46
48
|
end
|
47
49
|
|
48
|
-
JSON_RESPONSE_FORMAT = { type:
|
50
|
+
JSON_RESPONSE_FORMAT = { type: "json_object" }.freeze
|
49
51
|
|
50
|
-
|
52
|
+
protected
|
51
53
|
|
52
54
|
# @return [Hash]
|
53
55
|
def payload
|
@@ -20,7 +20,7 @@ module OmniAI
|
|
20
20
|
#
|
21
21
|
# client = OmniAI::Mistral::Client.new
|
22
22
|
class Client < OmniAI::Client
|
23
|
-
VERSION =
|
23
|
+
VERSION = "v1"
|
24
24
|
|
25
25
|
# @param api_key [String] optional - defaults to `OmniAI::Mistral.config.api_key`
|
26
26
|
# @param host [String] optional - defaults to `OmniAI::Mistral.config.host`
|
@@ -4,15 +4,15 @@ module OmniAI
|
|
4
4
|
module Mistral
|
5
5
|
# Configuration for Mistral.
|
6
6
|
class Config < OmniAI::Config
|
7
|
-
DEFAULT_HOST =
|
7
|
+
DEFAULT_HOST = "https://api.mistral.ai"
|
8
8
|
|
9
9
|
# @param api_key [String, nil] optional - defaults to `ENV['MISTRAL_API_KEY']`
|
10
10
|
# @param host [String, nil] optional - defaults to `ENV['MISTRAL_HOST'] w/ fallback to `DEFAULT_HOST`
|
11
11
|
# @param logger [Logger, nil] optional - defaults to
|
12
12
|
# @param timeout [Integer, Hash, nil] optional
|
13
13
|
def initialize(
|
14
|
-
api_key: ENV.fetch(
|
15
|
-
host: ENV.fetch(
|
14
|
+
api_key: ENV.fetch("MISTRAL_API_KEY", nil),
|
15
|
+
host: ENV.fetch("MISTRAL_HOST", DEFAULT_HOST),
|
16
16
|
logger: nil,
|
17
17
|
timeout: nil
|
18
18
|
)
|
data/lib/omniai/mistral/embed.rb
CHANGED
@@ -11,12 +11,12 @@ module OmniAI
|
|
11
11
|
# response.embedding [0.0, ...]
|
12
12
|
class Embed < OmniAI::Embed
|
13
13
|
module Model
|
14
|
-
EMBED =
|
14
|
+
EMBED = "mistral-embed"
|
15
15
|
end
|
16
16
|
|
17
17
|
DEFAULT_MODEL = Model::EMBED
|
18
18
|
|
19
|
-
|
19
|
+
protected
|
20
20
|
|
21
21
|
# @return [Hash]
|
22
22
|
def payload
|
data/lib/omniai/mistral.rb
CHANGED
metadata
CHANGED
@@ -1,14 +1,13 @@
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
2
2
|
name: omniai-mistral
|
3
3
|
version: !ruby/object:Gem::Version
|
4
|
-
version: 1.9.
|
4
|
+
version: 1.9.6
|
5
5
|
platform: ruby
|
6
6
|
authors:
|
7
7
|
- Kevin Sylvestre
|
8
|
-
autorequire:
|
9
8
|
bindir: exe
|
10
9
|
cert_chain: []
|
11
|
-
date:
|
10
|
+
date: 2025-02-27 00:00:00.000000000 Z
|
12
11
|
dependencies:
|
13
12
|
- !ruby/object:Gem::Dependency
|
14
13
|
name: event_stream_parser
|
@@ -28,16 +27,16 @@ dependencies:
|
|
28
27
|
name: omniai
|
29
28
|
requirement: !ruby/object:Gem::Requirement
|
30
29
|
requirements:
|
31
|
-
- - "
|
30
|
+
- - "~>"
|
32
31
|
- !ruby/object:Gem::Version
|
33
|
-
version: '
|
32
|
+
version: '1.9'
|
34
33
|
type: :runtime
|
35
34
|
prerelease: false
|
36
35
|
version_requirements: !ruby/object:Gem::Requirement
|
37
36
|
requirements:
|
38
|
-
- - "
|
37
|
+
- - "~>"
|
39
38
|
- !ruby/object:Gem::Version
|
40
|
-
version: '
|
39
|
+
version: '1.9'
|
41
40
|
- !ruby/object:Gem::Dependency
|
42
41
|
name: zeitwerk
|
43
42
|
requirement: !ruby/object:Gem::Requirement
|
@@ -74,7 +73,6 @@ metadata:
|
|
74
73
|
homepage_uri: https://github.com/ksylvest/omniai-mistral
|
75
74
|
changelog_uri: https://github.com/ksylvest/omniai-mistral/releases
|
76
75
|
rubygems_mfa_required: 'true'
|
77
|
-
post_install_message:
|
78
76
|
rdoc_options: []
|
79
77
|
require_paths:
|
80
78
|
- lib
|
@@ -89,8 +87,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
|
|
89
87
|
- !ruby/object:Gem::Version
|
90
88
|
version: '0'
|
91
89
|
requirements: []
|
92
|
-
rubygems_version: 3.
|
93
|
-
signing_key:
|
90
|
+
rubygems_version: 3.6.3
|
94
91
|
specification_version: 4
|
95
92
|
summary: A generalized framework for interacting with Mistral
|
96
93
|
test_files: []
|