[Fix] Add chat completion Example and simplify dependencies (#576)

This commit is contained in:
Zhuohan Li
2023-07-25 23:45:48 -07:00
committed by GitHub
parent df5dd3c68e
commit 82ad323dee
4 changed files with 52 additions and 11 deletions

View File

@@ -0,0 +1,28 @@
import openai
# Modify OpenAI's API key and API base to use vLLM's API server.
openai.api_key = "EMPTY"
openai.api_base = "http://localhost:8000/v1"
# List models API
models = openai.Model.list()
print("Models:", models)
model = models["data"][0]["id"]
# Completion API
stream = False
completion = openai.Completion.create(
model=model,
prompt="A robot may not injure a human being",
echo=False,
n=2,
stream=stream,
logprobs=3)
print("Completion results:")
if stream:
for c in completion:
print(c)
else:
print(completion)