Pārlūkot izejas kodu

refine token similarity (#2824)

### What problem does this PR solve?


### Type of change

- [x] Performance Improvement
tags/v0.13.0
Kevin Hu pirms 1 gada
vecāks
revīzija
b164116277
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
2 mainītis faili ar 3 papildinājumiem un 6 dzēšanām
  1. 1
    1
      agent/canvas.py
  2. 2
    5
      rag/nlp/query.py

+ 1
- 1
agent/canvas.py Parādīt failu

@@ -262,7 +262,7 @@ class Canvas(ABC):
convs = []
for role, obj in self.history[(window_size + 1) * -1:]:
convs.append({"role": role, "content": (obj if role == "user" else
'\n'.join(pd.DataFrame(obj)['content']))})
'\n'.join([str(s) for s in pd.DataFrame(obj)['content']]))})
return convs

def add_user_input(self, question):

+ 2
- 5
rag/nlp/query.py Parādīt failu

@@ -186,8 +186,5 @@ class EsQueryer:
s += v # * dtwt[k]
q = 1e-9
for k, v in qtwt.items():
q += v # * v
#d = 1e-9
# for k, v in dtwt.items():
# d += v * v
return s / q / max(1, math.sqrt(math.log10(max(len(qtwt.keys()), len(dtwt.keys())))))# math.sqrt(q) / math.sqrt(d)
q += v
return s / q

Notiek ielāde…
Atcelt
Saglabāt