Update prompt_demo.py

prompt_demo.py

@@ -19,39 +19,61 @@ if __name__ == "__main__":
     print(config)
     print()
     for prompt in config["prompts"]:
-        print(f'--- prompt mode: {prompt["mode"]} ---')
-        env = Environment(loader=FileSystemLoader("."))
-        template = env.get_template(prompt["template"])
-
-        source = template.environment.loader.get_source(template.environment, template.name)
-        variables = meta.find_undeclared_variables(env.parse(source[0]))
-
-        print("variables:", variables)
-        print("---")
-
-        data = {
-            "query": "Comment est votre blanquette ?",
-            "chunks" : [
-                {
-                    "url": "http://data.gouv.fr",
-                    "h": "hash49080805",
-                    "title": "A chunk title",
-                    "text": "text texs\ntext again ",
-                },
-                {
-                    "url": "http://...",
-                    "h": "hash49080806",
-                    "title": "A chunk title",
-                    "text": "text texs\ntext again ",
-                    "context": "I > am > a > context"
-                },
-
-            ]
-        }
-
-        if "system_prompt" in variables:
-            data["system_prompt"] = prompt["system_prompt"]
-
-        rendered_template = template.render(**data)
-        print(rendered_template)
-        print("---")
+        if prompt["mode"] == "rag":
+            print(f'--- prompt mode: {prompt["mode"]} ---')
+            env = Environment(loader=FileSystemLoader("."))
+            template = env.get_template(prompt["template"])
+
+            source = template.environment.loader.get_source(template.environment, template.name)
+            variables = meta.find_undeclared_variables(env.parse(source[0]))
+
+            print("variables:", variables)
+            print("---")
+
+            data = {
+                "query": "Quel est le meilleur moyen de cuire une blanquette?",
+                "chunks" : [
+                    {
+                        "url": "http://data.gouv.fr",
+                        "h": "hash49080805",
+                        "title": "A chunk title",
+                        "text": "Moi j'aime la blanquette avec du beurre dedans\nEt une sauce bien épaisse.",
+                    },
+                    {
+                        "url": "http://...",
+                        "h": "hash49080806",
+                        "title": "A chunk title",
+                        "text": "faîtes chauffer la blanquette à feu doux pendant 46 heures.",
+                        "context": "Recette de blanquette"
+                    },
+                                    {
+                        "url": "http://...",
+                        "h": "hash49080806",
+                        "title": "A chunk title",
+                        "text": "Les meilleures blanquettes se font avec des champignons de Paris",
+                        "context": "Avis de grand-mère"
+                    },
+                                    {
+                        "url": "http://...",
+                        "h": "hash49080806",
+                        "title": "A chunk title",
+                        "text": """Étape 1 Faire revenir la viande dans un peu de beurre doux jusqu'à ce que les morceaux soient un peu dorés.
+                                Étape 2: Saupoudrer de 2 cuillères de farine. Bien remuer.
+                                Étape 3: Ajouter 2 ou 3 verres d'eau, les cubes de bouillon, le vin et remuer. Ajouter de l'eau si nécessaire pour couvrir.
+                                Étape 4: Couper les carottes en rondelles et émincer les oignons puis les incorporer à la viande, ainsi que les champignons.
+                                Étape 5: Laisser mijoter à feu très doux environ 1h30 à 2h00 en remuant.
+                                Étape 6: Si nécessaire, ajouter de l'eau de temps en temps.
+                                Étape 7: Dans un bol, bien mélanger la crème fraîche, le jaune d’oeuf et le jus de citron. Ajouter ce mélange au dernier moment, bien remuer et servir tout de suite.
+                                """,
+                        "context": "Recette Marmiton"
+                    },
+
+                ]
+            }
+
+            if "system_prompt" in variables:
+                data["system_prompt"] = prompt["system_prompt"]
+    
+            rendered_template = template.render(**data)
+            print(rendered_template)
+            print("---")