import getpass
import os
#os.environ["MISTRAL_API_KEY"] = getpass.getpass()
#os.environ["HF_TOKEN"] = getpass.getpass()
api_key = os.getenv()

from mistralai import Mistral
api_key = os.environ["MISTRAL_API_KEY"]
#model = "open-mixtral-8x22b" #"pixtral-12b-2409"#"Mathstral" #"mistral-large-latest"
#client = Mistral(api_key='kK3vwEJn7YIPJeYg4Zvjz8S2i7v6Gb5H')
client = HuggingFaceHub(repo_id = 'mistralai/Mathstral-7B-v0.1') # 'mistralai/Mathstral-7B-v0.1')
prompt = "What is the best French cheese?"
response = client(prompt)
print(response)