ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-02-28 20:05:51 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

enhanced yes no

Browse Source
This commit is contained in:
Saifeddine ALOUI 2025-02-08 03:03:49 +01:00
parent 0dfe03fc7b
commit dfda143718
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lollms/personality.py
View File

@ -489,18 +489,18 @@ class AIPersonality:
- If return_explanation is False, returns a boolean (True for 'yes', False for 'no').
- If return_explanation is True, returns a dictionary with the answer and explanation.
"""
prompt = f"{conditionning}\nQuestion: {question}\nContext: {context}\nAnswer strictly with 'yes' or 'no'. If explanation is needed, provide it in a separate field."
prompt = f"{conditionning}\nQuestion: {question}\nContext: {context}\n"
template = """
{
"answer": "yes" | "no",
"answer": true | false,
"explanation": "Optional explanation if return_explanation is True"
}
"""
response = self.generate_code(
prompt=prompt,
template=template if return_explanation else None,
template=template,
language="json",
code_tag_format="markdown",
max_size=max_answer_length,
@ -515,9 +515,9 @@ class AIPersonality:
explanation = parsed_response.get("explanation", "")
if return_explanation:
return {"answer": answer == "yes", "explanation": explanation}
return {"answer": answer, "explanation": explanation}
else:
return answer == "yes"
return answer
except json.JSONDecodeError:
return False