File size: 600 Bytes
888f9aa
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
from openai import OpenAI

client = OpenAI(
    base_url="http://127.0.0.1:8000/v1",
    api_key="EMPTY",  # vLLM 默认通常不校验 API Key
)

response = client.chat.completions.create(
    model="qwen2.5-32b-instruct",  # 必须与 --served-model-name 一致
    messages=[
        {
            "role": "system",
            "content": "You are a helpful assistant.",
        },
        {
            "role": "user",
            "content": "简单介绍一下检索增强生成(RAG)。",
        },
    ],
    temperature=0,
    max_tokens=512,
)

print(response.choices[0].message.content)