快速开始

AWEL开发快速上手, 下面是一个最基本的开始教程。首先我们需要安装对应的包与依赖。

pip install dbgpt --upgrade
pip install openai

创建一个Python文件 simple_sdk_llm_example_dag.py 并写下下面的代码

from dbgpt.core import BaseOutputParser
from dbgpt.core.awel import DAG
from dbgpt.core.operators import (
    PromptBuilderOperator,
    RequestBuilderOperator
)
from dbgpt.model.proxy import OpenAILLMClient
from dbgpt.model.operators import LLMOperator
with DAG("simple_sdk_llm_example_dag") as dag:
    prompt_task = PromptBuilderOperator(
        "Write a SQL of {dialect} to query all dat of {table_name}"
    )
    model_pre_handle_task = RequestBuilderOperator(model="gpt-3.5-turbo")
    llm_task = LLMOperator(OpenAILLMClient())
    out_parse_task = BaseOutputParser()
    print(out_parse_task)
    prompt_task >> model_pre_handle_task >> llm_task >> out_parse_task

提供OpenAI key与接口地址

export OPENAI_API_KEY=sk-xx
export OPENAI_API_BASE=https://xx:80/v1

运行脚本进行测试

python simple_sdk_llm_example_dag.py

到这里，说明你已经掌握了AWEL最基本的用法。更多的使用案例，可以参考: 更多案例