Commit
•
bd6e3f8
1
Parent(s):
0b60c78
Update prompt_demo.py
Browse files- prompt_demo.py +58 -36
prompt_demo.py
CHANGED
@@ -19,39 +19,61 @@ if __name__ == "__main__":
|
|
19 |
print(config)
|
20 |
print()
|
21 |
for prompt in config["prompts"]:
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
26 |
-
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
54 |
-
|
55 |
-
|
56 |
-
|
57 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
19 |
print(config)
|
20 |
print()
|
21 |
for prompt in config["prompts"]:
|
22 |
+
if prompt["mode"] == "rag":
|
23 |
+
print(f'--- prompt mode: {prompt["mode"]} ---')
|
24 |
+
env = Environment(loader=FileSystemLoader("."))
|
25 |
+
template = env.get_template(prompt["template"])
|
26 |
+
|
27 |
+
source = template.environment.loader.get_source(template.environment, template.name)
|
28 |
+
variables = meta.find_undeclared_variables(env.parse(source[0]))
|
29 |
+
|
30 |
+
print("variables:", variables)
|
31 |
+
print("---")
|
32 |
+
|
33 |
+
data = {
|
34 |
+
"query": "Quel est le meilleur moyen de cuire une blanquette?",
|
35 |
+
"chunks" : [
|
36 |
+
{
|
37 |
+
"url": "http://data.gouv.fr",
|
38 |
+
"h": "hash49080805",
|
39 |
+
"title": "A chunk title",
|
40 |
+
"text": "Moi j'aime la blanquette avec du beurre dedans\nEt une sauce bien épaisse.",
|
41 |
+
},
|
42 |
+
{
|
43 |
+
"url": "http://...",
|
44 |
+
"h": "hash49080806",
|
45 |
+
"title": "A chunk title",
|
46 |
+
"text": "faîtes chauffer la blanquette à feu doux pendant 46 heures.",
|
47 |
+
"context": "Recette de blanquette"
|
48 |
+
},
|
49 |
+
{
|
50 |
+
"url": "http://...",
|
51 |
+
"h": "hash49080806",
|
52 |
+
"title": "A chunk title",
|
53 |
+
"text": "Les meilleures blanquettes se font avec des champignons de Paris",
|
54 |
+
"context": "Avis de grand-mère"
|
55 |
+
},
|
56 |
+
{
|
57 |
+
"url": "http://...",
|
58 |
+
"h": "hash49080806",
|
59 |
+
"title": "A chunk title",
|
60 |
+
"text": """Étape 1 Faire revenir la viande dans un peu de beurre doux jusqu'à ce que les morceaux soient un peu dorés.
|
61 |
+
Étape 2: Saupoudrer de 2 cuillères de farine. Bien remuer.
|
62 |
+
É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.
|
63 |
+
Étape 4: Couper les carottes en rondelles et émincer les oignons puis les incorporer à la viande, ainsi que les champignons.
|
64 |
+
Étape 5: Laisser mijoter à feu très doux environ 1h30 à 2h00 en remuant.
|
65 |
+
Étape 6: Si nécessaire, ajouter de l'eau de temps en temps.
|
66 |
+
É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.
|
67 |
+
""",
|
68 |
+
"context": "Recette Marmiton"
|
69 |
+
},
|
70 |
+
|
71 |
+
]
|
72 |
+
}
|
73 |
+
|
74 |
+
if "system_prompt" in variables:
|
75 |
+
data["system_prompt"] = prompt["system_prompt"]
|
76 |
+
|
77 |
+
rendered_template = template.render(**data)
|
78 |
+
print(rendered_template)
|
79 |
+
print("---")
|