fix: fix PR problems

Author: jameszyao

examples/assistant/chat_with_assistant.ipynb

@@ -31,7 +31,7 @@
     "from taskingai.inference import *\n",
     "import json\n",
     "# choose an available chat_completion model from your project\n",
-    "model_id = \"YOUR_MODEL_ID\""
+    "model_id = \"YOUR_CHAT_COMPLETION_MODEL_ID\""
    ],
    "metadata": {
     "collapsed": false
@@ -265,6 +265,16 @@
     "collapsed": false
    },
    "id": "4f3290f87635152a"
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "outputs": [],
+   "source": [],
+   "metadata": {
+    "collapsed": false
+   },
+   "id": "f9a7066d18575dc9"
   }
  ],
  "metadata": {

examples/crud/assistant_crud.ipynb

@@ -30,8 +30,6 @@
 from .chat_completion_system_message import *
 from .chat_completion_usage import *
 from .chat_completion_user_message import *
-from .chat_memory import *
-from .chat_memory_message import *
 from .chunk import *
 from .collection import *
 from .file_id_data import *

examples/crud/retrieval_crud.ipynb

@@ -20,5 +20,4 @@
 
 class AssistantMemory(BaseModel):
     type: AssistantMemoryType = Field(...)
-    max_messages: Optional[int] = Field(None, ge=1, le=1024)
-    max_tokens: Optional[int] = Field(None, ge=1, le=8192)
+    max_tokens: Optional[int] = Field(None, ge=0, le=8192)

examples/inference/chat_completion.ipynb

@@ -17,4 +17,5 @@
 
 
 class ToolType(str, Enum):
+    ACTION = "action"
     PLUGIN = "plugin"

taskingai/client/models/entities/__init__.py

@@ -117,6 +117,7 @@ def create_collection(
     :param embedding_model_id: The ID of an available embedding model in the project.
     :param capacity: The maximum number of embeddings that can be stored in the collection.
     :param name: The name of the collection.
+    :param type: The type of the collection.
     :param description: The description of the collection.
     :param metadata: The collection metadata. It can store up to 16 key-value pairs where each key's length is less than 64 and value's length is less than 512.
     :return: The created collection object.
@@ -149,12 +150,12 @@ async def a_create_collection(
     :param embedding_model_id: The ID of an available embedding model in the project.
     :param capacity: The maximum number of embeddings that can be stored in the collection.
     :param name: The name of the collection.
+    :param type: The type of the collection.
     :param description: The description of the collection.
     :param metadata: The collection metadata. It can store up to 16 key-value pairs where each key's length is less than 64 and value's length is less than 512.
     :return: The created collection object.
     """
 
-    # todo verify parameters
     body = CollectionCreateRequest(
         embedding_model_id=embedding_model_id,
         capacity=capacity,