Skip to content

Commit 9656c8c

Browse files
committed
feat: add upload file api
1 parent 19a73f9 commit 9656c8c

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

43 files changed

+2067
-537
lines changed

.env

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,4 @@ TASKINGAI_HOST=https://api.test199.com
44

55
TEXT_EMBEDDING_MODEL_ID=TpEZlEOK
66

7-
TASKINGAI_API_KEY=taxy8i3OCfeJfh0eXW0h00cF2QT7nWyy
7+
TASKINGAI_API_KEY=taxy8i3OCfeJfh0eXW0h00cF2QT7nWyy

.gitignore

Lines changed: 44 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -63,13 +63,52 @@ target/
6363
#Ipython Notebook
6464
.ipynb_checkpoints
6565

66+
# PyCharm
6667
.idea/
6768

69+
### macOS ###
70+
# General
6871
.DS_Store
69-
.venv
72+
.AppleDouble
73+
.LSOverride
7074

71-
# test
72-
test/.pytest_cache/
73-
test/log/
75+
# Icon must end with two \r
76+
Icon
7477

75-
**/allure-report
78+
79+
# Thumbnails
80+
._*
81+
82+
# Files that might appear in the root of a volume
83+
.DocumentRevisions-V100
84+
.fseventsd
85+
.Spotlight-V100
86+
.TemporaryItems
87+
.Trashes
88+
.VolumeIcon.icns
89+
.com.apple.timemachine.donotpresent
90+
91+
# Directories potentially created on remote AFP share
92+
.AppleDB
93+
.AppleDesktop
94+
Network Trash Folder
95+
Temporary Items
96+
.apdisk
97+
98+
### macOS Patch ###
99+
# iCloud generated files
100+
*.icloud
101+
102+
### VisualStudioCode ###
103+
.vscode/*
104+
105+
# Local History for Visual Studio Code
106+
.history/
107+
108+
# Built Visual Studio Code Extensions
109+
*.vsix
110+
111+
### VisualStudioCode Patch ###
112+
# Ignore all local history of files
113+
.history
114+
.ionide

examples/crud/assistant_crud.ipynb

Lines changed: 64 additions & 58 deletions
Original file line numberDiff line numberDiff line change
@@ -12,20 +12,20 @@
1212
},
1313
{
1414
"cell_type": "markdown",
15-
"source": [
16-
"# TaskingAI Assistant Module CRUD Example"
17-
],
1815
"metadata": {
1916
"collapsed": false
20-
}
17+
},
18+
"source": [
19+
"# TaskingAI Assistant Module CRUD Example"
20+
]
2121
},
2222
{
2323
"cell_type": "code",
2424
"execution_count": null,
2525
"metadata": {},
2626
"outputs": [],
2727
"source": [
28-
"from taskingai.models import Assistant, Chat\n",
28+
"from taskingai.assistant import Assistant, Chat\n",
2929
"from taskingai.assistant.memory import AssistantNaiveMemory\n",
3030
"\n",
3131
"# choose an available chat_completion model from your project\n",
@@ -34,19 +34,21 @@
3434
},
3535
{
3636
"cell_type": "markdown",
37-
"source": [
38-
"## Assistant Object"
39-
],
4037
"metadata": {
4138
"collapsed": false
42-
}
39+
},
40+
"source": [
41+
"## Assistant Object"
42+
]
4343
},
4444
{
4545
"cell_type": "code",
4646
"execution_count": null,
4747
"metadata": {},
4848
"outputs": [],
4949
"source": [
50+
"from taskingai.assistant import RetrievalConfig, RetrievalMethod\n",
51+
"\n",
5052
"# create an assistant\n",
5153
"def create_assistant() -> Assistant:\n",
5254
" assistant: Assistant = taskingai.assistant.create_assistant(\n",
@@ -57,6 +59,7 @@
5759
" memory=AssistantNaiveMemory(),\n",
5860
" tools=[],\n",
5961
" retrievals=[],\n",
62+
" retrieval_configs=RetrievalConfig(top_k=3, max_tokens=4096, method=RetrievalMethod.USER_MESSAGE),\n",
6063
" metadata={\"foo\": \"bar\"},\n",
6164
" )\n",
6265
" return assistant\n",
@@ -68,6 +71,9 @@
6871
{
6972
"cell_type": "code",
7073
"execution_count": null,
74+
"metadata": {
75+
"collapsed": false
76+
},
7177
"outputs": [],
7278
"source": [
7379
"# get assistant\n",
@@ -77,69 +83,70 @@
7783
")\n",
7884
"\n",
7985
"print(f\"got assistant: {assistant}\\n\")"
80-
],
81-
"metadata": {
82-
"collapsed": false
83-
}
86+
]
8487
},
8588
{
8689
"cell_type": "code",
8790
"execution_count": null,
91+
"metadata": {
92+
"collapsed": false
93+
},
8894
"outputs": [],
8995
"source": [
9096
"# update assistant\n",
9197
"assistant: Assistant = taskingai.assistant.update_assistant(\n",
9298
" assistant_id=assistant_id,\n",
9399
" name=\"My New Assistant\",\n",
94100
" description=\"This is my new assistant\",\n",
101+
" retrieval_configs=RetrievalConfig(top_k=4, max_tokens=8192, method=RetrievalMethod.USER_MESSAGE),\n",
95102
")\n",
96103
"\n",
97104
"print(f\"updated assistant: {assistant}\\n\")\n"
98-
],
99-
"metadata": {
100-
"collapsed": false
101-
}
105+
]
102106
},
103107
{
104108
"cell_type": "code",
105109
"execution_count": null,
110+
"metadata": {
111+
"collapsed": false
112+
},
106113
"outputs": [],
107114
"source": [
108115
"# delete assistant\n",
109116
"taskingai.assistant.delete_assistant(assistant_id=assistant_id)\n",
110117
"print(f\"deleted assistant: {assistant_id}\\n\")"
111-
],
112-
"metadata": {
113-
"collapsed": false
114-
}
118+
]
115119
},
116120
{
117121
"cell_type": "code",
118122
"execution_count": null,
123+
"metadata": {
124+
"collapsed": false
125+
},
119126
"outputs": [],
120127
"source": [
121128
"# list assistants\n",
122129
"assistants = taskingai.assistant.list_assistants()\n",
123130
"assistant_ids = [assistant.assistant_id for assistant in assistants]\n",
124131
"# ensure the assistant we deleted is not in the list\n",
125132
"print(f\"f{assistant_id} in assistant_ids: {assistant_id in assistant_ids}\\n\")"
126-
],
127-
"metadata": {
128-
"collapsed": false
129-
}
133+
]
130134
},
131135
{
132136
"cell_type": "markdown",
133-
"source": [
134-
"## Chat Object"
135-
],
136137
"metadata": {
137138
"collapsed": false
138-
}
139+
},
140+
"source": [
141+
"## Chat Object"
142+
]
139143
},
140144
{
141145
"cell_type": "code",
142146
"execution_count": null,
147+
"metadata": {
148+
"collapsed": false
149+
},
143150
"outputs": [],
144151
"source": [
145152
"# create a new assistant\n",
@@ -148,16 +155,17 @@
148155
"# create a chat\n",
149156
"chat: Chat = taskingai.assistant.create_chat(\n",
150157
" assistant_id=assistant.assistant_id,\n",
158+
" name=\"my chat\",\n",
151159
")\n",
152160
"print(f\"created chat: {chat.chat_id} for assistant: {assistant.assistant_id}\\n\")"
153-
],
154-
"metadata": {
155-
"collapsed": false
156-
}
161+
]
157162
},
158163
{
159164
"cell_type": "code",
160165
"execution_count": null,
166+
"metadata": {
167+
"collapsed": false
168+
},
161169
"outputs": [],
162170
"source": [
163171
"# get chat\n",
@@ -167,31 +175,32 @@
167175
" chat_id=chat_id,\n",
168176
")\n",
169177
"print(f\"chat: {chat}\\n\")"
170-
],
171-
"metadata": {
172-
"collapsed": false
173-
}
178+
]
174179
},
175180
{
176181
"cell_type": "code",
177182
"execution_count": null,
183+
"metadata": {
184+
"collapsed": false
185+
},
178186
"outputs": [],
179187
"source": [
180188
"# update chat\n",
181189
"chat: Chat = taskingai.assistant.update_chat(\n",
182190
" assistant_id=assistant.assistant_id,\n",
183191
" chat_id=chat_id,\n",
192+
" name=\"my chat new name\",\n",
184193
" metadata={\"foo\": \"bar\"},\n",
185194
")\n",
186195
"print(f\"updated chat: {chat}\\n\")"
187-
],
188-
"metadata": {
189-
"collapsed": false
190-
}
196+
]
191197
},
192198
{
193199
"cell_type": "code",
194200
"execution_count": null,
201+
"metadata": {
202+
"collapsed": false
203+
},
195204
"outputs": [],
196205
"source": [
197206
"# delete chat\n",
@@ -200,14 +209,14 @@
200209
" chat_id=chat_id,\n",
201210
")\n",
202211
"print(f\"deleted chat: {chat_id}\\n\")"
203-
],
204-
"metadata": {
205-
"collapsed": false
206-
}
212+
]
207213
},
208214
{
209215
"cell_type": "code",
210216
"execution_count": null,
217+
"metadata": {
218+
"collapsed": false
219+
},
211220
"outputs": [],
212221
"source": [
213222
"# list chats \n",
@@ -220,32 +229,29 @@
220229
" assistant_id=assistant.assistant_id,\n",
221230
")\n",
222231
"print(f\"num chats = {len(chats)}\\n\")"
223-
],
224-
"metadata": {
225-
"collapsed": false
226-
}
232+
]
227233
},
228234
{
229235
"cell_type": "code",
230236
"execution_count": null,
237+
"metadata": {
238+
"collapsed": false
239+
},
231240
"outputs": [],
232241
"source": [
233242
"# delete assistant\n",
234243
"taskingai.assistant.delete_assistant(assistant_id=assistant.assistant_id)"
235-
],
236-
"metadata": {
237-
"collapsed": false
238-
}
244+
]
239245
}
240246
],
241247
"metadata": {
242-
"language_info": {
243-
"name": "python"
244-
},
245248
"kernelspec": {
246-
"name": "python3",
249+
"display_name": "Python 3 (ipykernel)",
247250
"language": "python",
248-
"display_name": "Python 3 (ipykernel)"
251+
"name": "python3"
252+
},
253+
"language_info": {
254+
"name": "python"
249255
}
250256
},
251257
"nbformat": 4,

0 commit comments

Comments
 (0)