rails_ai 0.1.6 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 53d48e7589680a3868fd3f1941a782067cf8a223cd22180776bed4651ef64743
-  data.tar.gz: 6367732d15ece684b5b80791842e84aae8605c3d285e6984682671f42bcf332b
+  metadata.gz: 909d277e3d7288cb423211be454525d2ba2137aa17eeb437ee1c7df4f1f68d51
+  data.tar.gz: 7c7b3597e7cae56b30d6b9f24206a8c0658d36a12021cbf231ffac8cc559c588
 SHA512:
-  metadata.gz: cc442b24a76848c614942215d623e74aa1acdf82049c5e100ed2de430a91c79c4539e2be78dd5a26c2691d8e67f9e35629dd5dcee80c7feeb97ba91c6d85fbf5
-  data.tar.gz: 7d09b4883eb116e951868c0cb88a50bdc4a745112a9769d24003828f497391473f07e531f3accd102b7fa2059191c63df4a903d82325b6d3a31d65487a5f6d48
+  metadata.gz: 8fb52882e68fbccb0d87a6ab9539c76d63de7e95476bb1eb255637da2450069fdc12d8e949370b0fc2992b1853a01b11413c9df59a855f9818631d591b0666d8
+  data.tar.gz: 2c2767b19fdd5979fce2f1f5c594929ae02dad406658a189095b2dc62c48acb5fbb692a5ea1c56acd3dc2a1ccbf8ad40c8323b0ced1a131b5ad13bdc4141dc17

data/lib/rails_ai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RailsAi
-  VERSION = "0.1.6"
+  VERSION = "0.1.8"
 end

data/lib/rails_ai/web_search.rb

@@ -0,0 +1,115 @@
+# frozen_string_literal: true
+
+require 'net/http'
+require 'json'
+require 'uri'
+
+module RailsAi
+  module WebSearch
+    class SearchError < StandardError; end
+    
+    class GoogleSearch
+      def initialize(api_key: nil, search_engine_id: nil)
+        @api_key = api_key || ENV['GOOGLE_SEARCH_API_KEY']
+        @search_engine_id = search_engine_id || ENV['GOOGLE_SEARCH_ENGINE_ID']
+        @base_url = 'https://www.googleapis.com/customsearch/v1'
+      end
+      
+      def search(query, num_results: 5)
+        raise SearchError, "Google Search API key not configured" unless @api_key
+        raise SearchError, "Google Search Engine ID not configured" unless @search_engine_id
+        
+        uri = URI(@base_url)
+        params = {
+          key: @api_key,
+          cx: @search_engine_id,
+          q: query,
+          num: num_results
+        }
+        uri.query = URI.encode_www_form(params)
+        
+        response = Net::HTTP.get_response(uri)
+        raise SearchError, "Search failed: #{response.code}" unless response.code == '200'
+        
+        data = JSON.parse(response.body)
+        format_results(data)
+      rescue => e
+        raise SearchError, "Web search error: #{e.message}"
+      end
+      
+      private
+      
+      def format_results(data)
+        results = data['items'] || []
+        results.map do |item|
+          {
+            title: item['title'],
+            link: item['link'],
+            snippet: item['snippet']
+          }
+        end
+      end
+    end
+    
+    class DuckDuckGoSearch
+      def search(query, num_results: 5)
+        # Simple web scraping approach (for demo purposes)
+        # In production, you'd want to use a proper API
+        uri = URI("https://html.duckduckgo.com/html/?q=#{URI.encode_www_form_component(query)}")
+        
+        response = Net::HTTP.get_response(uri)
+        raise SearchError, "Search failed: #{response.code}" unless response.code == '200'
+        
+        # Parse HTML and extract results (simplified)
+        parse_duckduckgo_results(response.body, num_results)
+      rescue => e
+        raise SearchError, "Web search error: #{e.message}"
+      end
+      
+      private
+      
+      def parse_duckduckgo_results(html, num_results)
+        # This is a simplified parser - in production you'd use Nokogiri
+        results = []
+        lines = html.split("\n")
+        
+        lines.each_with_index do |line, index|
+          if line.include?('class="result__title"') && results.length < num_results
+            title_line = lines[index + 1] rescue ""
+            snippet_line = lines[index + 3] rescue ""
+            
+            results << {
+              title: title_line.strip,
+              link: "https://duckduckgo.com",
+              snippet: snippet_line.strip
+            }
+          end
+        end
+        
+        results
+      end
+    end
+    
+    class WebSearch
+      def initialize(provider: :google, **options)
+        @provider = case provider
+                   when :google
+                     GoogleSearch.new(**options)
+                   when :duckduckgo
+                     DuckDuckGoSearch.new(**options)
+                   else
+                     raise SearchError, "Unsupported search provider: #{provider}"
+                   end
+      end
+      
+      def search(query, **options)
+        @provider.search(query, **options)
+      end
+    end
+    
+    # Convenience method
+    def self.search(query, provider: :google, **options)
+      WebSearch.new(provider: provider, **options).search(query)
+    end
+  end
+end

data/lib/rails_ai.rb

@@ -500,3 +500,35 @@ end
   def self.handle_security_error(error, context = {})
     Security::ErrorHandler.handle_security_error(error, context)
   end
+require_relative "rails_ai/web_search"
+
+  # Web-enhanced chat with real-time information
+  def self.chat_with_web_search(prompt, model: config.default_model, **opts)
+    # Check if the prompt needs web search
+    web_keywords = ['current', 'latest', 'today', 'now', 'recent', 'weather', 'news', 'stock', 'price']
+    needs_web_search = web_keywords.any? { |keyword| prompt.downcase.include?(keyword) }
+    
+    if needs_web_search
+      begin
+        # Perform web search
+        search_results = WebSearch.search(prompt, num_results: 3)
+        
+        # Enhance the prompt with web results
+        web_context = "\n\nRecent web search results:\n"
+        search_results.each_with_index do |result, index|
+          web_context += "#{index + 1}. #{result[:title]}\n   #{result[:snippet]}\n   Source: #{result[:link]}\n\n"
+        end
+        
+        enhanced_prompt = "#{prompt}\n\nPlease use the following web search results to provide current, up-to-date information:#{web_context}"
+        
+        # Get AI response with web context
+        chat(enhanced_prompt, model: model, **opts)
+      rescue WebSearch::SearchError => e
+        # Fallback to regular chat if web search fails
+        chat(prompt, model: model, **opts)
+      end
+    else
+      # Regular chat for non-time-sensitive queries
+      chat(prompt, model: model, **opts)
+    end
+  end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rails_ai
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.8
 platform: ruby
 authors:
 - Daniel Amah
@@ -339,6 +339,7 @@ files:
 - lib/rails_ai/security/secure_file_handler.rb
 - lib/rails_ai/security/secure_http_client.rb
 - lib/rails_ai/version.rb
+- lib/rails_ai/web_search.rb
 - lib/rails_ai/window_context.rb
 - monitoring/ci_setup_guide.md
 - monitoring/enhanced_monitoring_script.rb