rm notebooks

games/angrybirds/notebooks/ai_v3.0.ipynb

@@ -1,663 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "id": "245fbc34-352d-4945-a689-d04f9365bb99",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "%load_ext autoreload\n",
-    "%autoreload 2"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "id": "eaed216c-160a-4e31-96fe-015c140ff18b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# from pprint import pprint"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "id": "97953551-8ef1-42c1-9317-99d5e65f535e",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import json\n",
-    "import pandas as pd\n",
-    "\n",
-    "import openai\n",
-    "from openai.embeddings_utils import get_embedding, cosine_similarity\n",
-    "\n",
-    "# from games.angrybirds.util import util\n",
-    "from games.angrybirds.config import config\n",
-    "# from games.angrybirds.prompt import prompt\n",
-    "# from games.angrybirds.query import query, yes_no_query\n",
-    "from games.angrybirds.angrybirds import angrybird"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "fc213f4c-cb07-468b-b15f-60e5120b3d94",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 88,
-   "id": "0c876614-a9bf-4d0b-a728-07770f1ad9c5",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "bird = angrybird()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 89,
-   "id": "1411de96-97d5-4bac-b8ed-e279fb1edaa0",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "0"
-      ]
-     },
-     "execution_count": 89,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 93,
-   "id": "163545c6-c707-401a-93b8-fb2cb9d65136",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "initializing...\n",
-      "loading embeddings from a file...\n",
-      "/Users/shaojun/playgo/abc123/games/angrybirds/../../embeddings.pkl\n",
-      "pkl size: 72\n",
-      "embedding size: 74\n",
-      "creating embeddings by query OpenAI...\n"
-     ]
-    },
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "  0%|                                                    | 0/76 [00:00<?, ?it/s]"
-     ]
-    },
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "query Change background volumn...\n"
-     ]
-    },
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "100%|██████████████████████████████████████████| 76/76 [00:00<00:00, 164.48it/s]"
-     ]
-    },
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "embedding size: 75\n"
-     ]
-    },
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "\n"
-     ]
-    }
-   ],
-   "source": [
-    "bird.init_emb()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 94,
-   "id": "31c02b28-0078-425d-87b6-702caeea02e3",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "75"
-      ]
-     },
-     "execution_count": 94,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 95,
-   "id": "9389ebb2-de88-4c23-95ee-c8beb83af600",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import pickle"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "808c1f89-2b7e-4134-a73e-94689f4cce75",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 96,
-   "id": "02478e9b-dd11-406a-b405-ea4ee56975b1",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import os"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 97,
-   "id": "7b506b8c-0b60-44e4-bb6e-f4af8439f04f",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'/Users/shaojun/playgo/abc123'"
-      ]
-     },
-     "execution_count": 97,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "root = os.environ['PYTHONPATH']\n",
-    "root"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "30fb4dd7-35d8-442f-bd6e-29e15374e3d6",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 98,
-   "id": "c2b17478-f627-4ad4-8efb-aa20690a312d",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'/Users/shaojun/playgo/abc123/games/angrybirds/notebooks'"
-      ]
-     },
-     "execution_count": 98,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "pwd"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 99,
-   "id": "dd8c489e-cd3f-4df6-81c0-fe3fd2cc0d33",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'{root}/embeddings.pkl', 'wb') as f:\n",
-    "    pickle.dump(bird.embeddings, f)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 100,
-   "id": "07032487-3fe7-4729-8ca0-c57d6a485c28",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'{root}/embeddings.pkl', 'rb') as f:\n",
-    "    tmp = pickle.load(f)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 101,
-   "id": "ab7ec218-0e02-4a2b-9c16-ec26704b3519",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "True"
-      ]
-     },
-     "execution_count": 101,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "tmp == bird.embeddings"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 102,
-   "id": "bc896434-21ae-438c-90b1-6b590103e731",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "75"
-      ]
-     },
-     "execution_count": 102,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(tmp)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 103,
-   "id": "9532b604-b7d6-4bd4-a0b9-c8c36470447b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "75"
-      ]
-     },
-     "execution_count": 103,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 41,
-   "id": "18888a7b-d32d-45b4-a9b3-3ce29b18a7f7",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# sim query"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 42,
-   "id": "4703a269-a1c0-48e4-8848-6b5cdb9b814b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "68"
-      ]
-     },
-     "execution_count": 42,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "567ca6d0-9291-4876-a3a2-77b12576f7f8",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "de48b250-651c-4183-ad41-506f891d4a7d",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 113,
-   "id": "5bf13de5-bc24-4804-8a7b-4fac0bef8799",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "0.9333 14/15\tn\n",
-      "[('n', 14), ('y', 1)]\n"
-     ]
-    },
-    {
-     "data": {
-      "text/plain": [
-       "'n'"
-      ]
-     },
-     "execution_count": 113,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "yes_no_query(\n",
-    "    bird, \n",
-    "    input_command, \n",
-    "    \"The game developer mentioned the number of birds he/she wants to add\"\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 110,
-   "id": "07fd3efa-ab40-48fb-8cc0-9cb49b9486e4",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "0.5333 8/15\ty\n",
-      "[('y', 8), ('n', 7)]\n"
-     ]
-    }
-   ],
-   "source": [
-    "s = yes_no_query(\n",
-    "    bird,\n",
-    "    input_command, \n",
-    "    \"\"\"\n",
-    "The game developer mentioned big bird, \n",
-    "magnet bird, \n",
-    "superheavy bird,\n",
-    "catapult bird,\n",
-    "flash bird,\n",
-    "antigravity bird.\n",
-    "\"\"\"\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "549675ad-dadb-48c1-a72a-abc729374c25",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 105,
-   "id": "26ac79de-dcda-4fae-99b6-a755667d7340",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'add six pig'"
-      ]
-     },
-     "execution_count": 105,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "input_command = 'add six pig'\n",
-    "input_command"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 106,
-   "id": "963beef6-296d-446a-a9fe-d741faf8712b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['OK', ['pig', 'add', {}], 'Add a pig.', 0.978199749271284]"
-      ]
-     },
-     "execution_count": 106,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.query_emb('add a pig')"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 107,
-   "id": "f24b37cd-5d03-4395-8a35-d06e0714c42a",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['OK',\n",
-       " ['pig', 'add', {'quantiy': 6}],\n",
-       " 'Please add six pigs.',\n",
-       " 0.9358702373920165]"
-      ]
-     },
-     "execution_count": 107,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.query_emb('add six pig')"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 108,
-   "id": "003a4073-81a2-4e64-879e-3b43ffc37042",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['OK',\n",
-       " ['change background', {'args': 'picture'}],\n",
-       " 'Change background color',\n",
-       " 1.0000000000000002]"
-      ]
-     },
-     "execution_count": 108,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.query_emb('Change background color')"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 109,
-   "id": "1c782904-f139-4db4-b792-a012fb104d1b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['OK',\n",
-       " ['change background', {'args': 'picture'}],\n",
-       " 'Change background picture',\n",
-       " 0.9748124378288634]"
-      ]
-     },
-     "execution_count": 109,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.query_emb('Change background photo')"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 110,
-   "id": "d5388d71-bb51-412d-b21e-78816163c4b0",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['OK',\n",
-       " ['change background', {'args': 'picture'}],\n",
-       " 'Change background picture',\n",
-       " 1.0000000000000002]"
-      ]
-     },
-     "execution_count": 110,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.query_emb('Change background picture')"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 111,
-   "id": "58c6b518-8aad-467b-8a1b-757f17a1a2e3",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['OK',\n",
-       " [['change background', {'args': 'picture'}],\n",
-       "  ['change background', {'args': 'music'}]],\n",
-       " 'Change background',\n",
-       " 0.9996814755558403]"
-      ]
-     },
-     "execution_count": 111,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.query_emb('Change background')"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 112,
-   "id": "658907ca-27ba-4c00-9cfa-3abec9e15019",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['OK',\n",
-       " [['change background', {'args': 'music'}],\n",
-       "  ['change background', {'args': 'volumn'}]],\n",
-       " 'Change background sound',\n",
-       " 0.931697521066715]"
-      ]
-     },
-     "execution_count": 112,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.query_emb('Change sound')"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "c4219d23-5a3e-4b5e-9f57-e2be83e9145b",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.4"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}

games/angrybirds/notebooks/ai_v3.5.ipynb

@@ -1,1146 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 295,
-   "id": "245fbc34-352d-4945-a689-d04f9365bb99",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "The autoreload extension is already loaded. To reload it, use:\n",
-      "  %reload_ext autoreload\n"
-     ]
-    }
-   ],
-   "source": [
-    "%load_ext autoreload\n",
-    "%autoreload 2"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 296,
-   "id": "eaed216c-160a-4e31-96fe-015c140ff18b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# from pprint import pprint"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 297,
-   "id": "97953551-8ef1-42c1-9317-99d5e65f535e",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import json\n",
-    "import pandas as pd\n",
-    "\n",
-    "import openai\n",
-    "from openai.embeddings_utils import get_embedding, cosine_similarity\n",
-    "\n",
-    "# from games.angrybirds.util import util\n",
-    "from games.angrybirds.config import config\n",
-    "# from games.angrybirds.prompt import prompt\n",
-    "# from games.angrybirds.query import query, yes_no_query\n",
-    "from games.angrybirds.angrybirds import angrybird"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "fc213f4c-cb07-468b-b15f-60e5120b3d94",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 298,
-   "id": "0c876614-a9bf-4d0b-a728-07770f1ad9c5",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "bird = angrybird()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 299,
-   "id": "1411de96-97d5-4bac-b8ed-e279fb1edaa0",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "0"
-      ]
-     },
-     "execution_count": 299,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 300,
-   "id": "129403ce-69f4-4f2a-9b22-7b3e36d78f24",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{}"
-      ]
-     },
-     "execution_count": 300,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.embeddings"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 301,
-   "id": "163545c6-c707-401a-93b8-fb2cb9d65136",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "initializing...\n",
-      "loading embeddings from a file...\n",
-      "/Users/shaojun/playgo/abc123/embeddings.pkl\n",
-      "pkl size: 766\n",
-      "embedding size: 766\n",
-      "creating embeddings by query OpenAI...\n",
-      "query 'move birds'...\n",
-      "query 'move pigs'...\n",
-      "query 'move houses'...\n",
-      "query 'move obstacles'...\n",
-      "embedding size: 770\n",
-      "train size: 770\n"
-     ]
-    }
-   ],
-   "source": [
-    "bird.init_emb()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 265,
-   "id": "31c02b28-0078-425d-87b6-702caeea02e3",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "767"
-      ]
-     },
-     "execution_count": 265,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 266,
-   "id": "9389ebb2-de88-4c23-95ee-c8beb83af600",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import pickle"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "808c1f89-2b7e-4134-a73e-94689f4cce75",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "067905b6-b030-4df1-9570-d45885f775eb",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 227,
-   "id": "02478e9b-dd11-406a-b405-ea4ee56975b1",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import os"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 228,
-   "id": "7b506b8c-0b60-44e4-bb6e-f4af8439f04f",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'/Users/shaojun/playgo/abc123'"
-      ]
-     },
-     "execution_count": 228,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "root = os.environ['PYTHONPATH']\n",
-    "root"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "e977cf56-3bfc-4c87-884e-58577b3808cd",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 229,
-   "id": "30fb4dd7-35d8-442f-bd6e-29e15374e3d6",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'{root}/game_train.json', 'w') as f:\n",
-    "    json.dump(config.game_train, f, indent=4)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 230,
-   "id": "3c051dbf-d547-4297-8f14-f464e58e9a64",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'{root}/game_train.json', 'r') as f:\n",
-    "    tmp = json.load(f)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 231,
-   "id": "77789684-e7b1-4e28-b9b7-5fbca3b6d376",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "True"
-      ]
-     },
-     "execution_count": 231,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "config.game_train == tmp"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "ce058298-2236-4ede-adca-164d6d1249e1",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 232,
-   "id": "dd8c489e-cd3f-4df6-81c0-fe3fd2cc0d33",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'{root}/embeddings.pkl', 'wb') as f:\n",
-    "    pickle.dump(bird.embeddings, f)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 233,
-   "id": "07032487-3fe7-4729-8ca0-c57d6a485c28",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'{root}/embeddings.pkl', 'rb') as f:\n",
-    "    tmp = pickle.load(f)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 234,
-   "id": "ab7ec218-0e02-4a2b-9c16-ec26704b3519",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "True"
-      ]
-     },
-     "execution_count": 234,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "tmp == bird.embeddings"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 235,
-   "id": "bc896434-21ae-438c-90b1-6b590103e731",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "471"
-      ]
-     },
-     "execution_count": 235,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(tmp)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 236,
-   "id": "9532b604-b7d6-4bd4-a0b9-c8c36470447b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "471"
-      ]
-     },
-     "execution_count": 236,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 237,
-   "id": "18888a7b-d32d-45b4-a9b3-3ce29b18a7f7",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# sim query"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 238,
-   "id": "4703a269-a1c0-48e4-8848-6b5cdb9b814b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "471"
-      ]
-     },
-     "execution_count": 238,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "4ff48663-e767-4936-807d-07fb8c42ce92",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 302,
-   "id": "567ca6d0-9291-4876-a3a2-77b12576f7f8",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "skip init\n"
-     ]
-    },
-    {
-     "data": {
-      "text/plain": [
-       "['OK', ['pig', 'change position', {}], 'move pigs', 0.9765097225482317]"
-      ]
-     },
-     "execution_count": 302,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "# x = 'i want to add birds'\n",
-    "# x = 'Add birds.'\n",
-    "# x = 'Add a bird.'\n",
-    "# x = 'bird list'\n",
-    "# x = 'reduce a bird'\n",
-    "x = 'remove a bird'\n",
-    "x = 'The game is too difficult. add more birds'\n",
-    "# x = 'birds'\n",
-    "x = 'reduce a pig'\n",
-    "x = 'Add Magnet Bird'\n",
-    "x = 'Anti-gravity bird'\n",
-    "x = 'move pig'\n",
-    "\n",
-    "bird.query_emb(x)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 289,
-   "id": "d53944cf-437f-4081-a0ce-3bc67abd67c6",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from openai.embeddings_utils import cosine_similarity as openai_cosine_similarity\n"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 292,
-   "id": "c90fd66f-5d1b-4ea1-9dfa-801c24469440",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "add one big bird 0.7962442494188581\n",
-      "add a big bird 0.7965765066894088\n",
-      "add two big birds 0.8103686756491976\n",
-      "add three big birds 0.8135757009624476\n",
-      "add magnet birds 0.8164196001609102\n",
-      "add catapult birds 0.8200955417278051\n",
-      "delete magnet birds 0.8216137811946468\n",
-      "delete birds 0.8228437232796108\n",
-      "add one elderly pig 0.8976585322979147\n",
-      "add an elderly pig 0.9035534949298423\n",
-      "add elderly pigs 0.9102856910269412\n",
-      "add one pig 0.9170617809681969\n",
-      "add a pig 0.9257477545332493\n",
-      "add pigs 0.9412379122704083\n",
-      "change the position of birds 0.8047532670923457\n",
-      "change the position of pigs 0.9107673506300684\n",
-      "change the position of houses 0.7765230737101322\n",
-      "change the position of obstacles 0.7781499359161194\n"
-     ]
-    }
-   ],
-   "source": [
-    "imput_emb = bird._compute_emb(x)\n",
-    "top_cmd = None\n",
-    "top_sim = 0\n",
-    "for cmd in bird.embeddings:\n",
-    "    sim = openai_cosine_similarity(\n",
-    "        imput_emb, \n",
-    "        bird.embeddings[cmd],\n",
-    "    )\n",
-    "    if cmd.find('position') != -1:\n",
-    "        print(cmd, sim)\n",
-    "    if sim == 1.0:\n",
-    "        continue\n",
-    "    if sim > top_sim:\n",
-    "        top_cmd = cmd\n",
-    "        top_sim = sim\n",
-    "        print(top_cmd, top_sim)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 286,
-   "id": "18a90c6c-9b5e-4e36-a98b-beb91d2b4e0f",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "('change the position of pigs', 0.946176004901703)"
-      ]
-     },
-     "execution_count": 286,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "top_cmd, top_sim"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 125,
-   "id": "de48b250-651c-4183-ad41-506f891d4a7d",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['OK',\n",
-       " ['pig', 'increase health points', {}],\n",
-       " 'the pigs are too weak. increase the hp for them',\n",
-       " 1.0]"
-      ]
-     },
-     "execution_count": 125,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "# x = 'i want to add birds'\n",
-    "# x = 'Add birds.'\n",
-    "# x = 'Add a bird.'\n",
-    "# x = \"bird picture\"\n",
-    "x = 'The pigs are too weak. Increase the HP for them'\n",
-    "\n",
-    "bird.query_emb(x)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "f5786cfe-8499-44fc-9220-4f494d2fe8f1",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 126,
-   "id": "8a3ff0d8-2220-4b0d-bebd-a048804abc83",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from pathlib import Path"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 129,
-   "id": "9f0a9d29-0abe-41c5-ac3a-65c73ae5ea20",
-   "metadata": {},
-   "outputs": [
-    {
-     "ename": "NameError",
-     "evalue": "name '__file__' is not defined",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
-      "\u001b[0;31mNameError\u001b[0m                                 Traceback (most recent call last)",
-      "Cell \u001b[0;32mIn[129], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m Path(\u001b[38;5;124mf\u001b[39m\u001b[38;5;124m'\u001b[39m\u001b[38;5;132;01m{\u001b[39;00mos\u001b[38;5;241m.\u001b[39mpath\u001b[38;5;241m.\u001b[39mdirname(\u001b[38;5;18;43m__file__\u001b[39;49m)\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m/../../embeddings.pkl\u001b[39m\u001b[38;5;124m'\u001b[39m)\n",
-      "\u001b[0;31mNameError\u001b[0m: name '__file__' is not defined"
-     ]
-    }
-   ],
-   "source": [
-    "Path(f'{os.path.dirname(__file__)}/../../embeddings.pkl')"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 73,
-   "id": "4affbcb6-0959-4871-b5e9-dc9a9591ebf7",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "command = 'add 6 birds'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 70,
-   "id": "d99b6d4d-5b29-4bc6-9731-fad79d4f7643",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[{'role': 'system', 'content': 'You are a helpful assistant.'},\n",
-       " {'role': 'user', 'content': 'add 6 birds'}]"
-      ]
-     },
-     "execution_count": 70,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird._get_full_prompt(command)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 24,
-   "id": "97d52d34-4a29-418a-b16b-ce2343e9f618",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "----------\n",
-      "\n",
-      "When the user says:\n",
-      "\n",
-      "'add antigravity birds'\n",
-      "\n",
-      "Did the user mention antigravity birds?\n",
-      "\n",
-      "Please answer yes or no.\n",
-      "Please do not explain.\n",
-      "\n",
-      "----------\n",
-      "0.8000 12/15\ty\n",
-      "[('y', 12), ('n', 3)]\n"
-     ]
-    },
-    {
-     "data": {
-      "text/plain": [
-       "'y'"
-      ]
-     },
-     "execution_count": 24,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.query_chatgpt_yes_no(\n",
-    "    \"add antigravity birds\", \n",
-    "    # \"Did the user mention a number?\"\n",
-    "    \"Did the user mention antigravity birds?\",\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 25,
-   "id": "baa47858-baf0-40cf-b4ec-43f4c0a5e303",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "----------\n",
-      "\n",
-      "When the user says:\n",
-      "\n",
-      "'add 4 antigrevity birds'\n",
-      "\n",
-      "Did the user want to add some birds?\n",
-      "\n",
-      "Please answer yes or no.\n",
-      "Please do not explain.\n",
-      "\n",
-      "----------\n",
-      "0.8667 13/15\ty\n",
-      "[('y', 13), ('n', 2)]\n"
-     ]
-    },
-    {
-     "data": {
-      "text/plain": [
-       "'y'"
-      ]
-     },
-     "execution_count": 25,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.query_chatgpt_yes_no(\n",
-    "    \"add 4 antigrevity birds\", \n",
-    "    # \"Did the user mention a number?\"\n",
-    "    \"Did the user want to add some birds?\",\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "e0417f76-71dc-4b47-b5a3-fa1dfafabcf6",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 26,
-   "id": "895d4714-e164-46a1-bdbc-21a52af5e2f2",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "----------\n",
-      "\n",
-      "When the user says:\n",
-      "\n",
-      "'add 221 antigravity birds'\n",
-      "\n",
-      "How many birds did the user want to add?\n",
-      "\n",
-      "Just provide the number in numerical form.\n",
-      "Please do not explain.\n",
-      "\n",
-      "----------\n",
-      "1.0000 15/15\t221\n",
-      "[('221', 15)]\n"
-     ]
-    },
-    {
-     "data": {
-      "text/plain": [
-       "'221'"
-      ]
-     },
-     "execution_count": 26,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.query_chatgpt_number(\n",
-    "    \"add 221 antigravity birds\", \n",
-    "    # \"Did the user mention a number?\"\n",
-    "    \"How many birds did the user want to add?\",\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "637c12b4-24d8-42b2-9c0a-3438a41d7aeb",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 27,
-   "id": "5a878ccf-6575-4440-a5fa-d81d046f9dee",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# print(json.dumps(bird.list_functionality(), indent=4))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 28,
-   "id": "06c7907e-0e2f-44be-9b75-692a5354f1c0",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# print(json.dumps(bird.show_json(), indent=4))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 168,
-   "id": "85717861-b3d7-4637-ae6a-dccd25ef0ccc",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "ok\n"
-     ]
-    }
-   ],
-   "source": [
-    "print('ok')"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "42d0d053-b523-4cd7-9571-063dc8b5ebc2",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "3806172f-6f72-415d-9e71-80d64a0638c6",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 180,
-   "id": "3ff452d6-0bc6-4a8b-b113-df47ceded955",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "from chatGPT (1): To add a pig to the game, you would need to create a new character model and implement its behavior and interactions within the game environment.\n",
-      "from chatGPT (2): To add a pig to the game, the game developer will need to create a new character model and associated behaviors for the pig. This includes defining its appearance, movements, interactions, and any unique abilities or behaviors it may have in the game.\n",
-      "from chatGPT (3): Sure! When you say \"add a pig,\" it means that you want to include a pig character in the game. This pig will be one of the targets for the Angry Birds to eliminate.\n",
-      "from chatGPT (4): To add a pig to the game, you would need to create a pig character with its unique behavior, physics, and visual representation within the game's environment.\n",
-      "from chatGPT (5): To add a pig to the game, you would need to incorporate pig character design, behavior, and spawn locations within the game's code.\n",
-      "from chatGPT (6): To add a pig to the game, you will need to create a new character or object that represents the pig. This includes designing its appearance, behavior, and any abilities or interactions it may have within the game.\n",
-      "from chatGPT (7): As a game developer, when you say \"add a pig,\" the concise response would be to add a pig character to the game with the necessary attributes, behavior, and visuals.\n",
-      "from chatGPT (8): To add a pig, the game developer needs to create a pig character in the game's code or design a pig asset that can be added to the game environment.\n",
-      "from chatGPT (9): To add a pig, you need to create a pig character with its own attributes such as size, weight, and abilities. This character should be visually represented in the game and programmed to interact with the birds and the game environment.\n",
-      "from chatGPT (10): Sure, I understand that as a game developer, you want to add a pig to the game.\n"
-     ]
-    }
-   ],
-   "source": [
-    "for i, x in enumerate(bird.query_chatgpt_nlp_response('add a pig', n=10)):\n",
-    "    print(f'from chatGPT ({i+1}): {x}')"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 242,
-   "id": "8e55af03-d673-452b-b70d-b69b4e231dee",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "prompt1 = f\"\"\"\n",
-    "{config.game_desc}\n",
-    "\"\"\"\n",
-    "full_prompt = bird._get_full_prompt(prompt1)\n",
-    "# print(prompt)\n",
-    "choices = bird._query_chatgpt(bird._get_full_prompt(prompt1), n=1)\n",
-    "\n",
-    "\n",
-    "prompt2 = f\"\"\"\n",
-    "As a game developer, when I say, \n",
-    "\n",
-    "{command}\n",
-    "\n",
-    "give me a concise response.\n",
-    "\"\"\"\n"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 243,
-   "id": "40ea1f8c-0cd8-4570-b5c0-c815dd02c5a1",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[\"Certainly! I'll be happy to assist you. How can I help you with developing the Angry Birds game?\"]"
-      ]
-     },
-     "execution_count": 243,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "choices"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 244,
-   "id": "b56959d9-7885-4482-a26f-aebfbf5a3a87",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "\"Certainly! I'll be happy to assist you. How can I help you with developing the Angry Birds game?\""
-      ]
-     },
-     "execution_count": 244,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "choices[0]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 245,
-   "id": "d652917a-1985-4c15-9082-697599795afd",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[{'role': 'system', 'content': 'You are a helpful assistant.'},\n",
-       " {'role': 'user',\n",
-       "  'content': '\\n\\nImaging we are developing the game of the Angry Birds, \\nand the game is described as the following: \\n\\nIn Angry Birds, players take control of a group of birds who are seeking \\nrevenge on a group of green pigs that have stolen their eggs. \\nThe gameplay involves using a slingshot to launch the birds at various \\nstructures where the pigs are hiding. The objective is to \\neliminate all the pigs on each level and score as many points as possible.\\n\\nA game developer is building the game, \\nand I need you to help me figure out what is going on.\\n\\n'}]"
-      ]
-     },
-     "execution_count": 245,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "full_prompt"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "f9a53efc-cb26-4d9e-b425-347a14e7901e",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 246,
-   "id": "0ef8cdfd-9c79-4530-ba08-1eca474ef420",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "full_prompt.append({\n",
-    "            \"role\": \"assistant\", \n",
-    "            \"content\": choices[0]})"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 247,
-   "id": "2df8919b-4e89-4f78-ae4c-dcaac5eec9fd",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[{'role': 'system', 'content': 'You are a helpful assistant.'},\n",
-       " {'role': 'user',\n",
-       "  'content': '\\n\\nImaging we are developing the game of the Angry Birds, \\nand the game is described as the following: \\n\\nIn Angry Birds, players take control of a group of birds who are seeking \\nrevenge on a group of green pigs that have stolen their eggs. \\nThe gameplay involves using a slingshot to launch the birds at various \\nstructures where the pigs are hiding. The objective is to \\neliminate all the pigs on each level and score as many points as possible.\\n\\nA game developer is building the game, \\nand I need you to help me figure out what is going on.\\n\\n'},\n",
-       " {'role': 'assistant',\n",
-       "  'content': \"Certainly! I'll be happy to assist you. How can I help you with developing the Angry Birds game?\"}]"
-      ]
-     },
-     "execution_count": 247,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "full_prompt"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 248,
-   "id": "4289159c-ce4c-48a7-bfcb-c6f904119a66",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "full_prompt.append({\n",
-    "            \"role\": \"user\", \n",
-    "            \"content\": \"\"\"\n",
-    "As a game developer, \n",
-    "Add 5 pink pigs. \n",
-    "\n",
-    "Please give a one sentence response.\"\"\"})"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 249,
-   "id": "8c4f7555-24cc-4b5f-badd-a89b32b55aa7",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[{'role': 'system', 'content': 'You are a helpful assistant.'},\n",
-       " {'role': 'user',\n",
-       "  'content': '\\n\\nImaging we are developing the game of the Angry Birds, \\nand the game is described as the following: \\n\\nIn Angry Birds, players take control of a group of birds who are seeking \\nrevenge on a group of green pigs that have stolen their eggs. \\nThe gameplay involves using a slingshot to launch the birds at various \\nstructures where the pigs are hiding. The objective is to \\neliminate all the pigs on each level and score as many points as possible.\\n\\nA game developer is building the game, \\nand I need you to help me figure out what is going on.\\n\\n'},\n",
-       " {'role': 'assistant',\n",
-       "  'content': \"Certainly! I'll be happy to assist you. How can I help you with developing the Angry Birds game?\"},\n",
-       " {'role': 'user',\n",
-       "  'content': '\\nAs a game developer, \\nAdd 5 pink pigs. \\n\\nPlease give a one sentence response.'}]"
-      ]
-     },
-     "execution_count": 249,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "full_prompt"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 250,
-   "id": "cf888b66-b154-4ba9-aaed-b21137bce213",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "choices = bird._query_chatgpt(full_prompt, n=5)\n"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 251,
-   "id": "3cfc4984-72f8-4d2e-904b-8627ef6b6d4d",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['Sure, I will add 5 pink pigs to the game.',\n",
-       " 'Sure, I will add 5 pink pigs to the game.',\n",
-       " 'Sure, I will add 5 pink pigs to the game.',\n",
-       " 'I have added 5 pink pigs to the game.',\n",
-       " 'Sure, I will add 5 pink pigs to the game.']"
-      ]
-     },
-     "execution_count": 251,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "choices"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "8ae1af73-d0c3-49a0-9c68-2ebd37539778",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 26,
-   "id": "c92de3b4-e6ef-42d7-811f-5c1dde2c8c46",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'version': '2.0.1-2023-07-28-d',\n",
-       " 'data': ['add one or more birds',\n",
-       "  'add one or more houses',\n",
-       "  'add one or more pigs',\n",
-       "  'change background',\n",
-       "  'change bird size',\n",
-       "  'change pig size',\n",
-       "  'change position',\n",
-       "  'change skin',\n",
-       "  'decrease HP for a pig',\n",
-       "  'delete a bird',\n",
-       "  'delete a pig',\n",
-       "  'increase HP for a pig',\n",
-       "  'list birds',\n",
-       "  'list pigs']}"
-      ]
-     },
-     "execution_count": 26,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.list_functionality()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "22769a08-a9c6-44d2-b539-e2e3b99f5a3e",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.4"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}

games/angrybirds/notebooks/ai_v3.7-config.ipynb

