n2b 0.2.3 → 0.2.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4d669156f98a861c284f95d3f56f9fdad92b5820d288e8f3227e885ed5d13b2a
-  data.tar.gz: e20b7578ff3ca46d9062137a930656e7598c4b613329da1984df09b80adffceb
+  metadata.gz: bf304be008b27bf4336085ec39f75fc8ac5207c6e17d442d5129a366795629eb
+  data.tar.gz: f436a7dc4a59f2840b023d7e406aa5fa2aa02dee7ebb10f48c982efaaf489546
 SHA512:
-  metadata.gz: ee2d6aaceaea02c667e03f5ff30e6f08c88e27cd9fa3b3fba97f9cd90b1410684d253bf729993a769ef1b3151839cde6f08f7fa99636001027c8e2cfc1713b59
-  data.tar.gz: f10b4515f9191e9f8adb4cacbc6096384f9b6be1697b94f2304130ede32a4360d81f90b9808c4462b8024aeff1982cd54842be64d22085938dbe7cd0b81bd5ab
+  metadata.gz: 770f49a5df340f3f9eae3f9072eafcbd9e8de6e2a5a227da3a96c67ebb22ca998411d1098921a375a9203fe446593ac8b15cfc3b6c6e6d13503a48957fed29c9
+  data.tar.gz: 88720e0bae3adf499830681214c142ea96787f3fbae01fcfe276593ce3860f65127783293cf2c1c993725747ad54165e260473f84d172fb290490165cd587708

data/README.md

@@ -56,16 +56,29 @@ n2rscrum "Create a user authentication system"
 
 ## Configuration
 
-Create a config file at `~/.n2b/config.yml` with your API keys:
+Create a config file at `~/.n2b/config.yml` with your API keys. You can also use a custom config file by setting the `N2B_CONFIG_FILE` environment variable:
 
+```bash
+export N2B_CONFIG_FILE=/path/to/your/config.yml
+```
+
+Example config file:
 ```yaml
-llm: claude  # or openai
+llm: claude  # or openai, gemini
 claude:
   key: your-anthropic-api-key
   model: claude-3-opus-20240229 # or opus, haiku, sonnet
 openai:
   key: your-openai-api-key
   model: gpt-4 # or gpt-3.5-turbo
+gemini:
+  key: your-google-api-key
+  model: gemini-flash # uses gemini-2.0-flash model
+```
+
+You can also set the history file location using the `N2B_HISTORY_FILE` environment variable:
+```bash
+export N2B_HISTORY_FILE=/path/to/your/history
 ```
 
 ## Quick Example N2B

data/lib/n2b/base.rb

@@ -1,8 +1,8 @@
 module N2B
   class Base
 
-    CONFIG_FILE = File.expand_path('~/.n2b/config.yml')
-    HISTORY_FILE = File.expand_path('~/.n2b/history')
+    CONFIG_FILE = ENV['N2B_CONFIG_FILE'] || File.expand_path('~/.n2b/config.yml')
+    HISTORY_FILE = ENV['N2B_HISTORY_FILE'] || File.expand_path('~/.n2b/history')
 
     def load_config
       if File.exist?(CONFIG_FILE)
@@ -18,16 +18,24 @@ module N2B
       model = config['model'] || 'sonnet35'
   
       if api_key.nil? || api_key == '' ||  reconfigure
-        print "choose a language model to use (1:claude, 2:openai) #{ config['llm'] }: "
+        print "choose a language model to use (1:claude, 2:openai, 3:gemini) #{ config['llm'] }: "
         llm = $stdin.gets.chomp
         llm = config['llm'] if llm.empty?
-        unless ['claude', 'openai','1','2'].include?(llm)
-          puts "Invalid language model. Choose from: claude, openai"
+        unless ['claude', 'openai', 'gemini', '1', '2', '3'].include?(llm)
+          puts "Invalid language model. Choose from: claude, openai, gemini"
           exit 1
         end
         llm = 'claude' if llm == '1'
         llm = 'openai' if llm == '2'
-        llm_class = llm == 'openai' ? N2M::Llm::OpenAi : N2M::Llm::Claude
+        llm = 'gemini' if llm == '3'
+        llm_class = case llm
+                   when 'openai'
+                     N2M::Llm::OpenAi
+                   when 'gemini'
+                     N2M::Llm::Gemini
+                   else
+                     N2M::Llm::Claude
+                   end
 
         print "Enter your #{llm} API key: #{ api_key.nil? || api_key.empty? ? '' : '(leave blank to keep the current key '+api_key[0..10]+'...)' }"
         api_key = $stdin.gets.chomp 
@@ -50,11 +58,8 @@ module N2B
         config['append_to_shell_history'] = false 
         puts "Current configuration: #{config['privacy']}"
         FileUtils.mkdir_p(File.dirname(CONFIG_FILE)) unless File.exist?(File.dirname(CONFIG_FILE))
-        File.open(CONFIG_FILE, 'w+') do |f|
-          f.write(config.to_yaml   )
-        end
+        File.write(CONFIG_FILE, config.to_yaml)
       end
-      
       config
     end
   end

data/lib/n2b/irb.rb

@@ -17,7 +17,14 @@ module N2B
 
     def n2r(input_string='', files: [], exception: nil, log: false)
       config = N2B::Base.new.get_config
