大模型到底怎么思考?一篇看懂 Prompt、思维链、思维树




角色定位:就是要让模型更专业的去分析你的问题,给出答案,例如你是一个资深大模型应用开发工程师,性格温和
上下文:背景信息,比如连续问答,遇到有歧义的问题,他可以根据刚刚在聊的内容来猜测你更可能想问什么
任务指令:明确核心动作,内容总结、分析推理、计算结果还是判断对错,避免答非所问
输入数据:具体问题描述、需总结的文档
输出要求:格式化输出便于后续代码解析,字数限制,风格等
约束条件:避免一些prompt攻击,比如禁止回答与主题无关的问题
ollama serve# Ollama原生OpenAPI接口
curl http://127.0.0.1:11434/api/chat -d '{
"model": "qwen3:4b",
"messages": [{"role": "user", "content": "你好"}],
"stream": false
}'
# OpenAI兼容接口
curl http://127.0.0.1:11434/v1/chat/completions \
-H "Content-Type: application/json" \
-H "Authorization: Bearer ollama" \
-d '{
"model": "qwen3:4b",
"messages": [{"role": "user", "content": "你好"}]
}'
from openai import OpenAI
# ===================== 配置项(OpenAI兼容模式) =====================
# 指向本地Ollama的OpenAI兼容接口
client = OpenAI(
base_url="http://127.0.0.1:11434/v1", # Ollama的OpenAI兼容前缀
api_key="ollama"# 本地模型任意值即可,仅用于格式兼容
)
MODEL_NAME = "qwen3:4b"
# ===================== OpenAPI规范调用(兼容OpenAI) =====================
def call_local_llm_openai_compatible(prompt, stream=False):
"""
用OpenAI OpenAPI规范调用本地Ollama模型
"""
try:
# 非流式调用
ifnot stream:
completion = client.chat.completions.create(
model=MODEL_NAME,
messages=[{"role": "user", "content": prompt}],
temperature=0.7,
stream=False
)
return completion.choices[0].message.content
# 流式调用
else:
completion = client.chat.completions.create(
model=MODEL_NAME,
messages=[{"role": "user", "content": prompt}],
temperature=0.7,
stream=True
)
def stream_generator():
for chunk in completion:
if chunk.choices[0].delta.content:
yield chunk.choices[0].delta.content #生成器,每次迭代都会返回一个结果
return stream_generator()
except Exception as e:
returnf"调用失败:{str(e)}"
# ===================== 测试兼容模式调用 =====================
if __name__ == "__main__":
# 1. 非流式调用(OpenAI OpenAPI规范)
print("=== OpenAI兼容模式-非流式调用 ===")
prompt1 = "请严格按JSON格式输出:{\"name\":\"张三\",\"age\":25,\"job\":\"程序员\"}"
result1 = call_local_llm_openai_compatible(prompt1)
print(result1)
print("-" * 60)
# 2. 流式调用(OpenAI SSE规范)
print("=== OpenAI兼容模式-流式调用 ===")
prompt2 = "请解释OpenAPI接口的核心特点"
stream_gen = call_local_llm_openai_compatible(prompt2, stream=True)
print("模型响应:", end="")
for chunk in stream_gen:
print(chunk, end="", flush=True)
print()temperature:温度,他是用来控制输出的稳定性,范围 0~2,数值越小,越严谨,例如数学推理计算,数值越大,越发散,适合文章扩写等等。
stream:True/False,代表是否流式输出,True是逐字输出,False是等全部生成好之后一次性输出。


