How to Create Custom Artificial Intelligence Models in FlutterFlow

1// Firebase Cloud Function: chatWithAI
2// Callable function — invoked from FlutterFlow Custom Action
3// firebase deploy --only functions:chatWithAI
4
5const functions = require('firebase-functions');
6const admin = require('firebase-admin');
7const OpenAI = require('openai');
8const Anthropic = require('@anthropic-ai/sdk');
9
10exports.chatWithAI = functions.https.onCall(async (data, context) => {
11  if (!context.auth) {
12    throw new functions.https.HttpsError('unauthenticated', 'Login required');
13  }
14
15  const { conversationId, userMessage, provider = 'openai' } = data;
16  if (!conversationId || !userMessage) {
17    throw new functions.https.HttpsError('invalid-argument', 'Missing required fields');
18  }
19
20  const db = admin.firestore();
21  const uid = context.auth.uid;
22
23  // Save user message to Firestore
24  const messagesRef = db.collection('conversations')
25    .doc(conversationId).collection('messages');
26  await messagesRef.add({
27    role: 'user',
28    content: userMessage,
29    timestamp: admin.firestore.FieldValue.serverTimestamp(),
30  });
31
32  // Load last 20 messages for context
33  const historySnap = await messagesRef
34    .orderBy('timestamp', 'desc').limit(20).get();
35  const history = historySnap.docs.reverse().map(d => ({
36    role: d.data().role,
37    content: d.data().content,
38  }));
39
40  let assistantMessage = '';
41
42  if (provider === 'openai') {
43    const openai = new OpenAI({ apiKey: functions.config().openai.key });
44    const completion = await openai.chat.completions.create({
45      model: 'gpt-4o',
46      messages: [
47        { role: 'system', content: 'You are a helpful assistant.' },
48        ...history,
49      ],
50      max_tokens: 1000,
51    });
52    assistantMessage = completion.choices[0].message.content;
53
54  } else if (provider === 'anthropic') {
55    const anthropic = new Anthropic({ apiKey: functions.config().anthropic.key });
56    const message = await anthropic.messages.create({
57      model: 'claude-3-5-sonnet-20241022',
58      max_tokens: 1024,
59      system: 'You are a helpful assistant.',
60      messages: history,
61    });
62    assistantMessage = message.content[0].text;
63
64  } else if (provider === 'gemini') {
65    const { GoogleGenerativeAI } = require('@google/generative-ai');
66    const genAI = new GoogleGenerativeAI(functions.config().gemini.key);
67    const model = genAI.getGenerativeModel({ model: 'gemini-1.5-pro' });
68    const chat = model.startChat({
69      history: history.slice(0, -1).map(m => ({
70        role: m.role === 'assistant' ? 'model' : m.role,
71        parts: [{ text: m.content }],
72      })),
73    });
74    const result = await chat.sendMessage(userMessage);
75    assistantMessage = result.response.text();
76  }
77
78  // Save assistant response to Firestore
79  await messagesRef.add({
80    role: 'assistant',
81    content: assistantMessage,
82    timestamp: admin.firestore.FieldValue.serverTimestamp(),
83  });
84
85  // Update conversation updatedAt
86  await db.collection('conversations').doc(conversationId).update({
87    updatedAt: admin.firestore.FieldValue.serverTimestamp(),
88  });
89
90  return { message: assistantMessage };
91});

1// Custom Action: sendMessageToAI
2// Arguments: userMessage (String), conversationId (String)
3// Return type: String (assistant response or error message)
4
5Future<String> sendMessageToAI(
6  String userMessage,
7  String conversationId,
8) async {
9  if (userMessage.trim().isEmpty) return '';
10
11  try {
12    final callable = FirebaseFunctions.instance
13        .httpsCallable(
14          'chatWithAI',
15          options: HttpsCallableOptions(
16            timeout: const Duration(seconds: 30),
17          ),
18        );
19
20    final result = await callable.call({
21      'userMessage': userMessage.trim(),
22      'conversationId': conversationId,
23      'provider': 'openai', // or 'anthropic', 'gemini'
24    });
25
26    return result.data['message'] as String? ?? '';
27  } on FirebaseFunctionsException catch (e) {
28    return 'Error: ${e.message}';
29  } on TimeoutException {
30    return 'Request timed out. Please try again.';
31  } catch (e) {
32    return 'An unexpected error occurred.';
33  }
34}