Austin/QChatGPT
1
0
Fork 0
mirror of https://github.com/RockChinQ/QChatGPT.git synced 2024-11-16 11:42:44 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

perf(ollama): 优化命令显示细节

Browse Source
This commit is contained in:
RockChinQ 2024-07-10 00:07:32 +08:00
parent 3dc413638b
commit 7c06141ce2
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pkg/command/operators/ollama.py
View File

@ -21,8 +21,8 @@ class OllamaOperator(operator.CommandOperator):
for model in model_list: for model in model_list:
content += f"name: {model['name']}\n" content += f"name: {model['name']}\n"
content += f"modified_at: {model['modified_at']}\n" content += f"modified_at: {model['modified_at']}\n"
content += f"size: {bytes_to_mb(model['size'])}mb\n\n" content += f"size: {bytes_to_mb(model['size'])}MB\n\n"
yield entities.CommandReturn(text=f"{content}") yield entities.CommandReturn(text=f"{content.strip()}")
def bytes_to_mb(num_bytes): def bytes_to_mb(num_bytes):
@ -56,7 +56,7 @@ class OllamaShowOperator(operator.CommandOperator):
except ollama.ResponseError as e: except ollama.ResponseError as e:
content = f"{e.error}" content = f"{e.error}"
yield entities.CommandReturn(text=content) yield entities.CommandReturn(text=content.strip())
@operator.operator_class( @operator.operator_class(
@ -93,7 +93,7 @@ class OllamaPullOperator(operator.CommandOperator):
if percentage_completed > progress_count: if percentage_completed > progress_count:
progress_count += 10 progress_count += 10
yield entities.CommandReturn( yield entities.CommandReturn(
text=f"下载进度: {completed}/{total} = {percentage_completed:.2f}%") text=f"下载进度: {completed}/{total} ({percentage_completed:.2f}%)")
except ollama.ResponseError as e: except ollama.ResponseError as e:
yield entities.CommandReturn(text=f"拉取失败: {e.error}") yield entities.CommandReturn(text=f"拉取失败: {e.error}")