@@ -1,844 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 36,
-   "id": "245fbc34-352d-4945-a689-d04f9365bb99",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "The autoreload extension is already loaded. To reload it, use:\n",
-      "  %reload_ext autoreload\n"
-     ]
-    }
-   ],
-   "source": [
-    "%load_ext autoreload\n",
-    "%autoreload 2"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 37,
-   "id": "eaed216c-160a-4e31-96fe-015c140ff18b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# from pprint import pprint"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 38,
-   "id": "97953551-8ef1-42c1-9317-99d5e65f535e",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import json\n",
-    "import pandas as pd\n",
-    "import pickle\n",
-    "\n",
-    "import openai\n",
-    "from openai.embeddings_utils import get_embedding, cosine_similarity\n",
-    "\n",
-    "# from games.angrybirds.util import util\n",
-    "from games.angrybirds.config import config\n",
-    "# from games.angrybirds.prompt import prompt\n",
-    "# from games.angrybirds.query import query, yes_no_query\n",
-    "from games.angrybirds.angrybirds import angrybird"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "fc213f4c-cb07-468b-b15f-60e5120b3d94",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 39,
-   "id": "0c876614-a9bf-4d0b-a728-07770f1ad9c5",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "before add_config\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "after add_config\n",
-      "779\n",
-      "final\n",
-      "779\n"
-     ]
-    }
-   ],
-   "source": [
-    "bird = angrybird()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 9,
-   "id": "1411de96-97d5-4bac-b8ed-e279fb1edaa0",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "0"
-      ]
-     },
-     "execution_count": 9,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 10,
-   "id": "129403ce-69f4-4f2a-9b22-7b3e36d78f24",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{}"
-      ]
-     },
-     "execution_count": 10,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.embeddings"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 11,
-   "id": "50e372eb-48ba-4378-b866-f2b7120bc94e",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "779"
-      ]
-     },
-     "execution_count": 11,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.train)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "f78b2c43-bfc7-478f-894d-d757b0e9927f",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 12,
-   "id": "163545c6-c707-401a-93b8-fb2cb9d65136",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "initializing...\n",
-      "loading embeddings from a file...\n",
-      "pkl size: 779\n",
-      "embedding size: 779\n",
-      "creating embeddings by query OpenAI...\n",
-      "embedding size: 779\n",
-      "train size: 779\n"
-     ]
-    }
-   ],
-   "source": [
-    "bird.init_emb()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 13,
-   "id": "31c02b28-0078-425d-87b6-702caeea02e3",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "779"
-      ]
-     },
-     "execution_count": 13,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "067905b6-b030-4df1-9570-d45885f775eb",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 11,
-   "id": "02478e9b-dd11-406a-b405-ea4ee56975b1",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import os"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 12,
-   "id": "7b506b8c-0b60-44e4-bb6e-f4af8439f04f",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'/Users/shaojun/playgo/bbc123'"
-      ]
-     },
-     "execution_count": 12,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "root = os.environ['PYTHONPATH']\n",
-    "root"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "e977cf56-3bfc-4c87-884e-58577b3808cd",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 13,
-   "id": "5662bd80-03c5-4793-94c3-4d0392c08102",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "before add_config\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "779\n",
-      "after add_config\n",
-      "779\n",
-      "final\n",
-      "779\n"
-     ]
-    }
-   ],
-   "source": [
-    "conf = config()\n",
-    "conf.init_game_train()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 177,
-   "id": "30fb4dd7-35d8-442f-bd6e-29e15374e3d6",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'../game_train.json', 'w') as f:\n",
-    "    json.dump(config.game_train, f, indent=4)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 178,
-   "id": "3c051dbf-d547-4297-8f14-f464e58e9a64",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'../game_train.json', 'r') as f:\n",
-    "    tmp = json.load(f)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 179,
-   "id": "77789684-e7b1-4e28-b9b7-5fbca3b6d376",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "True"
-      ]
-     },
-     "execution_count": 179,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "config.game_train == tmp"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "ce058298-2236-4ede-adca-164d6d1249e1",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 180,
-   "id": "dd8c489e-cd3f-4df6-81c0-fe3fd2cc0d33",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'{root}/embeddings.pkl', 'wb') as f:\n",
-    "    pickle.dump(bird.embeddings, f)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 181,
-   "id": "07032487-3fe7-4729-8ca0-c57d6a485c28",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'{root}/embeddings.pkl', 'rb') as f:\n",
-    "    tmp = pickle.load(f)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 182,
-   "id": "ab7ec218-0e02-4a2b-9c16-ec26704b3519",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "True"
-      ]
-     },
-     "execution_count": 182,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "tmp == bird.embeddings"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 183,
-   "id": "bc896434-21ae-438c-90b1-6b590103e731",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "779"
-      ]
-     },
-     "execution_count": 183,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(tmp)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 184,
-   "id": "9532b604-b7d6-4bd4-a0b9-c8c36470447b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "779"
-      ]
-     },
-     "execution_count": 184,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 163,
-   "id": "18888a7b-d32d-45b4-a9b3-3ce29b18a7f7",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# sim query"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 164,
-   "id": "4703a269-a1c0-48e4-8848-6b5cdb9b814b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "774"
-      ]
-     },
-     "execution_count": 164,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 126,
-   "id": "78b30acb-50eb-48a9-9d57-e8af47a5de2d",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "False"
-      ]
-     },
-     "execution_count": 126,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "'move the pigs' in bird.embeddings"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 127,
-   "id": "9bc42449-80a7-4365-964d-910dc3151396",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "True"
-      ]
-     },
-     "execution_count": 127,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "'change the position of pigs' in bird.embeddings"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 128,
-   "id": "d54fcdcc-2db4-4cba-839c-361c30a5743f",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "False"
-      ]
-     },
-     "execution_count": 128,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "'change the positon of pigs' in bird.embeddings"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 14,
-   "id": "acda65fa-803d-432f-b8d3-79863e0b4e9e",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "True"
-      ]
-     },
-     "execution_count": 14,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "'move pigs' in bird.embeddings"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "c885da07-e53d-4586-b340-8f9de0515fc2",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 130,
-   "id": "4ff48663-e767-4936-807d-07fb8c42ce92",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "{\n",
-      "    \"version\": \"3.8.0.2023-08-09-e\",\n",
-      "    \"data\": [\n",
-      "        \"(1) [unified] add/delete one or more (special) birds/pigs\",\n",
-      "        \"(2) add one or more houses\",\n",
-      "        \"(3) change HP\",\n",
-      "        \"(4) change background\",\n",
-      "        \"(5) change positions\",\n",
-      "        \"(6) change size\",\n",
-      "        \"(7) change skin\",\n",
-      "        \"(8) list birds or pigs\"\n",
-      "    ]\n",
-      "}\n"
-     ]
-    }
-   ],
-   "source": [
-    "print(json.dumps(bird.list_functionality(), indent=4))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 131,
-   "id": "982c3666-7b6a-4300-a2ac-b1a42c216c09",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "0 (1) [unified] add/delete one or more (special) birds/pigs\n",
-      "1 (2) add one or more houses\n",
-      "2 (3) change HP\n",
-      "3 (4) change background\n",
-      "4 (5) change positions\n",
-      "5 (6) change size\n",
-      "6 (7) change skin\n",
-      "7 (8) list birds or pigs\n"
-     ]
-    }
-   ],
-   "source": [
-    "for i, x in enumerate(bird.list_functionality()['data']):\n",
-    "    print(i,x)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 132,
-   "id": "3a11c8b8-4cea-42fc-930b-66dd7330c074",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# print(json.dumps(bird.show_json(),indent=4))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "75cf385f-ba62-452d-a98a-fd3d0bd36031",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 11,
-   "id": "e8c93b82-86ca-4c2a-aa79-2b9b1d4e6351",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "skip init\n",
-      "\n",
-      "--------------------\n",
-      "query count: 1\n",
-      "input: add two birds\n",
-      "{\n",
-      "  \"status\": \"OK\",\n",
-      "  \"result\": [\n",
-      "    \"bird\",\n",
-      "    \"add\",\n",
-      "    {\n",
-      "      \"quantity\": 2\n",
-      "    }\n",
-      "  ],\n",
-      "  \"command\": \"add two birds\",\n",
-      "  \"timestamp\": \"2023-09-16 15:55:07.691027\",\n",
-      "  \"ai-version\": \"3.8.0.2023-09-14-a\"\n",
-      "}\n",
-      "[\n",
-      "  [\n",
-      "    0,\n",
-      "    \"1.00000000\",\n",
-      "    \"add two birds\"\n",
-      "  ],\n",
-      "  [\n",
-      "    1,\n",
-      "    \"0.97266663\",\n",
-      "    \"add 2 birds\"\n",
-      "  ],\n",
-      "  [\n",
-      "    2,\n",
-      "    \"0.96591405\",\n",
-      "    \"add two big birds\"\n",
-      "  ],\n",
-      "  [\n",
-      "    3,\n",
-      "    \"0.94957413\",\n",
-      "    \"add one bird\"\n",
-      "  ],\n",
-      "  [\n",
-      "    4,\n",
-      "    \"0.94873926\",\n",
-      "    \"add 2 big birds\"\n",
-      "  ]\n",
-      "]\n",
-      "top: add two birds\n"
-     ]
-    },
-    {
-     "data": {
-      "text/plain": [
-       "({'status': 'OK',\n",
-       "  'result': ['bird', 'add', {'quantity': 2}],\n",
-       "  'command': 'add two birds',\n",
-       "  'timestamp': '2023-09-16 15:55:07.691027',\n",
-       "  'ai-version': '3.8.0.2023-09-14-a'},\n",
-       " [[0, '1.00000000', 'add two birds'],\n",
-       "  [1, '0.97266663', 'add 2 birds'],\n",
-       "  [2, '0.96591405', 'add two big birds'],\n",
-       "  [3, '0.94957413', 'add one bird'],\n",
-       "  [4, '0.94873926', 'add 2 big birds']])"
-      ]
-     },
-     "execution_count": 11,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.query_entry_point('add two birds')"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 32,
-   "id": "4403f4a2-bf66-479b-a70d-d0df9a1b2c1b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "command = 'check'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 35,
-   "id": "97b73dd8-0bbf-498d-b992-573a2a30480a",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "skip init\n",
-      "\n",
-      "--------------------\n",
-      "angrybird query count: 3\n",
-      "input: check\n",
-      "{\n",
-      "  \"status\": \"Failure\",\n",
-      "  \"result\": [\n",
-      "    \"We don't understand your request.\",\n",
-      "    \"Our best guess is: 'move birds'\"\n",
-      "  ],\n",
-      "  \"command\": \"check\",\n",
-      "  \"timestamp\": \"2023-09-17 23:41:16.912413\",\n",
-      "  \"ai-version\": \"3.8.0.2023-09-17-a\"\n",
-      "}\n",
-      "[\n",
-      "  [\n",
-      "    0,\n",
-      "    \"0.76896785\",\n",
-      "    \"move birds\"\n",
-      "  ],\n",
-      "  [\n",
-      "    1,\n",
-      "    \"0.76711535\",\n",
-      "    \"add two birds\"\n",
-      "  ],\n",
-      "  [\n",
-      "    2,\n",
-      "    \"0.76695086\",\n",
-      "    \"list birds\"\n",
-      "  ],\n",
-      "  [\n",
-      "    3,\n",
-      "    \"0.76688633\",\n",
-      "    \"change pigs\"\n",
-      "  ],\n",
-      "  [\n",
-      "    4,\n",
-      "    \"0.76666684\",\n",
-      "    \"change volumn\"\n",
-      "  ]\n",
-      "]\n",
-      "top: move birds\n"
-     ]
-    },
-    {
-     "data": {
-      "text/plain": [
-       "({'status': 'Failure',\n",
-       "  'result': [\"We don't understand your request.\",\n",
-       "   \"Our best guess is: 'move birds'\"],\n",
-       "  'command': 'check',\n",
-       "  'timestamp': '2023-09-17 23:41:16.912413',\n",
-       "  'ai-version': '3.8.0.2023-09-17-a'},\n",
-       " [[0, '0.76896785', 'move birds'],\n",
-       "  [1, '0.76711535', 'add two birds'],\n",
-       "  [2, '0.76695086', 'list birds'],\n",
-       "  [3, '0.76688633', 'change pigs'],\n",
-       "  [4, '0.76666684', 'change volumn']])"
-      ]
-     },
-     "execution_count": 35,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.query_entry_point(command)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 34,
-   "id": "d5d67ed9-e901-4cc4-b796-cbcf13d3a5a2",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "/Users/shaojun/playgo/bbc123/games/angrybirds/notebooks\n"
-     ]
-    }
-   ],
-   "source": [
-    "!pwd"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "01ec8620-dc1d-4de6-88ee-d6aec794fce6",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.4"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}

games/angrybirds/notebooks/demo.ipynb

@@ -1,502 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 18,
-   "id": "245fbc34-352d-4945-a689-d04f9365bb99",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "The autoreload extension is already loaded. To reload it, use:\n",
-      "  %reload_ext autoreload\n"
-     ]
-    }
-   ],
-   "source": [
-    "%load_ext autoreload\n",
-    "%autoreload 2"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 28,
-   "id": "eaed216c-160a-4e31-96fe-015c140ff18b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from pprint import pprint"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 39,
-   "id": "f6aa8ec8-ad1f-4cd5-819a-2b5786724a5b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from games.angrybirds.angrybirds import bird\n",
-    "from games.angrybirds.config import config"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 40,
-   "id": "1b0a6bee-18a3-4f50-9065-6e5e5eb4cad0",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "game = bird()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "a15d43e9-802f-4385-82ef-b052e46d63aa",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 25,
-   "id": "bfa4c3b9-538f-43c5-ad8e-4400bbd81b8d",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "res = bird.query(bird.input)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 30,
-   "id": "68dccb67-3f86-4b10-9dfd-12df7357f5b4",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "('The user wants to add more birds to the game because they find it too '\n",
-      " 'difficult. Therefore, the answer is (A) birds.')\n"
-     ]
-    }
-   ],
-   "source": [
-    "pprint(res)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 33,
-   "id": "a93d24dd-098a-4604-bdcc-30abe098f417",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "str"
-      ]
-     },
-     "execution_count": 33,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "type(res)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 49,
-   "id": "f53461f4-7718-4eb0-b047-ba4ae88c3016",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "prompt = \"The game is too difficult. Please add more birds.\""
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 92,
-   "id": "d3909a61-5bdf-4b77-89db-f41928112013",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "prompt = \"Make the bird bigger, so it can run it over to the enemies.\""
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "a79562aa-b978-474e-87e3-4eb3ac6c7316",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "485e9947-c982-4d2c-a170-bd6bd79f6d02",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "ae9bb6ad-cfbc-4666-adc9-3a354110f0f9",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 125,
-   "id": "41cbe328-3a8b-4fca-b54a-3720a654ffbb",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "q = [\n",
-    "    {\"role\": \"system\", \"content\": \"You are a helpful assistant.\"},\n",
-    "    {'role': 'user', 'content': f\"\"\"\n",
-    "{config.game_desc}\n",
-    "\n",
-    "In this user prompt:  \\\"{prompt}\\\".\n",
-    "\n",
-    "The user is talking about \n",
-    "(A) birds (B) pigs (C) eggs (D) structures (E) none of the above.\n",
-    "\n",
-    "Please answer it in the most concise way.\n",
-    "\"\"\"}\n",
-    "]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 143,
-   "id": "c7901c49-90d2-4837-aaf7-54055e924608",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "q = [\n",
-    "    {\"role\": \"system\", \"content\": \"You are a helpful assistant.\"},\n",
-    "    {'role': 'user', 'content': f\"\"\"\n",
-    "{config.game_desc}\n",
-    "\n",
-    "In this user prompt:  \\\"{prompt}\\\".\n",
-    "\n",
-    "The user wants to make changes to the __ of the pigs\n",
-    "(A) color (B) type (C) quantity (D) size (E) none of the above.\n",
-    "\n",
-    "Please answer it in the following format:\n",
-    "Answer: (<choice>)\n",
-    "\"\"\"}\n",
-    "]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 144,
-   "id": "59bb772b-5cc1-47ed-87e1-fdd4f6808227",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "res = game.query(q)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 145,
-   "id": "afa4cd1b-4b5e-4c9d-8d58-a4e274bb66f6",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'Answer: (D) size'"
-      ]
-     },
-     "execution_count": 145,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "res"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 131,
-   "id": "df30476b-3b1b-4b57-b458-5d2560422dd2",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'Make the bird bigger, so it can run it over to the enemies.'"
-      ]
-     },
-     "execution_count": 131,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "prompt"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 124,
-   "id": "239aee58-c53a-4938-9f3b-aeb3749efbe7",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[{'role': 'system', 'content': 'You are a helpful assistant.'},\n",
-       " {'role': 'user',\n",
-       "  'content': '\\n\\nImaging we are playing the angry birds game described as the following: \\n\\nIn Angry Birds, players take control of a group of birds who are seeking revenge on a group of green pigs that have stolen their eggs. The gameplay involves using a slingshot to launch the birds at various structures where the pigs are hiding. The objective is to eliminate all the pigs on each level and score as many points as possible.\\n\\nYou are assistant to help me figure out what is going on.\\n\\n\\nIn this user prompt:  \"Make the bird bigger, so it can run it over to the enemies.\".\\n\\nDoes the user wants to make changes to the size of the pigs?\\n\\nPlease answer \"Yes\" or \"No\"\\n\\n'}]"
-      ]
-     },
-     "execution_count": 124,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "q"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "id": "b8b0423f-58f2-4dad-a062-a0ffba33cb16",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "Make the bird red.\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: bird\n",
-      "[app]: color\n",
-      "[app]: red\n"
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "Make the bird red.\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 23,
-   "id": "6fe2bdc8-4324-4a4a-bcce-6b1ccf1a07ea",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "str = \"\"\"\n",
-    "color = \"red\"\n",
-    "bird.set_color(color)\n",
-    "\"\"\""
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 24,
-   "id": "22ed7d59-8845-4a96-8451-90056ed96246",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "\n",
-      "color = \"red\"\n",
-      "bird.set_color(color)\n",
-      "\n"
-     ]
-    }
-   ],
-   "source": [
-    "print(str)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 25,
-   "id": "c3cd1824-63d4-48f3-a347-5194e1f27351",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# eval(str)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 13,
-   "id": "38e842b8-ef0a-4fd9-bf6d-e2cba9721f7a",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "I want to make the bird much much biger.\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: bird\n",
-      "[app]: size\n",
-      "[app]: increase a lot\n"
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "I want to make the bird much much biger.\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "9d29a9d4-128b-46a0-bce9-13cff90de2e9",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "bird.set_size_by(1.5)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 15,
-   "id": "e7ee7f38-cf01-475d-82ee-454d4f65f96b",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "I want to rotate the pipes faster.\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: pipes\n",
-      "[app]: rotate\n"
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "I want to rotate the pipes faster.\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 28,
-   "id": "8eb9bb6c-8e2b-43df-8504-226700dc6adf",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "I want to change the bird's color to piink.\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: bird\n",
-      "[app]: color\n",
-      "[app]: pink\n"
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "I want to change the bird's color to piink.\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 29,
-   "id": "cc2f00d6-4e6c-478e-a409-664e87f39054",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "Add new species of birds\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: bird\n",
-      "[app]: none of the above\n",
-      "not implemented\n"
-     ]
-    },
-    {
-     "ename": "AssertionError",
-     "evalue": "",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
-      "\u001b[0;31mAssertionError\u001b[0m                            Traceback (most recent call last)",
-      "Cell \u001b[0;32mIn[29], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m msg \u001b[38;5;241m=\u001b[39m \u001b[43mbird\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mquery\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;124;43m\"\"\"\u001b[39;49m\n\u001b[1;32m      2\u001b[0m \u001b[38;5;124;43mAdd new species of birds\u001b[39;49m\n\u001b[1;32m      3\u001b[0m \u001b[38;5;124;43m\"\"\"\u001b[39;49m\u001b[43m)\u001b[49m\n",
-      "File \u001b[0;32m~/playgo/playgo.py:149\u001b[0m, in \u001b[0;36mbird.query\u001b[0;34m(self, user_request_prompt)\u001b[0m\n\u001b[1;32m    146\u001b[0m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39m_config()\n\u001b[1;32m    148\u001b[0m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39m_query_util(user_request_prompt, quiet\u001b[38;5;241m=\u001b[39m\u001b[38;5;28;01mTrue\u001b[39;00m)\n\u001b[0;32m--> 149\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43m_follow_up\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\n",
-      "File \u001b[0;32m~/playgo/playgo.py:234\u001b[0m, in \u001b[0;36mbird._follow_up\u001b[0;34m(self)\u001b[0m\n\u001b[1;32m    232\u001b[0m \u001b[38;5;28;01melse\u001b[39;00m:\n\u001b[1;32m    233\u001b[0m     \u001b[38;5;28mprint\u001b[39m(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mnot implemented\u001b[39m\u001b[38;5;124m\"\u001b[39m)\n\u001b[0;32m--> 234\u001b[0m     \u001b[38;5;28;01massert\u001b[39;00m \u001b[38;5;28;01mFalse\u001b[39;00m\n",
-      "\u001b[0;31mAssertionError\u001b[0m: "
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "Add new species of birds\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "5e3ef720-7948-49ca-ba94-fa1359271f3a",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.4"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}

games/angrybirds/notebooks/query.ipynb

