more fixes

lemmatizer.py

@@ -29,7 +29,7 @@ class RussianLemmatizer:
             """Компонент для исправления лемм терминов и их форм"""
             for token in doc:
                 lemma_lower = token.lemma_.lower()
-                canonical = VOCABULARY_MANAGER.extract_stem(lemma_lower)
+                canonical = VOCABULARY_MANAGER.find_word_in_terms(lemma_lower)
                 if canonical:
                     token.lemma_ = canonical.lower()
             return doc

vocabulary/parse_vocabulary.py

@@ -107,6 +107,16 @@ class ParserVocabulary:
         self.vocabulary[pattern.lemma().lower()] = definition
         self.patterns.append(pattern)
         
+    def find_word_in_terms(self, word: str) -> Optional[str]:
+        """Проверяет, является ли слово термином (учитывая словоформы).
+        Если да, возвращает лемму термина, иначе None.
+        """
+        word = word.lower()
+        for pattern in self.patterns:
+            if pattern.matches(word):
+                return pattern.lemma().lower()
+        return None
+    
     def find_terms(self, text: str) -> set[str]:
         """Возвращает список терминов в тексте (учитывая словоформы)
         
@@ -122,13 +132,9 @@ class ParserVocabulary:
         words = re.findall(r'\b\w+(?:[.-]\w+)*\b', text.lower())
         
         for word in words:
-            # проверяем по паттернам
-            for pattern in self.patterns:
-                if pattern.matches(word):
-                    lemma = pattern.lemma().lower()
-                    assert lemma in self.vocabulary, breakpoint()
-                    found_lemmas.add(lemma)
-                    break  # Слово найдено, переходим к следующему
+            res = self.find_word_in_terms(word)
+            if res:
+                found_lemmas.add(res)
         return found_lemmas
 
     def wrap_prompt(self, retrieved_text: str, query_text: str):