当前位置：首页 > backend >正文

一个最简单的 Model Context Protocol 的例子

backend 2025/5/19 10:21:21

在这里插入图片描述

Model Context Protocol（模型上下文协议）是一种用于定义和管理模型与其运行环境之间交互的协议。它通常用于人工智能、机器学习或复杂系统中，确保模型能够正确地接收输入、处理数据并返回输出。

以下是一个最简单的 Model Context Protocol 的例子：

场景描述

假设我们有一个简单的数学模型，用于计算两个数字的和。这个模型需要从环境中接收两个数字作为输入，并将它们的和返回给环境。

示例代码

1. 定义模型逻辑

class SimpleAdditionModel:def execute(self, context):# 从上下文中获取输入num1 = context.get("num1")num2 = context.get("num2")# 检查输入是否有效if num1 is None or num2 is None:raise ValueError("Input 'num1' and 'num2' must be provided in the context.")# 执行模型的核心逻辑result = num1 + num2# 将结果存回上下文context["result"] = result

2. 定义上下文

上下文是模型与环境之间的桥梁，用于传递输入和输出数据。

class Context:def __init__(self):self.data = {}def get(self, key):return self.data.get(key)def set(self, key, value):self.data[key] = value

3. 使用模型和上下文

# 创建上下文对象
context = Context()# 设置输入数据
context.set("num1", 5)
context.set("num2", 10)# 创建模型实例
model = SimpleAdditionModel()# 执行模型
model.execute(context)# 获取结果
result = context.get("result")
print(f"The result is: {result}")

输出结果

The result is: 15

解释

模型逻辑：SimpleAdditionModel 是一个简单的模型，它的核心功能是接收两个数字并返回它们的和。
上下文协议：Context 类用于存储和传递数据。模型通过 context.get() 获取输入数据，并通过 context["result"] 返回结果。
执行流程：
- 环境通过上下文向模型提供输入数据。
- 模型执行核心逻辑，并将结果写回上下文。
- 环境从上下文中读取结果。