github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-10-27 04:39:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(core): Support simple DB query for sdk (#917)

Browse Source 
Co-authored-by: chengfangyin2 <chengfangyin3@jd.com>
This commit is contained in:
FangYin Cheng
2023-12-11 18:33:54 +08:00
committed by GitHub
parent 43190ca333
commit cbba50ab1b
18 changed files with 467 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
examples/sdk/simple_sdk_llm_example.py
View File

@@ -3,16 +3,18 @@ from dbgpt.core.awel import DAG
from dbgpt.core import BaseOutputParser, OpenAILLM, RequestBuildOperator, PromptTemplate
with DAG("simple_sdk_llm_example_dag") as dag:
prompt = PromptTemplate.from_template(
prompt_task = PromptTemplate.from_template(
"Write a SQL of {dialect} to query all data of {table_name}."
)
req_builder = RequestBuildOperator(model="gpt-3.5-turbo")
llm = OpenAILLM()
out_parser = BaseOutputParser()
prompt >> req_builder >> llm >> out_parser
model_pre_handle_task = RequestBuildOperator(model="gpt-3.5-turbo")
llm_task = OpenAILLM()
out_parse_task = BaseOutputParser()
prompt_task >> model_pre_handle_task >> llm_task >> out_parse_task
if __name__ == "__main__":
output = asyncio.run(
out_parser.call(call_data={"data": {"dialect": "mysql", "table_name": "user"}})
out_parse_task.call(
call_data={"data": {"dialect": "mysql", "table_name": "user"}}
)
)
print(f"output: \n\n{output}")