cloudflare-ai 0.9.0 → 0.9.1

Sign up to get free protection for your applications and to get access to all the features.
checksums.yaml CHANGED
@@ -1,7 +1,7 @@
1
1
  ---
2
2
  SHA256:
3
- metadata.gz: '08feb777620d0767e6ab08f5366c9a207facc25783f9fe497f381e894e02876c'
4
- data.tar.gz: cacc6945ed2914fa62280e5afd8a0695f7c108139737e3785cbc74a24d26b79b
3
+ metadata.gz: c727f28c32fdfa8aace3c440ccd7c4f99e0edf9c60a35605dd386dfb1e83f3d9
4
+ data.tar.gz: 2a405251e064c93fc9de0bb4f27da57e0bdb342400cfc867ae4a92384494af4b
5
5
  SHA512:
6
- metadata.gz: 31b20441f2791a3c93a8ccde7d5674e2daaf8190cd348b2c941849b84edc6f04fcef9bfd2a3ef6de8e4c39c3a9d2f03e3388a0f8efd6fc6391ca83fafee111d5
7
- data.tar.gz: b695c2971236bb8aa182fe8c36bd63fd9c7303dfda70aee767a098532fbb274e720ed165e73515800b3b64a6a77d5e124b3048dcc2100bff7e039b0245577bbf
6
+ metadata.gz: f10b2c9ee7dbdfef2ac61c8a425bfb6d1f06ccddab9296c86cfcf097f243e499b5498dbcd0163ac03e31c5ae845f9ab54bcd506f73e2533ce172eb360336dfd4
7
+ data.tar.gz: a01a5743d7d845813582317b50d9bc131c981c884191eff0890286d2a85881801f9eda768ec6246693192aa086a5fd771580e2fcab4865f881e054bc810cdcaf
@@ -12,6 +12,13 @@ class Cloudflare::AI::Client
12
12
  @api_token = api_token
13
13
  end
14
14
 
15
+ def caption(image: nil, model_name: Cloudflare::AI::Models.image_to_text.first)
16
+ url = service_url_for(account_id: account_id, model_name: model_name)
17
+
18
+ image = File.open(image) if image.is_a?(String)
19
+ Cloudflare::AI::Results::ImageToText.new(post_request_with_binary_file(url, image).body)
20
+ end
21
+
15
22
  def chat(messages:, model_name: default_text_generation_model_name, max_tokens: default_max_tokens, &block)
16
23
  url = service_url_for(account_id: account_id, model_name: model_name)
17
24
  stream = block ? true : false
@@ -41,6 +48,13 @@ class Cloudflare::AI::Client
41
48
  post_streamable_request(url, payload, &block)
42
49
  end
43
50
 
51
+ def detect_objects(image: nil, model_name: Cloudflare::AI::Models.object_detection.first)
52
+ url = service_url_for(account_id: account_id, model_name: model_name)
53
+
54
+ image = File.open(image) if image.is_a?(String)
55
+ Cloudflare::AI::Results::ObjectDetection.new(post_request_with_binary_file(url, image).body)
56
+ end
57
+
44
58
  def draw(prompt:, num_steps: 20, model_name: Cloudflare::AI::Models.text_to_image.first)
45
59
  url = service_url_for(account_id: account_id, model_name: model_name)
46
60
  payload = {prompt: prompt, num_steps: num_steps}.to_json
@@ -59,6 +73,13 @@ class Cloudflare::AI::Client
59
73
  Cloudflare::AI::Results::TextEmbedding.new(connection.post(url, payload).body)
60
74
  end
61
75
 
76
+ def summarize(text:, model_name: Cloudflare::AI::Models.summarization.first, max_tokens: 1024)
77
+ url = service_url_for(account_id: account_id, model_name: model_name)
78
+ payload = {input_text: text, max_tokens: max_tokens}.to_json
79
+
80
+ Cloudflare::AI::Results::Summarization.new(connection.post(url, payload).body)
81
+ end
82
+
62
83
  def transcribe(source_url: nil, audio: nil, model_name: Cloudflare::AI::Models.automatic_speech_recognition.first)
