diff --git a/graph/component/categorize.py b/graph/component/categorize.py index d18cab6fd..c9ef19164 100644 --- a/graph/component/categorize.py +++ b/graph/component/categorize.py @@ -43,7 +43,7 @@ class CategorizeParam(GenerateParam): def get_prompt(self): cate_lines = [] for c, desc in self.category_description.items(): - for l in desc["examples"].split("\n"): + for l in desc.get("examples", "").split("\n"): if not l: continue cate_lines.append("Question: {}\tCategory: {}".format(l, c)) descriptions = [] diff --git a/rag/nlp/term_weight.py b/rag/nlp/term_weight.py index 720087acc..34765feaf 100644 --- a/rag/nlp/term_weight.py +++ b/rag/nlp/term_weight.py @@ -107,7 +107,7 @@ class Dealer: if re.match(p, t): tk = "#" break - tk = re.sub(r"([\+\\-])", r"\\\1", tk) + #tk = re.sub(r"([\+\\-])", r"\\\1", tk) if tk != "#" and tk: res.append(tk) return res