junyangwang0410 commited on
Commit
402ae99
·
verified ·
1 Parent(s): 40cc4c5

Update PCAgent/api.py

Browse files
Files changed (1) hide show
  1. PCAgent/api.py +1 -1
PCAgent/api.py CHANGED
@@ -49,7 +49,7 @@ def inference_chat(chat, model, api_url, token):
49
 
50
  response = requests.request("POST", api_url, headers=headers, data=payload)
51
  data = response.json()
52
- print(data)
53
  return data
54
 
55
  # client = OpenAI(
 
49
 
50
  response = requests.request("POST", api_url, headers=headers, data=payload)
51
  data = response.json()
52
+ print(data['choices'][0]['message']['content'])
53
  return data
54
 
55
  # client = OpenAI(