63
84
  raise ArgumentError, "Must provide either audio_url or audio" if [source_url, audio].compact.size != 1
64
85
 
@@ -1,43 +1,90 @@
1
1
  class Cloudflare::AI::Models
2
2
  class << self
3
- def text_generation
4
- %w[@hf/thebloke/llamaguard-7b-awq @hf/thebloke/neural-chat-7b-v3-1-awq @cf/meta/llama-2-7b-chat-fp16 @cf/mistral/mistral-7b-instruct-v0.1 @hf/thebloke/codellama-7b-instruct-awq @cf/meta/llama-2-7b-chat-int8 @hf/thebloke/mistral-7b-instruct-v0.1-awq @hf/thebloke/deepseek-coder-6.7b-base-awq @hf/thebloke/openhermes-2.5-mistral-7b-awq @hf/thebloke/deepseek-coder-6.7b-instruct-awq @hf/thebloke/zephyr-7b-beta-awq]
3
+ def all
4
+ {
5
+ automatic_speech_recognition: automatic_speech_recognition,
6
+ image_classification: image_classification,
7
+ image_to_text: image_to_text,
8
+ object_detection: object_detection,
9
+ summarization: summarization,
10
+ text_classification: text_classification,
11
+ text_embeddings: text_embedding,
12
+ text_generation: text_generation,
13
+ text_to_image: text_to_image,
14
+ translation: translation
15
+ }
5
16
  end
6
17
 
7
18
  def automatic_speech_recognition
8
19
  %w[@cf/openai/whisper]
9
20
  end
10
21
 
11
- def translation
12
- %w[@cf/meta/m2m100-1.2b]
22
+ def image_classification
23
+ %w[@cf/microsoft/resnet-50]
13
24
  end
14
25
 
15
- def text_classification
16
- %w[@cf/huggingface/distilbert-sst-2-int8]
26
+ def image_to_text
27
+ %w[@cf/unum/uform-gen2-qwen-500m]
17
28
  end
18
29
 
19
- def image_classification
20
- %w[@cf/microsoft/resnet-50]
30
+ def object_detection
31
+ %w[@cf/meta/det-resnet-50]
21
32
  end
22
33
 
23
- def text_to_image
24
- %w[@cf/stabilityai/stable-diffusion-xl-base-1.0]
34
+ def summarization
35
+ %w[@cf/facebook/bart-large-cnn]
36
+ end
37
+
38
+ def text_classification
39
+ %w[@cf/huggingface/distilbert-sst-2-int8]
25
40
  end
26
41
 
27
42
  def text_embedding
28
43
  %w[@cf/baai/bge-base-en-v1.5 @cf/baai/bge-large-en-v1.5 @cf/baai/bge-small-en-v1.5]
29
44
  end
30
45
 
