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

Open
#1,984 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
10.6k
Forks
1.4k
PR merge metrics
PR metrics pending

Description

# 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

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.