chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,35 @@
|
||||
"""Query client for an SGLang model served via Ray Serve LLM.
|
||||
|
||||
Prerequisites:
|
||||
Start a serving example first, e.g.:
|
||||
RAY_EXPERIMENTAL_NOSET_CUDA_VISIBLE_DEVICES=0 serve run serve_sglang_example:app
|
||||
|
||||
Usage:
|
||||
python query_example.py
|
||||
"""
|
||||
|
||||
from openai import OpenAI
|
||||
|
||||
client = OpenAI(base_url="http://localhost:8000/v1", api_key="fake-key")
|
||||
|
||||
# Chat completions
|
||||
print("=== Chat Completions ===")
|
||||
chat_response = client.chat.completions.create(
|
||||
model="Llama-3.1-8B-Instruct",
|
||||
messages=[
|
||||
{"role": "user", "content": "List 3 countries and their capitals."},
|
||||
],
|
||||
temperature=0,
|
||||
max_tokens=64,
|
||||
)
|
||||
print(chat_response.choices[0].message.content)
|
||||
|
||||
# Text completions
|
||||
print("\n=== Text Completions ===")
|
||||
completion_response = client.completions.create(
|
||||
model="Llama-3.1-8B-Instruct",
|
||||
prompt="San Francisco is a",
|
||||
temperature=0,
|
||||
max_tokens=30,
|
||||
)
|
||||
print(completion_response.choices[0].text)
|
||||
Reference in New Issue
Block a user