add type annotation

1 year ago · 2f6bfe8e30
parent 4f5a4e7194
commit 2f6bfe8e30
1 changed files with 1 additions and 1 deletions
--- a/api/core/rag/datasource/keyword/mecab/mecab_keyword_table_handler.py
+++ b/api/core/rag/datasource/keyword/mecab/mecab_keyword_table_handler.py
@ -61,7 +61,7 @@ class MeCabKeywordTableHandler:
            node = self.tagger.parseToNode(text)

            # Calculate term frequencies and scores
-            term_scores = defaultdict(float)
+            term_scores: defaultdict[str, float] = defaultdict(float)
            while node:
                features = node.feature.split(",")
                if len(features) > 0: