hadadrjt commited on
Commit
668b4c5
·
1 Parent(s): b48942e

Revert "ai: Simply outputs logic."

Browse files

This reverts commit 047f8a1dbecb358df0f0b07242da34a35dc596eb.

Files changed (1) hide show
  1. jarvis.py +1 -1
jarvis.py CHANGED
@@ -153,7 +153,7 @@ async def fetch_response_async(host, provider_key, selected_model, messages, mod
153
  response = await client.post(f"{host}", json={**data, "extra_body": extra, "session_id": session_id}, headers={"Authorization": f"Bearer {provider_key}"})
154
  response.raise_for_status()
155
  resp_json = response.json()
156
- ai_text = resp_json.get("choices", [{}])[0].get("message", {}).get("content", RESPONSES["RESPONSE_2"])
157
  return process_ai_response(ai_text)
158
  except Exception:
159
  marked_item(provider_key, LINUX_SERVER_PROVIDER_KEYS_MARKED, LINUX_SERVER_PROVIDER_KEYS_ATTEMPTS)
 
153
  response = await client.post(f"{host}", json={**data, "extra_body": extra, "session_id": session_id}, headers={"Authorization": f"Bearer {provider_key}"})
154
  response.raise_for_status()
155
  resp_json = response.json()
156
+ ai_text = resp_json["choices"][0]["message"]["content"] if resp_json.get("choices") and resp_json["choices"][0].get("message") and resp_json["choices"][0]["message"].get("content") else RESPONSES["RESPONSE_2"]
157
  return process_ai_response(ai_text)
158
  except Exception:
159
  marked_item(provider_key, LINUX_SERVER_PROVIDER_KEYS_MARKED, LINUX_SERVER_PROVIDER_KEYS_ATTEMPTS)