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

dashscope/__init__.py

@@ -1,24 +1,40 @@
+# Copyright (c) Alibaba, Inc. and its affiliates.
+
 import logging
 from logging import NullHandler
 
+from dashscope.aigc.code_generation import CodeGeneration
 from dashscope.aigc.conversation import Conversation, History, HistoryItem
-from dashscope.aigc.generation import Generation
+from dashscope.aigc.generation import AioGeneration, Generation
 from dashscope.aigc.image_synthesis import ImageSynthesis
-from dashscope.aigc.multimodal_conversation import MultiModalConversation
+from dashscope.aigc.multimodal_conversation import MultiModalConversation, AioMultiModalConversation
+from dashscope.aigc.video_synthesis import VideoSynthesis
+from dashscope.app.application import Application
+from dashscope.assistants import Assistant, AssistantList, Assistants
+from dashscope.assistants.assistant_types import AssistantFile, DeleteResponse
 from dashscope.audio.asr.transcription import Transcription
 from dashscope.audio.tts.speech_synthesizer import SpeechSynthesizer
 from dashscope.common.api_key import save_api_key
 from dashscope.common.env import (api_key, api_key_file_path,
                                   base_http_api_url, base_websocket_api_url)
-from dashscope.deployment import Deployment
+from dashscope.customize.deployments import Deployments
+from dashscope.customize.finetunes import FineTunes
+from dashscope.embeddings.batch_text_embedding import BatchTextEmbedding
+from dashscope.embeddings.batch_text_embedding_response import \
+    BatchTextEmbeddingResponse
+from dashscope.embeddings.multimodal_embedding import (
+    MultiModalEmbedding, MultiModalEmbeddingItemAudio,
+    MultiModalEmbeddingItemImage, MultiModalEmbeddingItemText, AioMultiModalEmbedding)
 from dashscope.embeddings.text_embedding import TextEmbedding
