appchat 0.0.5 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3071421a8c3c49e11f5c5186b9f17a00bd37d45229d0d95a3b466ae9b9032fee
-  data.tar.gz: 79284185b56a3efaeb60e1ec1ad144ada532fd499e686b4604db7a29aabccd16
+  metadata.gz: 4aca406ace060a7c7b9af445e20d5be1f3605614109320792523cba09b7f92fe
+  data.tar.gz: 8c1e290b46dfca74c66c94748d305386d67703291e455c066aed1df0d1dc5cc8
 SHA512:
-  metadata.gz: 9e642582fe817de68a171c14b272a0b0ceb2257b96021ae9a51e630c5ef562ed982c452fdc9694c2d6e6c81378720f636e0e8218f51d1704134d383602cf2df7
-  data.tar.gz: a71996c59642154416cb3178b9406d44242b5626ab7069aeb6341a15e1062d6c297131dfd19de2455c7e59be0cbf8af8c43e6daa0879afb269c001a88723f0a3
+  metadata.gz: 5aaa00c36101e08cb562924a7a3a0cca7da5ef49680eb6f9f202a8d17ebe0cbbe852fe86ef4481cb65ef8c2811d9933731074363875930ccb916e92d9fdd66d4
+  data.tar.gz: 876249a02dbec15d9d3ee4c9dbabcfbcc843970f8d85d7b4e9fad563a97af8a86c1a3600c7383ef3dea9fc2bfffef4c19f3fef70cb7009f0fc0a12401efbb462

data/lib/generators/appchat/appchat_generator.rb

@@ -9,17 +9,24 @@ class AppchatGenerator < Rails::Generators::Base
   source_root File.expand_path('templates', __dir__)
 
   def add_gems
-    unless gem_exists?('ollama-ai')
-      append_to_file 'Gemfile', "\ngem 'ollama-ai'\n"
-    end
-
-    unless gem_exists?('tailwindcss-rails')
-      append_to_file 'Gemfile', "\ngem 'tailwindcss-rails'\n"
+    gems = %w(
+      ollama-ai
+      tailwindcss-rails
+      watir
+    )
+
+    gems.each do |gem|
+      unless gem_exists?(gem)
+        append_to_file 'Gemfile', "\ngem '#{gem}'\n"
+      end
     end
 
     Bundler.with_unbundled_env do
       run 'bundle install'
     end
+  end
+
+  def install_tailwind
     run 'rails tailwindcss:install'
   end
 
@@ -30,7 +37,9 @@ class AppchatGenerator < Rails::Generators::Base
 
   def generate_models
     generate "model", "Chat context:text"
-    generate "model", "Message chat:references content:text role:integer"
+    generate "model", "Message chat:references content:text role:integer status:string"
+    generate "model", "AppchatFunction name:string description:text class_name:string"
+    generate "model", "FunctionParameter appchat_function:references name:string example_value:string"
   end
 
   def create_controllers
@@ -60,14 +69,16 @@ class AppchatGenerator < Rails::Generators::Base
   def copy_models
     copy_file "models/message.rb", "app/models/message.rb", force: true
     copy_file "models/chat.rb", "app/models/chat.rb", force: true
+    copy_file "models/appchat_function.rb", "app/models/appchat_function.rb", force: true
   end
 
-  def serialize_context
-    inject_into_class 'app/models/chat.rb', 'Chat', "serialize :context, coder:JSON, type: Array\n"
+  def copy_services
+    copy_file "services/appchat_function_service.rb", "app/services/appchat_function_service.rb", force: true
+    copy_file "services/web_search_service.rb", "app/services/web_search_service.rb", force: true
   end
 
-  def set_associations
-    inject_into_class 'app/models/chat.rb', 'Chat', "  has_many :messages, dependent: :destroy\n"
+  def serialize_context
+    inject_into_class 'app/models/chat.rb', 'Chat', "serialize :context, coder:JSON, type: Array\n"
   end
 
   def run_migrations
