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

refactor: Refactor for core SDK (#1092)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-01-21 09:57:57 +08:00
committed by GitHub
parent ba7248adbb
commit 2d905191f8
45 changed files with 236 additions and 133 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/sdk/simple_sdk_llm_example.py
View File

@@ -7,7 +7,7 @@ from dbgpt.core.operator import (
PromptBuilderOperator,
RequestBuilderOperator,
)
from dbgpt.model import OpenAILLMClient
from dbgpt.model.proxy import OpenAILLMClient
with DAG("simple_sdk_llm_example_dag") as dag:
prompt_task = PromptBuilderOperator(
@@ -20,8 +20,6 @@ with DAG("simple_sdk_llm_example_dag") as dag:
if __name__ == "__main__":
output = asyncio.run(
out_parse_task.call(
call_data={"data": {"dialect": "mysql", "table_name": "user"}}
)
out_parse_task.call(call_data={"dialect": "mysql", "table_name": "user"})
)
print(f"output: \n\n{output}")