Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature]: Online Inference on local model with OpenAI Python SDK #8631

Open
1 task done
pesc101 opened this issue Sep 19, 2024 · 2 comments
Open
1 task done

[Feature]: Online Inference on local model with OpenAI Python SDK #8631

pesc101 opened this issue Sep 19, 2024 · 2 comments

Comments

@pesc101
Copy link

pesc101 commented Sep 19, 2024

🚀 The feature, motivation and pitch

OpenAI recently provided a new endpoint batch inference (https://platform.openai.com/docs/guides/batch/overview?lang=curl). It would be nice if it works using the batch format from OpenAI but with a local model.
I created an usage Issue for that before (#8567)

Something like that:

from openai import OpenAI

client = OpenAI(
    api_key="EMPTY",
    base_url="http://localhost:8000/v1",
)

batch_input_file = client.files.create(
  file=open("batchinput.jsonl", "rb"),
  purpose="batch"
)

client.batches.create(
    input_file_id= batch_input_file.id,
    endpoint="/v1/chat/completions",
    completion_window="24h",
    metadata={
      "description": "nightly eval job"
    }
)

At the moment there will be an error:
NotFoundError: Error code: 404 - {'detail': 'Not Found'}

Advantages for the implementation:

  • vllm can be run as a docker container and function only as endpoint
  • It is compatible with the OpenAI Python SDK, so easier to use for newbies also the model can be easily switched from the OpenAI server to local models
  • Consistent workflow, if you use the docker for Chat

Alternatives

Internal Implementation:
There was a feature implemented using python -m vllm.entrypoints.openai_batch as described here (#4777), but that is not compatible with the OpenAI SDK and also not compatible with the docker setup.

Additional context

No response

Before submitting a new issue...

  • Make sure you already searched for relevant issues, and asked the chatbot living at the bottom right corner of the documentation page, which can answer lots of frequently asked questions.
@pesc101
Copy link
Author

pesc101 commented Sep 19, 2024

Seems to me here are some bots with suspect links 👀

@DarkLight1337
Copy link
Member

DarkLight1337 commented Sep 19, 2024

cc @wuisawesome @pooyadavoodi since you two have worked on batch API

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

4 participants
@DarkLight1337 @pesc101 and others