Skip to content

Commit ef26c83

Browse files
jameszyaoSimsonW
jameszyao
authored and
SimsonW
committed
feat: add inference
1 parent cb20701 commit ef26c83

File tree

8 files changed

+594
-106
lines changed

8 files changed

+594
-106
lines changed

‎examples/__init__.py‎

Whitespace-only changes.

‎taskingai/client/__init__.py‎

Lines changed: 0 additions & 105 deletions
Original file line numberDiff line numberDiff line change
@@ -1,105 +0,0 @@
1-
# coding: utf-8
2-
3-
# flake8: noqa
4-
5-
"""
6-
TaskingAI API
7-
8-
OpenAPI spec version: 0.1.0
9-
"""
10-
11-
from __future__ import absolute_import
12-
13-
# import apis into sdk package
14-
from taskingai.client.api.assistant_api import AssistantApi
15-
from taskingai.client.api.inference_api import InferenceApi
16-
from taskingai.client.api.retrieval_api import RetrievalApi
17-
from taskingai.client.api.tool_api import ToolApi
18-
# import ApiClient
19-
from taskingai.client.api_client import ApiClient
20-
from taskingai.client.configuration import Configuration
21-
# import models into sdk package
22-
from taskingai.client.models.action import Action
23-
from taskingai.client.models.action_authentication import ActionAuthentication
24-
from taskingai.client.models.action_authentication_type import ActionAuthenticationType
25-
from taskingai.client.models.action_create_request import ActionCreateRequest
26-
from taskingai.client.models.action_create_response import ActionCreateResponse
27-
from taskingai.client.models.action_delete_response import ActionDeleteResponse
28-
from taskingai.client.models.action_get_response import ActionGetResponse
29-
from taskingai.client.models.action_list_response import ActionListResponse
30-
from taskingai.client.models.action_run_request import ActionRunRequest
31-
from taskingai.client.models.action_run_response import ActionRunResponse
32-
from taskingai.client.models.action_update_request import ActionUpdateRequest
33-
from taskingai.client.models.action_update_response import ActionUpdateResponse
34-
from taskingai.client.models.assistant import Assistant
35-
from taskingai.client.models.assistant_create_request import AssistantCreateRequest
36-
from taskingai.client.models.assistant_create_response import AssistantCreateResponse
37-
from taskingai.client.models.assistant_delete_response import AssistantDeleteResponse
38-
from taskingai.client.models.assistant_get_response import AssistantGetResponse
39-
from taskingai.client.models.assistant_list_response import AssistantListResponse
40-
from taskingai.client.models.assistant_retrieval import AssistantRetrieval
41-
from taskingai.client.models.assistant_tool import AssistantTool
42-
from taskingai.client.models.assistant_update_request import AssistantUpdateRequest
43-
from taskingai.client.models.assistant_update_response import AssistantUpdateResponse
44-
from taskingai.client.models.chat import Chat
45-
from taskingai.client.models.chat_completion_assistant_message import ChatCompletionAssistantMessage
46-
from taskingai.client.models.chat_completion_function import ChatCompletionFunction
47-
from taskingai.client.models.chat_completion_function_call import ChatCompletionFunctionCall
48-
from taskingai.client.models.chat_completion_function_message import ChatCompletionFunctionMessage
49-
from taskingai.client.models.chat_completion_request import ChatCompletionRequest
50-
from taskingai.client.models.chat_completion_role import ChatCompletionRole
51-
from taskingai.client.models.chat_completion_system_message import ChatCompletionSystemMessage
52-
from taskingai.client.models.chat_completion_user_message import ChatCompletionUserMessage
53-
from taskingai.client.models.chat_create_request import ChatCreateRequest
54-
from taskingai.client.models.chat_create_response import ChatCreateResponse
55-
from taskingai.client.models.chat_delete_response import ChatDeleteResponse
56-
from taskingai.client.models.chat_get_response import ChatGetResponse
57-
from taskingai.client.models.chat_list_response import ChatListResponse
58-
from taskingai.client.models.chat_update_request import ChatUpdateRequest
59-
from taskingai.client.models.chat_update_response import ChatUpdateResponse
60-
from taskingai.client.models.chunk import Chunk
61-
from taskingai.client.models.chunk_query_request import ChunkQueryRequest
62-
from taskingai.client.models.chunk_query_response import ChunkQueryResponse
63-
from taskingai.client.models.collection import Collection
64-
from taskingai.client.models.collection_config import CollectionConfig
65-
from taskingai.client.models.collection_create_request import CollectionCreateRequest
66-
from taskingai.client.models.collection_create_response import CollectionCreateResponse
67-
from taskingai.client.models.collection_get_response import CollectionGetResponse
68-
from taskingai.client.models.collection_list_response import CollectionListResponse
69-
from taskingai.client.models.collection_update_request import CollectionUpdateRequest
70-
from taskingai.client.models.collection_update_response import CollectionUpdateResponse
71-
from taskingai.client.models.delete_collection_response import DeleteCollectionResponse
72-
from taskingai.client.models.function import Function
73-
from taskingai.client.models.function_create_request import FunctionCreateRequest
74-
from taskingai.client.models.function_create_response import FunctionCreateResponse
75-
from taskingai.client.models.function_delete_response import FunctionDeleteResponse
76-
from taskingai.client.models.function_get_response import FunctionGetResponse
77-
from taskingai.client.models.function_list_response import FunctionListResponse
78-
from taskingai.client.models.function_update_request import FunctionUpdateRequest
79-
from taskingai.client.models.function_update_response import FunctionUpdateResponse
80-
from taskingai.client.models.http_validation_error import HTTPValidationError
81-
from taskingai.client.models.message import Message
82-
from taskingai.client.models.message_content import MessageContent
83-
from taskingai.client.models.message_create_request import MessageCreateRequest
84-
from taskingai.client.models.message_create_response import MessageCreateResponse
85-
from taskingai.client.models.message_generate_request import MessageGenerateRequest
86-
from taskingai.client.models.message_generation_response_option import MessageGenerationResponseOption
87-
from taskingai.client.models.message_get_response import MessageGetResponse
88-
from taskingai.client.models.message_list_response import MessageListResponse
89-
from taskingai.client.models.message_role import MessageRole
90-
from taskingai.client.models.message_update_request import MessageUpdateRequest
91-
from taskingai.client.models.message_update_response import MessageUpdateResponse
92-
from taskingai.client.models.parameters_input import ParametersInput
93-
from taskingai.client.models.parameters_output import ParametersOutput
94-
from taskingai.client.models.record import Record
95-
from taskingai.client.models.record_create_request import RecordCreateRequest
96-
from taskingai.client.models.record_create_response import RecordCreateResponse
97-
from taskingai.client.models.record_delete_response import RecordDeleteResponse
98-
from taskingai.client.models.record_get_response import RecordGetResponse
99-
from taskingai.client.models.record_list_response import RecordListResponse
100-
from taskingai.client.models.record_update_request import RecordUpdateRequest
101-
from taskingai.client.models.record_update_response import RecordUpdateResponse
102-
from taskingai.client.models.text_embedding_output import TextEmbeddingOutput
103-
from taskingai.client.models.text_embedding_request import TextEmbeddingRequest
104-
from taskingai.client.models.text_embedding_response import TextEmbeddingResponse
105-
from taskingai.client.models.validation_error import ValidationError

