using gen-ai gem

BlazingRockStorm · Oct 18, 2024 · e50f0e8 · e50f0e8
1 parent e65f403
commit e50f0e8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/Gemfile b/Gemfile
@@ -2,5 +2,5 @@
 
 source 'https://rubygems.org'
 
-gem 'gemini-ai', '~> 4.2.0'
+gem 'gen-ai', '~> 0.4.2'
 gem 'dotenv'
diff --git a/lib/main.rb b/lib/main.rb
@@ -1,7 +1,6 @@
 # frozen_string_literal: true
 
-require 'dotenv/load'
-require 'gemini-ai'
+require 'gen_ai'
 
 # With an API key
 client = Gemini.new(
@@ -49,3 +48,5 @@
               .join
 
 puts mail_template
+model = GenAI::Language.new(:gemini, ENV['GOOGLE_GEMINI_TOKEN'])
+model.chat("日報例文を書いてください")