Add missing pluginName sendMessage() in AIService

ditrit · Sep 26, 2024 · f9871c9 · f9871c9
1 parent a37ac62
commit f9871c9
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/src/components/drawer/AIChatDrawer.vue b/src/components/drawer/AIChatDrawer.vue
@@ -257,7 +257,7 @@ async function submit(event) {
     props.pluginName,
     files,
   );
-  const aiMessage = await sendMessage(aiConversation.id, formattedText);
+  const aiMessage = await sendMessage(aiConversation.id, props.pluginName, formattedText);
 
   lastAIMessage.value = {
     isMine: false,

diff --git a/src/services/AIService.js b/src/services/AIService.js
@@ -94,13 +94,14 @@ export async function manageConversation(project, diagram, plugin, files) {
 /**
  * Send Message to AI and get its answers.
  * @param {string} conversationId - Id of conversation.
+ * @param {string} pluginName - Plugin name.
  * @param {string} message - Message to send.
  * @returns {Promise<object>} Answer of AI.
  */
-export async function sendMessage(conversationId, message) {
+export async function sendMessage(conversationId, pluginName, message) {
   const api = await prepareApiRequest();
 
-  return api.post(`/ai/conversations/${conversationId}/messages`, message, options)
+  return api.post(`/ai/conversations/${conversationId}/messages`, { plugin: pluginName, message }, options)
     .then((data) => ({
       ...data,
       message: uncompress(data.message),

diff --git a/tests/unit/services/AIService.spec.js b/tests/unit/services/AIService.spec.js
@@ -86,7 +86,7 @@ describe('AI Service tests', () => {
       }));
       prepareApiRequest.mockImplementation(() => Promise.resolve(api));
 
-      const result = await sendMessage('1', 'test');
+      const result = await sendMessage('1', 'plugin', 'test');
 
       expect(result.message).toEqual('OK');
     });