Jupyter推出免費AI助手，不只會寫代碼，多種大模型都能調用
本文經AI新媒體量子位（公眾號ID:QbitAI）授權轉載，轉載請聯系出處。
自然語言編程的家族，又添新成員！
近日，Jupyter宣布了推出免費開源的AI助手。
安裝簡單、使用方便，寫代碼、debug，一個窗口就能輕松搞定！
這個叫做Jupyter AI的編程神器是用LangChain打造的。
相比于VSCode里的Copilot，Jupyter AI支持的大模型更多。
除了我們耳熟能詳的ChatGPT和Claude之外，還有來自AWS、Cohere等廠商的模型。
甚至只要有API，HuggingFace上的模塊它都能調用，這意味著它的作用不僅僅是編程。
此外，Jupyter AI的配置也更為簡便，而且無需排隊、支持中文prompt。
支持的程序也不只是Jupyter自己，只要環境支持notebook或IPython就可以用，VSCode也包括在內。
這一重磅產品的推出，讓網友直言游戲規則要改變了：
我們進行了第一手實測，快來看看結果如何！
會寫又會跑，還能debug
安裝好帶AI插件的Jupyter Lab并打開之后，新建一個筆記文件。
然后輸入「%load_ext jupyter_ai」對插件進行加載。
之后是對API key進行配置，以ChatGPT為例，使用的命令是：
%env OPENAI_API_KEY=（填入OpenAI的API key）
其他大模型的配置方法也都相似。
完成這些配置之后，就可以上手寫代碼了，輸入內容的格式如下：
%%ai 模型名稱 --format=(code/math/html/image…)
要完成的任務
其中任務用中文描述就可以，format參數也可以不輸入。
我們從算法、Python繪圖以及小游戲三個方面對Jupyter AI進行了測試。
先來看算法，這里我們直接上難度，來一道困難級別的Leetcode題。
LeetCode平臺上，這道題的通過率為30.7%。
結果代碼很快就出現在了筆記的下方：
那么究竟對不對呢，我們在LeetCode上提交一下試試。
直接一步到位！
如果換成一道數學題呢？
老鐵，依舊沒毛病！
算法問題就展示到這里，接下來讓我們看看它的繪圖能力。
首先讓Jupyter AI隨便畫個什么圖形出來，比如兩顆心。
結果我們得到了一段調用海龜倉庫進行繪圖的代碼。
這里就要提到這款插件的另一個特性——生成好的代碼可以直接運行。
除了圖形，函數圖像的繪制也不是問題。
依舊是整個復雜的，y=x*sin[ln(x)]。
在這個區間之內的確沒毛病，不過沒能完整展示函數的單調性。
沒關系，我們可以對AI給出的代碼進行手工調整，改變一下取值范圍。
根據現有數據繪制統計圖表同樣是小菜一碟。
運行這段代碼，一張統計圖就大功告成了。
不過，這里我們還想展示一下它的多輪對話能力。
比如，這個藍色看起來不好看，我們想給它換個顏色。
除了算法和繪圖這些內容，交互式程序的編寫也難不倒Jupyter AI。
甚至可以做小游戲，比如貪吃蛇。
創制和運行程序的能力我們都已經見過了，那么debug的表現又如何呢？
我們讓ChatGPT幫忙出了道題——一段包含五處錯誤的Python代碼。
我們看到，ChatGPT生成的這五處錯誤全都沒能逃過插件的法眼。
除了修改操作，也可以要求它解釋修改的原因：
我們看到，or改成and的原因沒有進行解釋，不過沒關系，我們可以進行追問。
結果得到了更為詳細的解釋：
以上就是關于Jupyter AI編寫代碼和調試運行的全部測試了。
但它會的，不只是編程。
不只是編程
除了和Python程序相關的功能之外，這款AI插件還可以生成LaTex公式、SVG圖形、網頁等內容。
首先來看公式，我們想讓它生成一下麥克斯韋方程組。
只要把format的參數改成math，結果就會以LaTex排版之后的格式輸出。
為了證明這不是圖片，我們查看了相關區域的HTML代碼，是貨真價實的LaTex格式。
簡單的網頁，這個插件也能做得出來。
我們讓它生成一個時鐘頁面，這里需要把format改成html。
說到HTML，就會發現它還有一個妙用。
編程的一節提到過，這個AI插件可以生成Python繪圖代碼。
但它同樣可以不用Python，直接生成一些簡單圖形，以SVG的形式進行展示。
不過無論是Python還是SVG，畫出的圖形都比較簡單。
那么復雜的圖像它能畫嗎？答案是肯定的。
這款插件可以調用Stable Diffusion來畫圖。
不過由于沒有HuggingFace的API，這里我們只能展示一下官方給出的DEMO：
這里似乎需要把format設為image。
而且能調用的不只是Stable Diffusion，HuggingFace中的其他模型也可以！
除了在筆記中調用插件，Jupyter AI還在側邊欄中提供了聊天窗口。
配置好API之后，有什么問題直接問就可以！
還不趕快上手體驗一下？(GitHub頁面有安裝教程)
GitHub頁面：https://github.com/jupyterlab/jupyter-ai