@@ -88,6 +99,14 @@ class AppchatGenerator < Rails::Generators::Base
     end
   end
 
+  def copy_rake_tasks
+    copy_file "tasks/create_appchat_function.rake", "lib/tasks/create_appchat_function.rake"
+  end
+
+  def create_functions
+    rake "appchat_function:create_web_search"
+  end
+
   def show_art
     puts <<-ART
   

data/lib/generators/appchat/templates/get_ai_response_job.rb

@@ -1,27 +1,33 @@
 class GetAiResponseJob < ApplicationJob
   queue_as :default
 
-  attr_reader :chat, :user_prompt
+  attr_reader :client, :chat, :user_prompt, :informed_prompt, :message
 
   def perform(chat_id, user_prompt)
+    @client = new_client
     @chat = Chat.find(chat_id)
     @user_prompt = user_prompt
+    @message = chat.messages.create(role: 'assistant')
+    appchat_functions
     call_ollama
   end
 
   private
 
-  def call_ollama
-    client = Ollama.new(
+  def new_client
+    Ollama.new(
       credentials: { address: 'http://localhost:11434' },
       options: { server_sent_events: true }
     )
-    message = chat.messages.create(role: 'assistant')
+  end
+
+  def call_ollama
+    prompt = informed_prompt || user_prompt
 
     response = client.generate(
       {
         model: 'llama3.1',
-        prompt: user_prompt,
+        prompt: prompt,
         context: chat.context,
         stream: true,
       }
@@ -35,4 +41,20 @@ class GetAiResponseJob < ApplicationJob
       end
     chat.update(context: response.last["context"])
   end
+
+  def appchat_functions
+    appchat_function_service = AppchatFunctionService.new(chat.id, user_prompt).run
+    if appchat_function_service["match"] == "true" && appchat_function_service["appchat_function"].in?(AppchatFunction.pluck(:class_name))
+      puts "Function Match Found! --> #{appchat_function_service}"
+      function_class = appchat_function_service["appchat_function"].constantize
+      function_response = function_class.new(appchat_function_service["parameters"]).run do |status|
+        message.update(status: status)
+      end
+
+      return if function_response.nil?
+      @informed_prompt = "#{ user_prompt }, base your response on this data:
+        #{ appchat_function_service["name"] } responded with: #{ function_response },
+        :current_time => #{Date.current}"
+    end
+  end
 end

data/lib/generators/appchat/templates/messages/message.html.erb

@@ -1,5 +1,7 @@
 <div data-chat-message-target="message" id="<%= dom_id(message) %>" class="bg-gray-200 text-gray-900 shadow-lg p-4 rounded-lg">
-  <% if message.content? %>
+  <% if message.status? && !message.content? %>
+    <%= message.status %>
+  <% elsif message.content? %>
     <%=  message.content %>
   <% else %>
     <%= render 'messages/typing_bubbles' %>

data/lib/generators/appchat/templates/models/appchat_function.rb

@@ -0,0 +1,18 @@
+class AppchatFunction < ApplicationRecord
+  has_many :function_parameters, dependent: :destroy
+
+  def to_prompt_hash
+    {
+      name: name,
+      description: description,
+      class_name: class_name,
+      parameters: function_parameters.each_with_object({}) do |param, hash|
+        hash[param.name] = param.example_value
+      end
+    }
+  end
+
+  def self.all_to_prompt_json
+    all.includes(:function_parameters).map(&:to_prompt_hash).to_json
+  end
+end

data/lib/generators/appchat/templates/services/appchat_function_service.rb

@@ -0,0 +1,47 @@
+class AppchatFunctionService
+  attr_reader :chat, :user_prompt, :response_json
+
+  def initialize(chat_id, user_prompt)
+    @chat = Chat.find(chat_id)
+    @user_prompt = user_prompt
+  end
+
+  def run
+    call_ollama
+    response_json
+  end
+
+  def call_ollama
+    client = Ollama.new(
+      credentials: { address: 'http://localhost:11434' },
+      options: { server_sent_events: true }
+    )
+    response = client.generate(
+      {
+        model: 'llama3.1',
+        prompt: prompt,
+        context: chat.context,
+        "format": "json"
+      }
+    )
+    @response_json = JSON.parse(response.map { |r| r["response"] }.join)
+  end
+  def prompt
+    <<-PROMPT
+      Evaluate the following user prompt in chat context.
+      Your goal is to determine whether the user prompt is requesting or requires additional information from any of the available services.
+        - If the prompt is a general greeting, casual remark, or something that can be answered without further information, respond with JSON { 'match' => 'false' }.
+        - If the prompt asks for specific information, requires a search, or needs a service to generate the correct response, then respond with JSON like this example:
+        {
+          'match' => 'true',
+          'appchat_function' => 'WebSearchService',
+          'parameters' => {
+            'query' => 'a query based on the prompt and context'
+          }
+        }
+
+      Here is the user's prompt: #{user_prompt}
+      Here are the Available Services: #{ AppchatFunction.all_to_prompt_json }
+    PROMPT
+  end
+end

data/lib/generators/appchat/templates/services/web_search_service.rb

@@ -0,0 +1,23 @@
+require 'watir'
+require 'cgi'
+
+class WebSearchService
+  attr_reader :query
+
+  def initialize(args)
+    @query = args["query"]
+  end
+
+  def run
+    yield("Searching Google for #{@query}") if block_given?
+
+    browser = Watir::Browser.new :chrome, headless: true
+    search_url = "https://www.google.com/search?q=#{CGI.escape(query)}"
+    browser.goto(search_url)
+    response = "browser_text: #{browser.text} browser_links: #{browser.links}"
+    browser.close
+    response
+  rescue Selenium::WebDriver::Error::UnknownError => e
+    "Error: #{e.message}"
+  end
+end

data/lib/generators/appchat/templates/tasks/create_appchat_function.rake

@@ -0,0 +1,13 @@
+# lib/tasks/create_appchat_function.rake
+namespace :appchat_function do
+  desc "Create an AppchatFunction with name 'Web Search', class name 'WebSearchService', and a description"
+  task create_web_search: :environment do
+    af = AppchatFunction.create!(
+      name: "Web Search",
+      class_name: "WebSearchService",
+      description: "Searches Google with a user's query and returns the page text and links"
+    )
+    puts "AppchatFunction 'Web Search' created successfully."
+    af.function_parameters.create(name: 'query', example_value: 'Dog friendly vegan resturants in Austin, TX')
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: appchat
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors:
 - hackliteracy
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-08-29 00:00:00.000000000 Z
+date: 2024-09-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -66,7 +66,7 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 1.3.0
-description: The best and easiest framework for adding chats
+description: The best and easiest framework for adding AI chats
 email: hackliteracy@gmail.com
 executables: []
 extensions: []
@@ -86,8 +86,12 @@ files:
 - lib/generators/appchat/templates/messages/message.html.erb
 - lib/generators/appchat/templates/messages/new.html.erb
 - lib/generators/appchat/templates/messages_controller.rb
+- lib/generators/appchat/templates/models/appchat_function.rb
 - lib/generators/appchat/templates/models/chat.rb
 - lib/generators/appchat/templates/models/message.rb
+- lib/generators/appchat/templates/services/appchat_function_service.rb
+- lib/generators/appchat/templates/services/web_search_service.rb
+- lib/generators/appchat/templates/tasks/create_appchat_function.rake
 homepage: https://rubygems.org/gems/appchat
 licenses:
 - MIT
@@ -111,5 +115,5 @@ requirements: []
 rubygems_version: 3.5.17
 signing_key:
 specification_version: 4
-summary: Appchat makes it easy to add a chat to your app
+summary: Appchat makes it easy to add an AI chat to your app
 test_files: []