mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-07-27 05:47:47 +00:00
68 lines
1.8 KiB
Python
68 lines
1.8 KiB
Python
"""Client: Simple Flow CRUD example
|
|
|
|
This example demonstrates how to use the dbgpt client to create, get, update, and
|
|
delete datasource.
|
|
|
|
Example:
|
|
.. code-block:: python
|
|
|
|
DBGPT_API_KEY = "dbgpt"
|
|
client = Client(api_key=DBGPT_API_KEY)
|
|
# 1. Create a flow
|
|
res = await create_datasource(
|
|
client,
|
|
DatasourceModel(
|
|
db_name="dbgpt",
|
|
desc="for client datasource",
|
|
db_type="mysql",
|
|
db_type="mysql",
|
|
db_host="127.0.0.1",
|
|
db_user="root",
|
|
db_pwd="xxxx",
|
|
db_port=3306,
|
|
),
|
|
)
|
|
# 2. Update a flow
|
|
res = await update_datasource(
|
|
client,
|
|
DatasourceModel(
|
|
db_name="dbgpt",
|
|
desc="for client datasource",
|
|
db_type="mysql",
|
|
db_type="mysql",
|
|
db_host="127.0.0.1",
|
|
db_user="root",
|
|
db_pwd="xxxx",
|
|
db_port=3306,
|
|
),
|
|
)
|
|
# 3. Delete a flow
|
|
res = await delete_datasource(client, datasource_id="10")
|
|
# 4. Get a flow
|
|
res = await get_datasource(client, datasource_id="10")
|
|
# 5. List all datasource
|
|
res = await list_datasource(client)
|
|
|
|
"""
|
|
|
|
import asyncio
|
|
|
|
from dbgpt_client import Client
|
|
from dbgpt_client.datasource import list_datasource
|
|
|
|
|
|
async def main():
|
|
# initialize client
|
|
DBGPT_API_KEY = "dbgpt"
|
|
client = Client(api_key=DBGPT_API_KEY)
|
|
try:
|
|
res = await list_datasource(client)
|
|
print(res)
|
|
finally:
|
|
# explicitly close client to avoid event loop closed error
|
|
await client.aclose()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|