跟着Datawhale打卡·【ai-agent-book:从入门到 Agent 工程实践】学习笔记Task1

简介: 本节学习第二章“上下文工程”:从环境搭建(Homebrew→Ollama→Python依赖)到Qwen模型调用与升级;再通过Attention可视化实验,深入理解Transformer中Attention Sink机制——模型如何将冗余注意力锚定首token以稳定推理,并解析热力图层/头级注意力分布。(239字)

今天学习第二章 ·上下文工程

实验2-1是上下文工程,本期目标是:先安装统一 ch2 环境 → 启动 Ollama → 普通对话测试 → 提天气请求,观察完整工具调用链路(流式输出)

准备工作要做:step1:安装 Homebrew、再装 ollama

/bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"

安装结束后,一定要执行下面两行,把 brew 加到环境变量(M 系列 Mac 必做),确认好我们的环境有没有get ready。

echo 'eval "$(/opt/homebrew/bin/brew shellenv)"' >> ~/.zshrc
source ~/.zshrc

step2:验证我们已经安装好honebrew之后就可以执行下面这个命令,新开一个终端窗口安装ollama还有python的sgk包

brew install ollama
python3 -m pip install ollama #python代码import ollama用的库

step3:到这步之后你就可以打开你的第三个终端窗口,一个是Ollama 后台服务(这是常驻服务不要关掉),第二个是我们之前下载 ollama/pull 模型的终端,第三个就是安装我们试验要用的依赖环境,mac需要提前安装好homebrew

在仓库根目录 ai-agent-book里一次性安装第二章全部依赖,创建.venv

python3 -m pip install -e ".[ch2]"

安装好上述环境之后,就可以运行我们的试验

python3 -m pip install -r requirements.txt
python3 main.py --backend ollama

就顺利连上我们的Qwen拉!!!但是还没完,api调用告诉我模型能力不足,那么我们就升级!!!

吧模型拉到可以支持thinking的地步(运行到这步对ollama有了比较切实的认识)

ollama pull qwen3:0.6b

对话实验成功!!!

实验2-2是注意力机制可视化

首先进入到我们的【attention_visualization】目录里

cd chapter2/attention_visualization

注,这里咋们就不用跟着网页里的环境搭建再跑一遍全部命令了,我就是重跑了一遍吧所有环境依赖包都清空了(因为当时跑2-2实验我有做强制退出的动作),所以咋们就直接进去目录,运行代码就可以了

其次,我们要用transformers 库从 HuggingFace 下载模型,可以选择在国内镜像下载 Qwen3-0.6B 模型

这里又学到一个新名词【HuggingFace(简称 HF)】,他是全球最大的开源 AI 模型托管平台,相当于 "AI 界的 GitHub"。我们的试验要跑本地代码所以得用HF吧模型文件取下来

echo 'export HF_ENDPOINT=https://hf-mirror.com' >> ~/.zshrc
source ~/.zshrc

随后就可以跑可视化代码了

python3 attention_cli.py

到这一步,咋们又刷新了一个新的名词【attention sink】,他的概念就是LLM 会无意识地把大量注意力 "倾倒" 到句子第一个 token 上,作为信息 "缓冲锚点"。之后用代码跑出来的可视化图片就会保存在当前目录的【attention_heatmap.png 】,下图就是大模型底层 Transformer 的可视化

分析热力图:1)y轴的含义是发出注意力的 Token,每一行=一个token。x轴的含义是被关注的token,每一列=一个token,这一列代表:这个位置被前面哪些 token 关注 2)图片左上叫是<|im_start|>标记,这就是 Attention Sink。模型会强制保留一部分注意力在序列最起始的占位 token 上,当作一个 “信息锚点”,方便后续长文本的信息回溯,哪怕这个标记本身没有业务语义。也代表颜色亮的区域我们应该着重去关注 3)顺着对角线能看到一个由上到下蓝色的斜线图标,这部分的含义就是,每个token会更关注与他前面邻近的token3)左半边的深紫色三角形的含义是注意力权重≈0,也就是代表这两个 token 之间几乎没有关联,模型推理时完全不看。

补充说明:Attention Sink是什么?模型在预训练过程中学到:把多余、无处安放的注意力,全部丢给第 0 号 token 是最安全的选择,不会干扰语义计算。于是就形成了 sink。

那他的作用是什么?1)模型内部:稳定表示,防止信息过度混杂,避免因为数据噪点出现模型混乱的问题 2)工程推理层面:长上下文、KV 缓存优化。大模型推理会保存 KV Cache(每一个 token 的 K、V 向量),上下文越长内存占用越大。

有了初步能看懂图片的能力之后,我们继续往下走

python3 attention_cli.py --prompt "哈密 的 天气 怎么样" --layer 0 --head 3 --output layer0_head3.png

