|
12 | 12 | },
|
13 | 13 | {
|
14 | 14 | "cell_type": "markdown",
|
15 |
| - "source": [ |
16 |
| - "# TaskingAI Assistant Module CRUD Example" |
17 |
| - ], |
18 | 15 | "metadata": {
|
19 | 16 | "collapsed": false
|
20 |
| - } |
| 17 | + }, |
| 18 | + "source": [ |
| 19 | + "# TaskingAI Assistant Module CRUD Example" |
| 20 | + ] |
21 | 21 | },
|
22 | 22 | {
|
23 | 23 | "cell_type": "code",
|
24 | 24 | "execution_count": null,
|
25 | 25 | "metadata": {},
|
26 | 26 | "outputs": [],
|
27 | 27 | "source": [
|
28 |
| - "from taskingai.models import Assistant, Chat\n", |
| 28 | + "from taskingai.assistant import Assistant, Chat\n", |
29 | 29 | "from taskingai.assistant.memory import AssistantNaiveMemory\n",
|
30 | 30 | "\n",
|
31 | 31 | "# choose an available chat_completion model from your project\n",
|
|
34 | 34 | },
|
35 | 35 | {
|
36 | 36 | "cell_type": "markdown",
|
37 |
| - "source": [ |
38 |
| - "## Assistant Object" |
39 |
| - ], |
40 | 37 | "metadata": {
|
41 | 38 | "collapsed": false
|
42 |
| - } |
| 39 | + }, |
| 40 | + "source": [ |
| 41 | + "## Assistant Object" |
| 42 | + ] |
43 | 43 | },
|
44 | 44 | {
|
45 | 45 | "cell_type": "code",
|
46 | 46 | "execution_count": null,
|
47 | 47 | "metadata": {},
|
48 | 48 | "outputs": [],
|
49 | 49 | "source": [
|
| 50 | + "from taskingai.assistant import RetrievalConfig, RetrievalMethod\n", |
| 51 | + "\n", |
50 | 52 | "# create an assistant\n",
|
51 | 53 | "def create_assistant() -> Assistant:\n",
|
52 | 54 | " assistant: Assistant = taskingai.assistant.create_assistant(\n",
|
|
57 | 59 | " memory=AssistantNaiveMemory(),\n",
|
58 | 60 | " tools=[],\n",
|
59 | 61 | " retrievals=[],\n",
|
| 62 | + " retrieval_configs=RetrievalConfig(top_k=3, max_tokens=4096, method=RetrievalMethod.USER_MESSAGE),\n", |
60 | 63 | " metadata={\"foo\": \"bar\"},\n",
|
61 | 64 | " )\n",
|
62 | 65 | " return assistant\n",
|
|
68 | 71 | {
|
69 | 72 | "cell_type": "code",
|
70 | 73 | "execution_count": null,
|
| 74 | + "metadata": { |
| 75 | + "collapsed": false |
| 76 | + }, |
71 | 77 | "outputs": [],
|
72 | 78 | "source": [
|
73 | 79 | "# get assistant\n",
|
|
77 | 83 | ")\n",
|
78 | 84 | "\n",
|
79 | 85 | "print(f\"got assistant: {assistant}\\n\")"
|
80 |
| - ], |
81 |
| - "metadata": { |
82 |
| - "collapsed": false |
83 |
| - } |
| 86 | + ] |
84 | 87 | },
|
85 | 88 | {
|
86 | 89 | "cell_type": "code",
|
87 | 90 | "execution_count": null,
|
| 91 | + "metadata": { |
| 92 | + "collapsed": false |
| 93 | + }, |
88 | 94 | "outputs": [],
|
89 | 95 | "source": [
|
90 | 96 | "# update assistant\n",
|
91 | 97 | "assistant: Assistant = taskingai.assistant.update_assistant(\n",
|
92 | 98 | " assistant_id=assistant_id,\n",
|
93 | 99 | " name=\"My New Assistant\",\n",
|
94 | 100 | " description=\"This is my new assistant\",\n",
|
| 101 | + " retrieval_configs=RetrievalConfig(top_k=4, max_tokens=8192, method=RetrievalMethod.USER_MESSAGE),\n", |
95 | 102 | ")\n",
|
96 | 103 | "\n",
|
97 | 104 | "print(f\"updated assistant: {assistant}\\n\")\n"
|
98 |
| - ], |
99 |
| - "metadata": { |
100 |
| - "collapsed": false |
101 |
| - } |
| 105 | + ] |
102 | 106 | },
|
103 | 107 | {
|
104 | 108 | "cell_type": "code",
|
105 | 109 | "execution_count": null,
|
| 110 | + "metadata": { |
| 111 | + "collapsed": false |
| 112 | + }, |
106 | 113 | "outputs": [],
|
107 | 114 | "source": [
|
108 | 115 | "# delete assistant\n",
|
109 | 116 | "taskingai.assistant.delete_assistant(assistant_id=assistant_id)\n",
|
110 | 117 | "print(f\"deleted assistant: {assistant_id}\\n\")"
|
111 |
| - ], |
112 |
| - "metadata": { |
113 |
| - "collapsed": false |
114 |
| - } |
| 118 | + ] |
115 | 119 | },
|
116 | 120 | {
|
117 | 121 | "cell_type": "code",
|
118 | 122 | "execution_count": null,
|
| 123 | + "metadata": { |
| 124 | + "collapsed": false |
| 125 | + }, |
119 | 126 | "outputs": [],
|
120 | 127 | "source": [
|
121 | 128 | "# list assistants\n",
|
122 | 129 | "assistants = taskingai.assistant.list_assistants()\n",
|
123 | 130 | "assistant_ids = [assistant.assistant_id for assistant in assistants]\n",
|
124 | 131 | "# ensure the assistant we deleted is not in the list\n",
|
125 | 132 | "print(f\"f{assistant_id} in assistant_ids: {assistant_id in assistant_ids}\\n\")"
|
126 |
| - ], |
127 |
| - "metadata": { |
128 |
| - "collapsed": false |
129 |
| - } |
| 133 | + ] |
130 | 134 | },
|
131 | 135 | {
|
132 | 136 | "cell_type": "markdown",
|
133 |
| - "source": [ |
134 |
| - "## Chat Object" |
135 |
| - ], |
136 | 137 | "metadata": {
|
137 | 138 | "collapsed": false
|
138 |
| - } |
| 139 | + }, |
| 140 | + "source": [ |
| 141 | + "## Chat Object" |
| 142 | + ] |
139 | 143 | },
|
140 | 144 | {
|
141 | 145 | "cell_type": "code",
|
142 | 146 | "execution_count": null,
|
| 147 | + "metadata": { |
| 148 | + "collapsed": false |
| 149 | + }, |
143 | 150 | "outputs": [],
|
144 | 151 | "source": [
|
145 | 152 | "# create a new assistant\n",
|
|
148 | 155 | "# create a chat\n",
|
149 | 156 | "chat: Chat = taskingai.assistant.create_chat(\n",
|
150 | 157 | " assistant_id=assistant.assistant_id,\n",
|
| 158 | + " name=\"my chat\",\n", |
151 | 159 | ")\n",
|
152 | 160 | "print(f\"created chat: {chat.chat_id} for assistant: {assistant.assistant_id}\\n\")"
|
153 |
| - ], |
154 |
| - "metadata": { |
155 |
| - "collapsed": false |
156 |
| - } |
| 161 | + ] |
157 | 162 | },
|
158 | 163 | {
|
159 | 164 | "cell_type": "code",
|
160 | 165 | "execution_count": null,
|
| 166 | + "metadata": { |
| 167 | + "collapsed": false |
| 168 | + }, |
161 | 169 | "outputs": [],
|
162 | 170 | "source": [
|
163 | 171 | "# get chat\n",
|
|
167 | 175 | " chat_id=chat_id,\n",
|
168 | 176 | ")\n",
|
169 | 177 | "print(f\"chat: {chat}\\n\")"
|
170 |
| - ], |
171 |
| - "metadata": { |
172 |
| - "collapsed": false |
173 |
| - } |
| 178 | + ] |
174 | 179 | },
|
175 | 180 | {
|
176 | 181 | "cell_type": "code",
|
177 | 182 | "execution_count": null,
|
| 183 | + "metadata": { |
| 184 | + "collapsed": false |
| 185 | + }, |
178 | 186 | "outputs": [],
|
179 | 187 | "source": [
|
180 | 188 | "# update chat\n",
|
181 | 189 | "chat: Chat = taskingai.assistant.update_chat(\n",
|
182 | 190 | " assistant_id=assistant.assistant_id,\n",
|
183 | 191 | " chat_id=chat_id,\n",
|
| 192 | + " name=\"my chat new name\",\n", |
184 | 193 | " metadata={\"foo\": \"bar\"},\n",
|
185 | 194 | ")\n",
|
186 | 195 | "print(f\"updated chat: {chat}\\n\")"
|
187 |
| - ], |
188 |
| - "metadata": { |
189 |
| - "collapsed": false |
190 |
| - } |
| 196 | + ] |
191 | 197 | },
|
192 | 198 | {
|
193 | 199 | "cell_type": "code",
|
194 | 200 | "execution_count": null,
|
| 201 | + "metadata": { |
| 202 | + "collapsed": false |
| 203 | + }, |
195 | 204 | "outputs": [],
|
196 | 205 | "source": [
|
197 | 206 | "# delete chat\n",
|
|
200 | 209 | " chat_id=chat_id,\n",
|
201 | 210 | ")\n",
|
202 | 211 | "print(f\"deleted chat: {chat_id}\\n\")"
|
203 |
| - ], |
204 |
| - "metadata": { |
205 |
| - "collapsed": false |
206 |
| - } |
| 212 | + ] |
207 | 213 | },
|
208 | 214 | {
|
209 | 215 | "cell_type": "code",
|
210 | 216 | "execution_count": null,
|
| 217 | + "metadata": { |
| 218 | + "collapsed": false |
| 219 | + }, |
211 | 220 | "outputs": [],
|
212 | 221 | "source": [
|
213 | 222 | "# list chats \n",
|
|
220 | 229 | " assistant_id=assistant.assistant_id,\n",
|
221 | 230 | ")\n",
|
222 | 231 | "print(f\"num chats = {len(chats)}\\n\")"
|
223 |
| - ], |
224 |
| - "metadata": { |
225 |
| - "collapsed": false |
226 |
| - } |
| 232 | + ] |
227 | 233 | },
|
228 | 234 | {
|
229 | 235 | "cell_type": "code",
|
230 | 236 | "execution_count": null,
|
| 237 | + "metadata": { |
| 238 | + "collapsed": false |
| 239 | + }, |
231 | 240 | "outputs": [],
|
232 | 241 | "source": [
|
233 | 242 | "# delete assistant\n",
|
234 | 243 | "taskingai.assistant.delete_assistant(assistant_id=assistant.assistant_id)"
|
235 |
| - ], |
236 |
| - "metadata": { |
237 |
| - "collapsed": false |
238 |
| - } |
| 244 | + ] |
239 | 245 | }
|
240 | 246 | ],
|
241 | 247 | "metadata": {
|
242 |
| - "language_info": { |
243 |
| - "name": "python" |
244 |
| - }, |
245 | 248 | "kernelspec": {
|
246 |
| - "name": "python3", |
| 249 | + "display_name": "Python 3 (ipykernel)", |
247 | 250 | "language": "python",
|
248 |
| - "display_name": "Python 3 (ipykernel)" |
| 251 | + "name": "python3" |
| 252 | + }, |
| 253 | + "language_info": { |
| 254 | + "name": "python" |
249 | 255 | }
|
250 | 256 | },
|
251 | 257 | "nbformat": 4,
|
|
0 commit comments