在Jupyter中使用AI寫代碼,如有神助,太驚豔了

昨晚看到一個可以在JupyterLab中使用的AI代碼輔助工具jupyter-ai,它的交互確實非常棒,可以直接聊天,也可以就筆記中的代碼提問,最出彩的是生成筆記功能,還是蠻驚豔的。

這裏就極簡介紹一下安裝及用法

第一步是創建環境並安裝

conda create -n jupyter-ai python=3.11
conda activate jupyter-ai
pip install jupyter_ai

Jupyter AI 支持很多模型,這裏就以使用最廣泛的openai爲例

要使用 OpenAI 模型,請安裝openai的包,並在從終端啓動 JupyterLab 時設置環境變量:

pip install openai
OPENAI_API_KEY="這裏填api" jupyter lab

啓動之後界面是這樣

聊天界面

Jupyter AI 的最簡單方法是使用聊天界面,單擊左側面板中的新“聊天”圖標以打開聊天界面。

當你第一次打開聊天界面時,Jupyter AI 會詢問你想要使用哪些模型作爲語言模型和嵌入模型。做出選擇後,UI 可能會顯示一個或多個設置鍵的文本框。如果要切換模型,可以點擊右上角的齒輪。可以選擇語言模型,也可以選擇嵌入模型。您應該選擇其中之一,以便可以使用聊天界面的完整功能。

設置完所有必要的按鍵後,單擊 Jupyter AI 側面板左上角的“後退”(左箭頭)按鈕。現在出現聊天界面,可以使用底部的消息框提問。

詢問筆記本中的內容

Jupyter AI 的聊天界面可以在提示中包含筆記本的一部分。

比如選中代碼後在聊天界面提問代碼的用途

可以將 Jupyternaut 的響應複製到剪貼板,以便將其粘貼到筆記本或任何其他應用程序中。

還可以在發送消息之前單擊“替換選擇”,選擇用 Jupyternaut 的回答替換原內容。

連複製粘貼都省了!

生成一個新筆記本

可以使用 Jupyter AI 根據文本提示生成整個筆記本。

首先,打開聊天面板,然後向其發送一條以/generate開頭的消息

當 Jupyter AI 生成筆記本後,它會向向發送另一條消息,其中包含其生成的文件名。這裏沒有那麼智能哈,你需要手動打開該文件。

需要注意:可能需要幾分鐘才能生成筆記本,在此期間,你仍然可以像平常一樣使用 JupyterLab 和 Jupyter AI。

Jupyter AI 工作時請勿關閉 JupyterLab 實例。

總結

jupyter ai 總體還是蠻好用的,本文介紹的也是最常用的。

還有一些其他玩法比如/ask 還有 /learn 大家可以多試試。

還有 %ai 魔法命令在筆記本中使用,就是在將第一個提示發送到 AI 模型之前,通過在筆記本單元或 IPython shell 中運行以下代碼來加載 IPython 擴展:

%load_ext jupyter_ai_magics

這裏就不做過多介紹,感興趣可以看下官方文檔,個人感覺使用體驗不太好。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章