Merge branch '433137-mk-implementing-a-flag-to-clear-user-history-in-chat-api' into 'master'

Add resetting to chat rest endpoint

See merge request https://gitlab.com/gitlab-org/gitlab/-/merge_requests/138213



Merged-by: Nicolas Dular <ndular@gitlab.com>
Approved-by: Nicolas Dular <ndular@gitlab.com>
Reviewed-by: Nicolas Dular <ndular@gitlab.com>
Co-authored-by: Małgorzata Ksionek <mksionek@gitlab.com>
Co-authored-by: Alejandro Rodríguez <alejandro@gitlab.com>

ee/lib/api/chat.rb

@@ -28,6 +28,12 @@ def find_resource(parameters)
         object = parameters[:resource_type].camelize.safe_constantize
         object.find(parameters[:resource_id])
       end
+
+      def reset_chat(action_name, message_attributes)
+        message_attributes[:content] = '/reset'
+        prompt_message = ::Gitlab::Llm::AiMessage.for(action: action_name).new(message_attributes)
+        prompt_message.save!
+      end
     end
 
     namespace 'chat' do
@@ -38,6 +44,8 @@ def find_resource(parameters)
           optional :resource_id, type: Integer, desc: 'ID of resource.'
           optional :referer_url, type: String, limit: 1000, desc: 'Referer URL'
           optional :client_subscription_id, type: String, limit: 500, desc: 'Client Subscription ID'
+          optional :with_clean_history, type: Boolean,
+            desc: 'Indicates if we need to reset the history before and after the request'
         end
         post do
           safe_params = declared_params(include_missing: false)
@@ -46,9 +54,9 @@ def find_resource(parameters)
           not_found! unless user_allowed?(resource)
           action_name = 'chat'
 
+          options = safe_params.slice(:referer_url)
           message_attributes = {
             request_id: SecureRandom.uuid,
-            content: safe_params[:content],
             role: ::Gitlab::Llm::AiMessage::ROLE_USER,
             ai_action: action_name,
             user: current_user,
@@ -56,10 +64,15 @@ def find_resource(parameters)
             client_subscription_id: safe_params[:client_subscription_id]
           }
 
+          reset_chat(action_name, message_attributes) if safe_params[:with_clean_history]
+
+          message_attributes[:content] = safe_params[:content]
+
           prompt_message = ::Gitlab::Llm::AiMessage.for(action: action_name).new(message_attributes)
-          options = safe_params.slice(:referer_url)
           ai_response = Llm::Internal::CompletionService.new(prompt_message, options).execute
 
+          reset_chat(action_name, message_attributes) if safe_params[:with_clean_history]
+
           present ai_response.response_body
         end
       end

ee/spec/requests/api/chat_spec.rb

@@ -225,6 +225,25 @@
             post_api
           end
         end
+
+        context 'with reset_history' do
+          let(:params) { { content: content, with_clean_history: true } }
+          let(:resource) { current_user }
+          let(:reset_message) { instance_double(Gitlab::Llm::ChatMessage) }
+
+          it 'sends resource to the chat' do
+            reset_params = chat_message_params.dup
+            reset_params[:content] = '/reset'
+
+            expect(Gitlab::Llm::ChatMessage).to receive(:new).with(reset_params).twice.and_return(reset_message)
+            expect(reset_message).to receive(:save!).twice
+            expect(Gitlab::Llm::ChatMessage).to receive(:new).with(chat_message_params).and_return(chat_message)
+            expect(Llm::Internal::CompletionService).to receive(:new).with(chat_message, options).and_return(chat)
+            expect(chat).to receive(:execute)
+
+            post_api
+          end
+        end
       end
     end
   end