improved on sops generations on questions and added bot suggestion

This commit is contained in:
2024-09-16 23:38:33 +00:00
parent cd8f499f97
commit 47a274741f
10 changed files with 228 additions and 21 deletions
+4 -1
View File
@@ -22,6 +22,7 @@ class QuestionsGenerator:
assessment_type = input_data['assessment_type']
frequency_type = input_data['frequency_type']
total_duration = input_data['duration']
roles_data = input_data["roles_data"]
# Chunk the SOPs into smaller pieces
chunk_size = 1000 # Define your chunk size
@@ -47,7 +48,8 @@ class QuestionsGenerator:
{"role": "user", "content": f"Assessment Type: {assessment_type}"},
{"role": "user", "content": f"Frequency Type: {frequency_type}"},
{"role": "user", "content": f"Current Frequency Number to generate : {frequency_label}"},
{"role": "user", "content": f"Duration: {total_duration}"}
{"role": "user", "content": f"Duration: {total_duration}"},
{"role": "user", "content": f"roles data : {roles_data}"}
],
response_format=AssementQuestion, # Use the updated response schema
max_tokens=4096,
@@ -55,6 +57,7 @@ class QuestionsGenerator:
)
questions = json.loads(response.choices[0].message.content)
print(f"Generated questions: {questions}")
all_questions.append({
"frequency_number": frequency_label,
"questions": questions