Фикс регулярки

lib/llm_3_deepinfra.py

@@ -16,19 +16,12 @@ level = 'Продукт'
 level = 'Проблема'
 level = 'Проблема'
 
-def getType(str, level) :
+def getType(str, level):
     type = ''
-    t = level + ': [^>]+'
-    # t = r'Сектор\: \w+/\w+ >'
-    m = re.search(t, str)
-    if m :
-        substr = str[m.start() : m.end()]
-        substr
-        m = re.search('\: [\w ]+', substr)
-        subsubstr = substr[m.start() + 2 : ]
-        # subsubstr
-        type = subsubstr.strip()
-        
+    pattern = rf'{level}: ([^>]+)(?: >|$)'
+    m = re.search(pattern, str)
+    if m:
+        type = m.group(1).strip()
     return type
 
 def getQuestionAnswer(str) :