weiweiw
/
Langchain-Chatchat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix bug : 1、历史最后一条丢失；2、上下两行连读造成歧义 (#415) 

* fix _call bug

* fix

* fix bug : 上下两行连读造成歧义
This commit is contained in:
zhenkaivip 2023-05-20 01:19:22 +08:00 committed by GitHub
parent 18d6b68643
commit 9f5d1afc31
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
chains/local_doc_qa.py
View File

@ -121,7 +121,7 @@ def similarity_search_with_score_by_vector(
else: else:
_id0 = self.index_to_docstore_id[id] _id0 = self.index_to_docstore_id[id]
doc0 = self.docstore.search(_id0) doc0 = self.docstore.search(_id0)
doc.page_content += doc0.page_content doc.page_content += " " + doc0.page_content
if not isinstance(doc, Document): if not isinstance(doc, Document):
raise ValueError(f"Could not find document for id {_id}, got {doc}") raise ValueError(f"Could not find document for id {_id}, got {doc}")
doc_score = min([scores[0][id] for id in [indices[0].tolist().index(i) for i in id_seq if i in indices[0]]]) doc_score = min([scores[0][id] for id in [indices[0].tolist().index(i) for i in id_seq if i in indices[0]]])

10
models/chatglm_llm.py
View File

@ -69,19 +69,17 @@ class ChatGLM(LLM):
history: List[List[str]] = [], history: List[List[str]] = [],
streaming: bool = STREAMING): # -> Tuple[str, List[List[str]]]: streaming: bool = STREAMING): # -> Tuple[str, List[List[str]]]:
if streaming: if streaming:
for inum, (stream_resp, _) in enumerate(self.model.stream_chat( history += [[]]
for stream_resp, _ in self.model.stream_chat(
self.tokenizer, self.tokenizer,
prompt, prompt,
history=history[-self.history_len:-1] if self.history_len > 0 else [], history=history[-self.history_len:-1] if self.history_len > 0 else [],
max_length=self.max_token, max_length=self.max_token,
temperature=self.temperature, temperature=self.temperature,
top_p=self.top_p, top_p=self.top_p,
)): ):
torch_gc() torch_gc()
if inum == 0: history[-1] = [prompt, stream_resp]
history += [[prompt, stream_resp]]
else:
history[-1] = [prompt, stream_resp]
yield stream_resp, history yield stream_resp, history
torch_gc() torch_gc()
else: else: