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

Abierto
#1,984 1 comentario 0 reacciones 0 asignados Ver en GitHub
Lenguaje dominante
Python
Estrellas
10.6k
Forks
1.4k
Métricas de merge de PR
Métricas de PR pendientes

Descripción

# 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

Guía de contribución

Abrir la guía de contribución

Evaluación

Este issue todavía no se ha evaluado.

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.