Skip to content

GitLab

"docs/git@developer.sourcefind.cn:change/sglang.git" did not exist on "751e5ca2734287e5506e29d69d947e1912d70cdb"
Unverified Commit b0c74884 authored by Xiaomeng Zhao's avatar Xiaomeng Zhao Committed by GitHub
Browse files

Update llm_aided.py

parent 36999491
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mineru/utils/llm_aided.py
View file @ b0c74884
...@@ -90,7 +90,7 @@ Corrected title list: ...@@ -90,7 +90,7 @@ Corrected title list:
content_pieces = [] content_pieces = []
for chunk in completion: for chunk in completion:
if chunk.choices: if chunk.choices:
content_pieces.append(chunk.choices[0].delta.content) content_pieces.append(chunk.choices[0].delta.content.strip())
content = "".join(content_pieces) content = "".join(content_pieces)
# logger.info(f"Title completion: {content}") # logger.info(f"Title completion: {content}")
if "</think>" in content: if "</think>" in content:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment