From 70ebe45ec865df2a4b345a57d67af2b54843c73f Mon Sep 17 00:00:00 2001
From: Matt Lindsey <matthew.r.lindsey@gmail.com>
Date: Sun, 10 Nov 2024 06:18:15 -0500
Subject: [PATCH] Add print to assistant example

---
 examples/assistant_chat.rb | 56 ++++++++++++++++++++++++++++----------
 1 file changed, 41 insertions(+), 15 deletions(-)

diff --git a/examples/assistant_chat.rb b/examples/assistant_chat.rb
index 681c687fd..d534f44b6 100644
--- a/examples/assistant_chat.rb
+++ b/examples/assistant_chat.rb
@@ -6,30 +6,21 @@
 # gem install reline
 # or add `gem "reline"` to your Gemfile
 
-openai = Langchain::LLM::OpenAI.new(api_key: ENV["OPENAI_API_KEY"])
-assistant = Langchain::Assistant.new(
-  llm: openai,
-  instructions: "You are a Meteorologist Assistant that is able to pull the weather for any location",
-  tools: [
-    Langchain::Tool::Weather.new(api_key: ENV["OPEN_WEATHER_API_KEY"])
-  ]
-)
-
-DONE = %w[done end eof exit].freeze
-
-puts "Welcome to your Meteorological assistant!"
+DONE = %w[done end eof print exit].freeze
 
 def prompt_for_message
-  puts "(multiline input; type 'end' on its own line when done. or exit to exit)"
+  puts "\n(multiline input; type 'end' on its own line when done. or 'print' to print messages, or 'exit' to exit)\n\n"
 
-  user_message = Reline.readmultiline("Question: ", true) do |multiline_input|
+  user_message = Reline.readmultiline("Query: ", true) do |multiline_input|
     last = multiline_input.split.last
     DONE.include?(last)
   end
 
   return :noop unless user_message
+  return :print if user_message == "print"
 
   lines = user_message.split("\n")
+
   if lines.size > 1 && DONE.include?(lines.last)
     # remove the "done" from the message
     user_message = lines[0..-2].join("\n")
@@ -40,19 +31,54 @@ def prompt_for_message
   user_message
 end
 
+def print_messages(assistant)
+  puts "\n"
+  assistant.messages.each do |message|
+    puts "----"
+    puts message.role + " role content: " + message.content
+    case message.role
+    when "assistant"
+      message.tool_calls.each do |tool_call|
+        puts " " + tool_call["function"]["name"] + ": " << tool_call["function"]["arguments"]
+      end
+    when "tool"
+      puts " tool_call_id: " + message.tool_call_id
+    end
+  end
+  puts "-------"
+end
+
 begin
+  llm = Langchain::LLM::OpenAI.new(api_key: ENV["OPENAI_API_KEY"])
+
+  # Or use local Ollama. See https://ollama.com/search?c=tools for models that support tools.
+  # llm = Langchain::LLM::Ollama.new(default_options: {chat_model: "mistral"})
+
+  assistant = Langchain::Assistant.new(
+    llm: llm,
+    instructions: "You are a Meteorologist Assistant that is able to report the weather for any city in metric units.",
+    tools: [
+      Langchain::Tool::Weather.new(api_key: ENV["OPEN_WEATHER_API_KEY"])
+    ]
+  )
+
+  puts "Welcome to your Meteorological assistant!"
+
   loop do
     user_message = prompt_for_message
 
     case user_message
     when :noop
       next
+    when :print
+      print_messages(assistant)
+      next
     when :exit
       break
     end
 
     assistant.add_message_and_run content: user_message, auto_tool_execution: true
-    puts assistant.messages.last.content
+    puts assistant.messages.last.content + "\n"
   end
 rescue Interrupt
   exit 0