这张图可以明显的看到sink几乎消失,仅占3.5%。因为我们这次的代码写的是【layer0 head3】,这个头本职任务就是跟踪相邻词语、语序,它天生几乎不会把注意力分配给开头的<|im_start|>符号。写第一张图的代码他的含义就是不管我们些什么问题,llm的注意力都是会放在第一行的符号上,第二章图片的代码他的含义就是llm的注意力都会放在前一个词上面--head x 他的意思就是看某一个单独prompt的行为。

相关文章
|
2天前
|
人工智能 前端开发 API
跟着Datawhale打卡学习·【ai-agent-book:从入门到 Agent 工程实践】学习笔记
本课程手写笔记学习AI Agent原理,基于[ai-agent-book](https://github.com/bojieli/ai-agent-book/tree/main)第一章,实操搭建本地大模型环境(Python 3.11+),对比离线/联网调用Kimi-k3,深入理解ReAct推理循环、Context流转与工具协同机制。
|
2天前
|
人工智能 JSON 数据格式
把访谈逐字稿写成「姜Dora」风格的文章:我的过程与思考
这是一份真实记录:作者以姜Dora为范本,拆解6篇访谈成稿,构建可复用的AI写作Skill(`jiangdora-article-writer`),涵盖来源核验、逐篇分析、方法论迭代与全流程试跑,并深入反思表达精度、token管理、格式流转、知识沉淀等关键边界与踩坑经验。
|
24天前
|
人工智能 JSON API
全网刷屏的 Jev 模型正式开放!一手实战测评 + 保姆级教程
全网爆火的 Jev 模型是什么?有什么用?怎么使用?怎么接入 AI 编程工具?效果真的好么?傻子可懂的 Jev 保姆级实战教程 + 项目实战测评来啦
9439 31
|
22小时前
|
云安全 人工智能 运维
当攻击全面Agent化,你的安全防御能否跟上AI Speed?
多起客户侧AK泄露事件应急的真实观察
1153 0
|
10天前
|
存储 人工智能 并行计算
大模型本地部署终端选型方法论:以 Qwen3.8-27B 为例的四档分层完整流程
本文提出一套大模型本地部署终端选型方法论:定约束、定档位、定框架、定参数四步决策法,配合入门、主力、质量、无损四档分层模型。以 Qwen3.8-27B 实测数据为例,逐环节解读显存、带宽、存储、散热、系统、预算等要素,给出面向不同预算的优选方案、决策自查清单与市场观察框架。文末前瞻 AI 笔记本的 CPU+GPU 与统一内存两条路线,论证四步决策法在新品类上的延续性。
|
8天前
|
人工智能 JSON 自然语言处理
2026 年 Jev 决策模型深度拆解:原理解读、实战测评与保姆级落地教程
有一款特殊AI模型在开发者圈子刷屏,它摒弃传统大模型擅长的对话聊天能力,专注做高速结构化决策,它就是TypeSafe AI推出的Jev模型。该模型由ChatGPT共同发明人Diogo Almeida主导研发,定位为**System One Model(系统一模型)**,对标人类大脑快速直觉判断的思维模式,在响应延迟、调用成本、结构化输出稳定性上相比传统生成式大模型有着巨大差异。本文会完整拆解Jev底层原理、三大核心原语能力、适用业务场景,同时提供可直接运行的curl、Python代码示例,并且结合多组实测数据,客观分析模型优势与能力边界,帮助普通开发者和AI应用从业者快速上手落地。
687 1
|
5天前
|
人工智能 JSON 自然语言处理
2026 年 Jev 决策模型全解析:核心原理拆解,性能实战测评与落地部署教程
有一款特殊AI模型在开发者圈子刷屏,它摒弃传统大模型擅长的对话聊天能力,专注做高速结构化决策,它就是TypeSafe AI推出的Jev模型。该模型由ChatGPT共同发明人Diogo Almeida主导研发,定位为System One Model(系统一模型),对标人类大脑快速直觉判断的思维模式,在响应延迟、调用成本、结构化输出稳定性上相比传统生成式大模型有着巨大差异。本文会完整拆解Jev底层原理、三大核心原语能力、适用业务场景,同时提供可直接运行的curl、Python代码示例,并且结合多组实测数据,客观分析模型优势与能力边界,帮助普通开发者和AI应用从业者快速上手落地。
451 0
|
22天前
|
人工智能 测试技术 API
最近全网爆火的 Jev 到底是什么?适合干什么、怎么用,一篇讲透!
Jev是TypeSafe AI推出的“系统一模型”,不生成文本,专做毫秒级结构化决策:Choice(多选)、Score(打分)、Noul(是非概率)。响应快193倍、成本低444倍,适合工单路由、内容审核、测试定级等高频判断场景。
2519 4
最近全网爆火的 Jev 到底是什么?适合干什么、怎么用,一篇讲透!