GenerativeAgentsCN下載 - GenerativeAgentsCN原始碼下載

GenerativeAgentsCN

其他源碼

1.0.0

下載

簡體中文| English

生成式智能體（Generative Agents）深度漢化版

史丹佛AI小鎮由史丹佛大學和谷歌於2023年8月開源，由25個智能體組成的虛擬世界，模擬了真實的人類生活。

25個智能體完全由ChatGPT驅動，自主組織派對、參加會議、在情人節籌辦各種活動。他們能夠展現出與人類相似的生活模式和行為習慣。

Generative Agents的原始程式碼工程化程度較低，難以持續維護或拓展功能，且時隔一年多，中文LLM的能力也已勝任此類任務。因此，我們對原項目進行了重構+深度漢化，旨在為中文用戶提供一個利於維護的基礎版本，以便後續實驗或嘗試更多玩法。

wounderland專案是原始Generative Agents專案的重構版本，結構良好且程式碼品質遠優於原版，因此本專案基於wounderland開發。

主要工作：

重寫全部提示語，將智能體的「母語」切換為中文，以便對接Qwen-2.5或GLM-4等中文模型；
針對中文特色及Qwen2.5-7b/14b模型的能力，優化中文提示語及智能體之間的對話起止邏輯；
所有提示語模板化，便於後製；
修正原版的小問題（例如wounderland原版中智能體在入睡後便不再醒來）；
增加對本地Ollama API的支持，同時將LlamaIndex embedding也連接Ollama，實現完全本地部署，降低實驗成本。 Ollama安裝及配置可參考ollama.md ；
增加「斷點恢復」等特性；
回放介面基於原Generative Agents前端程式碼精簡，同時將智能體活動的時間軸及對話內容儲存至Markdown文件。

回放畫面：

snapshot

註：地圖及人物名稱也同步漢化，是為了避免LLM在遇到中英混雜的上下文時，切換到英文語境。

1. 準備工作

1.1 取得代碼：

 git clone https://github.com/x-glacier/GenerativeAgentsCN.git
cd GenerativeAgentsCN

1.2 配置大語言模型（LLM）

修改設定檔generative_agents/data/config.json :

預設使用Ollama載入本地量化模型，並提供OpenAI相容API。需先拉取量化模型（參考ollama.md），並確保base_url和model與Ollama中的配置一致。
如果希望呼叫其他廠商的API，需要在api_keys中填入對應的key，並根據API文件修改base_url和model 。

1.3 安裝python依賴

建議先使用anaconda3建立並啟動虛擬環境：

 conda create -n generative_agents_cn python=3.11
conda activate generative_agents_cn

安裝依賴：

 pip install -r requirements.txt

2. 運行虛擬小鎮

 cd generative_agents
python start.py --name sim-test --start "20240213-09:30" --step 10 --stride 10

參數說明:

name - 每次啟動虛擬小鎮，需要設定唯一的名稱，以便事後重播。
start - 虛擬小鎮的起始時間。
resume - 在運行結束或意外中斷後，從上次的「斷點」處，繼續運行虛擬小鎮。
step - 在迭代多少步驟之後停止運行。
stride - 每一步迭代在虛擬小鎮中對應的時間（分鐘）。假如設定--stride 10 ，虛擬小鎮在迭代過程中的時間變化將會是9:00，9:10，9:20 ...

3. 回放

3.1 產生回放數據

 python compress.py --name <simulation-name>

運行結束後將在results/compressed/<simulation-name>目錄下產生回放資料檔movement.json 。同時也將產生simulation.md ，以時間軸方式呈現每個智能體的狀態及對話內容。

3.2 啟動回放服務

 python replay.py

透過瀏覽器開啟回放頁面（網址： http://127.0.0.1:5000/?name=<simulation-name> ），可以看到虛擬小鎮中的居民在各個時段的活動。

可透過方向鍵移動畫面

參數說明

name - 啟動虛擬小鎮時設定的名稱。
step - 回放的起始步數，0代表從第一幀開始回放，預設值為0。
speed - 回放速度（0-5），0最慢，5最快，預設值為2。
zoom - 畫面縮放比例，預設值為0.8。

發布版本中內建了名為example的回放資料（由qwen2.5:32b-instruct-q4_K_M產生）。若希望以較快速度從頭開始回放，畫面縮放比例為0.6，則對應的url為：http://127.0.0.1:5000/?name=example&step=0&speed=2&zoom=0.6

也可直接開啟simulation.md，查看example中所有人物活動和對話訊息。