| 类型 | 特点 | 适用场景 |
|---|---|---|
| 零样本 CoT | 仅用提示词(如 “让我们一步步思考”)触发模型思考 | 不想写示例的快速场景 |
| 少样本 CoT | 给出 1-2 个 “问题 + 思考步骤 + 答案” 的示例,引导模型模仿 | 复杂问题(如数学竞赛题) |
| 自洽 CoT | 让模型生成多个思考路径,最后取一致答案 | 高精准要求场景(如财务计算) |
import requests
# 基础配置(适配本地Ollama)
OLLAMA_URL = "http://127.0.0.1:11434/api/generate"
MODEL_NAME = "qwen3:4b"
# 通用调用函数
def call_llm_with_cot(prompt, temperature=0.1):
"""调用本地模型,适配思维链Prompt"""
data = {
"model": MODEL_NAME,
"prompt": prompt,
"temperature": temperature, # 低温度保证推理严谨
"stream": False
}
try:
response = requests.post(OLLAMA_URL, json=data)
response.raise_for_status()
return response.json()["response"]
except Exception as e:
returnf"调用失败:{str(e)}"
# ===================== 场景1:基础CoT(分步思考) =====================
print("=== 场景1:基础CoT ===")
basic_cot_prompt = """
请严格按照「分析问题→拆解步骤→计算结果→总结答案」的步骤解决以下问题:
问题:某商店第一天卖出20件商品,第二天卖出的数量是第一天的1.5倍,第三天比第二天少卖5件,三天总共卖出多少件?
"""
basic_cot_result = call_llm_with_cot(basic_cot_prompt)
print(basic_cot_result)
print("-" * 70)
# ===================== 场景2:少样本CoT(示例引导) =====================
print("=== 场景2:少样本CoT ===")
few_shot_cot_prompt = """
示例1:
问题:小红有8支笔,买了5支,送给同学3支,还剩几支?
思考步骤:
1. 分析问题:需要计算笔的数量增减;
2. 拆解步骤:初始8支 + 新买5支 - 送出3支;
3. 计算:8+5-3=10;
4. 答案:还剩10支。
示例2:
问题:一本书有200页,第一天看了50页,第二天看了剩下的1/3,还剩多少页没看?
思考步骤:
1. 分析问题:先算剩余页数,再算第二天看的,最后算最终剩余;
2. 拆解步骤:
- 第一天看完剩余:200-50=150页;
- 第二天看的页数:150×1/3=50页;
- 最终剩余:150-50=100页;
3. 计算:200-50-(150×1/3)=100;
4. 答案:还剩100页。
请模仿上述示例的思考步骤,解决以下问题:
问题:一个长方形长12cm,宽比长少4cm,这个长方形的周长和面积分别是多少?
"""
few_shot_cot_result = call_llm_with_cot(few_shot_cot_prompt)
print(few_shot_cot_result)
print("-" * 70)
# ===================== 场景3:零样本CoT(无示例,仅提示) =====================
print("=== 场景3:零样本CoT ===")
zero_shot_cot_prompt = """
让我们一步步思考这个问题,确保每一步都准确:
问题:小明从家到学校需要走15分钟,速度是每分钟80米;放学时速度减慢到每分钟60米,放学回家需要走多久?
"""
zero_shot_cot_result = call_llm_with_cot(zero_shot_cot_prompt)
print(zero_shot_cot_result)
print("-" * 70)
# ===================== 场景4:复杂任务拆解(非数学题) =====================
print("=== 场景4:复杂任务拆解CoT ===")
task_cot_prompt = """
请用思维链的方式拆解「开发一个本地文档问答机器人」的任务,要求:
1. 先明确核心目标;
2. 拆解成5个以内的核心步骤;
3. 每个步骤说明关键动作和注意事项。
"""
task_cot_result = call_llm_with_cot(task_cot_prompt, temperature=0.5)
print(task_cot_result)

