github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-02 08:40:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat:chat_history add model_name

Browse Source
This commit is contained in:
aries_ckt 2023-09-11 21:38:08 +08:00
parent f358e1c8e5
commit 8320dc3cef
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pilot/openapi/api_v1/api_v1.py
View File

@ -340,11 +340,11 @@ async def chat_completions(dialogue: ConversationVo = Body()):
async def model_types(): async def model_types():
print(f"/controller/model/types") print(f"/controller/model/types")
try: try:
import requests import httpx
async with httpx.AsyncClient() as client:
response = requests.get( response = await client.get(
f"{CFG.MODEL_SERVER}/api/controller/models?healthy_only=true" f"{CFG.MODEL_SERVER}/api/controller/models?healthy_only=true",
) )
types = set() types = set()
if response.status_code == 200: if response.status_code == 200:
models = json.loads(response.text) models = json.loads(response.text)