-      llm = config['llm'] == 'openai' ? N2M::Llm::OpenAi.new(config) : N2M::Llm::Claude.new(config)
+      llm = case config['llm']
+            when 'openai'
+              N2M::Llm::OpenAi.new(config)
+            when 'gemini'
+              N2M::Llm::Gemini.new(config)
+            else
+              N2M::Llm::Claude.new(config)
+            end
       # detect if inside rails console
       console = case 
       when defined?(Rails) && Rails.respond_to?(:application)
@@ -586,7 +593,14 @@ module N2B
     
     def analyze_error(error_info, related_files)
       config = N2B::Base.new.get_config
-      llm = config['llm'] == 'openai' ? N2M::Llm::OpenAi.new(config) : N2M::Llm::Claude.new(config)
+      llm = case config['llm']
+            when 'openai'
+              N2M::Llm::OpenAi.new(config)
+            when 'gemini'
+              N2M::Llm::Gemini.new(config)
+            else
+              N2M::Llm::Claude.new(config)
+            end
       
       # Build file content section, showing context if available
       file_content_section = related_files.map do |file_path, content|
@@ -651,7 +665,14 @@ module N2B
     
     def generate_error_ticket(error_info, related_files = {}, url = nil)
       config = N2B::Base.new.get_config
-      llm = config['llm'] == 'openai' ? N2M::Llm::OpenAi.new(config) : N2M::Llm::Claude.new(config)
+      llm = case config['llm']
+            when 'openai'
+              N2M::Llm::OpenAi.new(config)
+            when 'gemini'
+              N2M::Llm::Gemini.new(config)
+            else
+              N2M::Llm::Claude.new(config)
+            end
       
       # Build file content section, showing context if available
       file_content_section = related_files.map do |file_path, content|

data/lib/n2b/llm/claude.rb

@@ -2,7 +2,7 @@ module N2M
   module Llm
     class Claude
       API_URI = URI.parse('https://api.anthropic.com/v1/messages')
-      MODELS = { 'haiku' =>  'claude-3-haiku-20240307', 'sonnet' => 'claude-3-sonnet-20240229', 'sonnet35' => 'claude-3-5-sonnet-20240620' }
+      MODELS = { 'haiku' =>  'claude-3-haiku-20240307', 'sonnet' => 'claude-3-sonnet-20240229', 'sonnet35' => 'claude-3-5-sonnet-20240620', "sonnet37" => "claude-3-7-sonnet-20250219" }
 
       def initialize(config)
         @config = config

data/lib/n2b/llm/gemini.rb

@@ -0,0 +1,69 @@
+require 'net/http'
+require 'json'
+require 'uri'
+
+module N2M
+  module Llm
+    class Gemini
+      API_URI = URI.parse('https://generativelanguage.googleapis.com/v1beta/models')
+      MODELS = { 
+        'gemini-flash' => 'gemini-2.0-flash'
+      }
+
+      def initialize(config)
+        @config = config
+      end
+
+      def make_request(content)
+        model = MODELS[@config['model']] || 'gemini-flash'
+        uri = URI.parse("#{API_URI}/#{model}:generateContent?key=#{@config['access_key']}")
+        
+        request = Net::HTTP::Post.new(uri)
+        request.content_type = 'application/json'
+
+        request.body = JSON.dump({
+          "contents" => [{
+            "parts" => [{
+              "text" => content
+            }]
+          }]
+        })
+
+        response = Net::HTTP.start(uri.hostname, uri.port, use_ssl: true) do |http|
+          http.request(request)
+        end
+
+        # check for errors
+        if response.code != '200'
+          puts "Error: #{response.code} #{response.message}"
+          puts response.body
+          puts "Config: #{@config.inspect}"
+          exit 1
+        end
+
+        parsed_response = JSON.parse(response.body)
+        answer = parsed_response['candidates'].first['content']['parts'].first['text']
+        
+        begin
+          # Try to parse as JSON if it looks like JSON
+          if answer.strip.start_with?('{') && answer.strip.end_with?('}')
+            answer = JSON.parse(answer)
+          else
+            # If not JSON, wrap it in our expected format
+            answer = {
+              'explanation' => answer,
+              'code' => nil
+            }
+          end
+        rescue JSON::ParserError
+          # If JSON parsing fails, wrap the text in our expected format
+          answer = {
+            'explanation' => answer,
+            'code' => nil
+          }
+        end
+        answer
+      end
+    end
+  end
+end 

data/lib/n2b/version.rb

@@ -1,4 +1,4 @@
 # lib/n2b/version.rb
 module N2B
-  VERSION = "0.2.3"
+  VERSION = "0.2.4"
 end

data/lib/n2b.rb

@@ -8,6 +8,7 @@ require 'fileutils'
 require 'n2b/version'
 require 'n2b/llm/claude'
 require 'n2b/llm/open_ai'
+require 'n2b/llm/gemini'
 require 'n2b/base'
 require 'n2b/cli'
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: n2b
 version: !ruby/object:Gem::Version
-  version: 0.2.3
+  version: 0.2.4
 platform: ruby
 authors:
 - Stefan Nothegger
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-03-20 00:00:00.000000000 Z
+date: 2025-03-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: json
@@ -69,6 +69,7 @@ files:
 - lib/n2b/cli.rb
 - lib/n2b/irb.rb
 - lib/n2b/llm/claude.rb
+- lib/n2b/llm/gemini.rb
 - lib/n2b/llm/open_ai.rb
 - lib/n2b/version.rb
 homepage: https://github.com/stefan-kp/n2b