import requests
import json
# ===================== 配置项 =====================
OLLAMA_URL = "http://127.0.0.1:11434/api/generate"
MODEL_NAME = "qwen3:4b"# 本地已下载的模型
TEMPERATURE = 0.5 # 生成多个思路时适度增加随机性
# ===================== 基础调用函数 =====================
def call_ollama(prompt, temperature=0.5):
"""调用本地Ollama模型,返回生成结果"""
data = {
"model": MODEL_NAME,
"prompt": prompt,
"temperature": temperature,
"stream": False,
"max_tokens": 1000# 限制生成长度,避免冗余
}
try:
response = requests.post(OLLAMA_URL, json=data)
response.raise_for_status()
return response.json()["response"].strip()
except Exception as e:
returnf"调用失败:{str(e)}"
# ===================== ToT核心函数 =====================
def tree_of_thoughts(problem, max_depth=3, top_k=2):
"""
思维树(ToT)核心实现
:param problem: 待解决的复杂问题
:param max_depth: 思维树最大深度(中间步骤数)
:param top_k: 每一步保留的最优思路数
:return: 最优解 + 完整推理路径
"""
# 存储完整的思维树(用于回溯和展示)
thought_tree = {
"root": problem,
"nodes": [], # 每个节点格式:{"depth": 深度, "thought": 思路, "score": 评分, "children": []}
}
def generate_thoughts(current_problem, depth):
"""生成当前步骤的多个可能思路"""
prompt = f"""
你需要解决的问题:{current_problem}
当前推理深度:{depth}/{max_depth}
要求:
1. 生成{top_k+1}个可能的解题思路(中间步骤);
2. 每个思路简洁明了(不超过50字);
3. 思路之间要有明显差异(不同的推理方向);
4. 仅输出思路列表,格式:1. 思路1\n2. 思路2\n...
"""
thoughts_str = call_ollama(prompt, temperature=TEMPERATURE)
# 解析思路列表
thoughts = []
for line in thoughts_str.split("\n"):
if line.strip().startswith(("1.", "2.", "3.", "4.", "5.")):
thought = line.strip().split(".", 1)[1].strip()
if thought:
thoughts.append(thought)
return thoughts[:top_k] # 取前top_k个思路
def evaluate_thoughts(thoughts, current_problem):
"""评估思路的可行性,返回(思路,评分)列表(评分1-10,越高越可行)"""
prompt = f"""
待解决问题:{current_problem}
需要评估的思路列表:
{chr(10).join([f"{i+1}. {t}" for i, t in enumerate(thoughts)])}
要求:
1. 为每个思路评分(1-10分,1=完全不可行,10=最优);
2. 评分理由简洁(不超过30字);
3. 仅输出评估结果,格式:
1. 思路1:评分(理由)
2. 思路2:评分(理由)
...
"""
eval_str = call_ollama(prompt, temperature=0.1) # 低温度保证评估严谨
# 解析评分
evaluated = []
for line in eval_str.split("\n"):
if line.strip().startswith(("1.", "2.", "3.", "4.", "5.")):
parts = line.strip().split(".", 1)[1].strip().split(":", 1)
if len(parts) == 2:
thought = parts[0].strip()
score_part = parts[1].split("(", 1)[0].strip()
try:
score = int(score_part)
evaluated.append((thought, score))
except ValueError:
continue
# 按评分排序(降序)
evaluated_sorted = sorted(evaluated, key=lambda x: x[1], reverse=True)
return evaluated_sorted
def expand_tree(current_problem, depth):
"""递归扩展思维树,返回最优思路和最终答案"""
# 终止条件:达到最大深度,生成最终答案
if depth >= max_depth:
final_prompt = f"""
基于当前思路解决问题:{current_problem}
要求:给出最终答案,简洁明了(不超过100字)。
"""
final_answer = call_ollama(final_prompt, temperature=0.1)
return final_answer, []
# 步骤1:生成当前步骤的多个思路
thoughts = generate_thoughts(current_problem, depth)
ifnot thoughts:
return"无法生成有效思路", []
# 步骤2:评估思路,选最优的top_k个
evaluated_thoughts = evaluate_thoughts(thoughts, current_problem)
ifnot evaluated_thoughts:
return"无法评估思路", []
# 步骤3:递归扩展最优思路的子节点
best_thought, best_score = evaluated_thoughts[0]
# 记录当前节点到思维树
thought_tree["nodes"].append({
"depth": depth,
"thought": best_thought,
"score": best_score,
"children": []
})
# 构建下一级问题(基于最优思路)
next_problem = f"基于思路「{best_thought}」,继续解决原问题:{problem}"
# 递归扩展
final_answer, child_paths = expand_tree(next_problem, depth + 1)
# 记录子路径
thought_tree["nodes"][-1]["children"] = child_paths
# 返回当前路径+最终答案
path = [f"深度{depth}:{best_thought}(评分{best_score})"] + child_paths
return final_answer, path
# 启动思维树推理
final_answer, reasoning_path = expand_tree(problem, depth=1)
return {
"final_answer": final_answer,
"reasoning_path": reasoning_path,
"thought_tree": thought_tree
}
# ===================== 测试ToT功能 =====================
if __name__ == "__main__":
# 示例:复杂逻辑谜题(适合ToT的典型场景)
complex_problem = """
逻辑谜题:有5个盒子,分别标有1-5号,每个盒子里装着红、黄、蓝三种颜色中的一种球,满足以下条件:
1. 1号盒子不是红球;
2. 2号盒子和4号盒子颜色相同;
3. 3号盒子是蓝球;
4. 5号盒子不是黄球;
5. 红球数量比黄球多1个。
请推理每个盒子里球的颜色,并说明推理过程。
"""
# 执行ToT推理
print("=== 思维树(ToT)推理开始 ===")
print(f"待解决问题:{complex_problem.strip()}")
print("-" * 80)
result = tree_of_thoughts(
problem=complex_problem,
max_depth=3, # 3层推理深度
top_k=2 # 每步保留2个最优思路
)
# 输出结果
print("=== 推理路径(思维树)===")
for idx, step in enumerate(result["reasoning_path"]):
print(f"{idx+1}. {step}")
print("-" * 80)
print("=== 最终答案 ===")
print(result["final_answer"])


往期推荐
本文首发于微信公众号 BigDataLab,转载请注明出处。
更多 AI / 大模型实战干货,欢迎微信搜索关注「BigDataLab」,或 点此回看原文。
作者:渔单渠 微信搜索“BigDataLab”
本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利.

浙公网安备 33010602011771号