github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-20 19:09:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(examples): update client usage to prefer await aclose() in async context" (#2839)

Browse Source
This commit is contained in:
huhao0926 2025-07-13 16:58:46 +08:00 committed by GitHub
parent 0ecfeabd8e
commit f364d10fb1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
6 changed files with 44 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
examples/client/app_crud_example.py
View File

@ -22,8 +22,12 @@ async def main():
# initialize client # initialize client
DBGPT_API_KEY = "dbgpt" DBGPT_API_KEY = "dbgpt"
client = Client(api_key=DBGPT_API_KEY) client = Client(api_key=DBGPT_API_KEY)
res = await list_app(client) try:
print(res) res = await list_app(client)
print(res)
finally:
# explicitly close client to avoid event loop closed error
await client.aclose()
if __name__ == "__main__": if __name__ == "__main__":

8
examples/client/client_chat_example.py
View File

@ -57,12 +57,16 @@ async def main():
# initialize client # initialize client
DBGPT_API_KEY = "dbgpt" DBGPT_API_KEY = "dbgpt"
client = Client(api_key=DBGPT_API_KEY) client = Client(api_key=DBGPT_API_KEY)
data = await client.chat(model="Qwen2.5-72B-Instruct", messages="hello") try:
data = await client.chat(model="Qwen2.5-72B-Instruct", messages="hello")
print(data)
finally:
# explicitly close client to avoid event loop closed error
await client.aclose()
# async for data in client.chat_stream( # async for data in client.chat_stream(
# model="chatgpt_proxyllm", # model="chatgpt_proxyllm",
# messages="hello", # messages="hello",
# ): # ):
print(data)
# res = await client.chat(model="chatgpt_proxyllm" ,messages="hello") # res = await client.chat(model="chatgpt_proxyllm" ,messages="hello")
# print(res) # print(res)

8
examples/client/client_evaluation.py
View File

@ -83,8 +83,12 @@ async def main():
} }
], ],
) )
data = await run_evaluation(client, request=request) try:
print(data) data = await run_evaluation(client, request=request)
print(data)
finally:
# explicitly close client to avoid event loop closed error
await client.aclose()
if __name__ == "__main__": if __name__ == "__main__":

8
examples/client/datasource_crud_example.py
View File

@ -55,8 +55,12 @@ async def main():
# initialize client # initialize client
DBGPT_API_KEY = "dbgpt" DBGPT_API_KEY = "dbgpt"
client = Client(api_key=DBGPT_API_KEY) client = Client(api_key=DBGPT_API_KEY)
res = await list_datasource(client) try:
print(res) res = await list_datasource(client)
print(res)
finally:
# explicitly close client to avoid event loop closed error
await client.aclose()
if __name__ == "__main__": if __name__ == "__main__":

8
examples/client/flow_crud_example.py
View File

@ -37,8 +37,12 @@ async def main():
# initialize client # initialize client
DBGPT_API_KEY = "dbgpt" DBGPT_API_KEY = "dbgpt"
client = Client(api_key=DBGPT_API_KEY) client = Client(api_key=DBGPT_API_KEY)
res = await list_flow(client) try:
print(res) res = await list_flow(client)
print(res)
finally:
# explicitly close client to avoid event loop closed error
await client.aclose()
if __name__ == "__main__": if __name__ == "__main__":

24
examples/client/knowledge_crud_example.py
View File

@ -75,16 +75,20 @@ async def main():
DBGPT_API_KEY = "dbgpt" DBGPT_API_KEY = "dbgpt"
client = Client(api_key=DBGPT_API_KEY) client = Client(api_key=DBGPT_API_KEY)
res = await create_space( try:
client, res = await create_space(
SpaceModel( client,
name="test_space_1", SpaceModel(
vector_type="Chroma", name="test_space_1",
desc="for client space desc", vector_type="Chroma",
owner="dbgpt", desc="for client space desc",
), owner="dbgpt",
) ),
print(res) )
print(res)
finally:
# explicitly close client to avoid event loop closed error
await client.aclose()
# list all spaces # list all spaces
# res = await list_space(client) # res = await list_space(client)