本课将涵盖:
- 探索不同的 Mistral 模型
- 了解每个模型的用例和场景
- 探索展示每个模型独特特性的代码示例。
本课我们将探索 3 种不同的 Mistral 模型:
Mistral Large、Mistral Small 和 Mistral Nemo。
这些模型均免费提供于 Microsoft Foundry Models。本笔记本中的代码将使用这些模型来运行。
注意: GitHub 模型服务将于 2026 年 7 月底退役。有关使用 Microsoft Foundry Models 进行 AI 模型原型设计的更多详情,请参见此链接。
Mistral Large 2 目前是 Mistral 的旗舰模型,面向企业使用设计。
该模型是对原始 Mistral Large 的升级,具备
- 更大的上下文窗口 - 128k 对比 32k
- 在数学和编码任务上表现更佳 - 平均准确率 76.9% 对比 60.4%
- 提升的多语言能力 - 支持语言包括:英语、法语、德语、西班牙语、意大利语、葡萄牙语、荷兰语、俄语、中文、日语、韩语、阿拉伯语和印地语。
凭借这些特性,Mistral Large 在以下方面表现优异:
- 检索增强生成(RAG) - 由于更大的上下文窗口
- 函数调用 - 该模型支持原生函数调用,允许与外部工具和 API 集成。调用方式可以并行或顺序执行。
- 代码生成 - 该模型在 Python、Java、TypeScript 和 C++ 代码生成方面表现出色。
在此示例中,我们使用 Mistral Large 2 对文本文件执行 RAG 模式。问题使用韩语书写,询问作者大学前的活动。
它使用 Cohere Embeddings 模型为文本和问题创建向量表示。此示例使用 faiss Python 包作为向量存储。
发给 Mistral 模型的提示包括问题和与问题相似的检索块。模型随后提供自然语言回复。
pip install faiss-cpuimport requests
import numpy as np
import faiss
import os
from azure.ai.inference import ChatCompletionsClient
from azure.ai.inference.models import SystemMessage, UserMessage
from azure.core.credentials import AzureKeyCredential
from azure.ai.inference import EmbeddingsClient
# 从您的Microsoft Foundry项目的“概览”页面获取这些
endpoint = os.environ["AZURE_INFERENCE_ENDPOINT"]
model_name = "Mistral-large"
token = os.environ["AZURE_INFERENCE_CREDENTIAL"]
client = ChatCompletionsClient(
endpoint=endpoint,
credential=AzureKeyCredential(token),
)
response = requests.get('https://raw.githubusercontent.com/run-llama/llama_index/main/docs/docs/examples/data/paul_graham/paul_graham_essay.txt')
text = response.text
chunk_size = 2048
chunks = [text[i:i + chunk_size] for i in range(0, len(text), chunk_size)]
len(chunks)
embed_model_name = "cohere-embed-v3-multilingual"
embed_client = EmbeddingsClient(
endpoint=endpoint,
credential=AzureKeyCredential(token)
)
embed_response = embed_client.embed(
input=chunks,
model=embed_model_name
)
text_embeddings = []
for item in embed_response.data:
length = len(item.embedding)
text_embeddings.append(item.embedding)
text_embeddings = np.array(text_embeddings)
d = text_embeddings.shape[1]
index = faiss.IndexFlatL2(d)
index.add(text_embeddings)
question = "저자가 대학에 오기 전에 주로 했던 두 가지 일은 무엇이었나요?"
question_embedding = embed_client.embed(
input=[question],
model=embed_model_name
)
question_embeddings = np.array(question_embedding.data[0].embedding)
D, I = index.search(question_embeddings.reshape(1, -1), k=2) # 距离,索引
retrieved_chunks = [chunks[i] for i in I.tolist()[0]]
prompt = f"""
Context information is below.
---------------------
{retrieved_chunks}
---------------------
Given the context information and not prior knowledge, answer the query.
Query: {question}
Answer:
"""
chat_response = client.complete(
messages=[
SystemMessage(content="You are a helpful assistant."),
UserMessage(content=prompt),
],
temperature=1.0,
top_p=1.0,
max_tokens=1000,
model=model_name
)
print(chat_response.choices[0].message.content)Mistral Small 是 Mistral 系列中另一款属于高级/企业分类的模型。顾名思义,该模型属于小型语言模型(SLM)。使用 Mistral Small 的优势包括:
- 与 Mistral 大型语言模型如 Mistral Large 和 NeMo 相比节省成本 - 价格下降 80%
- 低延迟 - 响应速度较 Mistral 的大型语言模型更快
- 灵活 - 可部署于不同环境,对资源需求限制较少。
Mistral Small 适合:
- 基于文本的任务,如摘要、情感分析和翻译。
- 因其成本效益适合频繁请求的应用
- 低延迟的代码任务,如代码审查和建议
为展示 Mistral Small 与 Large 在延迟上的差异,请运行下方单元格。
你应会看到响应时间差异在 3-5 秒之间。也请注意同一提示词下两者的回复长度和风格。
import os
endpoint = os.environ["AZURE_INFERENCE_ENDPOINT"]
model_name = "Mistral-small"
token = os.environ["AZURE_INFERENCE_CREDENTIAL"]
client = ChatCompletionsClient(
endpoint=endpoint,
credential=AzureKeyCredential(token),
)
response = client.complete(
messages=[
SystemMessage(content="You are a helpful coding assistant."),
UserMessage(content="Can you write a Python function to the fizz buzz test?"),
],
temperature=1.0,
top_p=1.0,
max_tokens=1000,
model=model_name
)
print(response.choices[0].message.content)import os
from azure.ai.inference import ChatCompletionsClient
from azure.ai.inference.models import SystemMessage, UserMessage
from azure.core.credentials import AzureKeyCredential
endpoint = os.environ["AZURE_INFERENCE_ENDPOINT"]
model_name = "Mistral-large"
token = os.environ["AZURE_INFERENCE_CREDENTIAL"]
client = ChatCompletionsClient(
endpoint=endpoint,
credential=AzureKeyCredential(token),
)
response = client.complete(
messages=[
SystemMessage(content="You are a helpful coding assistant."),
UserMessage(content="Can you write a Python function to the fizz buzz test?"),
],
temperature=1.0,
top_p=1.0,
max_tokens=1000,
model=model_name
)
print(response.choices[0].message.content)相较本课讨论的另外两款模型,Mistral NeMo 是唯一持有 Apache2 许可证的免费模型。
它被视作 Mistral 早期开源大型语言模型 Mistral 7B 的升级版本。
NeMo 模型的其他特点有:
-
更高效的分词: 该模型采用 Tekken 分词器,而非更常用的 tiktoken。这使得其在多语言和代码方面性能更优。
-
微调能力: 基础模型支持微调,允许在需要微调的场景中更具灵活性。
-
原生函数调用 - 与 Mistral Large 类似,该模型经过函数调用训练,是首批开源支持此功能的模型之一。
在此示例中,我们将比较 Mistral NeMo 与 Mistral Large 在分词处理上的差异。
两个示例都使用相同的提示词,但你会看到 NeMo 返回的令牌数少于 Mistral Large。
pip install mistral-common# 导入所需的包:
from mistral_common.protocol.instruct.messages import (
UserMessage,
)
from mistral_common.protocol.instruct.request import ChatCompletionRequest
from mistral_common.protocol.instruct.tool_calls import (
Function,
Tool,
)
from mistral_common.tokens.tokenizers.mistral import MistralTokenizer
# 加载Mistral分词器
model_name = "open-mistral-nemo"
tokenizer = MistralTokenizer.from_model(model_name)
# 对消息列表进行分词
tokenized = tokenizer.encode_chat_completion(
ChatCompletionRequest(
tools=[
Tool(
function=Function(
name="get_current_weather",
description="Get the current weather",
parameters={
"type": "object",
"properties": {
"location": {
"type": "string",
"description": "The city and state, e.g. San Francisco, CA",
},
"format": {
"type": "string",
"enum": ["celsius", "fahrenheit"],
"description": "The temperature unit to use. Infer this from the user's location.",
},
},
"required": ["location", "format"],
},
)
)
],
messages=[
UserMessage(content="What's the weather like today in Paris"),
],
model=model_name,
)
)
tokens, text = tokenized.tokens, tokenized.text
# 统计标记数量
print(len(tokens))# 导入所需的包:
from mistral_common.protocol.instruct.messages import (
UserMessage,
)
from mistral_common.protocol.instruct.request import ChatCompletionRequest
from mistral_common.protocol.instruct.tool_calls import (
Function,
Tool,
)
from mistral_common.tokens.tokenizers.mistral import MistralTokenizer
# 加载 Mistral 分词器
model_name = "mistral-large-latest"
tokenizer = MistralTokenizer.from_model(model_name)
# 对消息列表进行分词
tokenized = tokenizer.encode_chat_completion(
ChatCompletionRequest(
tools=[
Tool(
function=Function(
name="get_current_weather",
description="Get the current weather",
parameters={
"type": "object",
"properties": {
"location": {
"type": "string",
"description": "The city and state, e.g. San Francisco, CA",
},
"format": {
"type": "string",
"enum": ["celsius", "fahrenheit"],
"description": "The temperature unit to use. Infer this from the user's location.",
},
},
"required": ["location", "format"],
},
)
)
],
messages=[
UserMessage(content="What's the weather like today in Paris"),
],
model=model_name,
)
)
tokens, text = tokenized.tokens, tokenized.text
# 计算令牌数量
print(len(tokens))完成本课后,请访问我们的 生成式 AI 学习合集 继续提升你的生成式 AI 知识!
免责声明: 本文件由 AI 翻译服务 Co-op Translator 翻译完成。尽管我们力求准确,但请注意,自动翻译可能包含错误或不准确之处。原始语言版文件应视为权威来源。对于重要信息,建议使用专业人工翻译。我们对因使用本翻译而产生的任何误解或误释不承担责任。