-from dashscope.embeddings.multimodal_embedding import (MultiModalEmbedding,
-                                                       MultiModalEmbeddingItemAudio,
-                                                       MultiModalEmbeddingItemImage,
-                                                       MultiModalEmbeddingItemText)
-from dashscope.file import File
-from dashscope.finetune import FineTune
-from dashscope.model import Model
+from dashscope.files import Files
+from dashscope.models import Models
+from dashscope.nlp.understanding import Understanding
+from dashscope.rerank.text_rerank import TextReRank
+from dashscope.threads import (MessageFile, Messages, Run, RunList, Runs,
+                               RunStep, RunStepList, Steps, Thread,
+                               ThreadMessage, ThreadMessageList, Threads)
+from dashscope.tokenizers import (Tokenization, Tokenizer, get_tokenizer,
+                                  list_tokenizers)
 
 __all__ = [
     base_http_api_url,
@@ -26,23 +42,54 @@ __all__ = [
     api_key,
     api_key_file_path,
     save_api_key,
+    AioGeneration,
     Conversation,
     Generation,
     History,
     HistoryItem,
     ImageSynthesis,
     Transcription,
-    File,
-    Deployment,
-    FineTune,
-    Model,
+    Files,
+    Deployments,
+    FineTunes,
+    Models,
     TextEmbedding,
     MultiModalEmbedding,
+    AioMultiModalEmbedding,
     MultiModalEmbeddingItemAudio,
     MultiModalEmbeddingItemImage,
     MultiModalEmbeddingItemText,
     SpeechSynthesizer,
     MultiModalConversation,
+    AioMultiModalConversation,
+    BatchTextEmbedding,
+    BatchTextEmbeddingResponse,
+    Understanding,
+    CodeGeneration,
+    Tokenization,
+    Tokenizer,
+    get_tokenizer,
+    list_tokenizers,
+    Application,
+    TextReRank,
+    Assistants,
+    Threads,
+    Messages,
+    Runs,
+    Assistant,
+    ThreadMessage,
+    Run,
+    Steps,
+    AssistantList,
+    ThreadMessageList,
+    RunList,
+    RunStepList,
+    Thread,
+    DeleteResponse,
+    RunStep,
+    MessageFile,
+    AssistantFile,
+    VideoSynthesis,
 ]
 
 logging.getLogger(__name__).addHandler(NullHandler())

dashscope/aigc/__init__.py

@@ -1,13 +1,20 @@
+# Copyright (c) Alibaba, Inc. and its affiliates.
 from .conversation import Conversation, History, HistoryItem
-from .generation import Generation
-from .image_synthesis import ImageSynthesis
-from .multimodal_conversation import MultiModalConversation
+from .generation import Generation, AioGeneration
+from .image_synthesis import ImageSynthesis, AioImageSynthesis
+from .multimodal_conversation import MultiModalConversation, AioMultiModalConversation
+from .video_synthesis import VideoSynthesis, AioVideoSynthesis
 
 __all__ = [
     Generation,
+    AioGeneration,
     Conversation,
     HistoryItem,
     History,
     ImageSynthesis,
+    AioImageSynthesis,
     MultiModalConversation,
+    AioMultiModalConversation,
+    VideoSynthesis,
+    AioVideoSynthesis,
 ]

dashscope/aigc/chat_completion.py

@@ -0,0 +1,282 @@
+# Copyright (c) Alibaba, Inc. and its affiliates.
+
+import json
+from typing import Any, Dict, Generator, List, Union
+
+import dashscope
+from dashscope.aigc.generation import Generation
+from dashscope.api_entities.chat_completion_types import (ChatCompletion,
+                                                          ChatCompletionChunk)
+from dashscope.api_entities.dashscope_response import (GenerationResponse,
+                                                       Message)
+from dashscope.client.base_api import BaseAioApi, CreateMixin
+from dashscope.common.error import InputRequired, ModelRequired
+from dashscope.common.utils import _get_task_group_and_task
+
+
+class Completions(CreateMixin):
+    """Support openai compatible chat completion interface.
+
+    """
+    SUB_PATH = ''
+
+    @classmethod
+    def create(
+        cls,
+        *,
+        model: str,
+        messages: List[Message],
+        stream: bool = False,
+        temperature: float = None,
+        top_p: float = None,
+        top_k: int = None,
+        stop: Union[List[str], List[List[int]]] = None,
+        max_tokens: int = None,
+        repetition_penalty: float = None,
+        api_key: str = None,
+        workspace: str = None,
+        extra_headers: Dict = None,
+        extra_body: Dict = None,
+        **kwargs
+    ) -> Union[ChatCompletion, Generator[ChatCompletionChunk, None, None]]:
+        """Call openai compatible chat completion model service.
+
+        Args:
+            model (str): The requested model, such as qwen-long
+            messages (list): The generation messages.
+                examples:
+                    [{'role': 'user',
+                      'content': 'The weather is fine today.'},
+                      {'role': 'assistant', 'content': 'Suitable for outings'}]
+            stream(bool, `optional`): Enable server-sent events
+                (ref: https://developer.mozilla.org/en-US/docs/Web/API/Server-sent_events/Using_server-sent_events)  # noqa E501
+                the result will back partially[qwen-turbo,bailian-v1].
+            temperature(float, `optional`): Used to control the degree
+                of randomness and diversity. Specifically, the temperature
+                value controls the degree to which the probability distribution
+                of each candidate word is smoothed when generating text.
+                A higher temperature value will reduce the peak value of
+                the probability, allowing more low-probability words to be
+                selected, and the generated results will be more diverse;
+                while a lower temperature value will enhance the peak value
+                of the probability, making it easier for high-probability
+                words to be selected, the generated results are more
+                deterministic.
+            top_p(float, `optional`): A sampling strategy, called nucleus
+                sampling, where the model considers the results of the
+                tokens with top_p probability mass. So 0.1 means only
+                the tokens comprising the top 10% probability mass are
+                considered.
+            top_k(int, `optional`): The size of the sample candidate set when generated.  # noqa E501
+                For example, when the value is 50, only the 50 highest-scoring tokens  # noqa E501
+                in a single generation form a randomly sampled candidate set. # noqa E501
+                The larger the value, the higher the randomness generated;  # noqa E501
+                the smaller the value, the higher the certainty generated. # noqa E501
+                The default value is 0, which means the top_k policy is  # noqa E501
+                not enabled. At this time, only the top_p policy takes effect. # noqa E501
+            stop(list[str] or list[list[int]], `optional`): Used to control the generation to stop  # noqa E501
+                when encountering setting str or token ids, the result will not include # noqa E501
+                stop words or tokens.
+            max_tokens(int, `optional`): The maximum token num expected to be output. It should be # noqa E501
+                noted that the length generated by the model will only be less than max_tokens,  # noqa E501
+                not necessarily equal to it. If max_tokens is set too large, the service will # noqa E501
+                directly prompt that the length exceeds the limit. It is generally # noqa E501
+                not recommended to set this value.
+            repetition_penalty(float, `optional`): Used to control the repeatability when generating models.  # noqa E501
+                Increasing repetition_penalty can reduce the duplication of model generation.  # noqa E501
+                1.0 means no punishment.
+            api_key (str, optional): The api api_key, can be None,
+                if None, will get by default rule.
+            workspace (str, optional): The bailian workspace id.
+            **kwargs:
+                timeout: set request timeout.
+        Raises:
+            InvalidInput: The history and auto_history are mutually exclusive.
+
+        Returns:
+            Union[ChatCompletion,
+                  Generator[ChatCompletionChunk, None, None]]: If
+            stream is True, return Generator, otherwise ChatCompletion.
+        """
+        if messages is None or not messages:
+            raise InputRequired('Messages is required!')
+        if model is None or not model:
+            raise ModelRequired('Model is required!')
+        data = {}
+        data['model'] = model
+        data['messages'] = messages
+        if temperature is not None:
+            data['temperature'] = temperature
+        if top_p is not None:
+            data['top_p'] = top_p
+        if top_k is not None:
+            data['top_k'] = top_k
+        if stop is not None:
+            data['stop'] = stop
+        if max_tokens is not None:
+            data[max_tokens] = max_tokens
+        if repetition_penalty is not None:
+            data['repetition_penalty'] = repetition_penalty
+        if extra_body is not None and extra_body:
+            data = {**data, **extra_body}
+
+        if extra_headers is not None and extra_headers:
+            kwargs = {
+                'headers': extra_headers
+            } if kwargs else {
+                **kwargs,
+                **{
+                    'headers': extra_headers
+                }
+            }
+
+        response = super().call(data=data,
+                                path='chat/completions',
+                                base_address=dashscope.base_compatible_api_url,
+                                api_key=api_key,
+                                flattened_output=True,
+                                stream=stream,
+                                workspace=workspace,
+                                **kwargs)
+        if stream:
+            return (ChatCompletionChunk(**item) for _, item in response)
+        else:
+            return ChatCompletion(**response)
+
+
+class AioGeneration(BaseAioApi):
+    task = 'text-generation'
+    """API for AI-Generated Content(AIGC) models.
+
+    """
+    class Models:
+        """@deprecated, use qwen_turbo instead"""
+        qwen_v1 = 'qwen-v1'
+        """@deprecated, use qwen_plus instead"""
+        qwen_plus_v1 = 'qwen-plus-v1'
+
+        bailian_v1 = 'bailian-v1'
+        dolly_12b_v2 = 'dolly-12b-v2'
+        qwen_turbo = 'qwen-turbo'
+        qwen_plus = 'qwen-plus'
+        qwen_max = 'qwen-max'
+
+    @classmethod
+    async def call(
+        cls,
+        model: str,
+        prompt: Any = None,
+        history: list = None,
+        api_key: str = None,
+        messages: List[Message] = None,
+        plugins: Union[str, Dict[str, Any]] = None,
+        workspace: str = None,
+        **kwargs
+    ) -> Union[GenerationResponse, Generator[GenerationResponse, None, None]]:
+        """Call generation model service.
+
+        Args:
+            model (str): The requested model, such as qwen-turbo
+            prompt (Any): The input prompt.
+            history (list):The user provided history, deprecated
+                examples:
+                    [{'user':'The weather is fine today.',
+                    'bot': 'Suitable for outings'}].
+                Defaults to None.
+            api_key (str, optional): The api api_key, can be None,
+                if None, will get by default rule(TODO: api key doc).
+            messages (list): The generation messages.
+                examples:
+                    [{'role': 'user',
+                      'content': 'The weather is fine today.'},
+                      {'role': 'assistant', 'content': 'Suitable for outings'}]
+            plugins (Any): The plugin config. Can be plugins config str, or dict.
+            **kwargs:
+                stream(bool, `optional`): Enable server-sent events
+                    (ref: https://developer.mozilla.org/en-US/docs/Web/API/Server-sent_events/Using_server-sent_events)  # noqa E501
+                    the result will back partially[qwen-turbo,bailian-v1].
+                temperature(float, `optional`): Used to control the degree
+                    of randomness and diversity. Specifically, the temperature
+                    value controls the degree to which the probability distribution
+                    of each candidate word is smoothed when generating text.
+                    A higher temperature value will reduce the peak value of
+                    the probability, allowing more low-probability words to be
+                    selected, and the generated results will be more diverse;
+                    while a lower temperature value will enhance the peak value
+                    of the probability, making it easier for high-probability
+                    words to be selected, the generated results are more
+                    deterministic, range(0, 2) .[qwen-turbo,qwen-plus].
+                top_p(float, `optional`): A sampling strategy, called nucleus
+                    sampling, where the model considers the results of the
+                    tokens with top_p probability mass. So 0.1 means only
+                    the tokens comprising the top 10% probability mass are
+                    considered[qwen-turbo,bailian-v1].
+                top_k(int, `optional`): The size of the sample candidate set when generated.  # noqa E501
+                    For example, when the value is 50, only the 50 highest-scoring tokens  # noqa E501
+                    in a single generation form a randomly sampled candidate set. # noqa E501
+                    The larger the value, the higher the randomness generated;  # noqa E501
+                    the smaller the value, the higher the certainty generated. # noqa E501
+                    The default value is 0, which means the top_k policy is  # noqa E501
+                    not enabled. At this time, only the top_p policy takes effect. # noqa E501
+                enable_search(bool, `optional`): Whether to enable web search(quark).  # noqa E501
+                    Currently works best only on the first round of conversation.
+                    Default to False, support model: [qwen-turbo].
+                customized_model_id(str, required) The enterprise-specific
+                    large model id, which needs to be generated from the
+                    operation background of the enterprise-specific
+                    large model product, support model: [bailian-v1].
+                result_format(str, `optional`): [message|text] Set result result format. # noqa E501
+                    Default result is text
+                incremental_output(bool, `optional`): Used to control the streaming output mode. # noqa E501
+                    If true, the subsequent output will include the previously input content. # noqa E501
+                    Otherwise, the subsequent output will not include the previously output # noqa E501
+                    content. Default false.
+                stop(list[str] or list[list[int]], `optional`): Used to control the generation to stop  # noqa E501
+                    when encountering setting str or token ids, the result will not include # noqa E501
+                    stop words or tokens.
+                max_tokens(int, `optional`): The maximum token num expected to be output. It should be # noqa E501
+                    noted that the length generated by the model will only be less than max_tokens,  # noqa E501
+                    not necessarily equal to it. If max_tokens is set too large, the service will # noqa E501
+                    directly prompt that the length exceeds the limit. It is generally # noqa E501
+                    not recommended to set this value.
+                repetition_penalty(float, `optional`): Used to control the repeatability when generating models.  # noqa E501
+                    Increasing repetition_penalty can reduce the duplication of model generation.  # noqa E501
+                    1.0 means no punishment.
+            workspace (str): The dashscope workspace id.
+        Raises:
+            InvalidInput: The history and auto_history are mutually exclusive.
+
+        Returns:
+            Union[GenerationResponse,
+                  Generator[GenerationResponse, None, None]]: If
+            stream is True, return Generator, otherwise GenerationResponse.
+        """
+        if (prompt is None or not prompt) and (messages is None
+                                               or not messages):
+            raise InputRequired('prompt or messages is required!')
+        if model is None or not model:
+            raise ModelRequired('Model is required!')
+        task_group, function = _get_task_group_and_task(__name__)
+        if plugins is not None:
+            headers = kwargs.pop('headers', {})
+            if isinstance(plugins, str):
+                headers['X-DashScope-Plugin'] = plugins
+            else:
+                headers['X-DashScope-Plugin'] = json.dumps(plugins)
+            kwargs['headers'] = headers
+        input, parameters = Generation._build_input_parameters(
+            model, prompt, history, messages, **kwargs)
+        response = await super().call(model=model,
+                                      task_group=task_group,
+                                      task=Generation.task,
+                                      function=function,
+                                      api_key=api_key,
+                                      input=input,
+                                      workspace=workspace,
+                                      **parameters)
+        is_stream = kwargs.get('stream', False)
+        if is_stream:
+            return (GenerationResponse.from_api_response(rsp)
+                    async for rsp in response)
+        else:
+            return GenerationResponse.from_api_response(response)

dashscope/aigc/code_generation.py

@@ -0,0 +1,145 @@
+# Copyright (c) Alibaba, Inc. and its affiliates.
+
+from typing import Generator, List, Union
+
+from dashscope.api_entities.dashscope_response import (DashScopeAPIResponse,
+                                                       DictMixin, Role)
+from dashscope.client.base_api import BaseApi
+from dashscope.common.constants import MESSAGE, SCENE
+from dashscope.common.error import InputRequired, ModelRequired
+from dashscope.common.utils import _get_task_group_and_task
+
+
+class MessageParam(DictMixin):
+    role: str
+
+    def __init__(self, role: str, **kwargs):
+        super().__init__(role=role, **kwargs)
+
+
+class UserRoleMessageParam(MessageParam):
+    content: str
+
+    def __init__(self, content: str, **kwargs):
+        super().__init__(role=Role.USER, content=content, **kwargs)
+
+
+class AttachmentRoleMessageParam(MessageParam):
+    meta: dict
+
+    def __init__(self, meta: dict, **kwargs):
+        super().__init__(role=Role.ATTACHMENT, meta=meta, **kwargs)
+
+
+class OtherRoleContentMessageParam(MessageParam):
+    content: str
+
+    def __init__(self, role: str, content: str, **kwargs):
+        super().__init__(role=role, content=content, **kwargs)
+
+
+class OtherRoleMetaMessageParam(MessageParam):
+    meta: dict
+
+    def __init__(self, role: str, meta: dict, **kwargs):
+        super().__init__(role=role, meta=meta, **kwargs)
+
+
+class CodeGeneration(BaseApi):
+    function = 'generation'
+    """API for AI-Generated Content(AIGC) models.
+
+    """
+    class Models:
+        tongyi_lingma_v1 = 'tongyi-lingma-v1'
+
+    class Scenes:
+        custom = 'custom'
+        nl2code = 'nl2code'
+        code2comment = 'code2comment'
+        code2explain = 'code2explain'
+        commit2msg = 'commit2msg'
+        unit_test = 'unittest'
+        code_qa = 'codeqa'
+        nl2sql = 'nl2sql'
+
+    @classmethod
+    def call(
+        cls,
+        model: str,
+        scene: str = None,
+        api_key: str = None,
+        message: List[MessageParam] = None,
+        workspace: str = None,
+        **kwargs
+    ) -> Union[DashScopeAPIResponse, Generator[DashScopeAPIResponse, None,
+                                               None]]:
+        """Call generation model service.
+
+        Args:
+            model (str): The requested model, such as tongyi-lingma-v1
+            scene (str): Scene type, single choice, such as custom
+                examples:
+                    custom：User defined prompt
+                    nl2code：Natural language generated code
+                    code2comment：annotation
+                    code2explain：explain
+                    commit2msg：Automatically generate commit
+                    uinttest：Generating Unit Tests
+                    codeqa：Code Q&A
+                    nl2sql：Generate SQL code using natural language
+            api_key (str, optional): The api api_key, can be None,
+                if None, will get by default rule(TODO: api key doc).
+            message (list): The generation messages.
+                scene == custom, examples:
+                    [{"role": "user", "content": "根据下面的功能描述生成一个python函数。代码的功能是计算给定路径下所有文件的总大小。"}]  # noqa E501
+                scene == nl2code, examples:
+                    [{"role": "user", "content": "计算给定路径下所有文件的总大小"}, {"role": "attachment", "meta": {"language": "java"}}]  # noqa E501
+                scene == code2comment, examples:
+                    [{"role": "user", "content": "1. 生成中文注释\n2. 仅生成代码部分，不需要额外解释函数功能\n"}, {"role": "attachment", "meta": {"code": "\t\t@Override\n\t\tpublic  CancelExportTaskResponse  cancelExportTask(\n\t\t\t\tCancelExportTask  cancelExportTask)  {\n\t\t\tAmazonEC2SkeletonInterface  ec2Service  =  ServiceProvider.getInstance().getServiceImpl(AmazonEC2SkeletonInterface.class);\n\t\t\treturn  ec2Service.cancelExportTask(cancelExportTask);\n\t\t}", "language": "java"}}]  # noqa E501
+                scene == code2explain, examples:
+                    [{"role": "user", "content": "要求不低于200字"}, {"role": "attachment", "meta": {"code": "@Override\n                                public  int  getHeaderCacheSize()\n                                {\n                                        return  0;\n                                }\n\n", "language": "java"}}]  # noqa E501
+                scene == commit2msg, examples:
+                    [{"role": "attachment", "meta": {"diff_list": [{"diff": "--- src/com/siondream/core/PlatformResolver.java\n+++ src/com/siondream/core/PlatformResolver.java\n@@ -1,11 +1,8 @@\npackage com.siondream.core;\n-\n-import com.badlogic.gdx.files.FileHandle;\n\npublic interface PlatformResolver {\npublic void openURL(String url);\npublic void rateApp();\npublic void sendFeedback();\n-\tpublic FileHandle[] listFolder(String path);\n}\n", "old_file_path": "src/com/siondream/core/PlatformResolver.java", "new_file_path": "src/com/siondream/core/PlatformResolver.java"}]}}]  # noqa E501
+                scene == unittest, examples:
+                    [{"role": "attachment", "meta": {"code": "public static <T> TimestampMap<T> parseTimestampMap(Class<T> typeClass, String input, DateTimeZone timeZone) throws IllegalArgumentException {\n        if (typeClass == null) {\n            throw new IllegalArgumentException(\"typeClass required\");\n        }\n\n        if (input == null) {\n            return null;\n        }\n\n        TimestampMap result;\n\n        typeClass = AttributeUtils.getStandardizedType(typeClass);\n        if (typeClass.equals(String.class)) {\n            result = new TimestampStringMap();\n        } else if (typeClass.equals(Byte.class)) {\n            result = new TimestampByteMap();\n        } else if (typeClass.equals(Short.class)) {\n            result = new TimestampShortMap();\n        } else if (typeClass.equals(Integer.class)) {\n            result = new TimestampIntegerMap();\n        } else if (typeClass.equals(Long.class)) {\n            result = new TimestampLongMap();\n        } else if (typeClass.equals(Float.class)) {\n            result = new TimestampFloatMap();\n        } else if (typeClass.equals(Double.class)) {\n            result = new TimestampDoubleMap();\n        } else if (typeClass.equals(Boolean.class)) {\n            result = new TimestampBooleanMap();\n        } else if (typeClass.equals(Character.class)) {\n            result = new TimestampCharMap();\n        } else {\n            throw new IllegalArgumentException(\"Unsupported type \" + typeClass.getClass().getCanonicalName());\n        }\n\n        if (input.equalsIgnoreCase(EMPTY_VALUE)) {\n            return result;\n        }\n\n        StringReader reader = new StringReader(input + ' ');// Add 1 space so\n                                                            // reader.skip\n                                                            // function always\n                                                            // works when\n                                                            // necessary (end of\n                                                            // string not\n                                                            // reached).\n\n        try {\n            int r;\n            char c;\n            while ((r = reader.read()) != -1) {\n                c = (char) r;\n                switch (c) {\n                    case LEFT_BOUND_SQUARE_BRACKET:\n                    case LEFT_BOUND_BRACKET:\n                        parseTimestampAndValue(typeClass, reader, result, timeZone);\n                        break;\n                    default:\n                        // Ignore other chars outside of bounds\n                }\n            }\n        } catch (IOException ex) {\n            throw new RuntimeException(\"Unexpected expection while parsing timestamps\", ex);\n        }\n\n        return result;\n    }", "language": "java"}}]  # noqa E501
+                scene == codeqa, examples:
+                    [{"role": "user", "content": "I'm writing a small web server in Python, using BaseHTTPServer and a custom subclass of BaseHTTPServer.BaseHTTPRequestHandler. Is it possible to make this listen on more than one port?\nWhat I'm doing now:\nclass MyRequestHandler(BaseHTTPServer.BaseHTTPRequestHandler):\n  def doGET\n  [...]\n\nclass ThreadingHTTPServer(ThreadingMixIn, HTTPServer): \n    pass\n\nserver = ThreadingHTTPServer(('localhost', 80), MyRequestHandler)\nserver.serve_forever()"}]  # noqa E501
+                scene == nl2sql, examples:
+                    [{"role": "user", "content": "小明的总分数是多少"}, {"role": "attachment", "meta": {"synonym_infos": {"学生姓名": "姓名|名字|名称", "学生分数": "分数|得分"}, "recall_infos": [{"content": "student_score.id='小明'", "score": "0.83"}], "schema_infos": [{"table_id": "student_score", "table_desc": "学生分数表", "columns": [{"col_name": "id", "col_caption": "学生id", "col_desc": "例值为:1,2,3", "col_type": "string"}, {"col_name": "name", "col_caption": "学生姓名", "col_desc": "例值为:张三,李四,小明", "col_type": "string"}, {"col_name": "score", "col_caption": "学生分数", "col_desc": "例值为:98,100,66", "col_type": "string"}]}]}}]  # noqa E501
+            workspace (str): The dashscope workspace id.
+            **kwargs:
+                n(int, `optional`): The number of output results, currently only supports 1, with a default value of 1  # noqa E501
+
+        Returns:
+            Union[DashScopeAPIResponse,
+                  Generator[DashScopeAPIResponse, None, None]]: If
+            stream is True, return Generator, otherwise DashScopeAPIResponse.
+        """
+        if (scene is None or not scene) or (message is None or not message):
+            raise InputRequired('scene and message is required!')
+        if model is None or not model:
+            raise ModelRequired('Model is required!')
+        task_group, task = _get_task_group_and_task(__name__)
+        input, parameters = cls._build_input_parameters(
+            model, scene, message, **kwargs)
+        response = super().call(model=model,
+                                task_group=task_group,
+                                task=task,
+                                function=CodeGeneration.function,
+                                api_key=api_key,
+                                input=input,
+                                workspace=workspace,
+                                **parameters)
+
+        is_stream = kwargs.get('stream', False)
+        if is_stream:
+            return (rsp for rsp in response)
+        else:
+            return response
+
+    @classmethod
+    def _build_input_parameters(cls, model, scene, message, **kwargs):
+        parameters = {'n': kwargs.pop('n', 1)}
+        input = {SCENE: scene, MESSAGE: message}
+        return input, {**parameters, **kwargs}