wehub-resource-sync
d13100ebf3
Update draft releases / main (push) Has been cancelled
Build and push docs image / build-image (push) Has been cancelled
Build Web Application / build-web (macos-latest) (push) Has been cancelled
Build Web Application / build-web (ubuntu-latest) (push) Has been cancelled
Python Code Quality Checks / build (push) Has been cancelled
Test Python / test-python (macos-latest, 3.10) (push) Has been cancelled
Test Python / test-python (macos-latest, 3.11) (push) Has been cancelled
Test Python / test-python (ubuntu-latest, 3.10) (push) Has been cancelled
Test Python / test-python (ubuntu-latest, 3.11) (push) Has been cancelled
26 lines
824 B
Python
26 lines
824 B
Python
import asyncio
|
|
|
|
from dbgpt.core import BaseOutputParser
|
|
from dbgpt.core.awel import DAG
|
|
from dbgpt.core.operators import (
|
|
BaseLLMOperator,
|
|
PromptBuilderOperator,
|
|
RequestBuilderOperator,
|
|
)
|
|
from dbgpt.model.proxy import OpenAILLMClient
|
|
|
|
with DAG("simple_sdk_llm_example_dag") as dag:
|
|
prompt_task = PromptBuilderOperator(
|
|
"Write a SQL of {dialect} to query all data of {table_name}."
|
|
)
|
|
model_pre_handle_task = RequestBuilderOperator(model="gpt-3.5-turbo")
|
|
llm_task = BaseLLMOperator(OpenAILLMClient())
|
|
out_parse_task = BaseOutputParser()
|
|
prompt_task >> model_pre_handle_task >> llm_task >> out_parse_task
|
|
|
|
if __name__ == "__main__":
|
|
output = asyncio.run(
|
|
out_parse_task.call(call_data={"dialect": "mysql", "table_name": "user"})
|
|
)
|
|
print(f"output: \n\n{output}")
|