From a14df5521089ce122db2077965ff9ecfe7cd0385 Mon Sep 17 00:00:00 2001
From: Shivam Mishra <scm.mymail@gmail.com>
Date: Wed, 30 Jul 2025 20:58:53 +0530
Subject: [PATCH] fix: query

---
 .../lib/captain/tools/faq_lookup_tool.rb      | 26 ++++++++++++-------
 1 file changed, 16 insertions(+), 10 deletions(-)

diff --git a/enterprise/lib/captain/tools/faq_lookup_tool.rb b/enterprise/lib/captain/tools/faq_lookup_tool.rb
index e2ce85c86..68c80ea74 100644
--- a/enterprise/lib/captain/tools/faq_lookup_tool.rb
+++ b/enterprise/lib/captain/tools/faq_lookup_tool.rb
@@ -1,5 +1,4 @@
 class Captain::Tools::FaqLookupTool < Captain::Tools::BasePublicTool
-  name 'faq_lookup'
   description 'Search FAQ responses using semantic similarity to find relevant answers'
   param :query, type: 'string', desc: 'The question or topic to search for in the FAQ database'
 
@@ -7,29 +6,36 @@ class Captain::Tools::FaqLookupTool < Captain::Tools::BasePublicTool
     log_tool_usage('searching', { query: query })
 
     # Use existing vector search on approved responses
-    responses = @assistant.responses.approved.search(query)
+    responses = @assistant.responses.approved.search(query).to_a
 
     if responses.empty?
       log_tool_usage('no_results', { query: query })
       "No relevant FAQs found for: #{query}"
     else
-      log_tool_usage('found_results', { query: query, count: responses.count })
+      log_tool_usage('found_results', { query: query, count: responses.size })
       format_responses(responses)
     end
+  rescue StandardError => e
+    ChatwootExceptionTracker.new(e).capture_exception
   end
 
   private
 
   def format_responses(responses)
-    formatted = responses.map.with_index(1) do |response, index|
-      result = "\n#{index}. **Q:** #{response.question}\n   **A:** #{response.answer}"
+    responses.map { |response| format_response(response) }.join
+  end
 
-      # Include source if available
-      result += "\n   **Source:** #{response.documentable.external_link}" if response.documentable&.external_link
-
-      result
+  def format_response(response)
+    formatted_response = "
+        Question: #{response.question}
+        Answer: #{response.answer}
+        "
+    if response.documentable.present? && response.documentable.try(:external_link)
+      formatted_response += "
+          Source: #{response.documentable.external_link}
+          "
     end
 
-    "Found #{responses.count} relevant FAQ(s):#{formatted.join("\n")}"
+    formatted_response
   end
 end