test.py

import requests

# Define the URL of the API endpoint
url = "http://127.0.0.1:8000/generate-response/"

# Define the payload
payload = {
    "input": "give me a roadmap for frontend development using a mindmap.",
    "model": "llama"  # Change to "mixtral" to test the Mixtral model
}

# Send a POST request to the API
response = requests.post(url, json=payload)

# Print the response from the API
print(response.json())