abetlen / abetlen/llama-cpp-python

Stream last block not return token usage info when create_chat_completion_openai_v1 or create_chat_completion but server does

未關閉
#1,984 1 則留言 0 個 reaction 已指派 0 人 在 GitHub 檢視
主要語言
Python
星號
10.6k
分支
1.4k
PR 合併指標
PR 指標待擷取

描述

# Prerequisites

Stream last block not return token usage info when create_chat_completion_openai_v1 or create_chat_completion but server does.

Python api is better for me, but no token usage returned, is this by disign?

- [x] I am running the latest code. Development is very rapid so there are no tagged versions as of now.
- [x] I carefully followed the [README.md](https://github.com/abetlen/llama-cpp-python/blob/main/README.md).
- [x] I [searched using keywords relevant to my issue](https://docs.github.com/en/issues/tracking-your-work-with-issues/filtering-and-searching-issues-and-pull-requests) to make sure that I am creating a new issue that is not already open (or closed).
- [x] I reviewed the [Discussions](https://github.com/abetlen/llama-cpp-python/discussions), and have a new bug or useful enhancement to share.

# Expected Behavior

Called with

```
response = llm.create_chat_completion_openai_v1(
messages=[
{
"role": "system",
"content": "请生成该指令的场景标签和功能标签,并以JSON列表的格式回复。",
},
{"role": "user", "content": """"""},
],
temperature=0,
max_tokens=192,
stream=True,
)
```

# Current Behavior

![Image](https://github.com/user-attachments/assets/b771d5cd-603d-4113-913f-e4b2cbc9a6ed)

# Failure Information (for bugs)

Set stream_options={"include_usage": True}, seems not work, any other ways?

# Steps to Reproduce

Please provide detailed steps for reproducing the issue. We are not sitting in front of your screen, so the more detail the better.

1. load any model
2. call with create_chat_completion_openai_v1 or create_chat_completion and enable stream mode
3. check chunk response

貢獻指南

開啟貢獻指南

評估

這個 Issue 還沒有評估資料。

把新 issue 寄到你的電子郵件信箱

精選適合新手參與的 GitHub issue 摘要。