31
- def all
32
- {
33
- text_generation: text_generation,
34
- automatic_speech_recognition: automatic_speech_recognition,
35
- translation: translation,
36
- text_classification: text_classification,
37
- image_classification: image_classification,
38
- text_to_image: text_to_image,
39
- text_embeddings: text_embedding
40
- }
46
+ def text_generation
47
+ %w[
48
+ @hf/thebloke/codellama-7b-instruct-awq
49
+ @hf/thebloke/deepseek-coder-6.7b-base-awq
50
+ @hf/thebloke/deepseek-coder-6.7b-instruct-awq
51
+ @cf/deepseek-ai/deepseek-math-7b-base
52
+ @cf/deepseek-ai/deepseek-math-7b-instruct
53
+ @cf/thebloke/discolm-german-7b-v1-awq
54
+ @cf/tiiuae/falcon-7b-instruct
55
+ @hf/thebloke/llama-2-13b-chat-awq
56
+ @cf/meta/llama-2-7b-chat-fp16
57
+ @cf/meta/llama-2-7b-chat-int8
58
+ @hf/thebloke/llamaguard-7b-awq
59
+ @cf/mistral/mistral-7b-instruct-v0.1
60
+ @hf/thebloke/mistral-7b-instruct-v0.1-awq
61
+ @hf/thebloke/neural-chat-7b-v3-1-awq
62
+ @hf/thebloke/openchat_3.5-awq
63
+ @cf/openchat/openchat-3.5-0106
64
+ @hf/thebloke/openhermes-2.5-mistral-7b-awq
65
+ @cf/microsoft/phi-2
66
+ @cf/qwen/qwen1.5-0.5b-chat
67
+ @cf/qwen/qwen1.5-1.8b-chat
68
+ @cf/qwen/qwen1.5-14b-chat-awq
69
+ @cf/qwen/qwen1.5-7b-chat-awq
70
+ @cf/defog/sqlcoder-7b-2
71
+ @cf/tinyllama/tinyllama-1.1b-chat-v1.0
72
+ @hf/thebloke/zephyr-7b-beta-awq
73
+ ]
74
+ end
75
+
76
+ def text_to_image
77
+ %w[
78
+ @cf/lykon/dreamshaper-8-lcm
79
+ @cf/runwayml/stable-diffusion-v1-5-img2img
80
+ @cf/runwayml/stable-diffusion-v1-5-inpainting
81
+ @cf/stabilityai/stable-diffusion-xl-base-1.0
82
+ @cf/bytedance/stable-diffusion-xl-lightning
83
+ ]
84
+ end
85
+
86
+ def translation
87
+ %w[@cf/meta/m2m100-1.2b]
41
88
  end
42
89
  end
43
90
  end
@@ -0,0 +1,3 @@
1
+ class Cloudflare::AI::Results::ImageToText < Cloudflare::AI::Result
2
+ # Empty seam kept for consistency with other result objects that have more complexity.
3
+ end
@@ -0,0 +1,3 @@
1
+ class Cloudflare::AI::Results::ObjectDetection < Cloudflare::AI::Result
2
+ # Empty seam kept for consistency with other result objects that have more complexity.
3
+ end
@@ -0,0 +1,5 @@
1
+ class Cloudflare::AI::Results::Summarization < Cloudflare::AI::Result
2
+ def summary
3
+ result&.dig(:summary) # nil if no response
4
+ end
5
+ end
@@ -2,6 +2,6 @@
2
2
 
3
3
  module Cloudflare
4
4
  module AI
5
- VERSION = "0.9.0"
5
+ VERSION = "0.9.1"
6
6
  end
7
7
  end
metadata CHANGED
@@ -1,14 +1,14 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: cloudflare-ai
3
3
  version: !ruby/object:Gem::Version
4
- version: 0.9.0
4
+ version: 0.9.1
5
5
  platform: ruby
6
6
  authors:
7
7
  - Ajay Krishnan
8
8
  autorequire:
9
9
  bindir: exe
10
10
  cert_chain: []
11
- date: 2024-02-07 00:00:00.000000000 Z
11
+ date: 2024-02-28 00:00:00.000000000 Z
12
12
  dependencies:
13
13
  - !ruby/object:Gem::Dependency
14
14
  name: activemodel
@@ -115,6 +115,9 @@ files:
115
115
  - lib/cloudflare/ai/result.rb
116
116
  - lib/cloudflare/ai/results/automatic_speech_recognition.rb
117
117
  - lib/cloudflare/ai/results/image_classification.rb
118
+ - lib/cloudflare/ai/results/image_to_text.rb
119
+ - lib/cloudflare/ai/results/object_detection.rb
120
+ - lib/cloudflare/ai/results/summarization.rb
118
121
  - lib/cloudflare/ai/results/text_classification.rb
119
122
  - lib/cloudflare/ai/results/text_embedding.rb
120
123
  - lib/cloudflare/ai/results/text_generation.rb