fix: update utils

Author: jameszyao

examples/assistant.py

@@ -0,0 +1,37 @@
+import taskingai
+from taskingai.assistant import Assistant
+
+# Load TaskingAI API Key fron environment variable
+
+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}")
+
+
+# get assistant
+assistant_id: str = assistant.assistant_id
+assistant: Assistant = taskingai.assistant.get_assistant(
+    assistant_id=assistant_id
+)
+
+print(f"got assistant: {assistant}")
+
+
+
+
+
+
+
+

examples/__init__.py examples/inference.pyexamples/__init__.py renamed to examples/inference.py

@@ -1,2 +1,3 @@
 from .assistant import *
-from .chat import *
+from .chat import *
+from .message import *

examples/retrieval.py

@@ -2,7 +2,6 @@
 import logging
 from pathlib import Path
 import urllib3
-from taskingai.config import Config
 from taskingai.client.api import AssistantApi, ToolApi, RetrievalApi, InferenceApi
 from taskingai.client.api_client import ApiClient
 
@@ -25,6 +24,7 @@ def get_user_agent():
 
 
 def get_assistant_api_instance():
+    from taskingai.config import Config
     client_config = Config.OPENAPI_CONFIG
     client_config.api_key = client_config.api_key or {}
     api_client = ApiClient(configuration=client_config)
@@ -34,6 +34,7 @@ def get_assistant_api_instance():
 
 
 def get_tool_api_instance():
+    from taskingai.config import Config
     client_config = Config.OPENAPI_CONFIG
     client_config.api_key = client_config.api_key or {}
     api_client = ApiClient(configuration=client_config)
@@ -43,6 +44,7 @@ def get_tool_api_instance():
 
 
 def get_retrieval_api_instance():
+    from taskingai.config import Config
     client_config = Config.OPENAPI_CONFIG
     client_config.api_key = client_config.api_key or {}
     api_client = ApiClient(configuration=client_config)
@@ -52,6 +54,7 @@ def get_retrieval_api_instance():
 
 
 def get_inference_api_instance():
+    from taskingai.config import Config
     client_config = Config.OPENAPI_CONFIG
     client_config.api_key = client_config.api_key or {}
     api_client = ApiClient(configuration=client_config)

examples/tool.py

@@ -5,10 +5,10 @@
 
 import certifi
 
-from taskingai.client.utils import check_kwargs
-from taskingai.client.exceptions import ApiKeyError
-from taskingai.client.constants import PARENT_LOGGER_NAME, DEFAULT_PARENT_LOGGER_LEVEL
-from taskingai.client.configuration import Configuration as OpenApiConfiguration
+from .client.utils import check_kwargs
+from .client.exceptions import ApiKeyError
+from .client.constants import PARENT_LOGGER_NAME, DEFAULT_PARENT_LOGGER_LEVEL
+from .client.configuration import Configuration as OpenApiConfiguration
 
 __all__ = ["Config", "init"]
 

taskingai/assistant/__init__.py

@@ -0,0 +1,3 @@
+from .chunk import *
+from .collection import *
+from .record import *

taskingai/client/utils.py

@@ -0,0 +1 @@
+# todo

taskingai/config.py

@@ -0,0 +1,2 @@
+from action import *
+from function import *