diff --git a/tests/doc_example.py b/tests/doc_example.py deleted file mode 100644 index 9d7c9d34b..000000000 --- a/tests/doc_example.py +++ /dev/null @@ -1,20 +0,0 @@ -import openai - -openai.api_key = "EMPTY" -openai.base_url = "http://localhost:8000/v1/" - -model = "vicuna-7b-v1.5" -prompt = "Once upon a time" - -# create a completion -completion = openai.completions.create(model=model, prompt=prompt, max_tokens=64) -# print the completion -print(prompt + completion.choices[0].text) - -# create a chat completion -completion = openai.chat.completions.create( - model=model, - messages=[{"role": "user", "content": "Hello! What is your name?"}] -) -# print the completion -print(completion.choices[0].message.content)