Skip to content

GitLab

Unverified Commit 0530f599 authored by Xiaomeng Zhao's avatar Xiaomeng Zhao Committed by GitHub
Browse files

Update mineru/utils/llm_aided.py 


Co-authored-by: default avatarCopilot <175728472+Copilot@users.noreply.github.com>
parent b0c74884
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mineru/utils/llm_aided.py
View file @ 0530f599
......@@ -90,8 +90,8 @@ Corrected title list:
content_pieces = []
for chunk in completion:
if chunk.choices:
content_pieces.append(chunk.choices[0].delta.content.strip())
content = "".join(content_pieces)
content_pieces.append(chunk.choices[0].delta.content)
content = "".join(content_pieces).strip()
# logger.info(f"Title completion: {content}")
if "</think>" in content:
idx = content.index("</think>") + len("</think>")
......
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