def test_generation()

in llm_swarm/__init__.py [0:0]


def test_generation(endpoint):
    headers = {
        "Content-Type": "application/json",
    }
    data = {
        "inputs": "What is Deep Learning?",
        "parameters": {
            "max_new_tokens": 200,
        },
    }
    requests.post(endpoint, headers=headers, json=data)
    print("✅ test generation")