三亩地 三亩地SAN MU DI · CODE DIARY
ARTICLE DETAIL

日记详情

真实记录编程学习的某一天,欢迎挑你感兴趣的翻一翻。

如何在vs code使用Jupyter

如何在vs code使用Jupyter

在 VS Code 中运行 Jupyter Notebook 代码,主要分为以下几步:

第一步:安装必要扩展

在 VS Code 扩展市场(Ctrl+Shift+X)中搜索并安装Microsoft 官方发布的 Jupyter 扩展(ID:ms-toolsai.jupyter)。

安装后重启 VS Code。

第二步:配置 Python 环境

在你的虚拟环境中安装必要的包:

pip install jupyter ipykernel

然后注册内核(以你的 your-project 项目为例):

python-m ipykernel install--user--name your-project--display-name"Python (your-project)"

第三步:创建或打开 Notebook

有两种方式:

  • 新建:按Ctrl+Shift+P打开命令面板,输入Create: New Jupyter Notebook,或直接新建文件保存为.ipynb后缀
  • 打开已有:直接在 VS Code 中打开.ipynb文件,会自动以 Notebook 编辑器形式加载

第四步:选择内核

打开 Notebook 后,点击右上角的“Select Kernel”按钮,选择你刚才注册的Python (your-project)内核。

第五步:运行代码

操作快捷键
运行当前单元格Shift+Enter(运行并跳到下一个)
运行当前单元格(不跳转)Ctrl+Enter
运行当前单元格并插入新单元格Alt+Enter
运行所有单元格顶部工具栏的 “Run All” 按钮
中断执行顶部红色方块按钮

针对你的 LangGraph 项目

如果你想把之前的 LangGraph 代码放到 Notebook 里运行,可以新建一个langgraph_demo.ipynb,然后把代码拆成多个单元格:

单元格 1(环境初始化):

fromdotenvimportload_dotenv load_dotenv()fromlangchain_openaiimportChatOpenAIfromlanggraph.graphimportStateGraph,STARTfromlanggraph.graph.messageimportadd_messagesfromtypingimportAnnotatedfromtyping_extensionsimportTypedDictfromIPython.displayimportImage,display

单元格 2(定义模型和图):

model=ChatOpenAI(model="**",temperature=0,timeout=60,max_retries=3,)classState(TypedDict):messages:Annotated[list,add_messages]graph_builder=StateGraph(State)defchatbot(state:State):return{"messages":[model.invoke(state["messages"])]}graph_builder.add_node("chatbot",chatbot)graph_builder.add_edge(START,"chatbot")graph=graph_builder.compile()

单元格 3(可视化):

try:display(Image(graph.get_graph().draw_mermaid_png()))exceptExceptionase:print(f"可视化生成失败:{e}")

单元格 4(对话测试):

user_input="What do you know about LangGraph?"print("User:",user_input)foreventingraph.stream({"messages":[{"role":"user","content":user_input}]}):forvalueinevent.values():print("Assistant:",value["messages"][-1].content)

这样每个单元格可以独立运行和调试,比在.py文件里跑方便很多。


运行之后就可以看到下面这种的结构:

← 返回列表