llm_manager/llm_parser.py

import re

from form.form import work_categories


class LlmParser:
 _verification_prompt_answers_regex = re.compile(r"\|\s*([^|]*)\s?", re.MULTILINE)

 @classmethod
 def parse_verification_prompt_answers(cls, llm_answer) -> dict[int, str | None]:
 print(f"llm answer: {llm_answer}")
 expected_answers_count = 13
 answers = {}
 i = 0
 question_id = 0
 lines = [l for l in llm_answer.split("\n") if len(l.strip()) > 0]
 while i < len(lines):
 line = lines[i].strip()
 if len(line) == 0:
 i += 1
 elif line.endswith("?") and i+1<len(lines):
 i+=1
 elif "null" in lines[i]:
 answers[question_id] = None
 i += 1
 question_id += 1
 elif ":" in lines[i]:
 answers[question_id] = line.split(":")[1]
 i += 1
 question_id += 1
 else:
 answers[question_id] = line
 i+=1
 question_id += 1
 return answers




 @classmethod
 def parse_get_categories_answer(cls, category_answer) -> list[str]:
 categories = []
 for category in category_answer.split(";"):
 categories.extend([k for k, v in work_categories.items() if category in v])
 return categories