S5_AR01/parsingDemo.py
Guillaume David 85c7d96714 PoC2
2024-11-17 19:40:44 +01:00

38 lines
1.1 KiB
Python

import openai
# API OpenAI
client = openai.Client(api_key='XXXXXXXXX')
# Texte à convertir en JSON
text = 'Le cours "Python ML Avancé" est un programme dédié à l\'apprentissage du Machine Learning et de la régression en Python, proposé chaque lundi. Il est recommandé d\'avoir une bonne maîtrise des bases du Python et du Machine Learning débutant. Le tarif du cours est de 1000 euros.'
# Modèle JSON
model = '''
{
"name": "Data Science avec Python",
"description": "Cours introductif sur l'analyse de données et les bases du machine learning avec Python",
"day": "Mercredi",
"prerequest": "Aucun prérequis nécessaire",
"price": 800
}
'''
# Appel de l'API OpenAI
response = client.chat.completions.create(
model="gpt-3.5-turbo",
messages=[
{
"role": "system",
"content": f"Convert this text {text} according to this JSON model: {model}",
},
{
"role": "system",
"content": "Give only the output without text or explanation, I will use it in a JSON parser.",
}
],
max_tokens=2000,
)
# Affichage de la réponse
print(response.choices[0].message.content)