test.py

# test_app.py
import requests

# URL của FastAPI app của bạn
# API_URL = "http://localhost:8000/generate"
API_URL = "https://Hoangnam5904-LLM.hf.space/generate"

# Dữ liệu gửi đến endpoint
payload = {
    "model": "unsloth/Llama-3.2-3B-Instruct",
    "system": "Bạn là một trợ lý AI hữu ích và trả lời ngắn gọn, dễ hiểu.",
    "user": "Viết một đoạn văn ngắn về tác dụng của việc nghiện game.",
    "max_tokens": 200,
    "temperature": 0.7
}

def test_generate():
    try:
        response = requests.post(API_URL, json=payload)
        if response.status_code == 200:
            data = response.json()
            print(data)
            # if "generated_text" in data:
            #     print("=== Kết quả sinh ===")
            #     print(data["generated_text"])
            # else:
            #     print("Không có key 'generated_text', response:", data)
        else:
            print(f"Lỗi HTTP {response.status_code}: {response.text}")
    except Exception as e:
        print("Lỗi khi gọi API:", e)

if __name__ == "__main__":
    test_generate()