Publié le

Flux de travail de code avec les agents Mistral AI

3 min read
Auteurs
  • Profile picture of aithemes.net
    Nom
    aithemes.net
    Twitter
Image de l'article Dans cet article, je vais documenter un exemple de flux de travail pour la génération de code Python en utilisant les agents Mistral AI.

Aperçu des agents Mistral AI

Inspiré par le flux de travail de l'agent Python documenté dans les ressources officielles de Mistral, j'ai développé un flux de travail personnalisé nommé gen_code_workflow. L'objectif de ce flux de travail est d'automatiser la génération de scripts Python basés sur des instructions fournies par l'utilisateur. Une boucle est intégrée dans le flux de travail pour gérer l'exécution du script et les tentatives en cas d'erreurs.

Le flux de travail utilise trois agents Mistral AI :

  • Agent de génération de code Python : Génère des scripts Python basés sur les entrées de l'utilisateur.
  • Agent de correction de code Python : Tente de corriger les erreurs d'exécution dans le code généré.
  • Agent Judge : Évalue le script Python par rapport aux instructions de l'utilisateur et fournit un score entre 0 et 10.

Arguments du flux de travail

Le gen_code_workflow accepte les arguments suivants :

  • Instructions de l'utilisateur (str)
  • Optionnel : Chemin du fichier pour enregistrer le script Python résultant (str)
  • Optionnel : Mode débogage (bool)

Diagramme du flux de travail

Diagramme du flux de travail

Diagramme généré en utilisant l'intégration d'Excalidraw avec Mermaid.

Exemple d'exécution

gen_code_workflow("""
   Generate a method to search Wikipedia for a specific topic.
   You will receive the search topic as an argument. If no argument is passed, set a default.
   The method should return information found on Wikipedia.
   Use a publicly available Python library to access Wikipedia.
   Do not limit the number of sentences.
   Handle errors in case the topic is not found.
   Document the method with its arguments and return values.
""","test.py", False)

Score: 8.5

Script Python résultant

import wikipedia
import subprocess
import sys

# Install the wikipedia library if it's not already installed
try:
    import wikipedia
except ImportError:
    subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'wikipedia'])

def search_wikipedia(topic='Python programming language'):
    """
    Search Wikipedia for a given topic and return the information found.

    Args:
        topic (str): The topic to search on Wikipedia. Defaults to 'Python programming language'.

    Returns:
        str: The information found on Wikipedia for the specified topic.
              If the topic is not found, returns a user-friendly error message.
    """
    try:
        # Search Wikipedia for the provided topic
        page = wikipedia.page(topic)
        return page.content
    except wikipedia.exceptions.PageError:
        # Handle the case where the topic is not found
        return f"The topic '{topic}' was not found on Wikipedia."
    except wikipedia.exceptions.DisambiguationError as e:
        # Handle disambiguation errors by returning a message with possible options
        return f"The topic '{topic}' is ambiguous. Possible options are: {e.options}"

if __name__ == "__main__":
    print(search_wikipedia('Artificial Intelligence'))

Liens utiles

Vous avez apprécié cet article ? Vous l'avez trouvé utile ? N'hésitez pas à laisser un commentaire ci-dessous pour partager vos réflexions ou poser des questions. Un compte GitHub est requis pour participer à la discussion.

PARTAGER