Lisa_Stuff/Novel
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

pushed

Browse Source
This commit is contained in:
root 2025-03-24 20:18:56 +00:00
parent 90c5ab3465
commit 57d4f64989
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/integration/AI.py
View File

@ -62,7 +62,7 @@ def generate_response(data):
response = requests.post(generate_endpoint, json=payload, headers=headers) response = requests.post(generate_endpoint, json=payload, headers=headers)
response.raise_for_status() response.raise_for_status()
if use_groq: if use_groq:
return response.json()["choices"][0]["message"]["content"] return response.json()["choices"][0]["message"]["content"] + f"\nAI Model: {generate_model}"
else: else:
return response.json() return response.json()
@ -71,7 +71,7 @@ def generate_response(data):
Log.e(f"Using model: {generate_model}") Log.e(f"Using model: {generate_model}")
error_messages.append(f"Model {generate_model} failed: {e}") error_messages.append(f"Model {generate_model} failed: {e}")
return None return None
return f"All models failed. Errors: {error_messages}" return f"All models failed to generate response. Errors: {error_messages}"
def ai_analyse(src): def ai_analyse(src):