appchat 0.0.5 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3071421a8c3c49e11f5c5186b9f17a00bd37d45229d0d95a3b466ae9b9032fee
-  data.tar.gz: 79284185b56a3efaeb60e1ec1ad144ada532fd499e686b4604db7a29aabccd16
+  metadata.gz: 0f2e478412edcf3880d15f39ad4aeedbbf73a1a4819c0b270cf1505d3c47b826
+  data.tar.gz: 4cc000f36710cda913c9474e04c3c55fc221170811fde494392cb6d19c3795ac
 SHA512:
-  metadata.gz: 9e642582fe817de68a171c14b272a0b0ceb2257b96021ae9a51e630c5ef562ed982c452fdc9694c2d6e6c81378720f636e0e8218f51d1704134d383602cf2df7
-  data.tar.gz: a71996c59642154416cb3178b9406d44242b5626ab7069aeb6341a15e1062d6c297131dfd19de2455c7e59be0cbf8af8c43e6daa0879afb269c001a88723f0a3
+  metadata.gz: c6d2310e575c27c6401a2a74afe7c099c9e7dff546dd1220781dbcce74482e2f61af439c9634c1e06ae5c0589f2873ba919a9552f306507e0b2a03dcdaca2c0e
+  data.tar.gz: 2b5b16b39b719650c33ef334f3ceb415fcc445c4a7fa28284b19969ee2a8fd1d69735208148d961c4ebb5852bda67cb8e65a5d29830e1249d35cab53cc64141b

data/lib/generators/appchat/appchat_generator.rb

@@ -9,17 +9,23 @@ 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
+    )
+
+    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 +36,10 @@ 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"
+    generate "model", "FunctionLog message:references name:string prompt:text results:text"
   end
 
   def create_controllers
@@ -46,6 +55,7 @@ class AppchatGenerator < Rails::Generators::Base
     copy_file "messages/new.html.erb", "app/views/messages/new.html.erb", force: true
     copy_file "messages/message.html.erb", "app/views/messages/_message.html.erb", force: true
     copy_file "messages/_typing_bubbles.html.erb", "app/views/messages/_typing_bubbles.html.erb", force: true
+    copy_file "messages/_function_logs.html.erb", "app/views/messages/_function_logs.html.erb", force: true
   end
 
   def create_stylesheets
@@ -55,19 +65,22 @@ class AppchatGenerator < Rails::Generators::Base
   def create_stimulus_controllers
     copy_file "javascript/chat_message_controller.js", "app/javascript/controllers/chat_message_controller.js"
     copy_file "javascript/speech_to_text_controller.js", "app/javascript/controllers/speech_to_text_controller.js"
+    copy_file "javascript/toggle_controller.js", "app/javascript/controllers/toggle_controller.js"
   end
 
   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 +101,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,21 @@ 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_log = message.function_logs.create(name: appchat_function_service["appchat_function"], prompt: appchat_function_service["parameters"])
+      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?
+      function_log.update(results: function_response)
+      @informed_prompt = "The user prompted: #{ user_prompt }, to help you answer the user, an appchat function called #{ function_class } provided this data #{ function_response }"
+    end
+  end
 end

data/lib/generators/appchat/templates/javascript/toggle_controller.js

@@ -0,0 +1,15 @@
+import { Controller } from "@hotwired/stimulus"
+
+export default class extends Controller {
+  static targets = ["content", "icon"]
+
+  toggle() {
+    this.contentTarget.classList.toggle("hidden")
+    if (this.contentTarget.classList.contains("hidden")) {
+      this.iconTarget.textContent = "Expand"
+    } else {
+      this.iconTarget.textContent = "Collapse"
+    }
+  }
+}
+

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

@@ -0,0 +1,12 @@
+<div data-controller="toggle">
+  <%= message.function_logs.count %> Function calls informed this message.
+  <i data-toggle-target="icon" data-action="click->toggle#toggle" class="icon-class">Expand</i> 
+
+  <div data-toggle-target="content" class="hidden bg-white">
+    <% message.function_logs.each do |log| %>
+      <p> Name: <%= log.name %> </p>
+      <p> Parameters: <%= log.prompt %> </p>
+      <p> Response: <%= log.results %> </p>
+    <% end %>
+  </div>
+</div>

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

@@ -1,5 +1,9 @@
 <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? %>
+  <%= render partial: 'messages/function_logs', locals: { message: message } if message.function_logs.present? %>
+
+  <% 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/models/message.rb

@@ -1,6 +1,8 @@
 class Message < ApplicationRecord
   include ActionView::RecordIdentifier
   belongs_to :chat
+  has_many :function_logs
+
   after_create_commit :broadcast_message
   after_update_commit :broadcast_update_message
   

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,17 @@
+require 'cgi'
+require 'open-uri'
+
+class WebSearchService
+  attr_reader :query
+  def initialize(args)
+    @query = args["query"]
+  end
+
+  def run
+    yield("Searching Google for #{query}")
+
+    doc = Nokogiri::HTML(URI.open("https://www.google.com/search?q=#{CGI.escape(query)}").read.force_encoding('UTF-8'))
+    doc.css('script, style, noscript, comment').remove
+    doc.css('h1, h2, h3, h4, h5, h6, p').map(&:text).join("\n").strip
+  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.7
 platform: ruby
 authors:
 - hackliteracy
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-08-29 00:00:00.000000000 Z
+date: 2024-09-11 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: []
@@ -81,13 +81,19 @@ files:
 - lib/generators/appchat/templates/get_ai_response_job.rb
 - lib/generators/appchat/templates/javascript/chat_message_controller.js
 - lib/generators/appchat/templates/javascript/speech_to_text_controller.js
+- lib/generators/appchat/templates/javascript/toggle_controller.js
+- lib/generators/appchat/templates/messages/_function_logs.html.erb
 - lib/generators/appchat/templates/messages/_typing_bubbles.html.erb
 - lib/generators/appchat/templates/messages/index.html.erb
 - 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 +117,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: []