@5minds/node-red-dashboard-2-processcube-chat 0.1.1-add-functionality-dd2f3b-mdej6h1e → 0.1.1-add-functionality-043ee8-mdy8o5ki

package/ui/components/UIDeepChat.vue

@@ -187,55 +187,108 @@ export default {
         },
 
         formatForChatGPT(conversation, textOnly = false) {
-            return {
-                messages: conversation.map((msg) => {
-                    if (msg.text && (!msg.files || msg.files.length === 0)) {
-                        return {
-                            role: msg.role === 'ai' ? 'assistant' : 'user',
-                            content: msg.text,
-                        };
-                    } else if (msg.files && msg.files.length > 0) {
-                        const content = [{ type: 'text', text: msg.text || '' }];
-
-                        msg.files.forEach((file) => {
-                            if (file.type && file.type.startsWith('image/') && file.src) {
-                                content.push({
-                                    type: 'image_url',
-                                    image_url: { url: file.src },
-                                });
-                            } else if (!textOnly && file.type && file.type.startsWith('audio/') && file.base64Data) {
-                                const format = this.getAudioFormat(file.type);
-                                content.push({
-                                    type: 'input_audio',
-                                    input_audio: {
-                                        data: file.base64Data,
-                                        format: format,
-                                    },
-                                });
-                            } else if (!textOnly && file.fileData) {
-                                content.push({
-                                    type: 'file',
-                                    file: {
-                                        filename: file.name,
-                                        file_data: file.fileData,
-                                    },
-                                });
-                            }
-                        });
+            const payload = {
+                messages: conversation.map((msg) => this.formatMessage(msg, textOnly)),
+                model: this.props.model,
+            };
 
-                        return {
-                            role: msg.role === 'ai' ? 'assistant' : 'user',
-                            content: content,
-                        };
-                    }
+            // Add ChatGPT API extensions
+            if (this.props.tools) {
+                payload.tools = this.props.tools;
+            }
 
-                    return {
-                        role: msg.role === 'ai' ? 'assistant' : 'user',
-                        content: msg.text || '',
-                    };
-                }),
-                model: this.props.model,
+            if (this.props.tool_choice) {
+                payload.tool_choice = this.props.tool_choice;
+            }
+
+            if (this.props.temperature !== undefined) {
+                payload.temperature = this.props.temperature;
+            }
+
+            if (this.props.max_tokens) {
+                payload.max_tokens = this.props.max_tokens;
+            }
+
+            if (this.props.top_p !== undefined) {
+                payload.top_p = this.props.top_p;
+            }
+
+            if (this.props.frequency_penalty !== undefined) {
+                payload.frequency_penalty = this.props.frequency_penalty;
+            }
+
+            if (this.props.presence_penalty !== undefined) {
+                payload.presence_penalty = this.props.presence_penalty;
+            }
+
+            if (this.props.response_format) {
+                payload.response_format = this.props.response_format;
+            }
+
+            return payload;
+        },
+
+        formatMessage(msg, textOnly = false) {
+            const message = {
+                role: msg.role === 'ai' ? 'assistant' : 'user',
             };
+
+            // Handle tool calls (for assistant messages)
+            if (msg.tool_calls) {
+                message.tool_calls = msg.tool_calls;
+                message.content = msg.content || null;
+                return message;
+            }
+
+            // Handle tool responses (for tool messages)
+            if (msg.role === 'tool') {
+                message.role = 'tool';
+                message.content = msg.content;
+                message.tool_call_id = msg.tool_call_id;
+                return message;
+            }
+
+            // Handle messages with files
+            if (msg.files && msg.files.length > 0) {
+                message.content = this.buildContentArray(msg, textOnly);
+                return message;
+            }
+
+            // Handle text-only messages
+            message.content = msg.text || msg.content || '';
+            return message;
+        },
+
+        buildContentArray(msg, textOnly) {
+            const content = [{ type: 'text', text: msg.text || '' }];
+
+            msg.files.forEach((file) => {
+                if (file.type && file.type.startsWith('image/') && file.src) {
+                    content.push({
+                        type: 'image_url',
+                        image_url: { url: file.src },
+                    });
+                } else if (!textOnly && file.type && file.type.startsWith('audio/') && file.base64Data) {
+                    const format = this.getAudioFormat(file.type);
+                    content.push({
+                        type: 'input_audio',
+                        input_audio: {
+                            data: file.base64Data,
+                            format: format,
+                        },
+                    });
+                } else if (!textOnly && file.fileData) {
+                    content.push({
+                        type: 'file',
+                        file: {
+                            filename: file.name,
+                            file_data: file.fileData,
+                        },
+                    });
+                }
+            });
+
+            return content;
         },
 
         getAudioFormat(mimeType) {
@@ -256,16 +309,36 @@ export default {
 
         handleNodeREDResponse(msg, signals) {
             try {
+                // Store the complete ChatGPT response in conversation
+                const fullResponse = msg.payload;
+
+                // Create AI message with complete response data
                 const aiMessage = {
-                    text: msg.payload.text || msg.payload.content,
                     role: 'ai',
+                    content: fullResponse.message?.content || fullResponse.content,
+                    text: fullResponse.message?.content || fullResponse.content,
+                    fullResponse: fullResponse,
                 };
 
+                // Handle tool calls if present
+                if (fullResponse.message?.tool_calls) {
+                    aiMessage.tool_calls = fullResponse.message.tool_calls;
+                    aiMessage.content = fullResponse.message.content || null;
+                }
+
                 this.conversation.push(aiMessage);
 
-                if (msg.payload) {
+                // For the chat display, show appropriate response
+                if (fullResponse.message?.tool_calls) {
+                    // If ChatGPT wants to call tools, show a message
+                    signals.onResponse({
+                        text: 'Function call requested...',
+                        role: 'ai',
+                    });
+                } else if (fullResponse.message?.content || fullResponse.content) {
+                    // Normal text response
                     signals.onResponse({
-                        text: msg.payload.text || msg.payload.content,
+                        text: fullResponse.message?.content || fullResponse.content,
                         role: 'ai',
                     });
                 }