‎taskingai/client/models/__init__.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,8 @@
3333
from taskingai.client.models.assistant_update_request import AssistantUpdateRequest
3434
from taskingai.client.models.assistant_update_response import AssistantUpdateResponse
3535
from taskingai.client.models.chat import Chat
36+
from taskingai.client.models.chat_completion_response import ChatCompletionResponse
37+
from taskingai.client.models.chat_completion import ChatCompletion
3638
from taskingai.client.models.chat_completion_assistant_message import ChatCompletionAssistantMessage
3739
from taskingai.client.models.chat_completion_function import ChatCompletionFunction
3840
from taskingai.client.models.chat_completion_function_call import ChatCompletionFunctionCall
@@ -74,6 +76,7 @@
7476
from taskingai.client.models.message_create_request import MessageCreateRequest
7577
from taskingai.client.models.message_create_response import MessageCreateResponse
7678
from taskingai.client.models.message_generate_request import MessageGenerateRequest
79+
from taskingai.client.models.message_generate_response import MessageGenerateResponse
7780
from taskingai.client.models.message_generation_response_option import MessageGenerationResponseOption
7881
from taskingai.client.models.message_get_response import MessageGetResponse
7982
from taskingai.client.models.message_list_response import MessageListResponse

‎taskingai/client/models/chat_completion.py‎

