From 6a54124f38bc141c84b2e0f051ca3ec3bbc1c73e Mon Sep 17 00:00:00 2001 From: Shivam Mishra Date: Fri, 4 Apr 2025 22:56:28 +0530 Subject: [PATCH] fix: message response --- .../services/captain/llm/assistant_chat_service.rb | 11 ++--------- .../services/captain/tools/documentation_search.rb | 2 +- 2 files changed, 3 insertions(+), 10 deletions(-) diff --git a/enterprise/app/services/captain/llm/assistant_chat_service.rb b/enterprise/app/services/captain/llm/assistant_chat_service.rb index 8423b1687..ccb2db0b6 100644 --- a/enterprise/app/services/captain/llm/assistant_chat_service.rb +++ b/enterprise/app/services/captain/llm/assistant_chat_service.rb @@ -9,11 +9,7 @@ class Captain::Llm::AssistantChatService < Llm::BaseService def generate_response(input, previous_messages = [], _role = 'user') previous_messages.each do |msg| - if msg[:role] == 'system' - @chat.with_instructions(msg[:content]) - else - @chat.add_message(role: msg[:role], content: msg[:content]) - end + @chat.add_message(role: msg[:role], content: msg[:content]) end response = @chat.ask(input) if input.present? @@ -35,10 +31,7 @@ class Captain::Llm::AssistantChatService < Llm::BaseService begin ::JSON.parse(content) rescue ::JSON::ParserError - { - 'reasoning' => '', - 'response' => content - } + content end end end diff --git a/enterprise/app/services/captain/tools/documentation_search.rb b/enterprise/app/services/captain/tools/documentation_search.rb index dc8fef7dd..18326d1e4 100644 --- a/enterprise/app/services/captain/tools/documentation_search.rb +++ b/enterprise/app/services/captain/tools/documentation_search.rb @@ -10,7 +10,7 @@ class Captain::Tools::DocumentationSearch < RubyLLM::Tool @assistant = assistant end - def execute(query:) + def execute(query) @assistant .responses .approved