Flutter集成LangChain Mistral AI插件langchain_mistralai的使用

Flutter集成LangChain Mistral AI插件langchain_mistralai的使用

介绍

langchain_mistralai 是一个用于在 Dart 和 Flutter 项目中集成 Mistral AI 的插件。它提供了对 Mistral AI 的 Chat Completions API 和 Embeddings API 的封装，使得开发者可以轻松地在应用中使用 Mistral AI 的功能。

功能

• Chat 模型：

  • ChatMistralAI：封装了 Mistral AI 的 Chat Completions API，允许你与 Mistral AI 进行对话。

• Embeddings：

  • MistralAIEmbeddings：封装了 Mistral AI 的 Embeddings API，允许你生成文本的嵌入向量。

完整示例 Demo

以下是一个完整的 Flutter 示例，展示了如何使用 langchain_mistralai 插件来集成 Mistral AI 的聊天功能和嵌入功能。

1. 添加依赖

首先，在你的 pubspec.yaml 文件中添加 langchain_mistralai 依赖：

dependencies:
  flutter:
    sdk: flutter
  langchain_mistralai: ^0.1.0  # 请根据最新的版本号进行调整

然后运行 flutter pub get 来安装依赖。

2. 初始化 Mistral AI 客户端

在你的 Flutter 应用中，你需要初始化 Mistral AI 客户端。你可以通过提供 API 密钥来完成这一步骤。假设你已经在 Mistral AI 控制台获取了 API 密钥。

import 'package:langchain_mistralai/langchain_mistralai.dart';

void main() async {
  // 初始化 Mistral AI 客户端
  final mistralClient = MistralAI(apiKey: 'your-api-key');

  runApp(MyApp(mistralClient: mistralClient));
}

3. 创建聊天界面

接下来，我们创建一个简单的聊天界面，用户可以在其中输入消息并与 Mistral AI 进行对话。

import 'package:flutter/material.dart';
import 'package:langchain_mistralai/langchain_mistralai.dart';

class MyApp extends StatelessWidget {
  final MistralAI mistralClient;

  const MyApp({super.key, required this.mistralClient});

  [@override](/user/override)
  Widget build(BuildContext context) {
    return MaterialApp(
      home: Scaffold(
        appBar: AppBar(title: const Text('Mistral AI Chat')),
        body: ChatScreen(mistralClient: mistralClient),
      ),
    );
  }
}

class ChatScreen extends StatefulWidget {
  final MistralAI mistralClient;

  const ChatScreen({super.key, required this.mistralClient});

  [@override](/user/override)
  _ChatScreenState createState() => _ChatScreenState();
}

class _ChatScreenState extends State<ChatScreen> {
  final List<String> _messages = [];
  final TextEditingController _controller = TextEditingController();

  Future<void> _sendMessage(String message) async {
    if (message.isEmpty) return;

    setState(() {
      _messages.add('User: $message');
    });

    // 使用 Mistral AI 的 ChatMistralAI 模型生成回复
    final response = await widget.mistralClient.chatMistralAI.createChatCompletion(
      messages: [
        {'role': 'user', 'content': message},
      ],
    );

    setState(() {
      _messages.add('Mistral AI: ${response.choices.first.message.content}');
    });

    _controller.clear();
  }

  [@override](/user/override)
  Widget build(BuildContext context) {
    return Column(
      children: [
        Expanded(
          child: ListView.builder(
            itemCount: _messages.length,
            itemBuilder: (context, index) {
              return ListTile(
                title: Text(_messages[index]),
              );
            },
          ),
        ),
        Padding(
          padding: const EdgeInsets.all(8.0),
          child: Row(
            children: [
              Expanded(
                child: TextField(
                  controller: _controller,
                  decoration: const InputDecoration(hintText: 'Enter a message'),
                ),
              ),
              IconButton(
                icon: const Icon(Icons.send),
                onPressed: () => _sendMessage(_controller.text),
              ),
            ],
          ),
        ),
      ],
    );
  }
}

4. 生成文本嵌入

除了聊天功能，langchain_mistralai 还提供了生成文本嵌入的功能。你可以使用 MistralAIEmbeddings 来生成文本的嵌入向量。以下是一个简单的示例，展示如何生成嵌入并向用户显示结果。

import 'package:flutter/material.dart';
import 'package:langchain_mistralai/langchain_mistralai.dart';

class EmbeddingScreen extends StatefulWidget {
  final MistralAI mistralClient;

  const EmbeddingScreen({super.key, required this.mistralClient});

  [@override](/user/override)
  _EmbeddingScreenState createState() => _EmbeddingScreenState();
}

class _EmbeddingScreenState extends State<EmbeddingScreen> {
  String _embeddingResult = '';

  Future<void> _generateEmbedding(String text) async {
    if (text.isEmpty) return;

    final embedding = await widget.mistralClient.embeddings.createEmbedding(
      input: text,
    );

    setState(() {
      _embeddingResult = embedding.data.first.embedding.toString();
    });
  }

  [@override](/user/override)
  Widget build(BuildContext context) {
    return Scaffold(
      appBar: AppBar(title: const Text('Generate Embedding')),
      body: Padding(
        padding: const EdgeInsets.all(16.0),
        child: Column(
          children: [
            TextField(
              decoration: const InputDecoration(hintText: 'Enter text to embed'),
              onSubmitted: (text) => _generateEmbedding(text),
            ),
            const SizedBox(height: 20),
            ElevatedButton(
              onPressed: () => _generateEmbedding(_controller.text),
              child: const Text('Generate Embedding'),
            ),
            const SizedBox(height: 20),
            Text(_embeddingResult),
          ],
        ),
      ),
    );
  }
}