dashscope 1.8.0__py3-none-any.whl → 1.25.6__py3-none-any.whl

dashscope/common/api_key.py

@@ -1,3 +1,5 @@
+# Copyright (c) Alibaba, Inc. and its affiliates.
+
 import os
 from typing import Optional
 

dashscope/common/base_type.py

@@ -0,0 +1,135 @@
+# Copyright (c) Alibaba, Inc. and its affiliates.
+
+import dataclasses
+from dataclasses import dataclass
+from typing import Any, List
+
+import dashscope
+
+
+def get_object_type(name: str):
+    dashscope_objects = {
+        'assistant': dashscope.Assistant,
+        'assistant.deleted': dashscope.DeleteResponse,
+        'thread.message': dashscope.ThreadMessage,
+        'thread.run': dashscope.Run,
+        'thread.run.step': dashscope.RunStep,
+        'thread.message.file': dashscope.MessageFile,
+        'assistant.file': dashscope.AssistantFile,
+        'thread': dashscope.Thread,
+    }
+    return dashscope_objects.get(name, None)
+
+
+@dataclass(init=False)
+class BaseObjectMixin(object):
+    __slots__ = ()
+
+    def __init__(self, **kwargs):
+        field_type_map = self._get_fields_type()
+        for k, v in kwargs.items():
+            field = field_type_map.get(k, None)
+            if field and v is not None:
+                if dataclasses.is_dataclass(field.type):  # process dataclasses
+                    self.__setattr__(k, field.type(**v))
+                    continue
+
+            if isinstance(v, dict):
+                object_name = v.get('object', None)
+                if object_name:
+                    object_type = get_object_type(object_name)
+                    if object_type:
+                        self.__setattr__(k, object_type(**v))
+                    else:
+                        self.__setattr__(k, v)
+                else:
+                    self.__setattr__(k, v)
+            elif isinstance(v, list):
+                obj_list = self._init_list_element_recursive(field, v)
+                self.__setattr__(k, obj_list)
+            else:
+                self.__setattr__(k, v)
+
+    def _init_list_element_recursive(self, field, items: list) -> List[Any]:
+        obj_list = []
+        for item in items:
+            if field:
+                # current only support List[cls_name],
+                # not support List[cls_nam1, cls_name2]
+                element_type = field.type.__args__[0]
+                if dataclasses.is_dataclass(element_type):
+                    obj_list.append(element_type(**item))
+                    continue
+
+            if isinstance(item, dict):
+                object_name = item.get('object', None)
+                if object_name:
+                    object_type = get_object_type(object_name)
+                    if object_type:
+                        obj_list.append(object_type(**item))
+                    else:
+                        obj_list.append(item)
+                else:
+                    obj_list.append(item)
+            elif isinstance(item, list):
+                obj_list.append(self._init_list_element_recursive(item))
+            else:
+                obj_list.append(item)
+        return obj_list
+
+    def _get_fields_type(self):
+        field_type_map = {}
+        if dataclasses.is_dataclass(self):
+            for field in dataclasses.fields(self):
+                field_type_map[field.name] = field
+        return field_type_map
+
+    def __setitem__(self, __key: Any, __value: Any) -> None:
+        return self.__setattr__(__key, __value)
+
+    def __getitem__(self, __key: Any) -> Any:
+        return self.__getattribute__(__key)
+
+    def __contains__(self, item):
+        return hasattr(self, item)
+
+    def __delitem__(self, key):
+        self.__delattr__(key)
+
+    def _recursive_to_str__(self, input_object) -> Any:
+        if isinstance(input_object, list):
+            output_object = []
+            for item in input_object:
+                output_object.append(self._recursive_to_str__(item))
+            return output_object
+        elif dataclasses.is_dataclass(input_object):
+            output_object = {}
+            for field in dataclasses.fields(input_object):
+                if hasattr(input_object, field.name):
+                    output_object[field.name] = self._recursive_to_str__(
+                        getattr(input_object, field.name))
+            return output_object
+        else:
+            return input_object
+
+    def __str__(self) -> str:
+        real_dict = self.__dict__
+        self_fields = dataclasses.fields(self)
+        for field in self_fields:
+            if hasattr(self, field.name):
+                real_dict[field.name] = getattr(self, field.name)
+        output_object = {}
+        for key, value in real_dict.items():
+            output_object[key] = self._recursive_to_str__(value)
+        return str(output_object)
+
+
+@dataclass(init=False)
+class BaseList(BaseObjectMixin):
+    status_code: int
+    has_more: bool
+    last_id: str
+    first_id: str
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)

dashscope/common/constants.py

@@ -1,3 +1,5 @@
+# Copyright (c) Alibaba, Inc. and its affiliates.
+
 from http import HTTPStatus
 from pathlib import Path
 
@@ -23,11 +25,21 @@ NEGATIVE_PROMPT = 'negative_prompt'
 HISTORY = 'history'
 CUSTOMIZED_MODEL_ID = 'customized_model_id'
 IMAGES = 'images'
+REFERENCE_VIDEO_URLS = 'reference_video_urls'
 TEXT_EMBEDDING_INPUT_KEY = 'texts'
 SERVICE_503_MESSAGE = 'Service temporarily unavailable, possibly overloaded or not ready.'  # noqa E501
 WEBSOCKET_ERROR_CODE = 44
 SSE_CONTENT_TYPE = 'text/event-stream'
 DEPRECATED_MESSAGE = 'history and auto_history are deprecated for qwen serial models and will be remove in future, use messages'  # noqa E501