Lines changed: 200 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,200 @@
1+
# coding: utf-8
2+
3+
"""
4+
TaskingAI API
5+
6+
No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen) # noqa: E501
7+
8+
OpenAPI spec version: 0.1.0
9+
10+
Generated by: https://github.com/swagger-api/swagger-codegen.git
11+
"""
12+
13+
import pprint
14+
import re # noqa: F401
15+
16+
import six
17+
18+
class ChatCompletion(object):
19+
"""NOTE: This class is auto generated by the swagger code generator program.
20+
21+
Do not edit the class manually.
22+
"""
23+
"""
24+
Attributes:
25+
swagger_types (dict): The key is attribute name
26+
and the value is attribute type.
27+
attribute_map (dict): The key is attribute name
28+
and the value is json key in definition.
29+
"""
30+
swagger_types = {
31+
'object': 'object',
32+
'finish_reason': 'object',
33+
'message': 'object',
34+
'created_timestamp': 'object'
35+
}
36+
37+
attribute_map = {
38+
'object': 'object',
39+
'finish_reason': 'finish_reason',
40+
'message': 'message',
41+
'created_timestamp': 'created_timestamp'
42+
}
43+
44+
def __init__(self, object=None, finish_reason=None, message=None, created_timestamp=None): # noqa: E501
45+
"""ChatCompletion - a model defined in Swagger""" # noqa: E501
46+
self._object = None
47+
self._finish_reason = None
48+
self._message = None
49+
self._created_timestamp = None
50+
self.discriminator = None
51+
self.object = object
52+
self.finish_reason = finish_reason
53+
self.message = message
54+
self.created_timestamp = created_timestamp
55+
56+
@property
57+
def object(self):
58+
"""Gets the object of this ChatCompletion. # noqa: E501
59+
60+
The object type, which is always 'ChatCompletion'. # noqa: E501
61+
62+
:return: The object of this ChatCompletion. # noqa: E501
63+
:rtype: object
64+
"""
65+
return self._object
66+
67+
@object.setter
68+
def object(self, object):
69+
"""Sets the object of this ChatCompletion.
70+
71+
The object type, which is always 'ChatCompletion'. # noqa: E501
72+
73+
:param object: The object of this ChatCompletion. # noqa: E501
74+
:type: object
75+
"""
76+
if object is None:
77+
raise ValueError("Invalid value for `object`, must not be `None`") # noqa: E501
78+
79+
self._object = object
80+
81+
@property
82+
def finish_reason(self):
83+
"""Gets the finish_reason of this ChatCompletion. # noqa: E501
84+
85+
The reason why the generation is finished. # noqa: E501
86+
87+
:return: The finish_reason of this ChatCompletion. # noqa: E501
88+
:rtype: object
89+
"""
90+
return self._finish_reason
91+
92+
@finish_reason.setter
93+
def finish_reason(self, finish_reason):
94+
"""Sets the finish_reason of this ChatCompletion.
95+
96+
The reason why the generation is finished. # noqa: E501
97+
98+
:param finish_reason: The finish_reason of this ChatCompletion. # noqa: E501
99+
:type: object
100+
"""
101+
if finish_reason is None:
102+
raise ValueError("Invalid value for `finish_reason`, must not be `None`") # noqa: E501
103+
104+
self._finish_reason = finish_reason
105+
106+
@property
107+
def message(self):
108+
"""Gets the message of this ChatCompletion. # noqa: E501
109+
110+
The message generated by the assistant. # noqa: E501
111+
112+
:return: The message of this ChatCompletion. # noqa: E501
113+
:rtype: object
114+
"""
115+
return self._message
116+
117+
@message.setter
118+
def message(self, message):
119+
"""Sets the message of this ChatCompletion.
120+
121+
The message generated by the assistant. # noqa: E501
122+
123+
:param message: The message of this ChatCompletion. # noqa: E501
124+
:type: object
125+
"""
126+
if message is None:
127+
raise ValueError("Invalid value for `message`, must not be `None`") # noqa: E501
128+
129+
self._message = message
130+
131+
@property
132+
def created_timestamp(self):
133+
"""Gets the created_timestamp of this ChatCompletion. # noqa: E501
134+
135+
The timestamp in milliseconds when the response is created. # noqa: E501
136+
137+
:return: The created_timestamp of this ChatCompletion. # noqa: E501
138+
:rtype: object
139+
"""
140+
return self._created_timestamp
141+
142+
@created_timestamp.setter
143+
def created_timestamp(self, created_timestamp):
144+
"""Sets the created_timestamp of this ChatCompletion.
145+
146+
The timestamp in milliseconds when the response is created. # noqa: E501
147+
148+
:param created_timestamp: The created_timestamp of this ChatCompletion. # noqa: E501
149+
:type: object
150+
"""
151+
if created_timestamp is None:
152+
raise ValueError("Invalid value for `created_timestamp`, must not be `None`") # noqa: E501
153+
154+
self._created_timestamp = created_timestamp
155+
156+
def to_dict(self):
157+
"""Returns the model properties as a dict"""
158+
result = {}
159+
160+
for attr, _ in six.iteritems(self.swagger_types):
161+
value = getattr(self, attr)
162+
if isinstance(value, list):
163+
result[attr] = list(map(
164+
lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
165+
value
166+
))
167+
elif hasattr(value, "to_dict"):
168+
result[attr] = value.to_dict()
169+
elif isinstance(value, dict):
170+
result[attr] = dict(map(
171+
lambda item: (item[0], item[1].to_dict())
172+
if hasattr(item[1], "to_dict") else item,
173+
value.items()
174+
))
175+
else:
176+
result[attr] = value
177+
if issubclass(ChatCompletion, dict):
178+
for key, value in self.items():
179+
result[key] = value
180+
181+
return result
182+
183+
def to_str(self):
184+
"""Returns the string representation of the model"""
185+
return pprint.pformat(self.to_dict())
186+
187+
def __repr__(self):
188+
"""For `print` and `pprint`"""
189+
return self.to_str()
190+
191+
def __eq__(self, other):
192+
"""Returns true if both objects are equal"""
193+
if not isinstance(other, ChatCompletion):
194+
return False
195+
196+
return self.__dict__ == other.__dict__
197+
198+
def __ne__(self, other):
199+
"""Returns true if both objects are not equal"""
200+
return not self == other

0 commit comments

Comments
 (0)