@@ -1,358 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "id": "245fbc34-352d-4945-a689-d04f9365bb99",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "%load_ext autoreload\n",
-    "%autoreload 2"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 7,
-   "id": "f6aa8ec8-ad1f-4cd5-819a-2b5786724a5b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import playgo"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "id": "1b0a6bee-18a3-4f50-9065-6e5e5eb4cad0",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "bird = playgo.bird()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "id": "ba76086b-0aef-4af7-b77a-fb9ebda928aa",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "I want to make the bird look tiny.\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: bird\n",
-      "[app]: size\n",
-      "[app]: decrease for an unspecified amount\n"
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "I want to make the bird look tiny.\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "id": "953e901f-ea39-4fb7-ae58-9d7de13eec3e",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "I want to make the bird very very small.\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: bird\n",
-      "[app]: size\n",
-      "[app]: decrease a lot\n"
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "I want to make the bird very very small.\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 8,
-   "id": "8dc6d4f7-3c84-4e39-9804-1d0508d6c5f1",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "I want to make the bird very small.\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: bird\n",
-      "[app]: size\n",
-      "[app]: decrease a little\n"
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "I want to make the bird very small.\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 9,
-   "id": "fe4be8ed-15f1-4293-9627-7685ea0fc886",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "I want to make the bird much much smaller.\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: bird\n",
-      "[app]: size\n",
-      "[app]: decrease a lot\n"
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "I want to make the bird much much smaller.\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 10,
-   "id": "6fc9a34b-9197-489a-92fe-ac7bf8118b88",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "I want to make the bird biger.\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: bird\n",
-      "[app]: size\n",
-      "[app]: increase for an unspecified amount\n"
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "I want to make the bird biger.\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 11,
-   "id": "73a72c28-da0b-4722-947f-1dd3d4202e07",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "I want to make the bird bigger.\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: bird\n",
-      "[app]: size\n",
-      "[app]: increase for an unspecified amount\n"
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "I want to make the bird bigger.\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 12,
-   "id": "852b49e8-f15f-4202-a20f-dd835b54e60b",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "I want to change the bird's color to 13.\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: Please modify your request. Reason: \n",
-      "The user's request doesn't make sense because colors are usually represented by names or codes such as \"red\", \"#FF0000\" or \"rgb(255, 0, 0)\", not by numbers. Additionally, the game in the prompt is described as a simple side-scroller in which the player controls a bird flying between green pipes, and it doesn't mention the ability to change the bird's color.\n"
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "I want to change the bird's color to 13.\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 13,
-   "id": "e6e3fc93-7701-475a-b412-50fe2d32e7ce",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "I want to change the bird's color to pink\"\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: bird\n",
-      "[app]: color\n",
-      "[app]: pink\n"
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "I want to change the bird's color to pink\"\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 14,
-   "id": "eaf17c3c-9014-4d4d-bfcd-2fb1c54478a8",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "I want to change the bird's color to piink\"\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: bird\n",
-      "[app]: color\n",
-      "[app]: pink\n"
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "I want to change the bird's color to piink\"\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 15,
-   "id": "479eb160-952d-48c1-a502-821fafeace1c",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "I want to change the pipe's color red instead of green\"\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: pipes\n",
-      "[app]: color\n",
-      "[app]: red\n"
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "I want to change the pipe's color red instead of green\"\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 16,
-   "id": "35c8f908-f363-40f6-8cd2-bea208b4d7a0",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "[usr]: \n",
-      "I want to make the pipe rotate\"\n",
-      "\n",
-      "[sys]: please wait...\n",
-      "[app]: pipes\n",
-      "[app]: rotate\n"
-     ]
-    }
-   ],
-   "source": [
-    "msg = bird.query(\"\"\"\n",
-    "I want to make the pipe rotate\"\n",
-    "\"\"\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 17,
-   "id": "b8b0423f-58f2-4dad-a062-a0ffba33cb16",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# bird.messages"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "7d0822f0-5552-4739-8858-62bee4eb414f",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.4"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}

games/angrybirds/notebooks/rm-pkl2.ipynb

@@ -1,1008 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "id": "245fbc34-352d-4945-a689-d04f9365bb99",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "%load_ext autoreload\n",
-    "%autoreload 2"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "c088c757-ef2c-43f2-baa2-1a8691174925",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "id": "eaed216c-160a-4e31-96fe-015c140ff18b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# from pprint import pprint"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "id": "97953551-8ef1-42c1-9317-99d5e65f535e",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import json\n",
-    "# import pandas as pd\n",
-    "# import pickle\n",
-    "\n",
-    "import openai\n",
-    "from openai.embeddings_utils import get_embedding, cosine_similarity\n",
-    "\n",
-    "# from games.snake.util import util\n",
-    "from games.angrybirds.config import config\n",
-    "# from games.snake.prompt import prompt\n",
-    "# from games.snake.query import query, yes_no_query\n",
-    "from games.angrybirds.angrybirds import angrybird"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "fc213f4c-cb07-468b-b15f-60e5120b3d94",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "id": "0c876614-a9bf-4d0b-a728-07770f1ad9c5",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "before add_config\n",
-      "941\n",
-      "942\n",
-      "943\n",
-      "944\n",
-      "945\n",
-      "946\n",
-      "947\n",
-      "948\n",
-      "949\n",
-      "950\n",
-      "after add_config\n",
-      "950\n",
-      "final\n",
-      "1016\n"
-     ]
-    }
-   ],
-   "source": [
-    "game = angrybird()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "2f262d0d-287e-44ed-bb51-48f7e5d574fa",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "id": "1411de96-97d5-4bac-b8ed-e279fb1edaa0",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "0"
-      ]
-     },
-     "execution_count": 5,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(game.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "id": "129403ce-69f4-4f2a-9b22-7b3e36d78f24",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{}"
-      ]
-     },
-     "execution_count": 6,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "game.embeddings"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 7,
-   "id": "50e372eb-48ba-4378-b866-f2b7120bc94e",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "1016"
-      ]
-     },
-     "execution_count": 7,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(game.train)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "f78b2c43-bfc7-478f-894d-d757b0e9927f",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 8,
-   "id": "163545c6-c707-401a-93b8-fb2cb9d65136",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "initializing...\n",
-      "loading embeddings from a file...\n",
-      "pkl size: 1016\n",
-      "embedding size: 1016\n",
-      "creating embeddings by query OpenAI...\n",
-      "embedding size: 1016\n",
-      "train size: 1016\n"
-     ]
-    }
-   ],
-   "source": [
-    "game.init_emb()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 9,
-   "id": "31c02b28-0078-425d-87b6-702caeea02e3",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "1016"
-      ]
-     },
-     "execution_count": 9,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(game.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "067905b6-b030-4df1-9570-d45885f775eb",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 10,
-   "id": "02478e9b-dd11-406a-b405-ea4ee56975b1",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import os"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 11,
-   "id": "7b506b8c-0b60-44e4-bb6e-f4af8439f04f",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# root = os.environ['PYTHONPATH']\n",
-    "# root"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "e977cf56-3bfc-4c87-884e-58577b3808cd",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 12,
-   "id": "5662bd80-03c5-4793-94c3-4d0392c08102",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "before add_config\n",
-      "1016\n",
-      "1016\n",
-      "1016\n",
-      "1016\n",
-      "1016\n",
-      "1016\n",
-      "1016\n",
-      "1016\n",
-      "1016\n",
-      "1016\n",
-      "after add_config\n",
-      "1016\n",
-      "final\n",
-      "1016\n"
-     ]
-    }
-   ],
-   "source": [
-    "conf = config()\n",
-    "conf.init_game_train()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "73c89724-e5dd-4f08-b398-3ec78fcc4f82",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 13,
-   "id": "30fb4dd7-35d8-442f-bd6e-29e15374e3d6",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'../game_train.json', 'w') as f:\n",
-    "    json.dump(config.game_train, f, indent=4)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 14,
-   "id": "3c051dbf-d547-4297-8f14-f464e58e9a64",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'../game_train.json', 'r') as f:\n",
-    "    tmp = json.load(f)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 15,
-   "id": "77789684-e7b1-4e28-b9b7-5fbca3b6d376",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "True"
-      ]
-     },
-     "execution_count": 15,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "config.game_train == tmp"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "051b4e35-c84a-4ae9-b3db-80aee6772e2d",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 16,
-   "id": "02133e42-b076-4618-83ba-bf23f5112185",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "100%|███████████████████████████████████████| 1016/1016 [00:13<00:00, 77.17it/s]\n"
-     ]
-    }
-   ],
-   "source": [
-    "game.dump()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "c4fc4888-6084-4034-a920-a0d0372e7755",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "11a50d4a-1465-4b36-bc62-d8f7592298c1",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 17,
-   "id": "7c72275c-75a0-478a-b4e5-01c71f7b5c69",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import os"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "6aa5d783-f01d-400e-8868-55a0707457e8",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 18,
-   "id": "3d66b866-237c-485d-a466-fc32fea95158",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "total 79416\n",
-      "-rw-r--r--  1 shaojun  staff   34452 Sep 27 14:56 ab.json\n",
-      "-rw-r--r--  1 shaojun  staff   68893 Sep 27 14:56 ef.json\n",
-      "-rw-r--r--  1 shaojun  staff   68901 Sep 27 14:56 69.json\n",
-      "-rw-r--r--  1 shaojun  staff   34380 Sep 27 14:56 0a.json\n",
-      "-rw-r--r--  1 shaojun  staff   68823 Sep 27 14:56 11.json\n",
-      "-rw-r--r--  1 shaojun  staff  137767 Sep 27 14:56 81.json\n",
-      "-rw-r--r--  1 shaojun  staff  103298 Sep 27 14:56 5e.json\n",
-      "-rw-r--r--  1 shaojun  staff   34447 Sep 27 14:56 57.json\n",
-      "-rw-r--r--  1 shaojun  staff  103185 Sep 27 14:56 99.json\n",
-      "-rw-r--r--  1 shaojun  staff   34495 Sep 27 14:56 03.json\n",
-      "-rw-r--r--  1 shaojun  staff   68847 Sep 27 14:56 85.json\n",
-      "-rw-r--r--  1 shaojun  staff  103272 Sep 27 14:56 d6.json\n",
-      "-rw-r--r--  1 shaojun  staff  103254 Sep 27 14:56 71.json\n",
-      "-rw-r--r--  1 shaojun  staff  103356 Sep 27 14:56 a4.json\n",
-      "-rw-r--r--  1 shaojun  staff  172215 Sep 27 14:56 c0.json\n",
-      "-rw-r--r--  1 shaojun  staff   68853 Sep 27 14:56 46.json\n",
-      "-rw-r--r--  1 shaojun  staff  172146 Sep 27 14:56 5b.json\n",
-      "-rw-r--r--  1 shaojun  staff   34400 Sep 27 14:56 94.json\n",
-      "-rw-r--r--  1 shaojun  staff   68896 Sep 27 14:56 c1.json\n",
-      "-rw-r--r--  1 shaojun  staff  206481 Sep 27 14:56 bf.json\n",
-      "-rw-r--r--  1 shaojun  staff   68965 Sep 27 14:56 14.json\n",
-      "-rw-r--r--  1 shaojun  staff   68833 Sep 27 14:56 c6.json\n",
-      "-rw-r--r--  1 shaojun  staff   68869 Sep 27 14:56 cd.json\n",
-      "-rw-r--r--  1 shaojun  staff  206469 Sep 27 14:56 59.json\n",
-      "-rw-r--r--  1 shaojun  staff   68849 Sep 27 14:56 ee.json\n",
-      "-rw-r--r--  1 shaojun  staff   68925 Sep 27 14:56 20.json\n",
-      "-rw-r--r--  1 shaojun  staff  137698 Sep 27 14:56 df.json\n",
-      "-rw-r--r--  1 shaojun  staff   68883 Sep 27 14:56 fa.json\n",
-      "-rw-r--r--  1 shaojun  staff   68860 Sep 27 14:56 cb.json\n",
-      "-rw-r--r--  1 shaojun  staff  172188 Sep 27 14:56 63.json\n",
-      "-rw-r--r--  1 shaojun  staff  103252 Sep 27 14:56 e7.json\n",
-      "-rw-r--r--  1 shaojun  staff  172137 Sep 27 14:56 d1.json\n",
-      "-rw-r--r--  1 shaojun  staff  137754 Sep 27 14:56 80.json\n",
-      "-rw-r--r--  1 shaojun  staff  172119 Sep 27 14:56 0d.json\n",
-      "-rw-r--r--  1 shaojun  staff  137728 Sep 27 14:56 8d.json\n",
-      "-rw-r--r--  1 shaojun  staff   34412 Sep 27 14:56 77.json\n",
-      "-rw-r--r--  1 shaojun  staff  137524 Sep 27 14:56 89.json\n",
-      "-rw-r--r--  1 shaojun  staff   68785 Sep 27 14:56 f7.json\n",
-      "-rw-r--r--  1 shaojun  staff  103286 Sep 27 14:56 2e.json\n",
-      "-rw-r--r--  1 shaojun  staff  103284 Sep 27 14:56 cc.json\n",
-      "-rw-r--r--  1 shaojun  staff  103361 Sep 27 14:56 a5.json\n",
-      "-rw-r--r--  1 shaojun  staff  172162 Sep 27 14:56 b8.json\n",
-      "-rw-r--r--  1 shaojun  staff  206554 Sep 27 14:56 1e.json\n",
-      "-rw-r--r--  1 shaojun  staff  137706 Sep 27 14:56 52.json\n",
-      "-rw-r--r--  1 shaojun  staff  241040 Sep 27 14:56 d2.json\n",
-      "-rw-r--r--  1 shaojun  staff  137823 Sep 27 14:56 6b.json\n",
-      "-rw-r--r--  1 shaojun  staff   68845 Sep 27 14:56 50.json\n",
-      "-rw-r--r--  1 shaojun  staff  137872 Sep 27 14:56 f4.json\n",
-      "-rw-r--r--  1 shaojun  staff  103341 Sep 27 14:56 1c.json\n",
-      "-rw-r--r--  1 shaojun  staff  137725 Sep 27 14:56 b5.json\n",
-      "-rw-r--r--  1 shaojun  staff  172264 Sep 27 14:56 d8.json\n",
-      "-rw-r--r--  1 shaojun  staff  137704 Sep 27 14:56 24.json\n",
-      "-rw-r--r--  1 shaojun  staff  172163 Sep 27 14:56 37.json\n",
-      "-rw-r--r--  1 shaojun  staff   68848 Sep 27 14:56 66.json\n",
-      "-rw-r--r--  1 shaojun  staff   68844 Sep 27 14:56 01.json\n",
-      "-rw-r--r--  1 shaojun  staff   68865 Sep 27 14:56 55.json\n",
-      "-rw-r--r--  1 shaojun  staff   68792 Sep 27 14:56 35.json\n",
-      "-rw-r--r--  1 shaojun  staff   68927 Sep 27 14:56 c2.json\n",
-      "-rw-r--r--  1 shaojun  staff   68827 Sep 27 14:56 74.json\n",
-      "-rw-r--r--  1 shaojun  staff   68908 Sep 27 14:56 b3.json\n",
-      "-rw-r--r--  1 shaojun  staff  137767 Sep 27 14:56 86.json\n",
-      "-rw-r--r--  1 shaojun  staff  275521 Sep 27 14:56 b1.json\n",
-      "-rw-r--r--  1 shaojun  staff  172156 Sep 27 14:56 6c.json\n",
-      "-rw-r--r--  1 shaojun  staff  137790 Sep 27 14:56 16.json\n",
-      "-rw-r--r--  1 shaojun  staff   68860 Sep 27 14:56 5c.json\n",
-      "-rw-r--r--  1 shaojun  staff  103273 Sep 27 14:56 0c.json\n",
-      "-rw-r--r--  1 shaojun  staff  103417 Sep 27 14:56 97.json\n",
-      "-rw-r--r--  1 shaojun  staff  137817 Sep 27 14:56 fc.json\n",
-      "-rw-r--r--  1 shaojun  staff  172222 Sep 27 14:56 67.json\n",
-      "-rw-r--r--  1 shaojun  staff  206756 Sep 27 14:56 e6.json\n",
-      "-rw-r--r--  1 shaojun  staff  206553 Sep 27 14:56 8c.json\n",
-      "-rw-r--r--  1 shaojun  staff   34435 Sep 27 14:56 f1.json\n",
-      "-rw-r--r--  1 shaojun  staff   34439 Sep 27 14:56 aa.json\n",
-      "-rw-r--r--  1 shaojun  staff  137738 Sep 27 14:56 78.json\n",
-      "-rw-r--r--  1 shaojun  staff   68843 Sep 27 14:56 79.json\n",
-      "-rw-r--r--  1 shaojun  staff   68841 Sep 27 14:56 45.json\n",
-      "-rw-r--r--  1 shaojun  staff   68858 Sep 27 14:56 47.json\n",
-      "-rw-r--r--  1 shaojun  staff  206624 Sep 27 14:56 62.json\n",
-      "-rw-r--r--  1 shaojun  staff  241112 Sep 27 14:56 7a.json\n",
-      "-rw-r--r--  1 shaojun  staff   68907 Sep 27 14:56 fb.json\n",
-      "-rw-r--r--  1 shaojun  staff  137640 Sep 27 14:56 42.json\n",
-      "-rw-r--r--  1 shaojun  staff  206604 Sep 27 14:56 a6.json\n",
-      "-rw-r--r--  1 shaojun  staff   34483 Sep 27 14:56 dc.json\n",
-      "-rw-r--r--  1 shaojun  staff  103325 Sep 27 14:56 8a.json\n",
-      "-rw-r--r--  1 shaojun  staff  103302 Sep 27 14:56 5d.json\n",
-      "-rw-r--r--  1 shaojun  staff  137741 Sep 27 14:56 07.json\n",
-      "-rw-r--r--  1 shaojun  staff   68926 Sep 27 14:56 c5.json\n",
-      "-rw-r--r--  1 shaojun  staff  137689 Sep 27 14:56 dd.json\n",
-      "-rw-r--r--  1 shaojun  staff  206590 Sep 27 14:56 ac.json\n",
-      "-rw-r--r--  1 shaojun  staff  309939 Sep 27 14:56 83.json\n",
-      "-rw-r--r--  1 shaojun  staff  172106 Sep 27 14:56 f3.json\n",
-      "-rw-r--r--  1 shaojun  staff  137713 Sep 27 14:56 44.json\n",
-      "-rw-r--r--  1 shaojun  staff  137711 Sep 27 14:56 64.json\n",
-      "-rw-r--r--  1 shaojun  staff  275630 Sep 27 14:56 84.json\n",
-      "-rw-r--r--  1 shaojun  staff  241095 Sep 27 14:56 22.json\n",
-      "-rw-r--r--  1 shaojun  staff  206538 Sep 27 14:56 7f.json\n",
-      "-rw-r--r--  1 shaojun  staff  275535 Sep 27 14:56 4f.json\n",
-      "-rw-r--r--  1 shaojun  staff   34431 Sep 27 14:56 8e.json\n",
-      "-rw-r--r--  1 shaojun  staff  137644 Sep 27 14:56 5a.json\n",
-      "-rw-r--r--  1 shaojun  staff  137783 Sep 27 14:56 a9.json\n",
-      "-rw-r--r--  1 shaojun  staff  206504 Sep 27 14:56 8f.json\n",
-      "-rw-r--r--  1 shaojun  staff  137719 Sep 27 14:56 53.json\n",
-      "-rw-r--r--  1 shaojun  staff  103301 Sep 27 14:56 49.json\n",
-      "-rw-r--r--  1 shaojun  staff  103276 Sep 27 14:56 c9.json\n",
-      "-rw-r--r--  1 shaojun  staff  103283 Sep 27 14:56 7d.json\n",
-      "-rw-r--r--  1 shaojun  staff   68833 Sep 27 14:56 a1.json\n",
-      "-rw-r--r--  1 shaojun  staff   68900 Sep 27 14:56 bc.json\n",
-      "-rw-r--r--  1 shaojun  staff  103285 Sep 27 14:56 95.json\n",
-      "-rw-r--r--  1 shaojun  staff  137671 Sep 27 14:56 1f.json\n",
-      "-rw-r--r--  1 shaojun  staff  172106 Sep 27 14:56 43.json\n",
-      "-rw-r--r--  1 shaojun  staff  206653 Sep 27 14:56 91.json\n",
-      "-rw-r--r--  1 shaojun  staff  172226 Sep 27 14:56 25.json\n",
-      "-rw-r--r--  1 shaojun  staff  103391 Sep 27 14:56 ca.json\n",
-      "-rw-r--r--  1 shaojun  staff   68863 Sep 27 14:56 4b.json\n",
-      "-rw-r--r--  1 shaojun  staff  103268 Sep 27 14:56 f6.json\n",
-      "-rw-r--r--  1 shaojun  staff  137847 Sep 27 14:56 f2.json\n",
-      "-rw-r--r--  1 shaojun  staff   68883 Sep 27 14:56 88.json\n",
-      "-rw-r--r--  1 shaojun  staff   68845 Sep 27 14:56 51.json\n",
-      "-rw-r--r--  1 shaojun  staff   68806 Sep 27 14:56 61.json\n",
-      "-rw-r--r--  1 shaojun  staff  103350 Sep 27 14:56 ba.json\n",
-      "-rw-r--r--  1 shaojun  staff  103244 Sep 27 14:56 0b.json\n",
-      "-rw-r--r--  1 shaojun  staff  172225 Sep 27 14:56 a0.json\n",
-      "-rw-r--r--  1 shaojun  staff  137741 Sep 27 14:56 f9.json\n",
-      "-rw-r--r--  1 shaojun  staff  103316 Sep 27 14:56 3c.json\n",
-      "-rw-r--r--  1 shaojun  staff   68851 Sep 27 14:56 e8.json\n",
-      "-rw-r--r--  1 shaojun  staff  137642 Sep 27 14:56 0e.json\n",
-      "-rw-r--r--  1 shaojun  staff  103366 Sep 27 14:56 1d.json\n",
-      "-rw-r--r--  1 shaojun  staff  172252 Sep 27 14:56 db.json\n",
-      "-rw-r--r--  1 shaojun  staff  241013 Sep 27 14:56 06.json\n",
-      "-rw-r--r--  1 shaojun  staff  172004 Sep 27 14:56 a3.json\n",
-      "-rw-r--r--  1 shaojun  staff  137816 Sep 27 14:56 28.json\n",
-      "-rw-r--r--  1 shaojun  staff   68858 Sep 27 14:56 b7.json\n",
-      "-rw-r--r--  1 shaojun  staff  172149 Sep 27 14:56 13.json\n",
-      "-rw-r--r--  1 shaojun  staff  172254 Sep 27 14:56 58.json\n",
-      "-rw-r--r--  1 shaojun  staff  206714 Sep 27 14:57 ad.json\n",
-      "-rw-r--r--  1 shaojun  staff  103351 Sep 27 14:57 60.json\n",
-      "-rw-r--r--  1 shaojun  staff  103427 Sep 27 14:57 3a.json\n",
-      "-rw-r--r--  1 shaojun  staff  137620 Sep 27 14:57 9b.json\n",
-      "-rw-r--r--  1 shaojun  staff  137773 Sep 27 14:57 c4.json\n",
-      "-rw-r--r--  1 shaojun  staff  103339 Sep 27 14:57 9f.json\n",
-      "-rw-r--r--  1 shaojun  staff  137723 Sep 27 14:57 04.json\n",
-      "-rw-r--r--  1 shaojun  staff  103342 Sep 27 14:57 39.json\n",
-      "-rw-r--r--  1 shaojun  staff  137847 Sep 27 14:57 e4.json\n",
-      "-rw-r--r--  1 shaojun  staff  137679 Sep 27 14:57 6a.json\n",
-      "-rw-r--r--  1 shaojun  staff  172200 Sep 27 14:57 b0.json\n",
-      "-rw-r--r--  1 shaojun  staff  206547 Sep 27 14:57 6e.json\n",
-      "-rw-r--r--  1 shaojun  staff  172303 Sep 27 14:57 1b.json\n",
-      "-rw-r--r--  1 shaojun  staff  103265 Sep 27 14:57 e9.json\n",
-      "-rw-r--r--  1 shaojun  staff   68867 Sep 27 14:57 d9.json\n",
-      "-rw-r--r--  1 shaojun  staff  103318 Sep 27 14:57 21.json\n",
-      "-rw-r--r--  1 shaojun  staff  241129 Sep 27 14:57 40.json\n",
-      "-rw-r--r--  1 shaojun  staff  206637 Sep 27 14:57 d7.json\n",
-      "-rw-r--r--  1 shaojun  staff  172314 Sep 27 14:57 2a.json\n",
-      "-rw-r--r--  1 shaojun  staff  172157 Sep 27 14:57 36.json\n",
-      "-rw-r--r--  1 shaojun  staff  275522 Sep 27 14:57 73.json\n",
-      "-rw-r--r--  1 shaojun  staff  103333 Sep 27 14:57 d4.json\n",
-      "-rw-r--r--  1 shaojun  staff   68896 Sep 27 14:57 76.json\n",
-      "-rw-r--r--  1 shaojun  staff  137684 Sep 27 14:57 3e.json\n",
-      "-rw-r--r--  1 shaojun  staff  172392 Sep 27 14:57 d0.json\n",
-      "-rw-r--r--  1 shaojun  staff  172215 Sep 27 14:57 6f.json\n",
-      "-rw-r--r--  1 shaojun  staff  137757 Sep 27 14:57 33.json\n",
-      "-rw-r--r--  1 shaojun  staff  172135 Sep 27 14:57 65.json\n",
-      "-rw-r--r--  1 shaojun  staff  137740 Sep 27 14:57 54.json\n",
-      "-rw-r--r--  1 shaojun  staff  137764 Sep 27 14:57 00.json\n",
-      "-rw-r--r--  1 shaojun  staff  309961 Sep 27 14:57 7b.json\n",
-      "-rw-r--r--  1 shaojun  staff  241028 Sep 27 14:57 7c.json\n",
-      "-rw-r--r--  1 shaojun  staff  137745 Sep 27 14:57 75.json\n",
-      "-rw-r--r--  1 shaojun  staff  206498 Sep 27 14:57 10.json\n",
-      "-rw-r--r--  1 shaojun  staff  103268 Sep 27 14:57 98.json\n",
-      "-rw-r--r--  1 shaojun  staff  137722 Sep 27 14:57 23.json\n",
-      "-rw-r--r--  1 shaojun  staff  137687 Sep 27 14:57 a8.json\n",
-      "-rw-r--r--  1 shaojun  staff   68899 Sep 27 14:57 90.json\n",
-      "-rw-r--r--  1 shaojun  staff  137754 Sep 27 14:57 2b.json\n",
-      "-rw-r--r--  1 shaojun  staff  344431 Sep 27 14:57 d3.json\n",
-      "-rw-r--r--  1 shaojun  staff  137755 Sep 27 14:57 fd.json\n",
-      "-rw-r--r--  1 shaojun  staff  206556 Sep 27 14:57 4d.json\n",
-      "-rw-r--r--  1 shaojun  staff  103244 Sep 27 14:57 93.json\n",
-      "-rw-r--r--  1 shaojun  staff   68888 Sep 27 14:57 b6.json\n",
-      "-rw-r--r--  1 shaojun  staff  137709 Sep 27 14:57 3d.json\n",
-      "-rw-r--r--  1 shaojun  staff   34378 Sep 27 14:57 e3.json\n",
-      "-rw-r--r--  1 shaojun  staff  137797 Sep 27 14:57 38.json\n",
-      "-rw-r--r--  1 shaojun  staff  103484 Sep 27 14:57 bb.json\n",
-      "-rw-r--r--  1 shaojun  staff  206520 Sep 27 14:57 09.json\n",
-      "-rw-r--r--  1 shaojun  staff  137697 Sep 27 14:57 f0.json\n",
-      "-rw-r--r--  1 shaojun  staff  206502 Sep 27 14:57 48.json\n",
-      "-rw-r--r--  1 shaojun  staff  172162 Sep 27 14:57 2c.json\n",
-      "-rw-r--r--  1 shaojun  staff  172015 Sep 27 14:57 12.json\n",
-      "-rw-r--r--  1 shaojun  staff  206671 Sep 27 14:57 fe.json\n",
-      "-rw-r--r--  1 shaojun  staff  137729 Sep 27 14:57 da.json\n",
-      "-rw-r--r--  1 shaojun  staff  241042 Sep 27 14:57 34.json\n",
-      "-rw-r--r--  1 shaojun  staff   68847 Sep 27 14:57 d5.json\n",
-      "-rw-r--r--  1 shaojun  staff  206597 Sep 27 14:57 4c.json\n",
-      "-rw-r--r--  1 shaojun  staff  206774 Sep 27 14:57 72.json\n",
-      "-rw-r--r--  1 shaojun  staff  137697 Sep 27 14:57 9a.json\n",
-      "-rw-r--r--  1 shaojun  staff  206642 Sep 27 14:57 e2.json\n",
-      "-rw-r--r--  1 shaojun  staff  137795 Sep 27 14:57 bd.json\n",
-      "-rw-r--r--  1 shaojun  staff  103371 Sep 27 14:57 6d.json\n",
-      "-rw-r--r--  1 shaojun  staff  137769 Sep 27 14:57 e5.json\n",
-      "-rw-r--r--  1 shaojun  staff  172314 Sep 27 14:57 5f.json\n",
-      "-rw-r--r--  1 shaojun  staff  309865 Sep 27 14:57 be.json\n",
-      "-rw-r--r--  1 shaojun  staff  240982 Sep 27 14:57 ce.json\n",
-      "-rw-r--r--  1 shaojun  staff   68896 Sep 27 14:57 ea.json\n",
-      "-rw-r--r--  1 shaojun  staff  103251 Sep 27 14:57 92.json\n",
-      "-rw-r--r--  1 shaojun  staff  275642 Sep 27 14:57 68.json\n",
-      "-rw-r--r--  1 shaojun  staff   68858 Sep 27 14:57 2f.json\n",
-      "-rw-r--r--  1 shaojun  staff  241160 Sep 27 14:57 96.json\n",
-      "-rw-r--r--  1 shaojun  staff  137795 Sep 27 14:57 18.json\n",
-      "-rw-r--r--  1 shaojun  staff  137675 Sep 27 14:57 4a.json\n",
-      "-rw-r--r--  1 shaojun  staff  241006 Sep 27 14:57 af.json\n",
-      "-rw-r--r--  1 shaojun  staff  103422 Sep 27 14:57 e1.json\n",
-      "-rw-r--r--  1 shaojun  staff  137731 Sep 27 14:57 a7.json\n",
-      "-rw-r--r--  1 shaojun  staff   34446 Sep 27 14:57 c3.json\n",
-      "-rw-r--r--  1 shaojun  staff   68885 Sep 27 14:57 ec.json\n",
-      "-rw-r--r--  1 shaojun  staff  206571 Sep 27 14:57 e0.json\n",
-      "-rw-r--r--  1 shaojun  staff  172188 Sep 27 14:57 30.json\n",
-      "-rw-r--r--  1 shaojun  staff  206663 Sep 27 14:57 7e.json\n",
-      "-rw-r--r--  1 shaojun  staff  137848 Sep 27 14:57 f5.json\n",
-      "-rw-r--r--  1 shaojun  staff  172141 Sep 27 14:57 41.json\n",
-      "-rw-r--r--  1 shaojun  staff  137610 Sep 27 14:57 8b.json\n",
-      "-rw-r--r--  1 shaojun  staff  103309 Sep 27 14:57 c8.json\n",
-      "-rw-r--r--  1 shaojun  staff  206630 Sep 27 14:57 c7.json\n",
-      "-rw-r--r--  1 shaojun  staff  103359 Sep 27 14:57 32.json\n",
-      "-rw-r--r--  1 shaojun  staff   68889 Sep 27 14:57 19.json\n",
-      "-rw-r--r--  1 shaojun  staff  137774 Sep 27 14:57 17.json\n",
-      "-rw-r--r--  1 shaojun  staff  172198 Sep 27 14:57 4e.json\n",
-      "-rw-r--r--  1 shaojun  staff  172161 Sep 27 14:57 02.json\n",
-      "-rw-r--r--  1 shaojun  staff  275562 Sep 27 14:57 15.json\n",
-      "-rw-r--r--  1 shaojun  staff  137797 Sep 27 14:57 ff.json\n",
-      "-rw-r--r--  1 shaojun  staff  137735 Sep 27 14:57 9d.json\n",
-      "-rw-r--r--  1 shaojun  staff   68871 Sep 27 14:57 2d.json\n",
-      "-rw-r--r--  1 shaojun  staff  206515 Sep 27 14:57 de.json\n",
-      "-rw-r--r--  1 shaojun  staff  137603 Sep 27 14:57 f8.json\n",
-      "-rw-r--r--  1 shaojun  staff   68836 Sep 27 14:57 b2.json\n",
-      "-rw-r--r--  1 shaojun  staff  172220 Sep 27 14:57 9e.json\n",
-      "-rw-r--r--  1 shaojun  staff  172181 Sep 27 14:57 56.json\n",
-      "-rw-r--r--  1 shaojun  staff  413179 Sep 27 14:57 87.json\n",
-      "-rw-r--r--  1 shaojun  staff  206617 Sep 27 14:57 a2.json\n",
-      "-rw-r--r--  1 shaojun  staff  241003 Sep 27 14:57 ed.json\n",
-      "-rw-r--r--  1 shaojun  staff   68848 Sep 27 14:57 b4.json\n",
-      "-rw-r--r--  1 shaojun  staff  137642 Sep 27 14:57 3f.json\n",
-      "-rw-r--r--  1 shaojun  staff   68898 Sep 27 14:57 08.json\n",
-      "-rw-r--r--  1 shaojun  staff  206596 Sep 27 14:57 0f.json\n",
-      "-rw-r--r--  1 shaojun  staff  206643 Sep 27 14:57 ae.json\n",
-      "-rw-r--r--  1 shaojun  staff  275430 Sep 27 14:57 eb.json\n",
-      "-rw-r--r--  1 shaojun  staff  172207 Sep 27 14:57 1a.json\n",
-      "-rw-r--r--  1 shaojun  staff  206596 Sep 27 14:57 3b.json\n",
-      "-rw-r--r--  1 shaojun  staff  103198 Sep 27 14:57 31.json\n",
-      "-rw-r--r--  1 shaojun  staff  241067 Sep 27 14:57 b9.json\n",
-      "-rw-r--r--  1 shaojun  staff  172223 Sep 27 14:57 70.json\n",
-      "-rw-r--r--  1 shaojun  staff  172276 Sep 27 14:57 26.json\n"
-     ]
-    }
-   ],
-   "source": [
-    "!ls -lrt ../../../data/{game.PATH}/emb/"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 19,
-   "id": "ef4e291e-d228-4eb0-8c0e-1d0fefc53ba2",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "!rm -rf ../../../data/{game.PATH}/emb/"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "a99ec58e-8d08-4fc4-b1bb-c6bfa4392dbc",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 20,
-   "id": "84d3897c-235d-4c23-b6d1-3064efde1aa1",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "100%|██████████████████████████████████████| 1016/1016 [00:08<00:00, 125.41it/s]\n"
-     ]
-    }
-   ],
-   "source": [
-    "game.dump()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "47cba7a0-9492-46b2-b596-7dda56045ee0",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 21,
-   "id": "0ab453f0-f04d-4a40-97fb-d6aba6275757",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "PosixPath('/Users/shaojun/playgo/bbc123')"
-      ]
-     },
-     "execution_count": 21,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "game.ROOT"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 22,
-   "id": "42e1eaad-b66f-4bd3-85a6-f352d4296ab0",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'angrybird'"
-      ]
-     },
-     "execution_count": 22,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "game.PATH"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 23,
-   "id": "6f4dd0dd-927d-4b99-84b5-5a30b11f064d",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import glob\n",
-    "files = glob.glob(f\"{game.ROOT}/data/{game.PATH}/emb/*.json\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 24,
-   "id": "e5f485ca-d901-485b-adff-d348d0c2168b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "250"
-      ]
-     },
-     "execution_count": 24,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(files)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 25,
-   "id": "7f0dd810-f8d0-4c9e-b04b-c2ae7a010d19",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import glob\n",
-    "files = glob.glob(f\"{game.ROOT}/data/{game.PATH}/emb/*.json\")\n",
-    "tmp = {}\n",
-    "for f in files:\n",
-    "    with open(f, \"r\") as f:\n",
-    "        tmp |= json.load(f)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 26,
-   "id": "17078c2d-6455-433a-b03d-e23fdbb5755d",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "1016"
-      ]
-     },
-     "execution_count": 26,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(tmp)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "4e57785e-a841-4a43-b324-a4d07493e333",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 27,
-   "id": "18888a7b-d32d-45b4-a9b3-3ce29b18a7f7",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# sim query"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 28,
-   "id": "4703a269-a1c0-48e4-8848-6b5cdb9b814b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "1016"
-      ]
-     },
-     "execution_count": 28,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(game.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "c885da07-e53d-4586-b340-8f9de0515fc2",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 29,
-   "id": "37ba9cb2-83a7-45af-972c-6802ce48a57b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "True"
-      ]
-     },
-     "execution_count": 29,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "game.embeddings == tmp"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "63827499-7b11-4c3f-8527-38d3cc5066d6",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "8b06eff9-5382-4c28-aa7c-cb262793dd6f",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 30,
-   "id": "4ff48663-e767-4936-807d-07fb8c42ce92",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "{\n",
-      "    \"version\": \"3.8.0.2023-09-18-a\",\n",
-      "    \"data\": [\n",
-      "        \"(1) [unified] add/delete one or more (special) birds/pigs\",\n",
-      "        \"(2) add one or more houses\",\n",
-      "        \"(3) change HP\",\n",
-      "        \"(4) change background\",\n",
-      "        \"(5) change positions\",\n",
-      "        \"(6) change size\",\n",
-      "        \"(7) change skin\",\n",
-      "        \"(8) list birds or pigs\"\n",
-      "    ]\n",
-      "}\n"
-     ]
-    }
-   ],
-   "source": [
-    "print(json.dumps(game.list_functionality(), indent=4))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 31,
-   "id": "982c3666-7b6a-4300-a2ac-b1a42c216c09",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "0 (1) [unified] add/delete one or more (special) birds/pigs\n",
-      "1 (2) add one or more houses\n",
-      "2 (3) change HP\n",
-      "3 (4) change background\n",
-      "4 (5) change positions\n",
-      "5 (6) change size\n",
-      "6 (7) change skin\n",
-      "7 (8) list birds or pigs\n"
-     ]
-    }
-   ],
-   "source": [
-    "for i, x in enumerate(game.list_functionality()['data']):\n",
-    "    print(i,x)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "73820ce5-bade-4e08-9fc8-b60d8cdb4ba4",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 32,
-   "id": "75cf385f-ba62-452d-a98a-fd3d0bd36031",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# print(json.dumps(game.show_json(),indent=4))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "b3db3bf1-1b0d-443c-a1cd-cd8c10038b31",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.4"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}

games/angrybirds/notebooks/sim.ipynb

@@ -1,730 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 64,
-   "id": "245fbc34-352d-4945-a689-d04f9365bb99",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "The autoreload extension is already loaded. To reload it, use:\n",
-      "  %reload_ext autoreload\n"
-     ]
-    }
-   ],
-   "source": [
-    "%load_ext autoreload\n",
-    "%autoreload 2"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 65,
-   "id": "eaed216c-160a-4e31-96fe-015c140ff18b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# from pprint import pprint"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 66,
-   "id": "97953551-8ef1-42c1-9317-99d5e65f535e",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import json\n",
-    "import pandas as pd\n",
-    "import pickle\n",
-    "\n",
-    "import openai\n",
-    "from openai.embeddings_utils import get_embedding, cosine_similarity\n",
-    "\n",
-    "# from games.angrybirds.util import util\n",
-    "from games.angrybirds.config import config\n",
-    "# from games.angrybirds.prompt import prompt\n",
-    "# from games.angrybirds.query import query, yes_no_query\n",
-    "from games.angrybirds.angrybirds import angrybird"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "fc213f4c-cb07-468b-b15f-60e5120b3d94",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 67,
-   "id": "0c876614-a9bf-4d0b-a728-07770f1ad9c5",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "before add_config\n",
-      "773\n",
-      "773\n",
-      "773\n",
-      "773\n",
-      "774\n",
-      "after add_config\n",
-      "774\n",
-      "final\n",
-      "774\n"
-     ]
-    }
-   ],
-   "source": [
-    "bird = angrybird()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 68,
-   "id": "1411de96-97d5-4bac-b8ed-e279fb1edaa0",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "0"
-      ]
-     },
-     "execution_count": 68,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 69,
-   "id": "129403ce-69f4-4f2a-9b22-7b3e36d78f24",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{}"
-      ]
-     },
-     "execution_count": 69,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "bird.embeddings"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 70,
-   "id": "50e372eb-48ba-4378-b866-f2b7120bc94e",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "774"
-      ]
-     },
-     "execution_count": 70,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.train)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "f78b2c43-bfc7-478f-894d-d757b0e9927f",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 71,
-   "id": "163545c6-c707-401a-93b8-fb2cb9d65136",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "initializing...\n",
-      "loading embeddings from a file...\n",
-      "/Users/shaojun/playgo/abc123/embeddings.pkl\n",
-      "pkl size: 773\n",
-      "embedding size: 773\n",
-      "creating embeddings by query OpenAI...\n",
-      "query 'i don't like antigravity birds. delete all of them'...\n",
-      "embedding size: 774\n",
-      "train size: 774\n"
-     ]
-    }
-   ],
-   "source": [
-    "bird.init_emb()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 9,
-   "id": "31c02b28-0078-425d-87b6-702caeea02e3",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "773"
-      ]
-     },
-     "execution_count": 9,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "067905b6-b030-4df1-9570-d45885f775eb",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 10,
-   "id": "4703a269-a1c0-48e4-8848-6b5cdb9b814b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "773"
-      ]
-     },
-     "execution_count": 10,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(bird.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "75cf385f-ba62-452d-a98a-fd3d0bd36031",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 39,
-   "id": "d659ff93-eb35-4ef9-bf50-30e0c2206961",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "s = \"I don't like antigravity birds. Delete all of them\""
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 40,
-   "id": "c90b7510-09e0-4814-8327-26b56b9f9da5",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# s = 'delete antigravity birds'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 44,
-   "id": "2a0ddee9-883c-4e97-b62c-7a7c58b9fa42",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "skip init\n"
-     ]
-    },
-    {
-     "ename": "RetryError",
-     "evalue": "RetryError[<Future at 0x12a3decd0 state=finished raised APIConnectionError>]",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
-      "\u001b[0;31mConnectionRefusedError\u001b[0m                    Traceback (most recent call last)",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connection.py:200\u001b[0m, in \u001b[0;36mHTTPConnection._new_conn\u001b[0;34m(self)\u001b[0m\n\u001b[1;32m    199\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m--> 200\u001b[0m     sock \u001b[38;5;241m=\u001b[39m \u001b[43mconnection\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mcreate_connection\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    201\u001b[0m \u001b[43m        \u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43m_dns_host\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mport\u001b[49m\u001b[43m)\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    202\u001b[0m \u001b[43m        \u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mtimeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    203\u001b[0m \u001b[43m        \u001b[49m\u001b[43msource_address\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43msource_address\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    204\u001b[0m \u001b[43m        \u001b[49m\u001b[43msocket_options\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43msocket_options\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    205\u001b[0m \u001b[43m    \u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    206\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m socket\u001b[38;5;241m.\u001b[39mgaierror \u001b[38;5;28;01mas\u001b[39;00m e:\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/util/connection.py:85\u001b[0m, in \u001b[0;36mcreate_connection\u001b[0;34m(address, timeout, source_address, socket_options)\u001b[0m\n\u001b[1;32m     84\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m---> 85\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m err\n\u001b[1;32m     86\u001b[0m \u001b[38;5;28;01mfinally\u001b[39;00m:\n\u001b[1;32m     87\u001b[0m     \u001b[38;5;66;03m# Break explicitly a reference cycle\u001b[39;00m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/util/connection.py:73\u001b[0m, in \u001b[0;36mcreate_connection\u001b[0;34m(address, timeout, source_address, socket_options)\u001b[0m\n\u001b[1;32m     72\u001b[0m     sock\u001b[38;5;241m.\u001b[39mbind(source_address)\n\u001b[0;32m---> 73\u001b[0m \u001b[43msock\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mconnect\u001b[49m\u001b[43m(\u001b[49m\u001b[43msa\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m     74\u001b[0m \u001b[38;5;66;03m# Break explicitly a reference cycle\u001b[39;00m\n",
-      "\u001b[0;31mConnectionRefusedError\u001b[0m: [Errno 61] Connection refused",
-      "\nThe above exception was the direct cause of the following exception:\n",
-      "\u001b[0;31mNewConnectionError\u001b[0m                        Traceback (most recent call last)",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py:776\u001b[0m, in \u001b[0;36mHTTPConnectionPool.urlopen\u001b[0;34m(self, method, url, body, headers, retries, redirect, assert_same_host, timeout, pool_timeout, release_conn, chunked, body_pos, preload_content, decode_content, **response_kw)\u001b[0m\n\u001b[1;32m    775\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m--> 776\u001b[0m     \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43m_prepare_proxy\u001b[49m\u001b[43m(\u001b[49m\u001b[43mconn\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    777\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m (BaseSSLError, \u001b[38;5;167;01mOSError\u001b[39;00m, SocketTimeout) \u001b[38;5;28;01mas\u001b[39;00m e:\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py:1041\u001b[0m, in \u001b[0;36mHTTPSConnectionPool._prepare_proxy\u001b[0;34m(self, conn)\u001b[0m\n\u001b[1;32m   1035\u001b[0m conn\u001b[38;5;241m.\u001b[39mset_tunnel(\n\u001b[1;32m   1036\u001b[0m     scheme\u001b[38;5;241m=\u001b[39mtunnel_scheme,\n\u001b[1;32m   1037\u001b[0m     host\u001b[38;5;241m=\u001b[39m\u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39m_tunnel_host,\n\u001b[1;32m   1038\u001b[0m     port\u001b[38;5;241m=\u001b[39m\u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mport,\n\u001b[1;32m   1039\u001b[0m     headers\u001b[38;5;241m=\u001b[39m\u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mproxy_headers,\n\u001b[1;32m   1040\u001b[0m )\n\u001b[0;32m-> 1041\u001b[0m \u001b[43mconn\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mconnect\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connection.py:604\u001b[0m, in \u001b[0;36mHTTPSConnection.connect\u001b[0;34m(self)\u001b[0m\n\u001b[1;32m    603\u001b[0m sock: socket\u001b[38;5;241m.\u001b[39msocket \u001b[38;5;241m|\u001b[39m ssl\u001b[38;5;241m.\u001b[39mSSLSocket\n\u001b[0;32m--> 604\u001b[0m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39msock \u001b[38;5;241m=\u001b[39m sock \u001b[38;5;241m=\u001b[39m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43m_new_conn\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    605\u001b[0m server_hostname: \u001b[38;5;28mstr\u001b[39m \u001b[38;5;241m=\u001b[39m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mhost\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connection.py:215\u001b[0m, in \u001b[0;36mHTTPConnection._new_conn\u001b[0;34m(self)\u001b[0m\n\u001b[1;32m    214\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m \u001b[38;5;167;01mOSError\u001b[39;00m \u001b[38;5;28;01mas\u001b[39;00m e:\n\u001b[0;32m--> 215\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m NewConnectionError(\n\u001b[1;32m    216\u001b[0m         \u001b[38;5;28mself\u001b[39m, \u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mFailed to establish a new connection: \u001b[39m\u001b[38;5;132;01m{\u001b[39;00me\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m\"\u001b[39m\n\u001b[1;32m    217\u001b[0m     ) \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01me\u001b[39;00m\n\u001b[1;32m    219\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m sock\n",
-      "\u001b[0;31mNewConnectionError\u001b[0m: <urllib3.connection.HTTPSConnection object at 0x12a3dd350>: Failed to establish a new connection: [Errno 61] Connection refused",
-      "\nThe above exception was the direct cause of the following exception:\n",
-      "\u001b[0;31mProxyError\u001b[0m                                Traceback (most recent call last)",
-      "\u001b[0;31mProxyError\u001b[0m: ('Unable to connect to proxy', NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x12a3dd350>: Failed to establish a new connection: [Errno 61] Connection refused'))",
-      "\nThe above exception was the direct cause of the following exception:\n",
-      "\u001b[0;31mMaxRetryError\u001b[0m                             Traceback (most recent call last)",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/requests/adapters.py:486\u001b[0m, in \u001b[0;36mHTTPAdapter.send\u001b[0;34m(self, request, stream, timeout, verify, cert, proxies)\u001b[0m\n\u001b[1;32m    485\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m--> 486\u001b[0m     resp \u001b[38;5;241m=\u001b[39m \u001b[43mconn\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43murlopen\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    487\u001b[0m \u001b[43m        \u001b[49m\u001b[43mmethod\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrequest\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mmethod\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    488\u001b[0m \u001b[43m        \u001b[49m\u001b[43murl\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43murl\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    489\u001b[0m \u001b[43m        \u001b[49m\u001b[43mbody\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrequest\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mbody\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    490\u001b[0m \u001b[43m        \u001b[49m\u001b[43mheaders\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrequest\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    491\u001b[0m \u001b[43m        \u001b[49m\u001b[43mredirect\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43;01mFalse\u001b[39;49;00m\u001b[43m,\u001b[49m\n\u001b[1;32m    492\u001b[0m \u001b[43m        \u001b[49m\u001b[43massert_same_host\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43;01mFalse\u001b[39;49;00m\u001b[43m,\u001b[49m\n\u001b[1;32m    493\u001b[0m \u001b[43m        \u001b[49m\u001b[43mpreload_content\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43;01mFalse\u001b[39;49;00m\u001b[43m,\u001b[49m\n\u001b[1;32m    494\u001b[0m \u001b[43m        \u001b[49m\u001b[43mdecode_content\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43;01mFalse\u001b[39;49;00m\u001b[43m,\u001b[49m\n\u001b[1;32m    495\u001b[0m \u001b[43m        \u001b[49m\u001b[43mretries\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mmax_retries\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    496\u001b[0m \u001b[43m        \u001b[49m\u001b[43mtimeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mtimeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    497\u001b[0m \u001b[43m        \u001b[49m\u001b[43mchunked\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mchunked\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    498\u001b[0m \u001b[43m    \u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    500\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m (ProtocolError, \u001b[38;5;167;01mOSError\u001b[39;00m) \u001b[38;5;28;01mas\u001b[39;00m err:\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py:874\u001b[0m, in \u001b[0;36mHTTPConnectionPool.urlopen\u001b[0;34m(self, method, url, body, headers, retries, redirect, assert_same_host, timeout, pool_timeout, release_conn, chunked, body_pos, preload_content, decode_content, **response_kw)\u001b[0m\n\u001b[1;32m    871\u001b[0m     log\u001b[38;5;241m.\u001b[39mwarning(\n\u001b[1;32m    872\u001b[0m         \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mRetrying (\u001b[39m\u001b[38;5;132;01m%r\u001b[39;00m\u001b[38;5;124m) after connection broken by \u001b[39m\u001b[38;5;124m'\u001b[39m\u001b[38;5;132;01m%r\u001b[39;00m\u001b[38;5;124m'\u001b[39m\u001b[38;5;124m: \u001b[39m\u001b[38;5;132;01m%s\u001b[39;00m\u001b[38;5;124m\"\u001b[39m, retries, err, url\n\u001b[1;32m    873\u001b[0m     )\n\u001b[0;32m--> 874\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43murlopen\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    875\u001b[0m \u001b[43m        \u001b[49m\u001b[43mmethod\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    876\u001b[0m \u001b[43m        \u001b[49m\u001b[43murl\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    877\u001b[0m \u001b[43m        \u001b[49m\u001b[43mbody\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    878\u001b[0m \u001b[43m        \u001b[49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    879\u001b[0m \u001b[43m        \u001b[49m\u001b[43mretries\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    880\u001b[0m \u001b[43m        \u001b[49m\u001b[43mredirect\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    881\u001b[0m \u001b[43m        \u001b[49m\u001b[43massert_same_host\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    882\u001b[0m \u001b[43m        \u001b[49m\u001b[43mtimeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mtimeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    883\u001b[0m \u001b[43m        \u001b[49m\u001b[43mpool_timeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mpool_timeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    884\u001b[0m \u001b[43m        \u001b[49m\u001b[43mrelease_conn\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrelease_conn\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    885\u001b[0m \u001b[43m        \u001b[49m\u001b[43mchunked\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mchunked\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    886\u001b[0m \u001b[43m        \u001b[49m\u001b[43mbody_pos\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mbody_pos\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    887\u001b[0m \u001b[43m        \u001b[49m\u001b[43mpreload_content\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mpreload_content\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    888\u001b[0m \u001b[43m        \u001b[49m\u001b[43mdecode_content\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mdecode_content\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    889\u001b[0m \u001b[43m        \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mresponse_kw\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    890\u001b[0m \u001b[43m    \u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    892\u001b[0m \u001b[38;5;66;03m# Handle redirect?\u001b[39;00m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py:874\u001b[0m, in \u001b[0;36mHTTPConnectionPool.urlopen\u001b[0;34m(self, method, url, body, headers, retries, redirect, assert_same_host, timeout, pool_timeout, release_conn, chunked, body_pos, preload_content, decode_content, **response_kw)\u001b[0m\n\u001b[1;32m    871\u001b[0m     log\u001b[38;5;241m.\u001b[39mwarning(\n\u001b[1;32m    872\u001b[0m         \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mRetrying (\u001b[39m\u001b[38;5;132;01m%r\u001b[39;00m\u001b[38;5;124m) after connection broken by \u001b[39m\u001b[38;5;124m'\u001b[39m\u001b[38;5;132;01m%r\u001b[39;00m\u001b[38;5;124m'\u001b[39m\u001b[38;5;124m: \u001b[39m\u001b[38;5;132;01m%s\u001b[39;00m\u001b[38;5;124m\"\u001b[39m, retries, err, url\n\u001b[1;32m    873\u001b[0m     )\n\u001b[0;32m--> 874\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43murlopen\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    875\u001b[0m \u001b[43m        \u001b[49m\u001b[43mmethod\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    876\u001b[0m \u001b[43m        \u001b[49m\u001b[43murl\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    877\u001b[0m \u001b[43m        \u001b[49m\u001b[43mbody\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    878\u001b[0m \u001b[43m        \u001b[49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    879\u001b[0m \u001b[43m        \u001b[49m\u001b[43mretries\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    880\u001b[0m \u001b[43m        \u001b[49m\u001b[43mredirect\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    881\u001b[0m \u001b[43m        \u001b[49m\u001b[43massert_same_host\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    882\u001b[0m \u001b[43m        \u001b[49m\u001b[43mtimeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mtimeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    883\u001b[0m \u001b[43m        \u001b[49m\u001b[43mpool_timeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mpool_timeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    884\u001b[0m \u001b[43m        \u001b[49m\u001b[43mrelease_conn\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrelease_conn\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    885\u001b[0m \u001b[43m        \u001b[49m\u001b[43mchunked\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mchunked\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    886\u001b[0m \u001b[43m        \u001b[49m\u001b[43mbody_pos\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mbody_pos\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    887\u001b[0m \u001b[43m        \u001b[49m\u001b[43mpreload_content\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mpreload_content\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    888\u001b[0m \u001b[43m        \u001b[49m\u001b[43mdecode_content\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mdecode_content\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    889\u001b[0m \u001b[43m        \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mresponse_kw\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    890\u001b[0m \u001b[43m    \u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    892\u001b[0m \u001b[38;5;66;03m# Handle redirect?\u001b[39;00m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py:844\u001b[0m, in \u001b[0;36mHTTPConnectionPool.urlopen\u001b[0;34m(self, method, url, body, headers, retries, redirect, assert_same_host, timeout, pool_timeout, release_conn, chunked, body_pos, preload_content, decode_content, **response_kw)\u001b[0m\n\u001b[1;32m    842\u001b[0m     new_e \u001b[38;5;241m=\u001b[39m ProtocolError(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mConnection aborted.\u001b[39m\u001b[38;5;124m\"\u001b[39m, new_e)\n\u001b[0;32m--> 844\u001b[0m retries \u001b[38;5;241m=\u001b[39m \u001b[43mretries\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mincrement\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    845\u001b[0m \u001b[43m    \u001b[49m\u001b[43mmethod\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43murl\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43merror\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mnew_e\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43m_pool\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43m_stacktrace\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43msys\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mexc_info\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\u001b[43m[\u001b[49m\u001b[38;5;241;43m2\u001b[39;49m\u001b[43m]\u001b[49m\n\u001b[1;32m    846\u001b[0m \u001b[43m\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    847\u001b[0m retries\u001b[38;5;241m.\u001b[39msleep()\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/util/retry.py:515\u001b[0m, in \u001b[0;36mRetry.increment\u001b[0;34m(self, method, url, response, error, _pool, _stacktrace)\u001b[0m\n\u001b[1;32m    514\u001b[0m     reason \u001b[38;5;241m=\u001b[39m error \u001b[38;5;129;01mor\u001b[39;00m ResponseError(cause)\n\u001b[0;32m--> 515\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m MaxRetryError(_pool, url, reason) \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mreason\u001b[39;00m  \u001b[38;5;66;03m# type: ignore[arg-type]\u001b[39;00m\n\u001b[1;32m    517\u001b[0m log\u001b[38;5;241m.\u001b[39mdebug(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mIncremented Retry for (url=\u001b[39m\u001b[38;5;124m'\u001b[39m\u001b[38;5;132;01m%s\u001b[39;00m\u001b[38;5;124m'\u001b[39m\u001b[38;5;124m): \u001b[39m\u001b[38;5;132;01m%r\u001b[39;00m\u001b[38;5;124m\"\u001b[39m, url, new_retry)\n",
-      "\u001b[0;31mMaxRetryError\u001b[0m: HTTPSConnectionPool(host='api.openai.com', port=443): Max retries exceeded with url: /v1/engines/text-embedding-ada-002/embeddings (Caused by ProxyError('Unable to connect to proxy', NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x12a3dd350>: Failed to establish a new connection: [Errno 61] Connection refused')))",
-      "\nDuring handling of the above exception, another exception occurred:\n",
-      "\u001b[0;31mProxyError\u001b[0m                                Traceback (most recent call last)",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/openai/api_requestor.py:596\u001b[0m, in \u001b[0;36mAPIRequestor.request_raw\u001b[0;34m(self, method, url, params, supplied_headers, files, stream, request_id, request_timeout)\u001b[0m\n\u001b[1;32m    595\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m--> 596\u001b[0m     result \u001b[38;5;241m=\u001b[39m \u001b[43m_thread_context\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43msession\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mrequest\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    597\u001b[0m \u001b[43m        \u001b[49m\u001b[43mmethod\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    598\u001b[0m \u001b[43m        \u001b[49m\u001b[43mabs_url\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    599\u001b[0m \u001b[43m        \u001b[49m\u001b[43mheaders\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    600\u001b[0m \u001b[43m        \u001b[49m\u001b[43mdata\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mdata\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    601\u001b[0m \u001b[43m        \u001b[49m\u001b[43mfiles\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mfiles\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    602\u001b[0m \u001b[43m        \u001b[49m\u001b[43mstream\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mstream\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    603\u001b[0m \u001b[43m        \u001b[49m\u001b[43mtimeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrequest_timeout\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;28;43;01mif\u001b[39;49;00m\u001b[43m \u001b[49m\u001b[43mrequest_timeout\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;28;43;01melse\u001b[39;49;00m\u001b[43m \u001b[49m\u001b[43mTIMEOUT_SECS\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    604\u001b[0m \u001b[43m        \u001b[49m\u001b[43mproxies\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43m_thread_context\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43msession\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mproxies\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    605\u001b[0m \u001b[43m    \u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    606\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m requests\u001b[38;5;241m.\u001b[39mexceptions\u001b[38;5;241m.\u001b[39mTimeout \u001b[38;5;28;01mas\u001b[39;00m e:\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/requests/sessions.py:589\u001b[0m, in \u001b[0;36mSession.request\u001b[0;34m(self, method, url, params, data, headers, cookies, files, auth, timeout, allow_redirects, proxies, hooks, stream, verify, cert, json)\u001b[0m\n\u001b[1;32m    588\u001b[0m send_kwargs\u001b[38;5;241m.\u001b[39mupdate(settings)\n\u001b[0;32m--> 589\u001b[0m resp \u001b[38;5;241m=\u001b[39m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43msend\u001b[49m\u001b[43m(\u001b[49m\u001b[43mprep\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43msend_kwargs\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    591\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m resp\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/requests/sessions.py:703\u001b[0m, in \u001b[0;36mSession.send\u001b[0;34m(self, request, **kwargs)\u001b[0m\n\u001b[1;32m    702\u001b[0m \u001b[38;5;66;03m# Send the request\u001b[39;00m\n\u001b[0;32m--> 703\u001b[0m r \u001b[38;5;241m=\u001b[39m \u001b[43madapter\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43msend\u001b[49m\u001b[43m(\u001b[49m\u001b[43mrequest\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mkwargs\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    705\u001b[0m \u001b[38;5;66;03m# Total elapsed time of the request (approximately)\u001b[39;00m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/requests/adapters.py:513\u001b[0m, in \u001b[0;36mHTTPAdapter.send\u001b[0;34m(self, request, stream, timeout, verify, cert, proxies)\u001b[0m\n\u001b[1;32m    512\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;28misinstance\u001b[39m(e\u001b[38;5;241m.\u001b[39mreason, _ProxyError):\n\u001b[0;32m--> 513\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m ProxyError(e, request\u001b[38;5;241m=\u001b[39mrequest)\n\u001b[1;32m    515\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;28misinstance\u001b[39m(e\u001b[38;5;241m.\u001b[39mreason, _SSLError):\n\u001b[1;32m    516\u001b[0m     \u001b[38;5;66;03m# This branch is for urllib3 v1.22 and later.\u001b[39;00m\n",
-      "\u001b[0;31mProxyError\u001b[0m: HTTPSConnectionPool(host='api.openai.com', port=443): Max retries exceeded with url: /v1/engines/text-embedding-ada-002/embeddings (Caused by ProxyError('Unable to connect to proxy', NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x12a3dd350>: Failed to establish a new connection: [Errno 61] Connection refused')))",
-      "\nThe above exception was the direct cause of the following exception:\n",
-      "\u001b[0;31mAPIConnectionError\u001b[0m                        Traceback (most recent call last)",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/tenacity/__init__.py:382\u001b[0m, in \u001b[0;36mRetrying.__call__\u001b[0;34m(self, fn, *args, **kwargs)\u001b[0m\n\u001b[1;32m    381\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m--> 382\u001b[0m     result \u001b[38;5;241m=\u001b[39m \u001b[43mfn\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43margs\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mkwargs\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    383\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m \u001b[38;5;167;01mBaseException\u001b[39;00m:  \u001b[38;5;66;03m# noqa: B902\u001b[39;00m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/openai/embeddings_utils.py:23\u001b[0m, in \u001b[0;36mget_embedding\u001b[0;34m(text, engine, **kwargs)\u001b[0m\n\u001b[1;32m     21\u001b[0m text \u001b[38;5;241m=\u001b[39m text\u001b[38;5;241m.\u001b[39mreplace(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;124m\"\u001b[39m, \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124m \u001b[39m\u001b[38;5;124m\"\u001b[39m)\n\u001b[0;32m---> 23\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[43mopenai\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mEmbedding\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mcreate\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;28;43minput\u001b[39;49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43m[\u001b[49m\u001b[43mtext\u001b[49m\u001b[43m]\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mengine\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mengine\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mkwargs\u001b[49m\u001b[43m)\u001b[49m[\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mdata\u001b[39m\u001b[38;5;124m\"\u001b[39m][\u001b[38;5;241m0\u001b[39m][\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124membedding\u001b[39m\u001b[38;5;124m\"\u001b[39m]\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/openai/api_resources/embedding.py:33\u001b[0m, in \u001b[0;36mEmbedding.create\u001b[0;34m(cls, *args, **kwargs)\u001b[0m\n\u001b[1;32m     32\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m---> 33\u001b[0m     response \u001b[38;5;241m=\u001b[39m \u001b[38;5;28;43msuper\u001b[39;49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mcreate\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43margs\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mkwargs\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m     35\u001b[0m     \u001b[38;5;66;03m# If a user specifies base64, we'll just return the encoded string.\u001b[39;00m\n\u001b[1;32m     36\u001b[0m     \u001b[38;5;66;03m# This is only for the default case.\u001b[39;00m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/openai/api_resources/abstract/engine_api_resource.py:153\u001b[0m, in \u001b[0;36mEngineAPIResource.create\u001b[0;34m(cls, api_key, api_base, api_type, request_id, api_version, organization, **params)\u001b[0m\n\u001b[1;32m    138\u001b[0m (\n\u001b[1;32m    139\u001b[0m     deployment_id,\n\u001b[1;32m    140\u001b[0m     engine,\n\u001b[0;32m   (...)\u001b[0m\n\u001b[1;32m    150\u001b[0m     api_key, api_base, api_type, api_version, organization, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mparams\n\u001b[1;32m    151\u001b[0m )\n\u001b[0;32m--> 153\u001b[0m response, _, api_key \u001b[38;5;241m=\u001b[39m \u001b[43mrequestor\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mrequest\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    154\u001b[0m \u001b[43m    \u001b[49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43mpost\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m,\u001b[49m\n\u001b[1;32m    155\u001b[0m \u001b[43m    \u001b[49m\u001b[43murl\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    156\u001b[0m \u001b[43m    \u001b[49m\u001b[43mparams\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mparams\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    157\u001b[0m \u001b[43m    \u001b[49m\u001b[43mheaders\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    158\u001b[0m \u001b[43m    \u001b[49m\u001b[43mstream\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mstream\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    159\u001b[0m \u001b[43m    \u001b[49m\u001b[43mrequest_id\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrequest_id\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    160\u001b[0m \u001b[43m    \u001b[49m\u001b[43mrequest_timeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrequest_timeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    161\u001b[0m \u001b[43m\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    163\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m stream:\n\u001b[1;32m    164\u001b[0m     \u001b[38;5;66;03m# must be an iterator\u001b[39;00m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/openai/api_requestor.py:288\u001b[0m, in \u001b[0;36mAPIRequestor.request\u001b[0;34m(self, method, url, params, headers, files, stream, request_id, request_timeout)\u001b[0m\n\u001b[1;32m    277\u001b[0m \u001b[38;5;28;01mdef\u001b[39;00m \u001b[38;5;21mrequest\u001b[39m(\n\u001b[1;32m    278\u001b[0m     \u001b[38;5;28mself\u001b[39m,\n\u001b[1;32m    279\u001b[0m     method,\n\u001b[0;32m   (...)\u001b[0m\n\u001b[1;32m    286\u001b[0m     request_timeout: Optional[Union[\u001b[38;5;28mfloat\u001b[39m, Tuple[\u001b[38;5;28mfloat\u001b[39m, \u001b[38;5;28mfloat\u001b[39m]]] \u001b[38;5;241m=\u001b[39m \u001b[38;5;28;01mNone\u001b[39;00m,\n\u001b[1;32m    287\u001b[0m ) \u001b[38;5;241m-\u001b[39m\u001b[38;5;241m>\u001b[39m Tuple[Union[OpenAIResponse, Iterator[OpenAIResponse]], \u001b[38;5;28mbool\u001b[39m, \u001b[38;5;28mstr\u001b[39m]:\n\u001b[0;32m--> 288\u001b[0m     result \u001b[38;5;241m=\u001b[39m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mrequest_raw\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    289\u001b[0m \u001b[43m        \u001b[49m\u001b[43mmethod\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mlower\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    290\u001b[0m \u001b[43m        \u001b[49m\u001b[43murl\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    291\u001b[0m \u001b[43m        \u001b[49m\u001b[43mparams\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mparams\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    292\u001b[0m \u001b[43m        \u001b[49m\u001b[43msupplied_headers\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    293\u001b[0m \u001b[43m        \u001b[49m\u001b[43mfiles\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mfiles\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    294\u001b[0m \u001b[43m        \u001b[49m\u001b[43mstream\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mstream\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    295\u001b[0m \u001b[43m        \u001b[49m\u001b[43mrequest_id\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrequest_id\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    296\u001b[0m \u001b[43m        \u001b[49m\u001b[43mrequest_timeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrequest_timeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    297\u001b[0m \u001b[43m    \u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    298\u001b[0m     resp, got_stream \u001b[38;5;241m=\u001b[39m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39m_interpret_response(result, stream)\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/openai/api_requestor.py:609\u001b[0m, in \u001b[0;36mAPIRequestor.request_raw\u001b[0;34m(self, method, url, params, supplied_headers, files, stream, request_id, request_timeout)\u001b[0m\n\u001b[1;32m    608\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m requests\u001b[38;5;241m.\u001b[39mexceptions\u001b[38;5;241m.\u001b[39mRequestException \u001b[38;5;28;01mas\u001b[39;00m e:\n\u001b[0;32m--> 609\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m error\u001b[38;5;241m.\u001b[39mAPIConnectionError(\n\u001b[1;32m    610\u001b[0m         \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mError communicating with OpenAI: \u001b[39m\u001b[38;5;132;01m{}\u001b[39;00m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;241m.\u001b[39mformat(e)\n\u001b[1;32m    611\u001b[0m     ) \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01me\u001b[39;00m\n\u001b[1;32m    612\u001b[0m util\u001b[38;5;241m.\u001b[39mlog_debug(\n\u001b[1;32m    613\u001b[0m     \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mOpenAI API response\u001b[39m\u001b[38;5;124m\"\u001b[39m,\n\u001b[1;32m    614\u001b[0m     path\u001b[38;5;241m=\u001b[39mabs_url,\n\u001b[0;32m   (...)\u001b[0m\n\u001b[1;32m    617\u001b[0m     request_id\u001b[38;5;241m=\u001b[39mresult\u001b[38;5;241m.\u001b[39mheaders\u001b[38;5;241m.\u001b[39mget(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mX-Request-Id\u001b[39m\u001b[38;5;124m\"\u001b[39m),\n\u001b[1;32m    618\u001b[0m )\n",
-      "\u001b[0;31mAPIConnectionError\u001b[0m: Error communicating with OpenAI: HTTPSConnectionPool(host='api.openai.com', port=443): Max retries exceeded with url: /v1/engines/text-embedding-ada-002/embeddings (Caused by ProxyError('Unable to connect to proxy', NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x12a3dd350>: Failed to establish a new connection: [Errno 61] Connection refused')))",
-      "\nThe above exception was the direct cause of the following exception:\n",
-      "\u001b[0;31mRetryError\u001b[0m                                Traceback (most recent call last)",
-      "Cell \u001b[0;32mIn[44], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m xx \u001b[38;5;241m=\u001b[39m \u001b[43mbird\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mquery_emb\u001b[49m\u001b[43m(\u001b[49m\u001b[43ms\u001b[49m\u001b[43m)\u001b[49m\n",
-      "File \u001b[0;32m~/playgo/abc123/games/angrybirds/angrybirds.py:197\u001b[0m, in \u001b[0;36mangrybird.query_emb\u001b[0;34m(self, command)\u001b[0m\n\u001b[1;32m    194\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m command\u001b[38;5;241m.\u001b[39mstrip() \u001b[38;5;241m==\u001b[39m \u001b[38;5;124m'\u001b[39m\u001b[38;5;124m'\u001b[39m:\n\u001b[1;32m    195\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m status_failure, [\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mYour input is empty.\u001b[39m\u001b[38;5;124m\"\u001b[39m], \u001b[38;5;124m'\u001b[39m\u001b[38;5;124m'\u001b[39m, \u001b[38;5;241m0\u001b[39m\n\u001b[0;32m--> 197\u001b[0m imput_emb \u001b[38;5;241m=\u001b[39m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43m_compute_emb\u001b[49m\u001b[43m(\u001b[49m\u001b[43mcommand\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    198\u001b[0m top_cmd \u001b[38;5;241m=\u001b[39m \u001b[38;5;28;01mNone\u001b[39;00m\n\u001b[1;32m    199\u001b[0m top_sim \u001b[38;5;241m=\u001b[39m \u001b[38;5;241m0\u001b[39m\n",
-      "File \u001b[0;32m~/playgo/abc123/games/angrybirds/angrybirds.py:121\u001b[0m, in \u001b[0;36mangrybird._compute_emb\u001b[0;34m(self, command)\u001b[0m\n\u001b[1;32m    118\u001b[0m \u001b[38;5;28;01mdef\u001b[39;00m \u001b[38;5;21m_compute_emb\u001b[39m(\u001b[38;5;28mself\u001b[39m, command: \u001b[38;5;28mstr\u001b[39m):\n\u001b[1;32m    119\u001b[0m     \u001b[38;5;66;03m# always normalize text when query OpenAI embedding service\u001b[39;00m\n\u001b[1;32m    120\u001b[0m     command \u001b[38;5;241m=\u001b[39m command\u001b[38;5;241m.\u001b[39mstrip()\u001b[38;5;241m.\u001b[39mlower()\n\u001b[0;32m--> 121\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[43mopenai_get_embedding\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    122\u001b[0m \u001b[43m        \u001b[49m\u001b[43mcommand\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    123\u001b[0m \u001b[43m        \u001b[49m\u001b[43mengine\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43mtext-embedding-ada-002\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\n\u001b[1;32m    124\u001b[0m \u001b[43m    \u001b[49m\u001b[43m)\u001b[49m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/tenacity/__init__.py:289\u001b[0m, in \u001b[0;36mBaseRetrying.wraps.<locals>.wrapped_f\u001b[0;34m(*args, **kw)\u001b[0m\n\u001b[1;32m    287\u001b[0m \u001b[38;5;129m@functools\u001b[39m\u001b[38;5;241m.\u001b[39mwraps(f)\n\u001b[1;32m    288\u001b[0m \u001b[38;5;28;01mdef\u001b[39;00m \u001b[38;5;21mwrapped_f\u001b[39m(\u001b[38;5;241m*\u001b[39margs: t\u001b[38;5;241m.\u001b[39mAny, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mkw: t\u001b[38;5;241m.\u001b[39mAny) \u001b[38;5;241m-\u001b[39m\u001b[38;5;241m>\u001b[39m t\u001b[38;5;241m.\u001b[39mAny:\n\u001b[0;32m--> 289\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28;43mself\u001b[39;49m\u001b[43m(\u001b[49m\u001b[43mf\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43margs\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mkw\u001b[49m\u001b[43m)\u001b[49m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/tenacity/__init__.py:379\u001b[0m, in \u001b[0;36mRetrying.__call__\u001b[0;34m(self, fn, *args, **kwargs)\u001b[0m\n\u001b[1;32m    377\u001b[0m retry_state \u001b[38;5;241m=\u001b[39m RetryCallState(retry_object\u001b[38;5;241m=\u001b[39m\u001b[38;5;28mself\u001b[39m, fn\u001b[38;5;241m=\u001b[39mfn, args\u001b[38;5;241m=\u001b[39margs, kwargs\u001b[38;5;241m=\u001b[39mkwargs)\n\u001b[1;32m    378\u001b[0m \u001b[38;5;28;01mwhile\u001b[39;00m \u001b[38;5;28;01mTrue\u001b[39;00m:\n\u001b[0;32m--> 379\u001b[0m     do \u001b[38;5;241m=\u001b[39m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43miter\u001b[49m\u001b[43m(\u001b[49m\u001b[43mretry_state\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mretry_state\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    380\u001b[0m     \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;28misinstance\u001b[39m(do, DoAttempt):\n\u001b[1;32m    381\u001b[0m         \u001b[38;5;28;01mtry\u001b[39;00m:\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/tenacity/__init__.py:326\u001b[0m, in \u001b[0;36mBaseRetrying.iter\u001b[0;34m(self, retry_state)\u001b[0m\n\u001b[1;32m    324\u001b[0m     \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mreraise:\n\u001b[1;32m    325\u001b[0m         \u001b[38;5;28;01mraise\u001b[39;00m retry_exc\u001b[38;5;241m.\u001b[39mreraise()\n\u001b[0;32m--> 326\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m retry_exc \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mfut\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mexception\u001b[39;00m()\n\u001b[1;32m    328\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mwait:\n\u001b[1;32m    329\u001b[0m     sleep \u001b[38;5;241m=\u001b[39m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mwait(retry_state)\n",
-      "\u001b[0;31mRetryError\u001b[0m: RetryError[<Future at 0x12a3decd0 state=finished raised APIConnectionError>]"
-     ]
-    }
-   ],
-   "source": [
-    "xx = bird.query_emb(s)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 45,
-   "id": "06e53c7e-9c4b-4529-8e2f-ab90d29b0404",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "list"
-      ]
-     },
-     "execution_count": 45,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "type(xx)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 46,
-   "id": "8ca3cf57-469d-4632-a5b5-6da4ad4c0e2f",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['OK',\n",
-       " ['bird', 'delete', {'quantity': 'some', 'type': 'antigravity'}],\n",
-       " 'delete some antigravity birds',\n",
-       " 0.9348558221466966]"
-      ]
-     },
-     "execution_count": 46,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "xx"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 47,
-   "id": "622ffa96-98c1-432c-8cba-76facb4a9ddd",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "command = s"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 49,
-   "id": "0fe0cf9c-f395-4b17-bb11-121a326635dd",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "imput_emb = bird._compute_emb(command)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 50,
-   "id": "cbfdc092-88a6-4f57-b50a-a8929fc79b33",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import openai\n",
-    "from openai.embeddings_utils import get_embedding as openai_get_embedding\n",
-    "from openai.embeddings_utils import cosine_similarity as openai_cosine_similarity"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 51,
-   "id": "a2dff84a-9d48-4f7b-a053-0b73b57da9ad",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "top_cmd = None\n",
-    "top_sim = 0\n",
-    "for cmd in bird.embeddings:\n",
-    "    sim = openai_cosine_similarity(\n",
-    "        imput_emb, \n",
-    "        bird.embeddings[cmd],\n",
-    "    )\n",
-    "    if sim > top_sim:\n",
-    "        top_cmd = cmd\n",
-    "        top_sim = sim"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 52,
-   "id": "f3ed3ed1-98ee-4e42-82eb-4a6445edff3d",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'delete some antigravity birds'"
-      ]
-     },
-     "execution_count": 52,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "top_cmd"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 53,
-   "id": "7ffbc3e4-4933-4bac-89c1-ec00a9e6eb0a",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "0.9319764923699465"
-      ]
-     },
-     "execution_count": 53,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "top_sim"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 54,
-   "id": "0bc21e4a-e81c-4ab8-af45-f53eba709ab5",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "results = []\n",
-    "for cmd in bird.embeddings:\n",
-    "    sim = openai_cosine_similarity(\n",
-    "        imput_emb, \n",
-    "        bird.embeddings[cmd],\n",
-    "    )\n",
-    "    results.append([cmd, sim])"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 57,
-   "id": "973ae4ca-1eba-42c3-ba2b-ccf06d0ddbbe",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "results.sort(key = lambda x: x[1], reverse=True)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 59,
-   "id": "89523b8d-59f8-48da-a492-8659ccde3138",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "\"I don't like antigravity birds. Delete all of them\""
-      ]
-     },
-     "execution_count": 59,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "command"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 58,
-   "id": "2bba1891-5c0a-40c6-9b06-736553b55a3e",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[['delete some antigravity birds', 0.9319764923699465],\n",
-       " ['delete all antigravity birds', 0.926993589547008],\n",
-       " ['delete antigravity birds', 0.924798267679007],\n",
-       " ['delete more antigravity birds', 0.9231406436854316],\n",
-       " ['delete one antigravity bird', 0.9145274787932756]]"
-      ]
-     },
-     "execution_count": 58,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "results[:5]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 86,
-   "id": "3636b75c-de9c-4953-b3b2-61fc15ae0a6f",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "res2 = {}\n",
-    "for i, e in enumerate(results[:5]):\n",
-    "    tmp = {i: [f\"{e[1]:.8f}\", e[0]]}\n",
-    "    res2 |= tmp"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 87,
-   "id": "1470d109-edff-48b9-b81b-9c88bba842e3",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{0: ['0.93197649', 'delete some antigravity birds'],\n",
-       " 1: ['0.92699359', 'delete all antigravity birds'],\n",
-       " 2: ['0.92479827', 'delete antigravity birds'],\n",
-       " 3: ['0.92314064', 'delete more antigravity birds'],\n",
-       " 4: ['0.91452748', 'delete one antigravity bird']}"
-      ]
-     },
-     "execution_count": 87,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "res2"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "d4da3a5f-a0c8-4728-8bbb-9a21b9776cf8",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 90,
-   "id": "e14c27fc-881b-4c48-ae93-87cc7860d9c9",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "res2 = []\n",
-    "for i, e in enumerate(results[:5]):\n",
-    "    tmp = [i, f\"{e[1]:.8f}\", e[0]]\n",
-    "    res2.append(tmp)\n"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 91,
-   "id": "c80ff104-54d6-4a72-a9f7-ed980f4c4422",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[[0, '0.93197649', 'delete some antigravity birds'],\n",
-       " [1, '0.92699359', 'delete all antigravity birds'],\n",
-       " [2, '0.92479827', 'delete antigravity birds'],\n",
-       " [3, '0.92314064', 'delete more antigravity birds'],\n",
-       " [4, '0.91452748', 'delete one antigravity bird']]"
-      ]
-     },
-     "execution_count": 91,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "res2"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "eec6fc8a-e129-4383-bd7a-c4f9364c496c",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "e5de835b-9b8e-4f53-af05-d17581b6f46f",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 38,
-   "id": "8eb9449e-f0d0-40b9-b0f2-b51f5b5470c7",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "\"I don't like antigravity birds. Delete them\""
-      ]
-     },
-     "execution_count": 38,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "command"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 62,
-   "id": "77eb1df7-7640-43cb-a14b-1559611de854",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "example = 'Delete antigravity birds'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 63,
-   "id": "7018cf77-fbbc-471d-98ea-e550b2a59259",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "False"
-      ]
-     },
-     "execution_count": 63,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "example in bird.train"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "561ae08b-ccca-4dac-8154-bcd2f35babd6",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.4"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}

games/angrybirds/notebooks/test-json.ipynb

@@ -1,1158 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "id": "245fbc34-352d-4945-a689-d04f9365bb99",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "%load_ext autoreload\n",
-    "%autoreload 2"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "id": "eaed216c-160a-4e31-96fe-015c140ff18b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# from pprint import pprint"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "id": "97953551-8ef1-42c1-9317-99d5e65f535e",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "before add_config\n",
-      "723\n",
-      "724\n",
-      "725\n",
-      "726\n",
-      "after add_config\n",
-      "726\n",
-      "final\n",
-      "773\n"
-     ]
-    }
-   ],
-   "source": [
-    "import json\n",
-    "import pandas as pd\n",
-    "import pickle\n",
-    "\n",
-    "import openai\n",
-    "from openai.embeddings_utils import get_embedding, cosine_similarity\n",
-    "\n",
-    "# from games.angrybirds.util import util\n",
-    "from games.angrybirds.config import config\n",
-    "# from games.angrybirds.prompt import prompt\n",
-    "# from games.angrybirds.query import query, yes_no_query\n",
-    "from games.angrybirds.angrybirds import angrybird"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "d536db48-2ddf-476c-8e54-21263ccc1dde",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "id": "43318d58-8b5e-4544-89d2-81273ce39d82",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import os"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 7,
-   "id": "3f537173-a3c5-4f6d-8793-2904a20e46bd",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'/Users/shaojun/playgo/abc123'"
-      ]
-     },
-     "execution_count": 7,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "root = os.getenv('PYTHONPATH')\n",
-    "root"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 10,
-   "id": "d0a0a794-c6e8-48f7-859e-af9cc98c4806",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'{root}/embeddings.pkl', 'rb') as f:\n",
-    "    x = pickle.load(f)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 13,
-   "id": "885b8328-0812-419a-915a-aa952030aa54",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "keys = [s for s in x.keys()]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 15,
-   "id": "641d8409-8551-414a-abfa-d74292dfea76",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "772"
-      ]
-     },
-     "execution_count": 15,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(keys)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 16,
-   "id": "8edc26d9-d423-4108-9f17-8f732cec9acf",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "773"
-      ]
-     },
-     "execution_count": 16,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(config.game_train)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 18,
-   "id": "3665cd61-1e92-467e-86c4-bed67bfd8d27",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Change background photo\n"
-     ]
-    }
-   ],
-   "source": [
-    "for x in config.game_train:\n",
-    "    if x.lower() not in keys:\n",
-    "        print(x)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 20,
-   "id": "51fa6e2f-685c-494f-b269-b9a3ba7c4590",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['add 10 antigravity birds',\n",
-       " 'add 10 big birds',\n",
-       " 'add 10 birds',\n",
-       " 'add 10 catapult birds',\n",
-       " 'add 10 elderly pigs',\n",
-       " 'add 10 flash birds',\n",
-       " 'add 10 magnet birds',\n",
-       " 'add 10 pigs',\n",
-       " 'add 10 sport pigs',\n",
-       " 'add 10 superheavy birds',\n",
-       " 'add 2 antigravity birds',\n",
-       " 'add 2 big birds',\n",
-       " 'add 2 birds',\n",
-       " 'add 2 catapult birds',\n",
-       " 'add 2 elderly pigs',\n",
-       " 'add 2 flash birds',\n",
-       " 'add 2 magnet birds',\n",
-       " 'add 2 pigs',\n",
-       " 'add 2 sport pigs',\n",
-       " 'add 2 superheavy birds',\n",
-       " 'add 3 antigravity birds',\n",
-       " 'add 3 big birds',\n",
-       " 'add 3 birds',\n",
-       " 'add 3 catapult birds',\n",
-       " 'add 3 elderly pigs',\n",
-       " 'add 3 flash birds',\n",
-       " 'add 3 magnet birds',\n",
-       " 'add 3 pigs',\n",
-       " 'add 3 sport pigs',\n",
-       " 'add 3 superheavy birds',\n",
-       " 'add 4 antigravity birds',\n",
-       " 'add 4 big birds',\n",
-       " 'add 4 birds',\n",
-       " 'add 4 catapult birds',\n",
-       " 'add 4 elderly pigs',\n",
-       " 'add 4 flash birds',\n",
-       " 'add 4 magnet birds',\n",
-       " 'add 4 pigs',\n",
-       " 'add 4 sport pigs',\n",
-       " 'add 4 superheavy birds',\n",
-       " 'add 5 antigravity birds',\n",
-       " 'add 5 big birds',\n",
-       " 'add 5 birds',\n",
-       " 'add 5 catapult birds',\n",
-       " 'add 5 elderly pigs',\n",
-       " 'add 5 flash birds',\n",
-       " 'add 5 magnet birds',\n",
-       " 'add 5 pigs',\n",
-       " 'add 5 sport pigs',\n",
-       " 'add 5 superheavy birds',\n",
-       " 'add 6 antigravity birds',\n",
-       " 'add 6 big birds',\n",
-       " 'add 6 birds',\n",
-       " 'add 6 catapult birds',\n",
-       " 'add 6 elderly pigs',\n",
-       " 'add 6 flash birds',\n",
-       " 'add 6 magnet birds',\n",
-       " 'add 6 pigs',\n",
-       " 'add 6 sport pigs',\n",
-       " 'add 6 superheavy birds',\n",
-       " 'add 7 antigravity birds',\n",
-       " 'add 7 big birds',\n",
-       " 'add 7 birds',\n",
-       " 'add 7 catapult birds',\n",
-       " 'add 7 elderly pigs',\n",
-       " 'add 7 flash birds',\n",
-       " 'add 7 magnet birds',\n",
-       " 'add 7 pigs',\n",
-       " 'add 7 sport pigs',\n",
-       " 'add 7 superheavy birds',\n",
-       " 'add 8 antigravity birds',\n",
-       " 'add 8 big birds',\n",
-       " 'add 8 birds',\n",
-       " 'add 8 catapult birds',\n",
-       " 'add 8 elderly pigs',\n",
-       " 'add 8 flash birds',\n",
-       " 'add 8 magnet birds',\n",
-       " 'add 8 pigs',\n",
-       " 'add 8 sport pigs',\n",
-       " 'add 8 superheavy birds',\n",
-       " 'add 9 antigravity birds',\n",
-       " 'add 9 big birds',\n",
-       " 'add 9 birds',\n",
-       " 'add 9 catapult birds',\n",
-       " 'add 9 elderly pigs',\n",
-       " 'add 9 flash birds',\n",
-       " 'add 9 magnet birds',\n",
-       " 'add 9 pigs',\n",
-       " 'add 9 sport pigs',\n",
-       " 'add 9 superheavy birds',\n",
-       " 'add a big bird',\n",
-       " 'add a bird',\n",
-       " 'add a catapult bird',\n",
-       " 'add a flash bird',\n",
-       " 'add a magnet bird',\n",
-       " 'add a pig',\n",
-       " 'add a sport pig',\n",
-       " 'add a superheavy bird',\n",
-       " 'add all antigravity birds',\n",
-       " 'add all big birds',\n",
-       " 'add all birds',\n",
-       " 'add all catapult birds',\n",
-       " 'add all elderly pigs',\n",
-       " 'add all flash birds',\n",
-       " 'add all magnet birds',\n",
-       " 'add all pigs',\n",
-       " 'add all sport pigs',\n",
-       " 'add all superheavy birds',\n",
-       " 'add an antigravity bird',\n",
-       " 'add an elderly pig',\n",
-       " 'add antigravity birds',\n",
-       " 'add big birds',\n",
-       " 'add birds',\n",
-       " 'add catapult birds',\n",
-       " 'add eight antigravity birds',\n",
-       " 'add eight big birds',\n",
-       " 'add eight birds',\n",
-       " 'add eight catapult birds',\n",
-       " 'add eight elderly pigs',\n",
-       " 'add eight flash birds',\n",
-       " 'add eight magnet birds',\n",
-       " 'add eight pigs',\n",
-       " 'add eight sport pigs',\n",
-       " 'add eight superheavy birds',\n",
-       " 'add elderly pigs',\n",
-       " 'add five antigravity birds',\n",
-       " 'add five big birds',\n",
-       " 'add five birds',\n",
-       " 'add five catapult birds',\n",
-       " 'add five elderly pigs',\n",
-       " 'add five flash birds',\n",
-       " 'add five magnet birds',\n",
-       " 'add five pigs',\n",
-       " 'add five sport pigs',\n",
-       " 'add five superheavy birds',\n",
-       " 'add flash birds',\n",
-       " 'add four antigravity birds',\n",
-       " 'add four big birds',\n",
-       " 'add four birds',\n",
-       " 'add four catapult birds',\n",
-       " 'add four elderly pigs',\n",
-       " 'add four flash birds',\n",
-       " 'add four magnet birds',\n",
-       " 'add four pigs',\n",
-       " 'add four sport pigs',\n",
-       " 'add four superheavy birds',\n",
-       " 'add magnet birds',\n",
-       " 'add more antigravity birds',\n",
-       " 'add more big birds',\n",
-       " 'add more birds',\n",
-       " 'add more catapult birds',\n",
-       " 'add more elderly pigs',\n",
-       " 'add more flash birds',\n",
-       " 'add more magnet birds',\n",
-       " 'add more pigs',\n",
-       " 'add more sport pigs',\n",
-       " 'add more superheavy birds',\n",
-       " 'add nine antigravity birds',\n",
-       " 'add nine big birds',\n",
-       " 'add nine birds',\n",
-       " 'add nine catapult birds',\n",
-       " 'add nine elderly pigs',\n",
-       " 'add nine flash birds',\n",
-       " 'add nine magnet birds',\n",
-       " 'add nine pigs',\n",
-       " 'add nine sport pigs',\n",
-       " 'add nine superheavy birds',\n",
-       " 'add one antigravity bird',\n",
-       " 'add one big bird',\n",
-       " 'add one bird',\n",
-       " 'add one catapult bird',\n",
-       " 'add one elderly pig',\n",
-       " 'add one flash bird',\n",
-       " 'add one magnet bird',\n",
-       " 'add one pig',\n",
-       " 'add one sport pig',\n",
-       " 'add one superheavy bird',\n",
-       " 'add pigs',\n",
-       " 'add seven antigravity birds',\n",
-       " 'add seven big birds',\n",
-       " 'add seven birds',\n",
-       " 'add seven catapult birds',\n",
-       " 'add seven elderly pigs',\n",
-       " 'add seven flash birds',\n",
-       " 'add seven magnet birds',\n",
-       " 'add seven pigs',\n",
-       " 'add seven sport pigs',\n",
-       " 'add seven superheavy birds',\n",
-       " 'add six antigravity birds',\n",
-       " 'add six big birds',\n",
-       " 'add six birds',\n",
-       " 'add six catapult birds',\n",
-       " 'add six elderly pigs',\n",
-       " 'add six flash birds',\n",
-       " 'add six magnet birds',\n",
-       " 'add six pigs',\n",
-       " 'add six sport pigs',\n",
-       " 'add six superheavy birds',\n",
-       " 'add some antigravity birds',\n",
-       " 'add some big birds',\n",
-       " 'add some birds',\n",
-       " 'add some catapult birds',\n",
-       " 'add some elderly pigs',\n",
-       " 'add some flash birds',\n",
-       " 'add some magnet birds',\n",
-       " 'add some pigs',\n",
-       " 'add some sport pigs',\n",
-       " 'add some superheavy birds',\n",
-       " 'add sport pigs',\n",
-       " 'add superheavy birds',\n",
-       " 'add ten antigravity birds',\n",
-       " 'add ten big birds',\n",
-       " 'add ten birds',\n",
-       " 'add ten catapult birds',\n",
-       " 'add ten elderly pigs',\n",
-       " 'add ten flash birds',\n",
-       " 'add ten magnet birds',\n",
-       " 'add ten pigs',\n",
-       " 'add ten sport pigs',\n",
-       " 'add ten superheavy birds',\n",
-       " 'add three antigravity birds',\n",
-       " 'add three big birds',\n",
-       " 'add three birds',\n",
-       " 'add three catapult birds',\n",
-       " 'add three elderly pigs',\n",
-       " 'add three flash birds',\n",
-       " 'add three magnet birds',\n",
-       " 'add three pigs',\n",
-       " 'add three sport pigs',\n",
-       " 'add three superheavy birds',\n",
-       " 'add two antigravity birds',\n",
-       " 'add two big birds',\n",
-       " 'add two birds',\n",
-       " 'add two catapult birds',\n",
-       " 'add two elderly pigs',\n",
-       " 'add two flash birds',\n",
-       " 'add two magnet birds',\n",
-       " 'add two pigs',\n",
-       " 'add two sport pigs',\n",
-       " 'add two superheavy birds',\n",
-       " 'are there new kinds of birds?',\n",
-       " 'are there new kinds of pigs?',\n",
-       " 'bird list',\n",
-       " 'bird picture',\n",
-       " 'build a house',\n",
-       " 'build houses',\n",
-       " 'build houses, so i can protect my pigs',\n",
-       " 'change background',\n",
-       " 'change background color',\n",
-       " 'change background music',\n",
-       " 'change background picture',\n",
-       " 'change background sound',\n",
-       " 'change background text',\n",
-       " 'change background volumn',\n",
-       " \"change bird's picture\",\n",
-       " \"change bird's skin\",\n",
-       " 'change birds',\n",
-       " \"change pig's picture\",\n",
-       " \"change pig's skin\",\n",
-       " 'change pigs',\n",
-       " 'change the position of birds',\n",
-       " 'change the position of houses',\n",
-       " 'change the position of obstacles',\n",
-       " 'change the position of pigs',\n",
-       " 'change volumn',\n",
-       " 'delete 10 antigravity birds',\n",
-       " 'delete 10 big birds',\n",
-       " 'delete 10 birds',\n",
-       " 'delete 10 catapult birds',\n",
-       " 'delete 10 elderly pigs',\n",
-       " 'delete 10 flash birds',\n",
-       " 'delete 10 magnet birds',\n",
-       " 'delete 10 pigs',\n",
-       " 'delete 10 sport pigs',\n",
-       " 'delete 10 superheavy birds',\n",
-       " 'delete 2 antigravity birds',\n",
-       " 'delete 2 big birds',\n",
-       " 'delete 2 birds',\n",
-       " 'delete 2 catapult birds',\n",
-       " 'delete 2 elderly pigs',\n",
-       " 'delete 2 flash birds',\n",
-       " 'delete 2 magnet birds',\n",
-       " 'delete 2 pigs',\n",
-       " 'delete 2 sport pigs',\n",
-       " 'delete 2 superheavy birds',\n",
-       " 'delete 3 antigravity birds',\n",
-       " 'delete 3 big birds',\n",
-       " 'delete 3 birds',\n",
-       " 'delete 3 catapult birds',\n",
-       " 'delete 3 elderly pigs',\n",
-       " 'delete 3 flash birds',\n",
-       " 'delete 3 magnet birds',\n",
-       " 'delete 3 pigs',\n",
-       " 'delete 3 sport pigs',\n",
-       " 'delete 3 superheavy birds',\n",
-       " 'delete 4 antigravity birds',\n",
-       " 'delete 4 big birds',\n",
-       " 'delete 4 birds',\n",
-       " 'delete 4 catapult birds',\n",
-       " 'delete 4 elderly pigs',\n",
-       " 'delete 4 flash birds',\n",
-       " 'delete 4 magnet birds',\n",
-       " 'delete 4 pigs',\n",
-       " 'delete 4 sport pigs',\n",
-       " 'delete 4 superheavy birds',\n",
-       " 'delete 5 antigravity birds',\n",
-       " 'delete 5 big birds',\n",
-       " 'delete 5 birds',\n",
-       " 'delete 5 catapult birds',\n",
-       " 'delete 5 elderly pigs',\n",
-       " 'delete 5 flash birds',\n",
-       " 'delete 5 magnet birds',\n",
-       " 'delete 5 pigs',\n",
-       " 'delete 5 sport pigs',\n",
-       " 'delete 5 superheavy birds',\n",
-       " 'delete 6 antigravity birds',\n",
-       " 'delete 6 big birds',\n",
-       " 'delete 6 birds',\n",
-       " 'delete 6 catapult birds',\n",
-       " 'delete 6 elderly pigs',\n",
-       " 'delete 6 flash birds',\n",
-       " 'delete 6 magnet birds',\n",
-       " 'delete 6 pigs',\n",
-       " 'delete 6 sport pigs',\n",
-       " 'delete 6 superheavy birds',\n",
-       " 'delete 7 antigravity birds',\n",
-       " 'delete 7 big birds',\n",
-       " 'delete 7 birds',\n",
-       " 'delete 7 catapult birds',\n",
-       " 'delete 7 elderly pigs',\n",
-       " 'delete 7 flash birds',\n",
-       " 'delete 7 magnet birds',\n",
-       " 'delete 7 pigs',\n",
-       " 'delete 7 sport pigs',\n",
-       " 'delete 7 superheavy birds',\n",
-       " 'delete 8 antigravity birds',\n",
-       " 'delete 8 big birds',\n",
-       " 'delete 8 birds',\n",
-       " 'delete 8 catapult birds',\n",
-       " 'delete 8 elderly pigs',\n",
-       " 'delete 8 flash birds',\n",
-       " 'delete 8 magnet birds',\n",
-       " 'delete 8 pigs',\n",
-       " 'delete 8 sport pigs',\n",
-       " 'delete 8 superheavy birds',\n",
-       " 'delete 9 antigravity birds',\n",
-       " 'delete 9 big birds',\n",
-       " 'delete 9 birds',\n",
-       " 'delete 9 catapult birds',\n",
-       " 'delete 9 elderly pigs',\n",
-       " 'delete 9 flash birds',\n",
-       " 'delete 9 magnet birds',\n",
-       " 'delete 9 pigs',\n",
-       " 'delete 9 sport pigs',\n",
-       " 'delete 9 superheavy birds',\n",
-       " 'delete a big bird',\n",
-       " 'delete a bird',\n",
-       " 'delete a catapult bird',\n",
-       " 'delete a flash bird',\n",
-       " 'delete a magnet bird',\n",
-       " 'delete a pig',\n",
-       " 'delete a sport pig',\n",
-       " 'delete a superheavy bird',\n",
-       " 'delete all antigravity birds',\n",
-       " 'delete all big birds',\n",
-       " 'delete all birds',\n",
-       " 'delete all catapult birds',\n",
-       " 'delete all elderly pigs',\n",
-       " 'delete all flash birds',\n",
-       " 'delete all magnet birds',\n",
-       " 'delete all pigs',\n",
-       " 'delete all sport pigs',\n",
-       " 'delete all superheavy birds',\n",
-       " 'delete an antigravity bird',\n",
-       " 'delete an elderly pig',\n",
-       " 'delete antigravity birds',\n",
-       " 'delete big birds',\n",
-       " 'delete birds',\n",
-       " 'delete catapult birds',\n",
-       " 'delete eight antigravity birds',\n",
-       " 'delete eight big birds',\n",
-       " 'delete eight birds',\n",
-       " 'delete eight catapult birds',\n",
-       " 'delete eight elderly pigs',\n",
-       " 'delete eight flash birds',\n",
-       " 'delete eight magnet birds',\n",
-       " 'delete eight pigs',\n",
-       " 'delete eight sport pigs',\n",
-       " 'delete eight superheavy birds',\n",
-       " 'delete elderly pigs',\n",
-       " 'delete five antigravity birds',\n",
-       " 'delete five big birds',\n",
-       " 'delete five birds',\n",
-       " 'delete five catapult birds',\n",
-       " 'delete five elderly pigs',\n",
-       " 'delete five flash birds',\n",
-       " 'delete five magnet birds',\n",
-       " 'delete five pigs',\n",
-       " 'delete five sport pigs',\n",
-       " 'delete five superheavy birds',\n",
-       " 'delete flash birds',\n",
-       " 'delete four antigravity birds',\n",
-       " 'delete four big birds',\n",
-       " 'delete four birds',\n",
-       " 'delete four catapult birds',\n",
-       " 'delete four elderly pigs',\n",
-       " 'delete four flash birds',\n",
-       " 'delete four magnet birds',\n",
-       " 'delete four pigs',\n",
-       " 'delete four sport pigs',\n",
-       " 'delete four superheavy birds',\n",
-       " 'delete magnet birds',\n",
-       " 'delete more antigravity birds',\n",
-       " 'delete more big birds',\n",
-       " 'delete more birds',\n",
-       " 'delete more catapult birds',\n",
-       " 'delete more elderly pigs',\n",
-       " 'delete more flash birds',\n",
-       " 'delete more magnet birds',\n",
-       " 'delete more pigs',\n",
-       " 'delete more sport pigs',\n",
-       " 'delete more superheavy birds',\n",
-       " 'delete nine antigravity birds',\n",
-       " 'delete nine big birds',\n",
-       " 'delete nine birds',\n",
-       " 'delete nine catapult birds',\n",
-       " 'delete nine elderly pigs',\n",
-       " 'delete nine flash birds',\n",
-       " 'delete nine magnet birds',\n",
-       " 'delete nine pigs',\n",
-       " 'delete nine sport pigs',\n",
-       " 'delete nine superheavy birds',\n",
-       " 'delete one antigravity bird',\n",
-       " 'delete one big bird',\n",
-       " 'delete one bird',\n",
-       " 'delete one catapult bird',\n",
-       " 'delete one elderly pig',\n",
-       " 'delete one flash bird',\n",
-       " 'delete one magnet bird',\n",
-       " 'delete one pig',\n",
-       " 'delete one sport pig',\n",
-       " 'delete one superheavy bird',\n",
-       " 'delete pigs',\n",
-       " 'delete seven antigravity birds',\n",
-       " 'delete seven big birds',\n",
-       " 'delete seven birds',\n",
-       " 'delete seven catapult birds',\n",
-       " 'delete seven elderly pigs',\n",
-       " 'delete seven flash birds',\n",
-       " 'delete seven magnet birds',\n",
-       " 'delete seven pigs',\n",
-       " 'delete seven sport pigs',\n",
-       " 'delete seven superheavy birds',\n",
-       " 'delete six antigravity birds',\n",
-       " 'delete six big birds',\n",
-       " 'delete six birds',\n",
-       " 'delete six catapult birds',\n",
-       " 'delete six elderly pigs',\n",
-       " 'delete six flash birds',\n",
-       " 'delete six magnet birds',\n",
-       " 'delete six pigs',\n",
-       " 'delete six sport pigs',\n",
-       " 'delete six superheavy birds',\n",
-       " 'delete some antigravity birds',\n",
-       " 'delete some big birds',\n",
-       " 'delete some birds',\n",
-       " 'delete some catapult birds',\n",
-       " 'delete some elderly pigs',\n",
-       " 'delete some flash birds',\n",
-       " 'delete some magnet birds',\n",
-       " 'delete some pigs',\n",
-       " 'delete some sport pigs',\n",
-       " 'delete some superheavy birds',\n",
-       " 'delete sport pigs',\n",
-       " 'delete superheavy birds',\n",
-       " 'delete ten antigravity birds',\n",
-       " 'delete ten big birds',\n",
-       " 'delete ten birds',\n",
-       " 'delete ten catapult birds',\n",
-       " 'delete ten elderly pigs',\n",
-       " 'delete ten flash birds',\n",
-       " 'delete ten magnet birds',\n",
-       " 'delete ten pigs',\n",
-       " 'delete ten sport pigs',\n",
-       " 'delete ten superheavy birds',\n",
-       " 'delete three antigravity birds',\n",
-       " 'delete three big birds',\n",
-       " 'delete three birds',\n",
-       " 'delete three catapult birds',\n",
-       " 'delete three elderly pigs',\n",
-       " 'delete three flash birds',\n",
-       " 'delete three magnet birds',\n",
-       " 'delete three pigs',\n",
-       " 'delete three sport pigs',\n",
-       " 'delete three superheavy birds',\n",
-       " 'delete two antigravity birds',\n",
-       " 'delete two big birds',\n",
-       " 'delete two birds',\n",
-       " 'delete two catapult birds',\n",
-       " 'delete two elderly pigs',\n",
-       " 'delete two flash birds',\n",
-       " 'delete two magnet birds',\n",
-       " 'delete two pigs',\n",
-       " 'delete two sport pigs',\n",
-       " 'delete two superheavy birds',\n",
-       " 'list birds',\n",
-       " 'list pigs',\n",
-       " 'make pigs everywehre',\n",
-       " 'make the bird bigger',\n",
-       " 'make the bird bigger, so it can run over the enemies',\n",
-       " 'make the bird smaller',\n",
-       " 'make the pig bigger',\n",
-       " 'make the pig smaller',\n",
-       " 'make the pig weaker',\n",
-       " 'move birds',\n",
-       " 'move houses',\n",
-       " 'move obstacles',\n",
-       " 'move pigs',\n",
-       " 'our pigs are losing. add more pigs',\n",
-       " 'pig list',\n",
-       " 'pig picture',\n",
-       " 'protect the pigs by build houses for them',\n",
-       " 'reduce 10 antigravity birds',\n",
-       " 'reduce 10 big birds',\n",
-       " 'reduce 10 birds',\n",
-       " 'reduce 10 catapult birds',\n",
-       " 'reduce 10 elderly pigs',\n",
-       " 'reduce 10 flash birds',\n",
-       " 'reduce 10 magnet birds',\n",
-       " 'reduce 10 pigs',\n",
-       " 'reduce 10 sport pigs',\n",
-       " 'reduce 10 superheavy birds',\n",
-       " 'reduce 2 antigravity birds',\n",
-       " 'reduce 2 big birds',\n",
-       " 'reduce 2 birds',\n",
-       " 'reduce 2 catapult birds',\n",
-       " 'reduce 2 elderly pigs',\n",
-       " 'reduce 2 flash birds',\n",
-       " 'reduce 2 magnet birds',\n",
-       " 'reduce 2 pigs',\n",
-       " 'reduce 2 sport pigs',\n",
-       " 'reduce 2 superheavy birds',\n",
-       " 'reduce 3 antigravity birds',\n",
-       " 'reduce 3 big birds',\n",
-       " 'reduce 3 birds',\n",
-       " 'reduce 3 catapult birds',\n",
-       " 'reduce 3 elderly pigs',\n",
-       " 'reduce 3 flash birds',\n",
-       " 'reduce 3 magnet birds',\n",
-       " 'reduce 3 pigs',\n",
-       " 'reduce 3 sport pigs',\n",
-       " 'reduce 3 superheavy birds',\n",
-       " 'reduce 4 antigravity birds',\n",
-       " 'reduce 4 big birds',\n",
-       " 'reduce 4 birds',\n",
-       " 'reduce 4 catapult birds',\n",
-       " 'reduce 4 elderly pigs',\n",
-       " 'reduce 4 flash birds',\n",
-       " 'reduce 4 magnet birds',\n",
-       " 'reduce 4 pigs',\n",
-       " 'reduce 4 sport pigs',\n",
-       " 'reduce 4 superheavy birds',\n",
-       " 'reduce 5 antigravity birds',\n",
-       " 'reduce 5 big birds',\n",
-       " 'reduce 5 birds',\n",
-       " 'reduce 5 catapult birds',\n",
-       " 'reduce 5 elderly pigs',\n",
-       " 'reduce 5 flash birds',\n",
-       " 'reduce 5 magnet birds',\n",
-       " 'reduce 5 pigs',\n",
-       " 'reduce 5 sport pigs',\n",
-       " 'reduce 5 superheavy birds',\n",
-       " 'reduce 6 antigravity birds',\n",
-       " 'reduce 6 big birds',\n",
-       " 'reduce 6 birds',\n",
-       " 'reduce 6 catapult birds',\n",
-       " 'reduce 6 elderly pigs',\n",
-       " 'reduce 6 flash birds',\n",
-       " 'reduce 6 magnet birds',\n",
-       " 'reduce 6 pigs',\n",
-       " 'reduce 6 sport pigs',\n",
-       " 'reduce 6 superheavy birds',\n",
-       " 'reduce 7 antigravity birds',\n",
-       " 'reduce 7 big birds',\n",
-       " 'reduce 7 birds',\n",
-       " 'reduce 7 catapult birds',\n",
-       " 'reduce 7 elderly pigs',\n",
-       " 'reduce 7 flash birds',\n",
-       " 'reduce 7 magnet birds',\n",
-       " 'reduce 7 pigs',\n",
-       " 'reduce 7 sport pigs',\n",
-       " 'reduce 7 superheavy birds',\n",
-       " 'reduce 8 antigravity birds',\n",
-       " 'reduce 8 big birds',\n",
-       " 'reduce 8 birds',\n",
-       " 'reduce 8 catapult birds',\n",
-       " 'reduce 8 elderly pigs',\n",
-       " 'reduce 8 flash birds',\n",
-       " 'reduce 8 magnet birds',\n",
-       " 'reduce 8 pigs',\n",
-       " 'reduce 8 sport pigs',\n",
-       " 'reduce 8 superheavy birds',\n",
-       " 'reduce 9 antigravity birds',\n",
-       " 'reduce 9 big birds',\n",
-       " 'reduce 9 birds',\n",
-       " 'reduce 9 catapult birds',\n",
-       " 'reduce 9 elderly pigs',\n",
-       " 'reduce 9 flash birds',\n",
-       " 'reduce 9 magnet birds',\n",
-       " 'reduce 9 pigs',\n",
-       " 'reduce 9 sport pigs',\n",
-       " 'reduce 9 superheavy birds',\n",
-       " 'reduce a big bird',\n",
-       " 'reduce a bird',\n",
-       " 'reduce a catapult bird',\n",
-       " 'reduce a flash bird',\n",
-       " 'reduce a magnet bird',\n",
-       " 'reduce a pig',\n",
-       " 'reduce a sport pig',\n",
-       " 'reduce a superheavy bird',\n",
-       " 'reduce all antigravity birds',\n",
-       " 'reduce all big birds',\n",
-       " 'reduce all birds',\n",
-       " 'reduce all catapult birds',\n",
-       " 'reduce all elderly pigs',\n",
-       " 'reduce all flash birds',\n",
-       " 'reduce all magnet birds',\n",
-       " 'reduce all pigs',\n",
-       " 'reduce all sport pigs',\n",
-       " 'reduce all superheavy birds',\n",
-       " 'reduce an antigravity bird',\n",
-       " 'reduce an elderly pig',\n",
-       " 'reduce antigravity birds',\n",
-       " 'reduce big birds',\n",
-       " 'reduce birds',\n",
-       " 'reduce catapult birds',\n",
-       " 'reduce eight antigravity birds',\n",
-       " 'reduce eight big birds',\n",
-       " 'reduce eight birds',\n",
-       " 'reduce eight catapult birds',\n",
-       " 'reduce eight elderly pigs',\n",
-       " 'reduce eight flash birds',\n",
-       " 'reduce eight magnet birds',\n",
-       " 'reduce eight pigs',\n",
-       " 'reduce eight sport pigs',\n",
-       " 'reduce eight superheavy birds',\n",
-       " 'reduce elderly pigs',\n",
-       " 'reduce five antigravity birds',\n",
-       " 'reduce five big birds',\n",
-       " 'reduce five birds',\n",
-       " 'reduce five catapult birds',\n",
-       " 'reduce five elderly pigs',\n",
-       " 'reduce five flash birds',\n",
-       " 'reduce five magnet birds',\n",
-       " 'reduce five pigs',\n",
-       " 'reduce five sport pigs',\n",
-       " 'reduce five superheavy birds',\n",
-       " 'reduce flash birds',\n",
-       " 'reduce four antigravity birds',\n",
-       " 'reduce four big birds',\n",
-       " 'reduce four birds',\n",
-       " 'reduce four catapult birds',\n",
-       " 'reduce four elderly pigs',\n",
-       " 'reduce four flash birds',\n",
-       " 'reduce four magnet birds',\n",
-       " 'reduce four pigs',\n",
-       " 'reduce four sport pigs',\n",
-       " 'reduce four superheavy birds',\n",
-       " 'reduce magnet birds',\n",
-       " 'reduce more antigravity birds',\n",
-       " 'reduce more big birds',\n",
-       " 'reduce more birds',\n",
-       " 'reduce more catapult birds',\n",
-       " 'reduce more elderly pigs',\n",
-       " 'reduce more flash birds',\n",
-       " 'reduce more magnet birds',\n",
-       " 'reduce more pigs',\n",
-       " 'reduce more sport pigs',\n",
-       " 'reduce more superheavy birds',\n",
-       " 'reduce nine antigravity birds',\n",
-       " 'reduce nine big birds',\n",
-       " 'reduce nine birds',\n",
-       " 'reduce nine catapult birds',\n",
-       " 'reduce nine elderly pigs',\n",
-       " 'reduce nine flash birds',\n",
-       " 'reduce nine magnet birds',\n",
-       " 'reduce nine pigs',\n",
-       " 'reduce nine sport pigs',\n",
-       " 'reduce nine superheavy birds',\n",
-       " 'reduce one antigravity bird',\n",
-       " 'reduce one big bird',\n",
-       " 'reduce one bird',\n",
-       " 'reduce one catapult bird',\n",
-       " 'reduce one elderly pig',\n",
-       " 'reduce one flash bird',\n",
-       " 'reduce one magnet bird',\n",
-       " 'reduce one pig',\n",
-       " 'reduce one sport pig',\n",
-       " 'reduce one superheavy bird',\n",
-       " 'reduce pigs',\n",
-       " 'reduce seven antigravity birds',\n",
-       " 'reduce seven big birds',\n",
-       " 'reduce seven birds',\n",
-       " 'reduce seven catapult birds',\n",
-       " 'reduce seven elderly pigs',\n",
-       " 'reduce seven flash birds',\n",
-       " 'reduce seven magnet birds',\n",
-       " 'reduce seven pigs',\n",
-       " 'reduce seven sport pigs',\n",
-       " 'reduce seven superheavy birds',\n",
-       " 'reduce six antigravity birds',\n",
-       " 'reduce six big birds',\n",
-       " 'reduce six birds',\n",
-       " 'reduce six catapult birds',\n",
-       " 'reduce six elderly pigs',\n",
-       " 'reduce six flash birds',\n",
-       " 'reduce six magnet birds',\n",
-       " 'reduce six pigs',\n",
-       " 'reduce six sport pigs',\n",
-       " 'reduce six superheavy birds',\n",
-       " 'reduce some antigravity birds',\n",
-       " 'reduce some big birds',\n",
-       " 'reduce some birds',\n",
-       " 'reduce some catapult birds',\n",
-       " 'reduce some elderly pigs',\n",
-       " 'reduce some flash birds',\n",
-       " 'reduce some magnet birds',\n",
-       " 'reduce some pigs',\n",
-       " 'reduce some sport pigs',\n",
-       " 'reduce some superheavy birds',\n",
-       " 'reduce sport pigs',\n",
-       " 'reduce superheavy birds',\n",
-       " 'reduce ten antigravity birds',\n",
-       " 'reduce ten big birds',\n",
-       " 'reduce ten birds',\n",
-       " 'reduce ten catapult birds',\n",
-       " 'reduce ten elderly pigs',\n",
-       " 'reduce ten flash birds',\n",
-       " 'reduce ten magnet birds',\n",
-       " 'reduce ten pigs',\n",
-       " 'reduce ten sport pigs',\n",
-       " 'reduce ten superheavy birds',\n",
-       " 'reduce three antigravity birds',\n",
-       " 'reduce three big birds',\n",
-       " 'reduce three birds',\n",
-       " 'reduce three catapult birds',\n",
-       " 'reduce three elderly pigs',\n",
-       " 'reduce three flash birds',\n",
-       " 'reduce three magnet birds',\n",
-       " 'reduce three pigs',\n",
-       " 'reduce three sport pigs',\n",
-       " 'reduce three superheavy birds',\n",
-       " 'reduce two antigravity birds',\n",
-       " 'reduce two big birds',\n",
-       " 'reduce two birds',\n",
-       " 'reduce two catapult birds',\n",
-       " 'reduce two elderly pigs',\n",
-       " 'reduce two flash birds',\n",
-       " 'reduce two magnet birds',\n",
-       " 'reduce two pigs',\n",
-       " 'reduce two sport pigs',\n",
-       " 'reduce two superheavy birds',\n",
-       " 'show bird list',\n",
-       " 'show pig list',\n",
-       " \"the obstacles are too weak. let's build some solid houses\",\n",
-       " 'the pigs are too weak. increase the blood level for them',\n",
-       " 'the pigs are too weak. increase the blood levels for them',\n",
-       " 'the pigs are too weak. increase the health points for them',\n",
-       " 'the pigs are too weak. increase the hit points for them',\n",
-       " 'the pigs are too weak. increase the hp for them',\n",
-       " 'what kind of birds do you have?',\n",
-       " 'what kind of pigs do you have?']"
-      ]
-     },
-     "execution_count": 20,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "sorted(keys)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "54dfec68-3956-4b92-83a2-e1ab6a9cec9c",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "id": "29eab725-99f4-480f-9750-bf41758be527",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open('../game_train.json', 'r') as f:\n",
-    "    s = json.load(f)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "id": "559708ec-5adf-4648-a2ef-14bc0a70fea4",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "773"
-      ]
-     },
-     "execution_count": 5,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(s)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "id": "8cb63fe9-c8a0-4316-aace-92ee6477c97a",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "True"
-      ]
-     },
-     "execution_count": 6,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "'build a house' in s"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 7,
-   "id": "6908bc1e-4c93-422b-b392-d9a66863c8e9",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "773"
-      ]
-     },
-     "execution_count": 7,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(s)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 8,
-   "id": "987ccc02-9837-4ba1-8d02-0750d9ab94ea",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "773"
-      ]
-     },
-     "execution_count": 8,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len([x.lower() for x in s.keys()])"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 9,
-   "id": "ff240004-b842-4d99-8d55-4fc936bcaf31",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "773"
-      ]
-     },
-     "execution_count": 9,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(set([x.lower() for x in s.keys()]))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 10,
-   "id": "fb6990af-55c5-4c56-98f5-3816571323ce",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from collections import Counter"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 11,
-   "id": "8320aace-22b5-44a2-bab3-fe81d5cb9f1e",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# Counter([x.lower() for x in s.keys()]).most_common()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "69fec331-e930-449d-9be4-53b9aa01ea73",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "56c7ba52-8828-4cf9-a363-442a5b6058f5",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.4"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}

games/angrybirds/notebooks/test.ipynb

@@ -1,1265 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 7,
-   "id": "245fbc34-352d-4945-a689-d04f9365bb99",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "%load_ext autoreload\n",
-    "%autoreload 2"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 8,
-   "id": "eaed216c-160a-4e31-96fe-015c140ff18b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# from pprint import pprint"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 9,
-   "id": "8ddfb0a9-57bd-411a-98af-331ae5098c2d",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import json\n",
-    "import pandas as pd\n",
-    "\n",
-    "import openai\n",
-    "from openai.embeddings_utils import get_embedding, cosine_similarity"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 10,
-   "id": "6e3c3925-2fc6-412d-bb10-9492c9457273",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# from games.angrybirds.util import util\n",
-    "# from games.angrybirds.config import config\n",
-    "# from games.angrybirds.prompt import prompt\n",
-    "# from games.angrybirds.query import query, yes_no_query\n",
-    "from games.angrybirds.angrybirds import angrybird\n",
-    "from games.angrybirds.common import api_key"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "ae27ebd0-2d95-4ff0-b188-f6e46b189cd1",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 11,
-   "id": "2839129a-61bf-4182-a9e3-477935d18a5f",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "openai.api_key = api_key"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "id": "468f0a2f-a42d-4cf9-9c32-a1c0daaf7b2b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# optional"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 13,
-   "id": "dc3829e5-9896-46ca-ae49-6830ff99a15d",
-   "metadata": {},
-   "outputs": [
-    {
-     "ename": "Timeout",
-     "evalue": "Request timed out: HTTPSConnectionPool(host='api.openai.com', port=443): Max retries exceeded with url: /v1/models (Caused by ConnectTimeoutError(<urllib3.connection.HTTPSConnection object at 0x129e054d0>, 'Connection to api.openai.com timed out. (connect timeout=600)'))",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
-      "\u001b[0;31mTimeoutError\u001b[0m                              Traceback (most recent call last)",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connection.py:200\u001b[0m, in \u001b[0;36mHTTPConnection._new_conn\u001b[0;34m(self)\u001b[0m\n\u001b[1;32m    199\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m--> 200\u001b[0m     sock \u001b[38;5;241m=\u001b[39m \u001b[43mconnection\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mcreate_connection\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    201\u001b[0m \u001b[43m        \u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43m_dns_host\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mport\u001b[49m\u001b[43m)\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    202\u001b[0m \u001b[43m        \u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mtimeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    203\u001b[0m \u001b[43m        \u001b[49m\u001b[43msource_address\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43msource_address\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    204\u001b[0m \u001b[43m        \u001b[49m\u001b[43msocket_options\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43msocket_options\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    205\u001b[0m \u001b[43m    \u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    206\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m socket\u001b[38;5;241m.\u001b[39mgaierror \u001b[38;5;28;01mas\u001b[39;00m e:\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/util/connection.py:85\u001b[0m, in \u001b[0;36mcreate_connection\u001b[0;34m(address, timeout, source_address, socket_options)\u001b[0m\n\u001b[1;32m     84\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m---> 85\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m err\n\u001b[1;32m     86\u001b[0m \u001b[38;5;28;01mfinally\u001b[39;00m:\n\u001b[1;32m     87\u001b[0m     \u001b[38;5;66;03m# Break explicitly a reference cycle\u001b[39;00m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/util/connection.py:73\u001b[0m, in \u001b[0;36mcreate_connection\u001b[0;34m(address, timeout, source_address, socket_options)\u001b[0m\n\u001b[1;32m     72\u001b[0m     sock\u001b[38;5;241m.\u001b[39mbind(source_address)\n\u001b[0;32m---> 73\u001b[0m \u001b[43msock\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mconnect\u001b[49m\u001b[43m(\u001b[49m\u001b[43msa\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m     74\u001b[0m \u001b[38;5;66;03m# Break explicitly a reference cycle\u001b[39;00m\n",
-      "\u001b[0;31mTimeoutError\u001b[0m: [Errno 60] Operation timed out",
-      "\nThe above exception was the direct cause of the following exception:\n",
-      "\u001b[0;31mConnectTimeoutError\u001b[0m                       Traceback (most recent call last)",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py:790\u001b[0m, in \u001b[0;36mHTTPConnectionPool.urlopen\u001b[0;34m(self, method, url, body, headers, retries, redirect, assert_same_host, timeout, pool_timeout, release_conn, chunked, body_pos, preload_content, decode_content, **response_kw)\u001b[0m\n\u001b[1;32m    789\u001b[0m \u001b[38;5;66;03m# Make the request on the HTTPConnection object\u001b[39;00m\n\u001b[0;32m--> 790\u001b[0m response \u001b[38;5;241m=\u001b[39m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43m_make_request\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    791\u001b[0m \u001b[43m    \u001b[49m\u001b[43mconn\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    792\u001b[0m \u001b[43m    \u001b[49m\u001b[43mmethod\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    793\u001b[0m \u001b[43m    \u001b[49m\u001b[43murl\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    794\u001b[0m \u001b[43m    \u001b[49m\u001b[43mtimeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mtimeout_obj\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    795\u001b[0m \u001b[43m    \u001b[49m\u001b[43mbody\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mbody\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    796\u001b[0m \u001b[43m    \u001b[49m\u001b[43mheaders\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    797\u001b[0m \u001b[43m    \u001b[49m\u001b[43mchunked\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mchunked\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    798\u001b[0m \u001b[43m    \u001b[49m\u001b[43mretries\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mretries\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    799\u001b[0m \u001b[43m    \u001b[49m\u001b[43mresponse_conn\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mresponse_conn\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    800\u001b[0m \u001b[43m    \u001b[49m\u001b[43mpreload_content\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mpreload_content\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    801\u001b[0m \u001b[43m    \u001b[49m\u001b[43mdecode_content\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mdecode_content\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    802\u001b[0m \u001b[43m    \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mresponse_kw\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    803\u001b[0m \u001b[43m\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    805\u001b[0m \u001b[38;5;66;03m# Everything went great!\u001b[39;00m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py:491\u001b[0m, in \u001b[0;36mHTTPConnectionPool._make_request\u001b[0;34m(self, conn, method, url, body, headers, retries, timeout, chunked, response_conn, preload_content, decode_content, enforce_content_length)\u001b[0m\n\u001b[1;32m    490\u001b[0m         new_e \u001b[38;5;241m=\u001b[39m _wrap_proxy_error(new_e, conn\u001b[38;5;241m.\u001b[39mproxy\u001b[38;5;241m.\u001b[39mscheme)\n\u001b[0;32m--> 491\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m new_e\n\u001b[1;32m    493\u001b[0m \u001b[38;5;66;03m# conn.request() calls http.client.*.request, not the method in\u001b[39;00m\n\u001b[1;32m    494\u001b[0m \u001b[38;5;66;03m# urllib3.request. It also calls makefile (recv) on the socket.\u001b[39;00m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py:467\u001b[0m, in \u001b[0;36mHTTPConnectionPool._make_request\u001b[0;34m(self, conn, method, url, body, headers, retries, timeout, chunked, response_conn, preload_content, decode_content, enforce_content_length)\u001b[0m\n\u001b[1;32m    466\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m--> 467\u001b[0m     \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43m_validate_conn\u001b[49m\u001b[43m(\u001b[49m\u001b[43mconn\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    468\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m (SocketTimeout, BaseSSLError) \u001b[38;5;28;01mas\u001b[39;00m e:\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py:1092\u001b[0m, in \u001b[0;36mHTTPSConnectionPool._validate_conn\u001b[0;34m(self, conn)\u001b[0m\n\u001b[1;32m   1091\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m conn\u001b[38;5;241m.\u001b[39mis_closed:\n\u001b[0;32m-> 1092\u001b[0m     \u001b[43mconn\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mconnect\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m   1094\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;129;01mnot\u001b[39;00m conn\u001b[38;5;241m.\u001b[39mis_verified:\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connection.py:604\u001b[0m, in \u001b[0;36mHTTPSConnection.connect\u001b[0;34m(self)\u001b[0m\n\u001b[1;32m    603\u001b[0m sock: socket\u001b[38;5;241m.\u001b[39msocket \u001b[38;5;241m|\u001b[39m ssl\u001b[38;5;241m.\u001b[39mSSLSocket\n\u001b[0;32m--> 604\u001b[0m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39msock \u001b[38;5;241m=\u001b[39m sock \u001b[38;5;241m=\u001b[39m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43m_new_conn\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    605\u001b[0m server_hostname: \u001b[38;5;28mstr\u001b[39m \u001b[38;5;241m=\u001b[39m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mhost\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connection.py:209\u001b[0m, in \u001b[0;36mHTTPConnection._new_conn\u001b[0;34m(self)\u001b[0m\n\u001b[1;32m    208\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m SocketTimeout \u001b[38;5;28;01mas\u001b[39;00m e:\n\u001b[0;32m--> 209\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m ConnectTimeoutError(\n\u001b[1;32m    210\u001b[0m         \u001b[38;5;28mself\u001b[39m,\n\u001b[1;32m    211\u001b[0m         \u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mConnection to \u001b[39m\u001b[38;5;132;01m{\u001b[39;00m\u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mhost\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m timed out. (connect timeout=\u001b[39m\u001b[38;5;132;01m{\u001b[39;00m\u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mtimeout\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m)\u001b[39m\u001b[38;5;124m\"\u001b[39m,\n\u001b[1;32m    212\u001b[0m     ) \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01me\u001b[39;00m\n\u001b[1;32m    214\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m \u001b[38;5;167;01mOSError\u001b[39;00m \u001b[38;5;28;01mas\u001b[39;00m e:\n",
-      "\u001b[0;31mConnectTimeoutError\u001b[0m: (<urllib3.connection.HTTPSConnection object at 0x129e054d0>, 'Connection to api.openai.com timed out. (connect timeout=600)')",
-      "\nThe above exception was the direct cause of the following exception:\n",
-      "\u001b[0;31mMaxRetryError\u001b[0m                             Traceback (most recent call last)",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/requests/adapters.py:486\u001b[0m, in \u001b[0;36mHTTPAdapter.send\u001b[0;34m(self, request, stream, timeout, verify, cert, proxies)\u001b[0m\n\u001b[1;32m    485\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m--> 486\u001b[0m     resp \u001b[38;5;241m=\u001b[39m \u001b[43mconn\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43murlopen\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    487\u001b[0m \u001b[43m        \u001b[49m\u001b[43mmethod\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrequest\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mmethod\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    488\u001b[0m \u001b[43m        \u001b[49m\u001b[43murl\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43murl\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    489\u001b[0m \u001b[43m        \u001b[49m\u001b[43mbody\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrequest\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mbody\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    490\u001b[0m \u001b[43m        \u001b[49m\u001b[43mheaders\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrequest\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    491\u001b[0m \u001b[43m        \u001b[49m\u001b[43mredirect\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43;01mFalse\u001b[39;49;00m\u001b[43m,\u001b[49m\n\u001b[1;32m    492\u001b[0m \u001b[43m        \u001b[49m\u001b[43massert_same_host\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43;01mFalse\u001b[39;49;00m\u001b[43m,\u001b[49m\n\u001b[1;32m    493\u001b[0m \u001b[43m        \u001b[49m\u001b[43mpreload_content\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43;01mFalse\u001b[39;49;00m\u001b[43m,\u001b[49m\n\u001b[1;32m    494\u001b[0m \u001b[43m        \u001b[49m\u001b[43mdecode_content\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43;01mFalse\u001b[39;49;00m\u001b[43m,\u001b[49m\n\u001b[1;32m    495\u001b[0m \u001b[43m        \u001b[49m\u001b[43mretries\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mmax_retries\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    496\u001b[0m \u001b[43m        \u001b[49m\u001b[43mtimeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mtimeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    497\u001b[0m \u001b[43m        \u001b[49m\u001b[43mchunked\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mchunked\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    498\u001b[0m \u001b[43m    \u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    500\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m (ProtocolError, \u001b[38;5;167;01mOSError\u001b[39;00m) \u001b[38;5;28;01mas\u001b[39;00m err:\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py:874\u001b[0m, in \u001b[0;36mHTTPConnectionPool.urlopen\u001b[0;34m(self, method, url, body, headers, retries, redirect, assert_same_host, timeout, pool_timeout, release_conn, chunked, body_pos, preload_content, decode_content, **response_kw)\u001b[0m\n\u001b[1;32m    871\u001b[0m     log\u001b[38;5;241m.\u001b[39mwarning(\n\u001b[1;32m    872\u001b[0m         \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mRetrying (\u001b[39m\u001b[38;5;132;01m%r\u001b[39;00m\u001b[38;5;124m) after connection broken by \u001b[39m\u001b[38;5;124m'\u001b[39m\u001b[38;5;132;01m%r\u001b[39;00m\u001b[38;5;124m'\u001b[39m\u001b[38;5;124m: \u001b[39m\u001b[38;5;132;01m%s\u001b[39;00m\u001b[38;5;124m\"\u001b[39m, retries, err, url\n\u001b[1;32m    873\u001b[0m     )\n\u001b[0;32m--> 874\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43murlopen\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    875\u001b[0m \u001b[43m        \u001b[49m\u001b[43mmethod\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    876\u001b[0m \u001b[43m        \u001b[49m\u001b[43murl\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    877\u001b[0m \u001b[43m        \u001b[49m\u001b[43mbody\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    878\u001b[0m \u001b[43m        \u001b[49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    879\u001b[0m \u001b[43m        \u001b[49m\u001b[43mretries\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    880\u001b[0m \u001b[43m        \u001b[49m\u001b[43mredirect\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    881\u001b[0m \u001b[43m        \u001b[49m\u001b[43massert_same_host\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    882\u001b[0m \u001b[43m        \u001b[49m\u001b[43mtimeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mtimeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    883\u001b[0m \u001b[43m        \u001b[49m\u001b[43mpool_timeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mpool_timeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    884\u001b[0m \u001b[43m        \u001b[49m\u001b[43mrelease_conn\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrelease_conn\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    885\u001b[0m \u001b[43m        \u001b[49m\u001b[43mchunked\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mchunked\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    886\u001b[0m \u001b[43m        \u001b[49m\u001b[43mbody_pos\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mbody_pos\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    887\u001b[0m \u001b[43m        \u001b[49m\u001b[43mpreload_content\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mpreload_content\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    888\u001b[0m \u001b[43m        \u001b[49m\u001b[43mdecode_content\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mdecode_content\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    889\u001b[0m \u001b[43m        \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mresponse_kw\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    890\u001b[0m \u001b[43m    \u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    892\u001b[0m \u001b[38;5;66;03m# Handle redirect?\u001b[39;00m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py:874\u001b[0m, in \u001b[0;36mHTTPConnectionPool.urlopen\u001b[0;34m(self, method, url, body, headers, retries, redirect, assert_same_host, timeout, pool_timeout, release_conn, chunked, body_pos, preload_content, decode_content, **response_kw)\u001b[0m\n\u001b[1;32m    871\u001b[0m     log\u001b[38;5;241m.\u001b[39mwarning(\n\u001b[1;32m    872\u001b[0m         \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mRetrying (\u001b[39m\u001b[38;5;132;01m%r\u001b[39;00m\u001b[38;5;124m) after connection broken by \u001b[39m\u001b[38;5;124m'\u001b[39m\u001b[38;5;132;01m%r\u001b[39;00m\u001b[38;5;124m'\u001b[39m\u001b[38;5;124m: \u001b[39m\u001b[38;5;132;01m%s\u001b[39;00m\u001b[38;5;124m\"\u001b[39m, retries, err, url\n\u001b[1;32m    873\u001b[0m     )\n\u001b[0;32m--> 874\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43murlopen\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    875\u001b[0m \u001b[43m        \u001b[49m\u001b[43mmethod\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    876\u001b[0m \u001b[43m        \u001b[49m\u001b[43murl\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    877\u001b[0m \u001b[43m        \u001b[49m\u001b[43mbody\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    878\u001b[0m \u001b[43m        \u001b[49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    879\u001b[0m \u001b[43m        \u001b[49m\u001b[43mretries\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    880\u001b[0m \u001b[43m        \u001b[49m\u001b[43mredirect\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    881\u001b[0m \u001b[43m        \u001b[49m\u001b[43massert_same_host\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    882\u001b[0m \u001b[43m        \u001b[49m\u001b[43mtimeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mtimeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    883\u001b[0m \u001b[43m        \u001b[49m\u001b[43mpool_timeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mpool_timeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    884\u001b[0m \u001b[43m        \u001b[49m\u001b[43mrelease_conn\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrelease_conn\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    885\u001b[0m \u001b[43m        \u001b[49m\u001b[43mchunked\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mchunked\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    886\u001b[0m \u001b[43m        \u001b[49m\u001b[43mbody_pos\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mbody_pos\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    887\u001b[0m \u001b[43m        \u001b[49m\u001b[43mpreload_content\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mpreload_content\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    888\u001b[0m \u001b[43m        \u001b[49m\u001b[43mdecode_content\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mdecode_content\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    889\u001b[0m \u001b[43m        \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mresponse_kw\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    890\u001b[0m \u001b[43m    \u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    892\u001b[0m \u001b[38;5;66;03m# Handle redirect?\u001b[39;00m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py:844\u001b[0m, in \u001b[0;36mHTTPConnectionPool.urlopen\u001b[0;34m(self, method, url, body, headers, retries, redirect, assert_same_host, timeout, pool_timeout, release_conn, chunked, body_pos, preload_content, decode_content, **response_kw)\u001b[0m\n\u001b[1;32m    842\u001b[0m     new_e \u001b[38;5;241m=\u001b[39m ProtocolError(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mConnection aborted.\u001b[39m\u001b[38;5;124m\"\u001b[39m, new_e)\n\u001b[0;32m--> 844\u001b[0m retries \u001b[38;5;241m=\u001b[39m \u001b[43mretries\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mincrement\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    845\u001b[0m \u001b[43m    \u001b[49m\u001b[43mmethod\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43murl\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43merror\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mnew_e\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43m_pool\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43m_stacktrace\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43msys\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mexc_info\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\u001b[43m[\u001b[49m\u001b[38;5;241;43m2\u001b[39;49m\u001b[43m]\u001b[49m\n\u001b[1;32m    846\u001b[0m \u001b[43m\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    847\u001b[0m retries\u001b[38;5;241m.\u001b[39msleep()\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/urllib3/util/retry.py:515\u001b[0m, in \u001b[0;36mRetry.increment\u001b[0;34m(self, method, url, response, error, _pool, _stacktrace)\u001b[0m\n\u001b[1;32m    514\u001b[0m     reason \u001b[38;5;241m=\u001b[39m error \u001b[38;5;129;01mor\u001b[39;00m ResponseError(cause)\n\u001b[0;32m--> 515\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m MaxRetryError(_pool, url, reason) \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mreason\u001b[39;00m  \u001b[38;5;66;03m# type: ignore[arg-type]\u001b[39;00m\n\u001b[1;32m    517\u001b[0m log\u001b[38;5;241m.\u001b[39mdebug(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mIncremented Retry for (url=\u001b[39m\u001b[38;5;124m'\u001b[39m\u001b[38;5;132;01m%s\u001b[39;00m\u001b[38;5;124m'\u001b[39m\u001b[38;5;124m): \u001b[39m\u001b[38;5;132;01m%r\u001b[39;00m\u001b[38;5;124m\"\u001b[39m, url, new_retry)\n",
-      "\u001b[0;31mMaxRetryError\u001b[0m: HTTPSConnectionPool(host='api.openai.com', port=443): Max retries exceeded with url: /v1/models (Caused by ConnectTimeoutError(<urllib3.connection.HTTPSConnection object at 0x129e054d0>, 'Connection to api.openai.com timed out. (connect timeout=600)'))",
-      "\nDuring handling of the above exception, another exception occurred:\n",
-      "\u001b[0;31mConnectTimeout\u001b[0m                            Traceback (most recent call last)",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/openai/api_requestor.py:596\u001b[0m, in \u001b[0;36mAPIRequestor.request_raw\u001b[0;34m(self, method, url, params, supplied_headers, files, stream, request_id, request_timeout)\u001b[0m\n\u001b[1;32m    595\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m--> 596\u001b[0m     result \u001b[38;5;241m=\u001b[39m \u001b[43m_thread_context\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43msession\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mrequest\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    597\u001b[0m \u001b[43m        \u001b[49m\u001b[43mmethod\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    598\u001b[0m \u001b[43m        \u001b[49m\u001b[43mabs_url\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    599\u001b[0m \u001b[43m        \u001b[49m\u001b[43mheaders\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    600\u001b[0m \u001b[43m        \u001b[49m\u001b[43mdata\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mdata\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    601\u001b[0m \u001b[43m        \u001b[49m\u001b[43mfiles\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mfiles\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    602\u001b[0m \u001b[43m        \u001b[49m\u001b[43mstream\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mstream\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    603\u001b[0m \u001b[43m        \u001b[49m\u001b[43mtimeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrequest_timeout\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;28;43;01mif\u001b[39;49;00m\u001b[43m \u001b[49m\u001b[43mrequest_timeout\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;28;43;01melse\u001b[39;49;00m\u001b[43m \u001b[49m\u001b[43mTIMEOUT_SECS\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    604\u001b[0m \u001b[43m        \u001b[49m\u001b[43mproxies\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43m_thread_context\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43msession\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mproxies\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    605\u001b[0m \u001b[43m    \u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    606\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m requests\u001b[38;5;241m.\u001b[39mexceptions\u001b[38;5;241m.\u001b[39mTimeout \u001b[38;5;28;01mas\u001b[39;00m e:\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/requests/sessions.py:589\u001b[0m, in \u001b[0;36mSession.request\u001b[0;34m(self, method, url, params, data, headers, cookies, files, auth, timeout, allow_redirects, proxies, hooks, stream, verify, cert, json)\u001b[0m\n\u001b[1;32m    588\u001b[0m send_kwargs\u001b[38;5;241m.\u001b[39mupdate(settings)\n\u001b[0;32m--> 589\u001b[0m resp \u001b[38;5;241m=\u001b[39m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43msend\u001b[49m\u001b[43m(\u001b[49m\u001b[43mprep\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43msend_kwargs\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    591\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m resp\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/requests/sessions.py:703\u001b[0m, in \u001b[0;36mSession.send\u001b[0;34m(self, request, **kwargs)\u001b[0m\n\u001b[1;32m    702\u001b[0m \u001b[38;5;66;03m# Send the request\u001b[39;00m\n\u001b[0;32m--> 703\u001b[0m r \u001b[38;5;241m=\u001b[39m \u001b[43madapter\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43msend\u001b[49m\u001b[43m(\u001b[49m\u001b[43mrequest\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mkwargs\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    705\u001b[0m \u001b[38;5;66;03m# Total elapsed time of the request (approximately)\u001b[39;00m\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/requests/adapters.py:507\u001b[0m, in \u001b[0;36mHTTPAdapter.send\u001b[0;34m(self, request, stream, timeout, verify, cert, proxies)\u001b[0m\n\u001b[1;32m    506\u001b[0m     \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;129;01mnot\u001b[39;00m \u001b[38;5;28misinstance\u001b[39m(e\u001b[38;5;241m.\u001b[39mreason, NewConnectionError):\n\u001b[0;32m--> 507\u001b[0m         \u001b[38;5;28;01mraise\u001b[39;00m ConnectTimeout(e, request\u001b[38;5;241m=\u001b[39mrequest)\n\u001b[1;32m    509\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;28misinstance\u001b[39m(e\u001b[38;5;241m.\u001b[39mreason, ResponseError):\n",
-      "\u001b[0;31mConnectTimeout\u001b[0m: HTTPSConnectionPool(host='api.openai.com', port=443): Max retries exceeded with url: /v1/models (Caused by ConnectTimeoutError(<urllib3.connection.HTTPSConnection object at 0x129e054d0>, 'Connection to api.openai.com timed out. (connect timeout=600)'))",
-      "\nThe above exception was the direct cause of the following exception:\n",
-      "\u001b[0;31mTimeout\u001b[0m                                   Traceback (most recent call last)",
-      "Cell \u001b[0;32mIn[13], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m models \u001b[38;5;241m=\u001b[39m \u001b[43mopenai\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mModel\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mlist\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m      2\u001b[0m [d\u001b[38;5;241m.\u001b[39mid \u001b[38;5;28;01mfor\u001b[39;00m d \u001b[38;5;129;01min\u001b[39;00m models\u001b[38;5;241m.\u001b[39mdata \u001b[38;5;28;01mif\u001b[39;00m d\u001b[38;5;241m.\u001b[39mid\u001b[38;5;241m.\u001b[39mstartswith(\u001b[38;5;124m'\u001b[39m\u001b[38;5;124mg\u001b[39m\u001b[38;5;124m'\u001b[39m)]\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/openai/api_resources/abstract/listable_api_resource.py:60\u001b[0m, in \u001b[0;36mListableAPIResource.list\u001b[0;34m(cls, api_key, request_id, api_version, organization, api_base, api_type, **params)\u001b[0m\n\u001b[1;32m     41\u001b[0m \u001b[38;5;129m@classmethod\u001b[39m\n\u001b[1;32m     42\u001b[0m \u001b[38;5;28;01mdef\u001b[39;00m \u001b[38;5;21mlist\u001b[39m(\n\u001b[1;32m     43\u001b[0m     \u001b[38;5;28mcls\u001b[39m,\n\u001b[0;32m   (...)\u001b[0m\n\u001b[1;32m     50\u001b[0m     \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mparams,\n\u001b[1;32m     51\u001b[0m ):\n\u001b[1;32m     52\u001b[0m     requestor, url \u001b[38;5;241m=\u001b[39m \u001b[38;5;28mcls\u001b[39m\u001b[38;5;241m.\u001b[39m__prepare_list_requestor(\n\u001b[1;32m     53\u001b[0m         api_key,\n\u001b[1;32m     54\u001b[0m         api_version,\n\u001b[0;32m   (...)\u001b[0m\n\u001b[1;32m     57\u001b[0m         api_type,\n\u001b[1;32m     58\u001b[0m     )\n\u001b[0;32m---> 60\u001b[0m     response, _, api_key \u001b[38;5;241m=\u001b[39m \u001b[43mrequestor\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mrequest\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m     61\u001b[0m \u001b[43m        \u001b[49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43mget\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43murl\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mparams\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mrequest_id\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrequest_id\u001b[49m\n\u001b[1;32m     62\u001b[0m \u001b[43m    \u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m     63\u001b[0m     openai_object \u001b[38;5;241m=\u001b[39m util\u001b[38;5;241m.\u001b[39mconvert_to_openai_object(\n\u001b[1;32m     64\u001b[0m         response, api_key, api_version, organization\n\u001b[1;32m     65\u001b[0m     )\n\u001b[1;32m     66\u001b[0m     openai_object\u001b[38;5;241m.\u001b[39m_retrieve_params \u001b[38;5;241m=\u001b[39m params\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/openai/api_requestor.py:288\u001b[0m, in \u001b[0;36mAPIRequestor.request\u001b[0;34m(self, method, url, params, headers, files, stream, request_id, request_timeout)\u001b[0m\n\u001b[1;32m    277\u001b[0m \u001b[38;5;28;01mdef\u001b[39;00m \u001b[38;5;21mrequest\u001b[39m(\n\u001b[1;32m    278\u001b[0m     \u001b[38;5;28mself\u001b[39m,\n\u001b[1;32m    279\u001b[0m     method,\n\u001b[0;32m   (...)\u001b[0m\n\u001b[1;32m    286\u001b[0m     request_timeout: Optional[Union[\u001b[38;5;28mfloat\u001b[39m, Tuple[\u001b[38;5;28mfloat\u001b[39m, \u001b[38;5;28mfloat\u001b[39m]]] \u001b[38;5;241m=\u001b[39m \u001b[38;5;28;01mNone\u001b[39;00m,\n\u001b[1;32m    287\u001b[0m ) \u001b[38;5;241m-\u001b[39m\u001b[38;5;241m>\u001b[39m Tuple[Union[OpenAIResponse, Iterator[OpenAIResponse]], \u001b[38;5;28mbool\u001b[39m, \u001b[38;5;28mstr\u001b[39m]:\n\u001b[0;32m--> 288\u001b[0m     result \u001b[38;5;241m=\u001b[39m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mrequest_raw\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m    289\u001b[0m \u001b[43m        \u001b[49m\u001b[43mmethod\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mlower\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    290\u001b[0m \u001b[43m        \u001b[49m\u001b[43murl\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    291\u001b[0m \u001b[43m        \u001b[49m\u001b[43mparams\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mparams\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    292\u001b[0m \u001b[43m        \u001b[49m\u001b[43msupplied_headers\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mheaders\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    293\u001b[0m \u001b[43m        \u001b[49m\u001b[43mfiles\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mfiles\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    294\u001b[0m \u001b[43m        \u001b[49m\u001b[43mstream\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mstream\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    295\u001b[0m \u001b[43m        \u001b[49m\u001b[43mrequest_id\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrequest_id\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    296\u001b[0m \u001b[43m        \u001b[49m\u001b[43mrequest_timeout\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrequest_timeout\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m    297\u001b[0m \u001b[43m    \u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    298\u001b[0m     resp, got_stream \u001b[38;5;241m=\u001b[39m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39m_interpret_response(result, stream)\n\u001b[1;32m    299\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m resp, got_stream, \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mapi_key\n",
-      "File \u001b[0;32m~/playgo/abc123/.venv/lib/python3.11/site-packages/openai/api_requestor.py:607\u001b[0m, in \u001b[0;36mAPIRequestor.request_raw\u001b[0;34m(self, method, url, params, supplied_headers, files, stream, request_id, request_timeout)\u001b[0m\n\u001b[1;32m    596\u001b[0m     result \u001b[38;5;241m=\u001b[39m _thread_context\u001b[38;5;241m.\u001b[39msession\u001b[38;5;241m.\u001b[39mrequest(\n\u001b[1;32m    597\u001b[0m         method,\n\u001b[1;32m    598\u001b[0m         abs_url,\n\u001b[0;32m   (...)\u001b[0m\n\u001b[1;32m    604\u001b[0m         proxies\u001b[38;5;241m=\u001b[39m_thread_context\u001b[38;5;241m.\u001b[39msession\u001b[38;5;241m.\u001b[39mproxies,\n\u001b[1;32m    605\u001b[0m     )\n\u001b[1;32m    606\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m requests\u001b[38;5;241m.\u001b[39mexceptions\u001b[38;5;241m.\u001b[39mTimeout \u001b[38;5;28;01mas\u001b[39;00m e:\n\u001b[0;32m--> 607\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m error\u001b[38;5;241m.\u001b[39mTimeout(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mRequest timed out: \u001b[39m\u001b[38;5;132;01m{}\u001b[39;00m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;241m.\u001b[39mformat(e)) \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01me\u001b[39;00m\n\u001b[1;32m    608\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m requests\u001b[38;5;241m.\u001b[39mexceptions\u001b[38;5;241m.\u001b[39mRequestException \u001b[38;5;28;01mas\u001b[39;00m e:\n\u001b[1;32m    609\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m error\u001b[38;5;241m.\u001b[39mAPIConnectionError(\n\u001b[1;32m    610\u001b[0m         \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mError communicating with OpenAI: \u001b[39m\u001b[38;5;132;01m{}\u001b[39;00m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;241m.\u001b[39mformat(e)\n\u001b[1;32m    611\u001b[0m     ) \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01me\u001b[39;00m\n",
-      "\u001b[0;31mTimeout\u001b[0m: Request timed out: HTTPSConnectionPool(host='api.openai.com', port=443): Max retries exceeded with url: /v1/models (Caused by ConnectTimeoutError(<urllib3.connection.HTTPSConnection object at 0x129e054d0>, 'Connection to api.openai.com timed out. (connect timeout=600)'))"
-     ]
-    }
-   ],
-   "source": [
-    "models = openai.Model.list()\n",
-    "[d.id for d in models.data if d.id.startswith('g')]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "id": "0668fb4b-b4c1-4be1-aa4a-c0c2ada7c128",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['gpt-3.5-turbo-16k-0613',\n",
-       " 'gpt-3.5-turbo-16k',\n",
-       " 'gpt-3.5-turbo',\n",
-       " 'gpt-3.5-turbo-0301',\n",
-       " 'gpt-3.5-turbo-0613']"
-      ]
-     },
-     "execution_count": 6,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "models = openai.Model.list()\n",
-    "[d.id for d in models.data if d.id.startswith('g')]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "daf7a5db-dd42-4452-8c3d-41a466e7da7e",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 21,
-   "id": "0a5c3ff3-2bb1-4f72-9aa3-04473a7b714e",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "space = 'stevez-abc123'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "21fe7b7b-eb92-45f4-8cf4-d42aedebf9e8",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "f589b196-b1ae-4cd5-8b67-15919ba43b3b",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "c8a50d0f-0e2a-4f1c-b258-1768f75937b5",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 14,
-   "id": "6b619c70-f9e8-46d2-9326-371559c95c64",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# modify_request = json.dumps(['Please modify your request.'])\n",
-    "# modify_request"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 15,
-   "id": "766fda6a-71c8-48a8-941b-0493e7f41584",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# json.loads(modify_request)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 16,
-   "id": "f7f67984-540b-4aa3-ba4d-f95d4570fce8",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "27"
-      ]
-     },
-     "execution_count": 16,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(list(config.game_examples.keys()))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 17,
-   "id": "4edf89fb-0b93-4d1a-b9c2-341417128a45",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['Add a bird.',\n",
-       " 'Delete a bird.',\n",
-       " 'Add more birds.',\n",
-       " 'Add a big bird.',\n",
-       " 'Add a magnet bird.',\n",
-       " 'Add a superheavy bird.',\n",
-       " 'Add a catapult bird.',\n",
-       " 'Add a flash bird.',\n",
-       " 'Add an antigravity bird.',\n",
-       " 'The game is too difficult. Please add more birds.',\n",
-       " 'Make the bird bigger.',\n",
-       " 'Make the bird bigger, so it can run over the enemies.',\n",
-       " 'List birds.',\n",
-       " 'Show bird list.',\n",
-       " 'What kind of birds do you have?',\n",
-       " 'Are there new kinds of birds?',\n",
-       " 'Add a pig.',\n",
-       " 'Our pigs are losing. Add more pigs.',\n",
-       " 'The pigs are too week. Increase the health points for them.',\n",
-       " 'The pigs are too week. Increase the HP for them.',\n",
-       " 'The pigs are too week. Increase the hit points for them.',\n",
-       " 'The pigs are too week. Increase the blood level for them.',\n",
-       " 'The pigs are too week. Increase the blood levels for them.',\n",
-       " 'I want to protect the pigs by build houses for them.',\n",
-       " 'I want to build houses.',\n",
-       " 'I want to build houses, so I can protect my pigs.',\n",
-       " \"The obstacles are too weak. Let's build some solid houses.\"]"
-      ]
-     },
-     "execution_count": 17,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "list(config.game_examples.keys())"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 18,
-   "id": "4ae67ee9-2907-4811-8dae-aeb2140c827d",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'Add a bird.'"
-      ]
-     },
-     "execution_count": 18,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "command = list(config.game_examples.keys())[0]\n",
-    "command"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 19,
-   "id": "b625856c-b9ea-4d57-83dd-99e57b0d3404",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# test"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 20,
-   "id": "be16a49a-7b4f-4842-b079-19728e50eecc",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['The game developer is talking about birds.',\n",
-       " 'The game developer wants to add a bird.']"
-      ]
-     },
-     "execution_count": 20,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "prompts = config.game_examples[command]['prompt']\n",
-    "prompts"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 21,
-   "id": "28e7d1bc-df58-4399-b2c3-d7b96aa1a780",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "bird = angrybird()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 22,
-   "id": "2d4abff5-2b19-4ea5-98ca-453a9d2139d0",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "1.0000 15/15\ty\n",
-      "[('y', 15)]\n"
-     ]
-    },
-    {
-     "data": {
-      "text/plain": [
-       "'y'"
-      ]
-     },
-     "execution_count": 22,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "yes_no_query(bird, command, prompts[0])"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "449c163a-4348-4000-a596-6c85d71f5d12",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 23,
-   "id": "8ca28c48-512b-41db-a470-a9337690ed30",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "debug = 2"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 25,
-   "id": "f3f8f3bd-d928-4de1-a8af-c0f6b3fcd111",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "--------------------\n",
-      "prompt:\n",
-      "\n",
-      "Imaging we are developing the game of the Angry Birds, \n",
-      "and the game is described as the following: \n",
-      "\n",
-      "In Angry Birds, players take control of a group of birds who are seeking \n",
-      "revenge on a group of green pigs that have stolen their eggs. \n",
-      "The gameplay involves using a slingshot to launch the birds at various \n",
-      "structures where the pigs are hiding. The objective is to \n",
-      "eliminate all the pigs on each level and score as many points as possible.\n",
-      "\n",
-      "A game developer is building the game, \n",
-      "and I need you to help me figure out what is going on.\n",
-      "\n",
-      "\n",
-      "The game developer says:  \"Add a bird.\".\n",
-      "\n",
-      "The game developer is talking about ___\n",
-      "(A) birds (B) houses (C) pigs (D) building houses for pigs (E) none of the above\n",
-      "\n",
-      "Do not explain.\n",
-      "\n",
-      "1.0000 15/15\t(A) birds\n",
-      "[('(A) birds', 15)]\n"
-     ]
-    }
-   ],
-   "source": [
-    "modify_request = 'Please modify your request.'\n",
-    "p = prompt()\n",
-    "\n",
-    "print('-'*20)\n",
-    "obj_lst = list(config.game_spec.keys())\n",
-    "obj_list = [f\"{i} {o}\" for i, o in zip(util.order, obj_lst)]\n",
-    "s = p.prompt_which_object(\n",
-    "    config.game_desc, \n",
-    "    command,\n",
-    "    \" \".join(obj_list))\n",
-    "full_prompt = p.get_full_prompt(s)\n",
-    "if debug >= 2:\n",
-    "    print(f'prompt:{s}')\n",
-    "\n",
-    "res = bird.query_chatgpt(full_prompt)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "615f67e3-9982-4fde-a2de-aec47f86f798",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 103,
-   "id": "593083b1-9d98-45e4-b3ff-55d00cd805f0",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "\n",
-      "====================\n",
-      "--------------------\n",
-      "add a big bird\n",
-      "The game developer is talking about birds.\n",
-      "0.8667 13/15\ty\n",
-      "[('y', 13), ('n', 2)]\n",
-      "--------------------\n",
-      "add a big bird\n",
-      "The game developer wants to add a bird.\n",
-      "1.0000 15/15\ty\n",
-      "[('y', 15)]\n",
-      "--------------------\n",
-      "add a big bird\n",
-      "The type of the bird the game developer wants to add is a big bird.\n",
-      "0.5333 8/15\ty\n",
-      "[('y', 8), ('n', 7)]\n",
-      "\n",
-      "====================\n",
-      "--------------------\n",
-      "Make the bird bigger, so it can run over the enemies.\n",
-      "The game developer is talking about birds.\n",
-      "0.7333 11/15\ty\n",
-      "[('y', 11), ('n', 4)]\n",
-      "--------------------\n",
-      "Make the bird bigger, so it can run over the enemies.\n",
-      "The game developer wants to change the size of a bird.\n",
-      "1.0000 15/15\ty\n",
-      "[('y', 15)]\n",
-      "--------------------\n",
-      "Make the bird bigger, so it can run over the enemies.\n",
-      "The game developer did not specify how much bigger\n",
-      "0.8000 12/15\ty\n",
-      "[('y', 12), ('n', 3)]\n",
-      "\n",
-      "====================\n",
-      "--------------------\n",
-      "Our pigs are losing. Add more pigs\n",
-      "The game developer is talking about pigs.\n",
-      "1.0000 15/15\ty\n",
-      "[('y', 15)]\n",
-      "--------------------\n",
-      "Our pigs are losing. Add more pigs\n",
-      "The game developer wants to add some pig.\n",
-      "1.0000 15/15\ty\n",
-      "[('y', 15)]\n"
-     ]
-    }
-   ],
-   "source": [
-    "for command in config.game_examples:\n",
-    "    if 'prompt' not in config.game_examples[command]:\n",
-    "        continue\n",
-    "\n",
-    "    if 'train' in config.game_examples[command]:\n",
-    "        continue\n",
-    "    prompts = config.game_examples[command]['prompt']\n",
-    "    print()\n",
-    "    print('='*20)\n",
-    "    for prompt in prompts:\n",
-    "        print('-'*20)\n",
-    "        print(command)\n",
-    "        print(prompt)\n",
-    "        assert yes_no_query(command, prompt) == 'y'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 86,
-   "id": "fbc675f6-ff70-442f-8b5d-8db4af917954",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "====================\n",
-      "\n",
-      "--------------------\n",
-      "add a big bird\n",
-      "The game developer is talking about birds.\n",
-      "\n",
-      "0.9333 14/15\ty\n",
-      "[('y', 14), ('n', 1)]\n",
-      "y\n",
-      "\n",
-      "--------------------\n",
-      "add a big bird\n",
-      "The game developer wants to add a bird.\n",
-      "\n",
-      "1.0000 15/15\ty\n",
-      "[('y', 15)]\n",
-      "y\n",
-      "\n",
-      "--------------------\n",
-      "add a big bird\n",
-      "The type of the bird the game developer wants to add is a big bird.\n",
-      "\n",
-      "0.8000 12/15\tn\n",
-      "[('n', 12), ('y', 3)]\n",
-      "n\n",
-      "\n",
-      "====================\n",
-      "\n",
-      "--------------------\n",
-      "Make the bird bigger, so it can run over the enemies.\n",
-      "The game developer is talking about birds.\n",
-      "\n",
-      "0.6000 9/15\ty\n",
-      "[('y', 9), ('n', 6)]\n",
-      "y\n",
-      "\n",
-      "--------------------\n",
-      "Make the bird bigger, so it can run over the enemies.\n",
-      "The game developer wants to change the size of a bird.\n",
-      "\n",
-      "1.0000 15/15\ty\n",
-      "[('y', 15)]\n",
-      "y\n",
-      "\n",
-      "--------------------\n",
-      "Make the bird bigger, so it can run over the enemies.\n",
-      "The game developer did not specify how much bigger\n",
-      "\n",
-      "0.9333 14/15\ty\n",
-      "[('y', 14), ('n', 1)]\n",
-      "y\n",
-      "\n"
-     ]
-    }
-   ],
-   "source": [
-    "for command in config.game_examples:\n",
-    "    if 'prompt' not in config.game_examples[command]:\n",
-    "        continue\n",
-    "\n",
-    "    if 'focus' not in config.game_examples[command]:\n",
-    "        continue\n",
-    "    prompts = config.game_examples[command]['prompt']\n",
-    "    print('='*20)\n",
-    "    print()\n",
-    "    for prompt in prompts:\n",
-    "        print('-'*20)\n",
-    "        print(command)\n",
-    "        print(prompt)\n",
-    "        print()\n",
-    "        print(yes_no_query(command, prompt))\n",
-    "        print()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 84,
-   "id": "89c4c4b0-c2c1-4566-a5ff-3fd4436bb027",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# command = \"add a flash bird.\"\n",
-    "input_command = \"add a red pig\""
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 22,
-   "id": "18888a7b-d32d-45b4-a9b3-3ce29b18a7f7",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# sim query"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 23,
-   "id": "5dd25e64-cf95-4831-95f1-b4a853b3ef1a",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "def get_emb(command):\n",
-    "    return get_embedding(\n",
-    "        command,\n",
-    "        engine=\"text-embedding-ada-002\"\n",
-    "    )"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 24,
-   "id": "c44cb708-d129-43f2-bb63-9287ee6fe899",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "input_embedding = get_emb(input_command)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 26,
-   "id": "d8847340-1d63-47df-bca2-b7a18fe5243b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[-0.009216495789587498,\n",
-       " -0.020423918962478638,\n",
-       " -0.007699600420892239,\n",
-       " 0.0027933907695114613,\n",
-       " -0.009676982648670673]"
-      ]
-     },
-     "execution_count": 26,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "input_embedding[:5]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 27,
-   "id": "374629b3-54c9-4a6f-9621-d0daa9d200b6",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "embeddings = {}\n",
-    "for command in config.examples.keys():\n",
-    "    embeddings[command] = get_emb(command)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "3dede08f-376f-4bd9-8b9c-5561f4c03bce",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 28,
-   "id": "c845321b-51c9-42a9-b645-277a3c630bb2",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "df = pd.DataFrame(columns=['sim', 'command'])"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 29,
-   "id": "98be3413-d5f1-4d32-ad50-7bb47384c270",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/html": [
-       "<div>\n",
-       "<style scoped>\n",
-       "    .dataframe tbody tr th:only-of-type {\n",
-       "        vertical-align: middle;\n",
-       "    }\n",
-       "\n",
-       "    .dataframe tbody tr th {\n",
-       "        vertical-align: top;\n",
-       "    }\n",
-       "\n",
-       "    .dataframe thead th {\n",
-       "        text-align: right;\n",
-       "    }\n",
-       "</style>\n",
-       "<table border=\"1\" class=\"dataframe\">\n",
-       "  <thead>\n",
-       "    <tr style=\"text-align: right;\">\n",
-       "      <th></th>\n",
-       "      <th>sim</th>\n",
-       "      <th>command</th>\n",
-       "    </tr>\n",
-       "  </thead>\n",
-       "  <tbody>\n",
-       "  </tbody>\n",
-       "</table>\n",
-       "</div>"
-      ],
-      "text/plain": [
-       "Empty DataFrame\n",
-       "Columns: [sim, command]\n",
-       "Index: []"
-      ]
-     },
-     "execution_count": 29,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "df"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "c0457c5f-149a-41f3-a153-c5fbc19573ef",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 30,
-   "id": "8b5c4536-f046-4872-b583-5f5be6b4bf41",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "for command in embeddings:\n",
-    "    sim = cosine_similarity(input_embedding, embeddings[command])\n",
-    "    # print(f\":{sim:.4f}\\t{command}\")\n",
-    "    df2 = pd.DataFrame({'sim': [sim], 'command': [command]})\n",
-    "    df = pd.concat([df, df2])"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 31,
-   "id": "c4413de1-844b-4bed-be27-177c11229117",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/html": [
-       "<div>\n",
-       "<style scoped>\n",
-       "    .dataframe tbody tr th:only-of-type {\n",
-       "        vertical-align: middle;\n",
-       "    }\n",
-       "\n",
-       "    .dataframe tbody tr th {\n",
-       "        vertical-align: top;\n",
-       "    }\n",
-       "\n",
-       "    .dataframe thead th {\n",
-       "        text-align: right;\n",
-       "    }\n",
-       "</style>\n",
-       "<table border=\"1\" class=\"dataframe\">\n",
-       "  <thead>\n",
-       "    <tr style=\"text-align: right;\">\n",
-       "      <th></th>\n",
-       "      <th>sim</th>\n",
-       "      <th>command</th>\n",
-       "    </tr>\n",
-       "  </thead>\n",
-       "  <tbody>\n",
-       "    <tr>\n",
-       "      <th>0</th>\n",
-       "      <td>0.853803</td>\n",
-       "      <td>add a bird.</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>0</th>\n",
-       "      <td>0.839128</td>\n",
-       "      <td>add more birds</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>0</th>\n",
-       "      <td>0.852401</td>\n",
-       "      <td>add a big bird</td>\n",
-       "    </tr>\n",
-       "  </tbody>\n",
-       "</table>\n",
-       "</div>"
-      ],
-      "text/plain": [
-       "        sim         command\n",
-       "0  0.853803     add a bird.\n",
-       "0  0.839128  add more birds\n",
-       "0  0.852401  add a big bird"
-      ]
-     },
-     "execution_count": 31,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "df[0:3]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 33,
-   "id": "436b68a4-70ff-495b-bcb5-e837b331606b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'add a red pig'"
-      ]
-     },
-     "execution_count": 33,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "input_command"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 34,
-   "id": "579a9788-fbea-4c2d-b5dd-9c6eaf20ee20",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/html": [
-       "<div>\n",
-       "<style scoped>\n",
-       "    .dataframe tbody tr th:only-of-type {\n",
-       "        vertical-align: middle;\n",
-       "    }\n",
-       "\n",
-       "    .dataframe tbody tr th {\n",
-       "        vertical-align: top;\n",
-       "    }\n",
-       "\n",
-       "    .dataframe thead th {\n",
-       "        text-align: right;\n",
-       "    }\n",
-       "</style>\n",
-       "<table border=\"1\" class=\"dataframe\">\n",
-       "  <thead>\n",
-       "    <tr style=\"text-align: right;\">\n",
-       "      <th></th>\n",
-       "      <th>sim</th>\n",
-       "      <th>command</th>\n",
-       "    </tr>\n",
-       "  </thead>\n",
-       "  <tbody>\n",
-       "    <tr>\n",
-       "      <th>0</th>\n",
-       "      <td>0.955520</td>\n",
-       "      <td>add a pig</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>0</th>\n",
-       "      <td>0.864960</td>\n",
-       "      <td>Our pigs are loosing. Add more pigs</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>0</th>\n",
-       "      <td>0.853803</td>\n",
-       "      <td>add a bird.</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>0</th>\n",
-       "      <td>0.852401</td>\n",
-       "      <td>add a big bird</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>0</th>\n",
-       "      <td>0.840981</td>\n",
-       "      <td>The pigs are too week. Increase the blood leve...</td>\n",
-       "    </tr>\n",
-       "  </tbody>\n",
-       "</table>\n",
-       "</div>"
-      ],
-      "text/plain": [
-       "        sim                                            command\n",
-       "0  0.955520                                          add a pig\n",
-       "0  0.864960                Our pigs are loosing. Add more pigs\n",
-       "0  0.853803                                        add a bird.\n",
-       "0  0.852401                                     add a big bird\n",
-       "0  0.840981  The pigs are too week. Increase the blood leve..."
-      ]
-     },
-     "execution_count": 34,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "df.sort_values('sim', ascending=False)[:5]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "41df04aa-4aaf-48ef-8fb8-fc4c0c743b9f",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "a6910b5c-ae29-4988-b4eb-218efb594689",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "assert False"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "93ed444b-e938-4923-be55-525481c23615",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "res = query(command)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "a0aa2da0-9560-4496-ac5f-610b4946e082",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "res"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "fa3287e7-a4fe-44ee-aa55-50ea88968d65",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "type(res)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "2ef74c20-8859-4280-acd2-4c181a83bbef",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "b1ad0047-e5b9-4aea-abfb-3881458531e1",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "x"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "59405568-618d-44c8-8c23-ec3a6fd1b347",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "x['data']"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "876bd705-c36c-44e8-a0ad-c3a1cb68237a",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "json.loads(x['data'][0])"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "9148abdc-8ddb-49de-a709-69ed838a53b6",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "b86d999b-30b6-48b1-8eca-589651392caf",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "x= json.loads(res)\n",
-    "x"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "0c2828ac-af6c-4592-8b96-9f318f16fa56",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "['please modify your request']"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "751e8f6d-5e5c-4d6f-90b1-08271c287fdf",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "type(x)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "c274efb2-d6fd-4dd0-beff-1ea3e0e74207",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "x[0]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "56446edc-1708-4387-ade8-4da5b217553b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "len(x)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "e3263229-c828-4ec4-9790-d90e08d4ca62",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "04e2ffc0-3451-44aa-abe0-879c30f4eded",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "res"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "dc16a485-71e0-4c09-8ee1-4b5ee5079d5f",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "json.loads(res)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "66b26cd3-f7ae-4652-8982-576069b23224",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "print(f\"{command}\\n\\t{json.loads(res)}\\n\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "3daf7e67-a149-4c9e-bf7d-b8e90764af4b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "res"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "55511792-72ed-48ee-8585-35699f118017",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import time"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "739321a6-1ba6-4124-b7cc-5d78a94e47a6",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "time.sleep(3)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "5a2d1414-cc65-42f8-ad74-196443dd918b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "datetime.today()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "1b827bad-2bdf-4f3a-85b8-3ab2243207fa",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "for i, command in enumerate(config.game_commands):\n",
-    "    res = query(command)\n",
-    "    print(f\"\\n{i+1}/{len(config.game_commands)}\\t{command}\\n\\t{json.loads(res)}\\n\")\n",
-    "    time.sleep(10)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "82ff07d2-33e3-4e01-8023-9941d5fd2714",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "a6b0aa67-5272-46f8-b065-8b9c97183bcd",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "for i, command in enumerate(config.game_commands):\n",
-    "    res = query(command)\n",
-    "    print(f\"\\n{i+1}/{len(config.game_commands)}\\t{command}\\n\\t{json.loads(res)}\\n\")\n",
-    "    time.sleep(3)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "ef911a45-51c3-4403-a91a-1d73ca536fb6",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "758ac7f0-50c6-462c-95f9-f40788878e44",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.4"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}

games/angrybirds/notebooks/to_anhang.ipynb

@@ -1,529 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "id": "245fbc34-352d-4945-a689-d04f9365bb99",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "%load_ext autoreload\n",
-    "%autoreload 2"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "id": "eaed216c-160a-4e31-96fe-015c140ff18b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# from pprint import pprint"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "id": "97953551-8ef1-42c1-9317-99d5e65f535e",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "723\n",
-      "770\n"
-     ]
-    }
-   ],
-   "source": [
-    "import json\n",
-    "import pandas as pd\n",
-    "\n",
-    "import openai\n",
-    "# from openai.embeddings_utils import get_embedding, cosine_similarity\n",
-    "\n",
-    "# from games.angrybirds.util import util\n",
-    "from games.angrybirds.config import config\n",
-    "# from games.angrybirds.prompt import prompt\n",
-    "# from games.angrybirds.query import query, yes_no_query\n",
-    "from games.angrybirds.angrybirds import angrybird"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "ae27ebd0-2d95-4ff0-b188-f6e46b189cd1",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 19,
-   "id": "766fda6a-71c8-48a8-941b-0493e7f41584",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# json.loads(modify_request)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 22,
-   "id": "f7f67984-540b-4aa3-ba4d-f95d4570fce8",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "50"
-      ]
-     },
-     "execution_count": 22,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(list(config.game_examples.keys()))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 23,
-   "id": "4edf89fb-0b93-4d1a-b9c2-341417128a45",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['Add a bird.',\n",
-       " 'Delete a bird.',\n",
-       " 'Add more birds.',\n",
-       " 'Add a big bird.',\n",
-       " 'Add a magnet bird.',\n",
-       " 'Add a superheavy bird.',\n",
-       " 'Add a catapult bird.',\n",
-       " 'Add a flash bird.',\n",
-       " 'Add an antigravity bird.',\n",
-       " 'The game is too difficult. Please add more birds.',\n",
-       " 'Make the bird bigger.',\n",
-       " 'Make the bird bigger, so it can run over the enemies.',\n",
-       " 'List birds.',\n",
-       " 'Show bird list.',\n",
-       " 'What kind of birds do you have?',\n",
-       " 'Are there new kinds of birds?',\n",
-       " 'I want to change the positon of birds',\n",
-       " 'I want to change the positon of pigs',\n",
-       " 'I want to change the positon of houses',\n",
-       " 'I want to change the positon of obstacles',\n",
-       " 'Add a pig.',\n",
-       " 'Please add two pigs.',\n",
-       " 'Please add 2 pigs.',\n",
-       " 'Please add three pigs.',\n",
-       " 'Please add 3 pigs.',\n",
-       " 'Please add four pigs.',\n",
-       " 'Please add 4 pigs.',\n",
-       " 'Please add five pigs.',\n",
-       " 'Please add 5 pigs.',\n",
-       " 'Please add six pigs.',\n",
-       " 'Please add 6 pigs.',\n",
-       " 'Please add seven pigs.',\n",
-       " 'Please add 7 pigs.',\n",
-       " 'Please add eight pigs.',\n",
-       " 'Please add 8 pigs.',\n",
-       " 'Please add nine pigs.',\n",
-       " 'Please add 9 pigs.',\n",
-       " 'Please add ten pigs.',\n",
-       " 'Please add 10 pigs.',\n",
-       " 'I want to make pigs everywehre.',\n",
-       " 'Our pigs are losing. Add more pigs.',\n",
-       " 'The pigs are too week. Increase the health points for them.',\n",
-       " 'The pigs are too week. Increase the HP for them.',\n",
-       " 'The pigs are too week. Increase the hit points for them.',\n",
-       " 'The pigs are too week. Increase the blood level for them.',\n",
-       " 'The pigs are too week. Increase the blood levels for them.',\n",
-       " 'I want to protect the pigs by build houses for them.',\n",
-       " 'I want to build houses.',\n",
-       " 'I want to build houses, so I can protect my pigs.',\n",
-       " \"The obstacles are too weak. Let's build some solid houses.\"]"
-      ]
-     },
-     "execution_count": 23,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "list(config.game_examples.keys())"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 24,
-   "id": "4ae67ee9-2907-4811-8dae-aeb2140c827d",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'Add a bird.'"
-      ]
-     },
-     "execution_count": 24,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "command = list(config.game_examples.keys())[0]\n",
-    "command"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 25,
-   "id": "b625856c-b9ea-4d57-83dd-99e57b0d3404",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# test"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 32,
-   "id": "be16a49a-7b4f-4842-b079-19728e50eecc",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['The game developer is talking about birds.',\n",
-       " 'The game developer wants to add a bird.']"
-      ]
-     },
-     "execution_count": 32,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "prompts = config.game_examples[command]['prompt']\n",
-    "prompts"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 34,
-   "id": "522394aa-7805-4ab3-a95e-b7826d0759e7",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['bird', 'add', {}]"
-      ]
-     },
-     "execution_count": 34,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "returns = config.game_examples[command]['result']\n",
-    "returns"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "e198dbe4-3b28-4b7b-a9b1-cb1245692fa9",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 26,
-   "id": "1f168079-4a85-4a09-bfa2-11cea105a21e",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "bird = angrybird()\n",
-    "# bird.init_emb()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 12,
-   "id": "2d4abff5-2b19-4ea5-98ca-453a9d2139d0",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "\n",
-      "====================\n",
-      "--------------------\n",
-      "Make the bird bigger, so it can run over the enemies.\n",
-      "The game developer is talking about birds.\n",
-      "0.6667 10/15\tn\n",
-      "[('n', 10), ('y', 5)]\n"
-     ]
-    },
-    {
-     "ename": "AssertionError",
-     "evalue": "",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
-      "\u001b[0;31mAssertionError\u001b[0m                            Traceback (most recent call last)",
-      "Cell \u001b[0;32mIn[12], line 14\u001b[0m\n\u001b[1;32m     12\u001b[0m \u001b[38;5;28mprint\u001b[39m(command)\n\u001b[1;32m     13\u001b[0m \u001b[38;5;28mprint\u001b[39m(prompt)\n\u001b[0;32m---> 14\u001b[0m \u001b[38;5;28;01massert\u001b[39;00m yes_no_query(bird, command, prompt) \u001b[38;5;241m==\u001b[39m \u001b[38;5;124m'\u001b[39m\u001b[38;5;124my\u001b[39m\u001b[38;5;124m'\u001b[39m\n",
-      "\u001b[0;31mAssertionError\u001b[0m: "
-     ]
-    }
-   ],
-   "source": [
-    "for command in config.game_examples:\n",
-    "    if 'prompt' not in config.game_examples[command]:\n",
-    "        continue\n",
-    "\n",
-    "    if 'train' in config.game_examples[command]:\n",
-    "        continue\n",
-    "    prompts = config.game_examples[command]['prompt']\n",
-    "    print()\n",
-    "    print('='*20)\n",
-    "    for prompt in prompts:\n",
-    "        print('-'*20)\n",
-    "        print(command)\n",
-    "        print(prompt)\n",
-    "        assert yes_no_query(bird, command, prompt) == 'y'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "6bc6d2fb-ff82-44c3-8a78-33f577f60d64",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 30,
-   "id": "fba16d64-8981-4bc1-bc3e-facc2e263f96",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "s = {}\n",
-    "for command in config.game_examples:\n",
-    "    if 'train' not in config.game_examples[command]:\n",
-    "        continue\n",
-    "    s[command] = {}\n",
-    "    s[command]['result'] = config.game_examples[command]['result']\n",
-    "    # s[command]['desc'] = config.game_examples[command]['desc']\n",
-    "res = {\n",
-    "    'version': '1.0-2023-07-26-a',\n",
-    "    'data': s,\n",
-    "}"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "572dfa7d-bdc7-4c5c-875e-7d2779d60d72",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 31,
-   "id": "c9c376d6-75e8-4ba0-9aa3-f0b673d1646c",
-   "metadata": {},
-   "outputs": [
-    {
-     "ename": "SyntaxError",
-     "evalue": "'return' outside function (3844427982.py, line 1)",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[0;36m  Cell \u001b[0;32mIn[31], line 1\u001b[0;36m\u001b[0m\n\u001b[0;31m    return json.dumps(res, indent=4), \"\"\u001b[0m\n\u001b[0m    ^\u001b[0m\n\u001b[0;31mSyntaxError\u001b[0m\u001b[0;31m:\u001b[0m 'return' outside function\n"
-     ]
-    }
-   ],
-   "source": [
-    "return json.dumps(res, indent=4), \"\""
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 49,
-   "id": "ca386aa5-762f-4c6c-b601-2388d87dfb97",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "s = {}\n",
-    "for command in config.game_examples:\n",
-    "    if 'train' not in config.game_examples[command]:\n",
-    "        continue\n",
-    "    s[command] = {}\n",
-    "    s[command]['result'] = config.game_examples[command]['result']"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 32,
-   "id": "60a451d8-0fc5-4d8b-935f-10016a9b1560",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "ss = json.dumps(res, indent=4)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "4d5aea15-e2d2-47d5-b579-8166ed426040",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 33,
-   "id": "52161707-0a42-4deb-849a-c07fcd9644f3",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'version': '1.0-2023-07-26-a',\n",
-       " 'data': {'Add a bird.': {'result': ['bird', 'add', {}]},\n",
-       "  'Delete a bird.': {'result': ['bird', 'add', {'quantiy': -1}]},\n",
-       "  'Add more birds.': {'result': ['bird', 'add', {'quantiy': 'some'}]},\n",
-       "  'Add a big bird.': {'result': ['bird', 'add', {'type': 'big'}]},\n",
-       "  'Add a magnet bird.': {'result': ['bird', 'add', {'type': 'magnet'}]},\n",
-       "  'Add a superheavy bird.': {'result': ['bird',\n",
-       "    'add',\n",
-       "    {'type': 'superheavy'}]},\n",
-       "  'Add a catapult bird.': {'result': ['bird', 'add', {'type': 'catapult'}]},\n",
-       "  'Add a flash bird.': {'result': ['bird', 'add', {'type': 'flash'}]},\n",
-       "  'Add an antigravity bird.': {'result': ['bird',\n",
-       "    'add',\n",
-       "    {'type': 'antigravity'}]},\n",
-       "  'The game is too difficult. Please add more birds.': {'result': ['bird',\n",
-       "    'add',\n",
-       "    {'quantiy': 'some'}]},\n",
-       "  'Make the bird bigger.': {'result': ['bird', 'size', {}]},\n",
-       "  'Make the bird bigger, so it can run over the enemies.': {'result': ['bird',\n",
-       "    'size',\n",
-       "    {}]},\n",
-       "  'List birds.': {'result': ['bird', 'list', {}]},\n",
-       "  'Show bird list.': {'result': ['bird', 'list', {}]},\n",
-       "  'What kind of birds do you have?': {'result': ['bird', 'list', {}]},\n",
-       "  'Are there new kinds of birds?': {'result': ['bird',\n",
-       "    'list',\n",
-       "    {'type': 'new'}]},\n",
-       "  'I want to change the positon of birds': {'result': ['change position',\n",
-       "    {'args': 'bird'}]},\n",
-       "  'I want to change the positon of pigs': {'result': ['change position',\n",
-       "    {'args': 'pig'}]},\n",
-       "  'I want to change the positon of houses': {'result': ['change position',\n",
-       "    {'args': 'house'}]},\n",
-       "  'I want to change the positon of obstacles': {'result': ['change position',\n",
-       "    {'args': 'obstacle'}]},\n",
-       "  'Add a pig.': {'result': ['pig', 'add', {}]},\n",
-       "  'Please add two pigs.': {'result': ['pig', 'add', {'quantiy': 2}]},\n",
-       "  'Please add 2 pigs.': {'result': ['pig', 'add', {'quantiy': 2}]},\n",
-       "  'Please add three pigs.': {'result': ['pig', 'add', {'quantiy': 3}]},\n",
-       "  'Please add 3 pigs.': {'result': ['pig', 'add', {'quantiy': 3}]},\n",
-       "  'Please add four pigs.': {'result': ['pig', 'add', {'quantiy': 4}]},\n",
-       "  'Please add 4 pigs.': {'result': ['pig', 'add', {'quantiy': 4}]},\n",
-       "  'Please add five pigs.': {'result': ['pig', 'add', {'quantiy': 5}]},\n",
-       "  'Please add 5 pigs.': {'result': ['pig', 'add', {'quantiy': 5}]},\n",
-       "  'Please add six pigs.': {'result': ['pig', 'add', {'quantiy': 6}]},\n",
-       "  'Please add 6 pigs.': {'result': ['pig', 'add', {'quantiy': 6}]},\n",
-       "  'Please add seven pigs.': {'result': ['pig', 'add', {'quantiy': 7}]},\n",
-       "  'Please add 7 pigs.': {'result': ['pig', 'add', {'quantiy': 7}]},\n",
-       "  'Please add eight pigs.': {'result': ['pig', 'add', {'quantiy': 8}]},\n",
-       "  'Please add 8 pigs.': {'result': ['pig', 'add', {'quantiy': 8}]},\n",
-       "  'Please add nine pigs.': {'result': ['pig', 'add', {'quantiy': 9}]},\n",
-       "  'Please add 9 pigs.': {'result': ['pig', 'add', {'quantiy': 9}]},\n",
-       "  'Please add ten pigs.': {'result': ['pig', 'add', {'quantiy': 10}]},\n",
-       "  'Please add 10 pigs.': {'result': ['pig', 'add', {'quantiy': 10}]},\n",
-       "  'I want to make pigs everywehre.': {'result': ['pig',\n",
-       "    'add',\n",
-       "    {'quantiy': 'everywhere'}]},\n",
-       "  'Our pigs are losing. Add more pigs.': {'result': ['pig',\n",
-       "    'add',\n",
-       "    {'quantiy': 'some'}]},\n",
-       "  'The pigs are too week. Increase the health points for them.': {'result': ['pig',\n",
-       "    'increase health points',\n",
-       "    {}]},\n",
-       "  'The pigs are too week. Increase the HP for them.': {'result': ['pig',\n",
-       "    'increase health points',\n",
-       "    {}]},\n",
-       "  'The pigs are too week. Increase the hit points for them.': {'result': ['pig',\n",
-       "    'increase health points',\n",
-       "    {}]},\n",
-       "  'The pigs are too week. Increase the blood level for them.': {'result': ['pig',\n",
-       "    'increase health points',\n",
-       "    {}]},\n",
-       "  'The pigs are too week. Increase the blood levels for them.': {'result': ['pig',\n",
-       "    'increase health points',\n",
-       "    {}]},\n",
-       "  'I want to protect the pigs by build houses for them.': {'result': ['house',\n",
-       "    'add',\n",
-       "    {'quantiy': 'some'}]},\n",
-       "  'I want to build houses.': {'result': ['house', 'add', {}]},\n",
-       "  'I want to build houses, so I can protect my pigs.': {'result': ['house',\n",
-       "    'add',\n",
-       "    {}]},\n",
-       "  \"The obstacles are too weak. Let's build some solid houses.\": {'result': ['house',\n",
-       "    'add',\n",
-       "    {'args': 'solid'}]}}}"
-      ]
-     },
-     "execution_count": 33,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "json.loads(ss)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "4618fed0-320b-4be2-bce6-73b6b051d9c5",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.4"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}

games/angrybirds/notebooks/tok.ipynb

@@ -1,106 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 9,
-   "id": "eceea364-b90c-4d42-bdf6-db3d940c5063",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from gensim.utils import tokenize"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 10,
-   "id": "0bf38b21-bb63-4a76-ae60-7cf073c1e955",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "['There', 'are', 'multiple', 'ways', 'we', 'can', 'perform', 'tokenization', 'on', 'given', 'text', 'data', 'We', 'can', 'choose', 'any', 'method', 'based', 'on', 'langauge', 'library', 'and', 'purpose', 'of', 'modeling']\n"
-     ]
-    }
-   ],
-   "source": [
-    "text = \"\"\"There are multiple ways we can perform tokenization on given text data. We can choose any method based on langauge, library and purpose of modeling.\"\"\"\n",
-    "\n",
-    "tokens = list(tokenize(text))\n",
-    "print(tokens)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 11,
-   "id": "bf2d2a7a-3d81-4118-b83c-f22cc5355caf",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['this', 'is', 'a', 'book']"
-      ]
-     },
-     "execution_count": 11,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "list(tokenize('this is a book.'))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 14,
-   "id": "5ca5aac7-18f1-4ada-8499-a6c9a301af50",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "['this', 'is', 'USA']"
-      ]
-     },
-     "execution_count": 14,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "list(tokenize('this is USA!'))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "939ed5e8-7547-41aa-ab32-52fa7abaea5c",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.4"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}

games/snake/notebooks/emb.ipynb

@@ -1,137 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "id": "1cc076bb-3248-466d-8eee-df79ded42c47",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import openai\n",
-    "from openai.embeddings_utils import get_embedding as openai_get_embedding\n",
-    "from openai.embeddings_utils import cosine_similarity as openai_cosine_similarity"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "id": "085a6255-d828-42b6-8da9-cc9a8cc02681",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "api_key = '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'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "id": "8e8fb3b6-a8ec-405f-a801-ba0435e4a767",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "openai.api_key = api_key"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "id": "08c2a609-91a2-4d62-a187-d29f3148b68e",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "def _compute_emb(command: str):\n",
-    "    # always normalize text when query OpenAI embedding service\n",
-    "    command = command.strip().lower()\n",
-    "    return openai_get_embedding(\n",
-    "        command,\n",
-    "        engine=\"text-embedding-ada-002\"\n",
-    "    )\n"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 7,
-   "id": "7a7c2dba-c54e-44a8-9466-adba94053b47",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "t = _compute_emb('hi')"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 8,
-   "id": "f1ea3765-3c99-477c-82d3-d539e6b79f95",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[-0.03515825420618057,\n",
-       " -0.02061408944427967,\n",
-       " -0.015398159623146057,\n",
-       " -0.03980923444032669,\n",
-       " -0.027459176257252693]"
-      ]
-     },
-     "execution_count": 8,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "t[:5]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 9,
-   "id": "3ab28dda-dff3-4921-9e24-f4ca05527ebe",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "1536"
-      ]
-     },
-     "execution_count": 9,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(t)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "564a9a79-5906-4b29-9063-6b4ee8663d2d",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.4"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}

games/snake/notebooks/query.ipynb

@@ -1,755 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 4,
-   "id": "245fbc34-352d-4945-a689-d04f9365bb99",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "The autoreload extension is already loaded. To reload it, use:\n",
-      "  %reload_ext autoreload\n"
-     ]
-    }
-   ],
-   "source": [
-    "%load_ext autoreload\n",
-    "%autoreload 2"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "id": "eaed216c-160a-4e31-96fe-015c140ff18b",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# from pprint import pprint"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "id": "97953551-8ef1-42c1-9317-99d5e65f535e",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import json\n",
-    "import pandas as pd\n",
-    "import pickle\n",
-    "\n",
-    "import openai\n",
-    "from openai.embeddings_utils import get_embedding, cosine_similarity\n",
-    "\n",
-    "# from games.angrybirds.util import util\n",
-    "from games.snake.config import config\n",
-    "# from games.angrybirds.prompt import prompt\n",
-    "# from games.angrybirds.query import query, yes_no_query\n",
-    "from games.snake.snake import snake"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "fc213f4c-cb07-468b-b15f-60e5120b3d94",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 7,
-   "id": "0c876614-a9bf-4d0b-a728-07770f1ad9c5",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "before add_config\n",
-      "3104\n",
-      "after add_config\n",
-      "3104\n",
-      "final\n",
-      "3104\n"
-     ]
-    }
-   ],
-   "source": [
-    "game = snake()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 8,
-   "id": "1411de96-97d5-4bac-b8ed-e279fb1edaa0",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "0"
-      ]
-     },
-     "execution_count": 8,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(game.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 9,
-   "id": "129403ce-69f4-4f2a-9b22-7b3e36d78f24",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# game.embeddings"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 10,
-   "id": "50e372eb-48ba-4378-b866-f2b7120bc94e",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "3104"
-      ]
-     },
-     "execution_count": 10,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(game.train)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "f78b2c43-bfc7-478f-894d-d757b0e9927f",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 11,
-   "id": "163545c6-c707-401a-93b8-fb2cb9d65136",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "initializing...\n",
-      "loading embeddings from a file...\n",
-      "pkl size: 3126\n",
-      "embedding size: 3104\n",
-      "creating embeddings by query OpenAI...\n",
-      "embedding size: 3104\n",
-      "train size: 3104\n"
-     ]
-    }
-   ],
-   "source": [
-    "game.init_emb()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 12,
-   "id": "31c02b28-0078-425d-87b6-702caeea02e3",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "3104"
-      ]
-     },
-     "execution_count": 12,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(game.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 13,
-   "id": "cb8e1af2-b0f8-4b74-bc3d-d02ed7a16912",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# list(game.embeddings.keys())"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "067905b6-b030-4df1-9570-d45885f775eb",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 14,
-   "id": "02478e9b-dd11-406a-b405-ea4ee56975b1",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import os"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 15,
-   "id": "7b506b8c-0b60-44e4-bb6e-f4af8439f04f",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# root = os.environ['PYTHONPATH']\n",
-    "# root"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "e977cf56-3bfc-4c87-884e-58577b3808cd",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 16,
-   "id": "5662bd80-03c5-4793-94c3-4d0392c08102",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "before add_config\n",
-      "3104\n",
-      "after add_config\n",
-      "3104\n",
-      "final\n",
-      "3104\n"
-     ]
-    }
-   ],
-   "source": [
-    "conf = config()\n",
-    "conf.init_game_train()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 17,
-   "id": "30fb4dd7-35d8-442f-bd6e-29e15374e3d6",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'../game_train.json', 'w') as f:\n",
-    "    json.dump(config.game_train, f, indent=4)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 18,
-   "id": "3c051dbf-d547-4297-8f14-f464e58e9a64",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'../game_train.json', 'r') as f:\n",
-    "    tmp = json.load(f)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 19,
-   "id": "82c2f34a-f84c-43fd-9a10-b572fd910f31",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# config.game_train = tmp"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 20,
-   "id": "77789684-e7b1-4e28-b9b7-5fbca3b6d376",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "True"
-      ]
-     },
-     "execution_count": 20,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "config.game_train == tmp"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "ce058298-2236-4ede-adca-164d6d1249e1",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 21,
-   "id": "dd8c489e-cd3f-4df6-81c0-fe3fd2cc0d33",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'{game.ROOT}/embeddings.pkl', 'wb') as f:\n",
-    "    pickle.dump(game.embeddings, f)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 22,
-   "id": "07032487-3fe7-4729-8ca0-c57d6a485c28",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "with open(f'{game.ROOT}/embeddings.pkl', 'rb') as f:\n",
-    "    tmp = pickle.load(f)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 23,
-   "id": "ab7ec218-0e02-4a2b-9c16-ec26704b3519",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "True"
-      ]
-     },
-     "execution_count": 23,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "tmp == game.embeddings"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 24,
-   "id": "bc896434-21ae-438c-90b1-6b590103e731",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "3104"
-      ]
-     },
-     "execution_count": 24,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(tmp)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 25,
-   "id": "9532b604-b7d6-4bd4-a0b9-c8c36470447b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "3104"
-      ]
-     },
-     "execution_count": 25,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(game.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 26,
-   "id": "18888a7b-d32d-45b4-a9b3-3ce29b18a7f7",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# sim query"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 27,
-   "id": "4703a269-a1c0-48e4-8848-6b5cdb9b814b",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "3104"
-      ]
-     },
-     "execution_count": 27,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "len(game.embeddings)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "c885da07-e53d-4586-b340-8f9de0515fc2",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 25,
-   "id": "4ff48663-e767-4936-807d-07fb8c42ce92",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "{\n",
-      "    \"version\": \"1.4.0.2023-11-10-a\",\n",
-      "    \"data\": [\n",
-      "        \"(1) add/delete bombs/swords/...\",\n",
-      "        \"(2) add/delete enemies/monsters\",\n",
-      "        \"(3) add/delete obstacles\",\n",
-      "        \"(4) add/delete one or more weapons/obstacle\",\n",
-      "        \"(5) add/delete props\",\n",
-      "        \"(6) add/delete weapons\",\n",
-      "        \"(7) change background\",\n",
-      "        \"(8) change map size\",\n",
-      "        \"(9) change skin\",\n",
-      "        \"(10) change snake size\",\n",
-      "        \"(11) disable patrol\",\n",
-      "        \"(12) enable patrol\",\n",
-      "        \"(13) enemies/obstacles attack snake/player\",\n",
-      "        \"(14) enter edit mode\",\n",
-      "        \"(15) help\",\n",
-      "        \"(16) increase/decrease props/weapons/food/obstacles/monsters(enemies) speed\",\n",
-      "        \"(17) list game modes\",\n",
-      "        \"(18) list props/weapons/food/obstacles/enemies\",\n",
-      "        \"(19) mute music\",\n",
-      "        \"(20) replace props\",\n",
-      "        \"(21) reset game\",\n",
-      "        \"(22) set game mode\",\n",
-      "        \"(23) set/increase/decrease snake speed\",\n",
-      "        \"(24) timer\",\n",
-      "        \"(25) winning condition\"\n",
-      "    ]\n",
-      "}\n"
-     ]
-    }
-   ],
-   "source": [
-    "print(json.dumps(game.list_functionality(), indent=4))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 26,
-   "id": "3bde59df-de5c-47ec-b131-c5a2c017a88e",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "/Users/shaojun/playgo/prod/ai/games/snake/notebooks\n"
-     ]
-    }
-   ],
-   "source": [
-    "!pwd"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "d7087f6d-103c-4104-b97c-d2eb64e39c0d",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 27,
-   "id": "b31e5edc-d9a5-4fd2-b7cc-adbec539d9e6",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# command = 'take away 10 guns'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 28,
-   "id": "21f3ad37-2a13-45c8-aa81-42df9aa46197",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# command = '10 less guns'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 29,
-   "id": "178d5332-5d68-4924-b6ae-872d0b400900",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "command = 'create more obstacles'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 30,
-   "id": "3cc479e9-3897-4a56-a9f4-a494ae608c7c",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "command = 'create more objects'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 31,
-   "id": "5919e519-1776-4433-a901-9c1089d8cb9d",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "command = 'allow obstacles to move'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 32,
-   "id": "bc9174da-46cc-428e-9306-51fff3db00ad",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "command = 'shhhh'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 33,
-   "id": "2446f322-b097-477d-81e3-492243013225",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "command = 'reset game'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 34,
-   "id": "d9fb2a0d-cfc6-48db-b7b1-50442298d143",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "command = 'change player speed'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 35,
-   "id": "73fa9c0e-b5b1-4120-a7d7-590ff8015817",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "command = 'slow down player'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 38,
-   "id": "7e7e39ed-2a8d-4bd1-b244-c8b62ecb58b2",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "command = 'add weapons'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 39,
-   "id": "7eb324ad-1040-47d1-8989-3e38fb130a65",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# command = 'tomorrow'"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 40,
-   "id": "0af1322d-7342-4cf0-9f81-63a6bd5a5d15",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "skip init\n",
-      "\n",
-      "--------------------\n",
-      "[snake] query count: 2\n",
-      "input: add weapons\n",
-      "{\n",
-      "  \"status\": \"OK\",\n",
-      "  \"result\": [\n",
-      "    \"weapon\",\n",
-      "    \"add\",\n",
-      "    {\n",
-      "      \"args\": \"some\"\n",
-      "    }\n",
-      "  ],\n",
-      "  \"command\": \"add weapons\",\n",
-      "  \"timestamp\": \"2023-11-10 18:26:24.587111+08:00\",\n",
-      "  \"ai-version\": \"1.4.0.2023-11-10-a\"\n",
-      "}\n",
-      "[\n",
-      "  [\n",
-      "    0,\n",
-      "    \"1.00000000\",\n",
-      "    \"add weapons\"\n",
-      "  ],\n",
-      "  [\n",
-      "    1,\n",
-      "    \"0.94908069\",\n",
-      "    \"increase weapons\"\n",
-      "  ],\n",
-      "  [\n",
-      "    2,\n",
-      "    \"0.94125815\",\n",
-      "    \"create weapons\"\n",
-      "  ],\n",
-      "  [\n",
-      "    3,\n",
-      "    \"0.93469667\",\n",
-      "    \"add guns\"\n",
-      "  ],\n",
-      "  [\n",
-      "    4,\n",
-      "    \"0.93148426\",\n",
-      "    \"remove weapons\"\n",
-      "  ]\n",
-      "]\n",
-      "%%%%%%%%%%%%%%%%%%%%\n",
-      "[snake] query count: 2\n",
-      "input: add weapons\n",
-      "OK\n",
-      "top: add weapons\n",
-      "--------------------\n",
-      "\n"
-     ]
-    },
-    {
-     "data": {
-      "text/plain": [
-       "({'status': 'OK',\n",
-       "  'result': ['weapon', 'add', {'args': 'some'}],\n",
-       "  'command': 'add weapons',\n",
-       "  'timestamp': '2023-11-10 18:26:24.587111+08:00',\n",
-       "  'ai-version': '1.4.0.2023-11-10-a'},\n",
-       " [[0, '1.00000000', 'add weapons'],\n",
-       "  [1, '0.94908069', 'increase weapons'],\n",
-       "  [2, '0.94125815', 'create weapons'],\n",
-       "  [3, '0.93469667', 'add guns'],\n",
-       "  [4, '0.93148426', 'remove weapons']])"
-      ]
-     },
-     "execution_count": 40,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "game.query_entry_point(command)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "7c151a5e-6bef-4847-a469-03f22b7af9d9",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "060f85f3-7e75-4ef5-9100-98df83ea0fd5",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
-   "language": "python",
-   "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.4"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}