+SCENE = 'scene'
+MESSAGE = 'message'
+REQUEST_CONTENT_TEXT = 'text'
+REQUEST_CONTENT_IMAGE = 'image'
+REQUEST_CONTENT_AUDIO = 'audio'
+FILE_PATH_SCHEMA = 'file://'
+
+ENCRYPTION_AES_SECRET_KEY_BYTES = 32
+ENCRYPTION_AES_IV_LENGTH = 12
 
 REPEATABLE_STATUS = [
     HTTPStatus.SERVICE_UNAVAILABLE, HTTPStatus.GATEWAY_TIMEOUT
@@ -36,22 +48,7 @@ REPEATABLE_STATUS = [
 
 class FilePurpose:
     fine_tune = 'fine_tune'
-
-
-class StreamResultMode:
-    """Stream result mode.
-        examples(I like apple)
-            accumulate: will output
-                I
-                I like
-                I like apple
-            divide: will output
-                I
-                like
-                apple
-    """
-    ACCUMULATE = 'accumulate'
-    DIVIDE = 'divide'
+    assistants = 'assistants'
 
 
 class DeploymentStatus:

dashscope/common/env.py

@@ -1,3 +1,5 @@
+# Copyright (c) Alibaba, Inc. and its affiliates.
+
 import os
 
 from dashscope.common.constants import (DASHSCOPE_API_KEY_ENV,

dashscope/common/error.py

@@ -1,61 +1,89 @@
-class AuthenticationError(Exception):
+# Copyright (c) Alibaba, Inc. and its affiliates.
+
+
+class DashScopeException(Exception):
+    pass
+
+
+class AuthenticationError(DashScopeException):
     pass
 
 
-class InvalidParameter(Exception):
+class InvalidParameter(DashScopeException):
     pass
 
 
-class InvalidTask(Exception):
+class InvalidTask(DashScopeException):
     pass
 
 
-class UnsupportedModel(Exception):
+class UnsupportedModel(DashScopeException):
     pass
 
 
-class UnsupportedTask(Exception):
+class UnsupportedTask(DashScopeException):
     pass
 
 
-class ModelRequired(Exception):
+class ModelRequired(DashScopeException):
     pass
 
 
-class InvalidModel(Exception):
+class InvalidModel(DashScopeException):
     pass
 
 
-class InvalidInput(Exception):
+class InvalidInput(DashScopeException):
     pass
 
 
-class InvalidFileFormat(Exception):
+class InvalidFileFormat(DashScopeException):
     pass
 
 
-class UnsupportedApiProtocol(Exception):
+class UnsupportedApiProtocol(DashScopeException):
     pass
 
 
-class NotImplemented(Exception):
+class NotImplemented(DashScopeException):
     pass
 
 
-class MultiInputsWithBinaryNotSupported(Exception):
+class MultiInputsWithBinaryNotSupported(DashScopeException):
     pass
 
 
-class UnexpectedMessageReceived(Exception):
+class UnexpectedMessageReceived(DashScopeException):
     pass
 
 
-class UnsupportedData(Exception):
+class UnsupportedData(DashScopeException):
     pass
 
 
+class AssistantError(DashScopeException):
+    def __init__(self, **kwargs):
+        self.message = None
+        self.code = None
+        self.request_id = None
+        if 'message' in kwargs:
+            import json
+            msg = json.loads(kwargs['message'])
+            if 'request_id' in msg:
+                self.request_id = msg['request_id']
+            if 'code' in msg:
+                self.code = msg['code']
+            if 'message' in msg:
+                self.message = msg['message']
+
+    def __str__(self):
+        msg = 'Request failed, request_id: %s, code: %s, message: %s' % (  # noqa E501
+            self.request_id, self.code, self.message)
+        return msg
+
+
 # for server send generation or inference error.
-class RequestFailure(Exception):
+class RequestFailure(DashScopeException):
     def __init__(self,
                  request_id=None,
                  message=None,
@@ -72,29 +100,37 @@ class RequestFailure(Exception):
         return msg
 
 
-class UnknownMessageReceived(Exception):
+class UnknownMessageReceived(DashScopeException):
+    pass
+
+
+class InputDataRequired(DashScopeException):
+    pass
+
+
+class InputRequired(DashScopeException):
     pass
 
 
-class InputDataRequired(Exception):
+class UnsupportedDataType(DashScopeException):
     pass
 
 
-class InputRequired(Exception):
+class ServiceUnavailableError(DashScopeException):
     pass
 
 
-class UnsupportedDataType(Exception):
+class UnsupportedHTTPMethod(DashScopeException):
     pass
 
 
-class ServiceUnavailableError(Exception):
+class AsyncTaskCreateFailed(DashScopeException):
     pass
 
 
-class UnsupportedHTTPMethod(Exception):
+class UploadFileException(DashScopeException):
     pass
 
 
-class AsyncTaskCreateFailed(Exception):
+class TimeoutException(DashScopeException):
     pass

dashscope/common/logging.py

@@ -1,3 +1,5 @@
+# Copyright (c) Alibaba, Inc. and its affiliates.
+
 import logging
 import os
 

dashscope/common/message_manager.py

@@ -1,3 +1,5 @@
+# Copyright (c) Alibaba, Inc. and its affiliates.
+
 from collections import deque
 from typing import List