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

Aperta
#1,984 1 commento 0 reazioni 0 assegnatari Vedi su GitHub
Lingua principale
Python
Stelle
10.6k
Fork
1.4k
Metriche di merge delle PR
Metriche PR in attesa

Descrizione

# 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

Guida per i contributori

Apri la guida per i contributori

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.