lumi-voice-companion-app / verify_vllm_local.py
Debdeep30's picture
Upload verify_vllm_local.py with huggingface_hub
d2fa827 verified
Raw
History Blame
932 Bytes
import requests
import json
def check_vllm():
url = "http://localhost:8000/v1/chat/completions"
model = "./lumi-qwen3-output"
payload = {
"model": model,
"messages": [
{"role": "system", "content": "You are Lumi."},
{"role": "user", "content": "Hello!"}
],
"temperature": 0.7,
"max_tokens": 50
}
print(f"Checking vLLM at {url}...")
try:
response = requests.post(url, json=payload, timeout=10)
if response.status_code == 200:
print("✅ SUCCESS! vLLM responded correctly.")
print(f"Response: {response.json()['choices'][0]['message']['content']}")
else:
print(f"❌ FAILED! Status Code: {response.status_code}")
print(f"Error: {response.text}")
except Exception as e:
print(f"❌ CONNECTION ERROR: {e}")
if __name__ == "__main__":
check_vllm()