feat: finish assistant example

Author: jameszyao

examples/assistant.py

@@ -0,0 +1,57 @@
+import taskingai
+from taskingai.assistant import Assistant
+
+# Load TaskingAI API Key fron environment variable
+
+# choose an available chat_completion model from your project
+model_id = "Gk1145Bl"
+
+
+# create an assistant
+assistant: Assistant = taskingai.assistant.create_assistant(
+    model_id=model_id,
+    name="My Assistant",
+    description="This is my assistant",
+    system_prompt_template=["You are a professional assistant speaking {{language}}."],
+    tools=[],
+    retrievals=[],
+    metadata={"foo": "bar"},
+)
+
+print(f"created assistant: {assistant}\n")
+
+
+# get assistant
+assistant_id: str = assistant.assistant_id
+assistant: Assistant = taskingai.assistant.get_assistant(
+    assistant_id=assistant_id
+)
+
+print(f"got assistant: {assistant}\n")
+
+
+# update assistant
+assistant: Assistant = taskingai.assistant.update_assistant(
+    assistant_id=assistant_id,
+    name="My New Assistant",
+    description="This is my new assistant",
+)
+
+print(f"updated assistant: {assistant}\n")
+
+
+# delete assistant
+taskingai.assistant.delete_assistant(assistant_id=assistant_id)
+print(f"deleted assistant: {assistant_id}\n")
+
+# list assistants
+assistants = taskingai.assistant.list_assistants()
+assistant_ids = [assistant.assistant_id for assistant in assistants]
+print(f"list assistants: {assistants}\n")
+print(f"f{assistant_id} in {assistant_ids}: {assistant_id in assistant_ids}\n")
+
+
+
+
+
+

examples/assistant/assistant.py

@@ -23,6 +23,7 @@ class ActionAuthenticationType(object):
     BASIC = "basic"
     BEARER = "bearer"
     CUSTOM = "custom"
+    NONE = "none"
     """
     Attributes:
       swagger_types (dict): The key is attribute name

examples/inference.py renamed to examples/assistant/chat.py

@@ -168,6 +168,9 @@ def name(self, name):
         :param name: The name of this Assistant.  # noqa: E501
         :type: object
         """
+        if name is None:
+            raise ValueError("Invalid value for `name`, must not be `None`")
+
         self._name = name
 
     @property
@@ -190,6 +193,9 @@ def description(self, description):
         :param description: The description of this Assistant.  # noqa: E501
         :type: object
         """
+        if description is None:
+            raise ValueError("Invalid value for `description`, must not be `None`")
+
         self._description = description
 
     @property
@@ -212,6 +218,9 @@ def system_prompt_template(self, system_prompt_template):
         :param system_prompt_template: The system_prompt_template of this Assistant.  # noqa: E501
         :type: object
         """
+        if system_prompt_template is None:
+            raise ValueError("Invalid value for `system_prompt_template`, must not be `None`")
+
         self._system_prompt_template = system_prompt_template
 
     @property
@@ -234,6 +243,9 @@ def tools(self, tools):
         :param tools: The tools of this Assistant.  # noqa: E501
         :type: object
         """
+        if tools is None:
+            raise ValueError("Invalid value for `tools`, must not be `None`")  # noqa: E501
+
         self._tools = tools
 
     @property
@@ -256,6 +268,9 @@ def retrievals(self, retrievals):
         :param retrievals: The retrievals of this Assistant.  # noqa: E501
         :type: object
         """
+        if retrievals is None:
+            raise ValueError("Invalid value for `retrievals`, must not be `None`")  # noqa: E501
+
         self._retrievals = retrievals
 
     @property
@@ -278,6 +293,9 @@ def metadata(self, metadata):
         :param metadata: The metadata of this Assistant.  # noqa: E501
         :type: object
         """
+        if metadata is None:
+            raise ValueError("Invalid value for `metadata`, must not be `None`")
+
         self._metadata = metadata
 
     @property
@@ -300,6 +318,9 @@ def created_timestamp(self, created_timestamp):
         :param created_timestamp: The created_timestamp of this Assistant.  # noqa: E501
         :type: object
         """
+        if created_timestamp is None:
+            raise ValueError("Invalid value for `created_timestamp`, must not be `None`")  # noqa: E501
+
         self._created_timestamp = created_timestamp
 
     def to_dict(self):