fix: fix action schema pydantic error and remove MessageGenerationLog

Author: jameszyao

examples/assistant/chat_with_assistant.ipynb

@@ -173,7 +173,7 @@
    "source": [
     "from taskingai.assistant import Message, MessageChunk\n",
     "user_input = input(\"User Input: \")\n",
-    "while user_input != \"q\":\n",
+    "while user_input.strip() and user_input != \"q\":\n",
     "    # create user message\n",
     "    taskingai.assistant.create_message(\n",
     "        assistant_id=assistant.assistant_id,\n",
@@ -206,7 +206,7 @@
    "outputs": [],
    "source": [
     "user_input = input(\"User Input: \")\n",
-    "while user_input != \"q\":\n",
+    "while user_input.strip() and user_input != \"q\":\n",
     "    # create user message\n",
     "    taskingai.assistant.create_message(\n",
     "        assistant_id=assistant.assistant_id,\n",
@@ -273,6 +273,34 @@
     "collapsed": false
    },
    "id": "ed39836bbfdc7a4e"
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "outputs": [],
+   "source": [
+    "# list messages\n",
+    "messages = taskingai.assistant.list_messages(\n",
+    "    assistant_id=assistant.assistant_id,\n",
+    "    chat_id=chat.chat_id,\n",
+    ")\n",
+    "for message in messages:\n",
+    "    print(f\"{message.role}: {message.content.text}\")"
+   ],
+   "metadata": {
+    "collapsed": false
+   },
+   "id": "e94e3adb0d15373b"
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "outputs": [],
+   "source": [],
+   "metadata": {
+    "collapsed": false
+   },
+   "id": "6380260c47c9a4c"
   }
  ],
  "metadata": {

taskingai/assistant/message.py

@@ -6,7 +6,6 @@
     MessageRole,
     MessageContent,
     MessageChunk,
-    MessageGenerationLog,
 )
 from taskingai.client.models import (
     MessageCreateRequest,
@@ -22,7 +21,6 @@
 __all__ = [
     "Message",
     "MessageChunk",
-    "MessageGenerationLog",
     "get_message",
     "list_messages",
     "create_message",

taskingai/client/models/entity/assistant/message.py

@@ -7,7 +7,6 @@
     "Message",
     "MessageContent",
     "MessageRole",
-    "MessageGenerationLog",
     "MessageChunk"
 ]
 
@@ -17,14 +16,6 @@ class MessageRole(str, Enum):
     assistant = "assistant"
 
 
-class MessageGenerationLog(TaskingaiBaseModel):
-    object: str
-    type: str
-    session_id: str
-    created_timestamp: int
-    content: Dict
-
-
 class MessageContent(TaskingaiBaseModel):
     text: Optional[str]
 

taskingai/client/models/entity/tool/action.py

@@ -29,7 +29,10 @@ class Action(TaskingaiBaseModel):
     action_id: str
     name: str
     description: str
-    schema: Dict[str, Any]
+    schema: Dict[str, Any] = Field(..., alias='schema')
     authentication: ActionAuthentication
     created_timestamp: int
 
+    class Config:
+        allow_population_by_field_name = True
+

taskingai/client/stream.py

@@ -6,7 +6,6 @@
 from .models import (
     MessageChunk,
     Message,
-    MessageGenerationLog,
     ChatCompletion,
     ChatCompletionChunk
 )
@@ -25,8 +24,6 @@ def _cast(item: Dict):
         return Message(**item)
     elif item.get("object") == "MessageChunk":
         return MessageChunk(**item)
-    elif item.get("object") == "MessageGenerationLog":
-        return MessageGenerationLog(**item)
     else:
         # cannot cast, keep the original dict
         return item