Utiliser les prompts dans les chat completions

Le registre de prompts vous permet de stocker, de versionner et de gérer vos prompts de manière centralisée dans Studio. Ce guide explique comment récupérer un prompt enregistré et l’utiliser comme message système dans une chat completion.

Utilisation de base

Récupérez le prompt par nom ou par ID, extrayez son contenu, puis transmettez-le comme message système.

import os
from mistralai.client import Mistral

client = Mistral(api_key=os.environ["MISTRAL_API_KEY"])

prompt = client.beta.prompts.get(prompt_id="coding-assistant")
system_content = prompt.definition.content

response = client.chat.complete(
    model="mistral-large-latest",
    messages=[
        {"role": "system", "content": system_content},
        {"role": "user", "content": "Help me debug this function."},
    ],
)
print(response.choices[0].message.content)

import os
from mistralai.client import Mistral

client = Mistral(api_key=os.environ["MISTRAL_API_KEY"])

prompt = client.beta.prompts.get(prompt_id="coding-assistant")
system_content = prompt.definition.content

response = client.chat.complete(
    model="mistral-large-latest",
    messages=[
        {"role": "system", "content": system_content},
        {"role": "user", "content": "Help me debug this function."},
    ],
)
print(response.choices[0].message.content)

Variables de modèle

Les prompts peuvent contenir des emplacements réservés {{variable}}. Remplacez-les avant de transmettre le contenu au chat completion.

Par exemple, un prompt stocké sous cette forme :

You are a coding assistant. You specialise in {{language}}.

You are a coding assistant. You specialise in {{language}}.

Peut être utilisé comme ceci :

prompt = client.beta.prompts.get(prompt_id="coding-assistant")
system_content = prompt.definition.content.replace("{{language}}", "Python")

prompt = client.beta.prompts.get(prompt_id="coding-assistant")
system_content = prompt.definition.content.replace("{{language}}", "Python")

Épingler une version

Par défaut, get renvoie la dernière version d'un prompt. Épinglez une version ou un alias spécifique pour éviter que le comportement en production ne change lorsqu'un prompt est mis à jour.

# Pin by version number
prompt = client.beta.prompts.get(prompt_id="coding-assistant", version=3)

# Pin by alias
prompt = client.beta.prompts.get(prompt_id="coding-assistant", alias="production")

# Pin by version number
prompt = client.beta.prompts.get(prompt_id="coding-assistant", version=3)

# Pin by alias
prompt = client.beta.prompts.get(prompt_id="coding-assistant", alias="production")

Astuce

Utilisez des alias nommés comme production ou staging pour dissocier votre code des numéros de version. Vous pouvez modifier la version vers laquelle pointe un alias depuis Studio, sans changer votre application.