Florian.Moret commited on
Commit
38cd346
·
1 Parent(s): b847856

remove biblio dans le suffixe

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -11,7 +11,7 @@ from mistralai import Mistral
11
  # from llama_index.core import VectorStoreIndex, SimpleDirectoryReader
12
  # from dotenv import load_dotenv
13
 
14
- MISTRAL_API_KEY = 'DFtRafUrJnASg1cqrEWNtSMKAjZT4Njp' #os.getenv("api_mistral")
15
  model = "mistral-large-latest" #"ministral-8b-latest" # ancien model : 'mistral-large-latest'
16
  mistral_client = Mistral(api_key=MISTRAL_API_KEY)
17
  MAX_TOKENS = 1500
@@ -76,7 +76,7 @@ def generate_prompts(score:str, type: str, annee_min: str, annee_max:str, contex
76
  Réponds en vulgarisant les informations.
77
  Pour fournir la réponse, tu dois te baser sur des publications/articles qui ont une date de publication entre {annee_min} et {annee_max}."""
78
  suffix_prompt = """Réponds en français et donne une réponse directe et claire.
79
- Fini par faire une bibliographie avec les références bibliographiquesque tu as utilisé."""
80
  elif score == "2":
81
  prefix_prompt = f"""Tu es un assistant IA spécialisé en nutrition de la volaille. Ton utilisateur est un chercheur travaillant sur
82
  l'amélioration des régimes alimentaires pour optimiser la santé et la croissance des {type_description}. Voici les informations extraites des documents à utiliser avec priorité : {context}.
@@ -86,7 +86,7 @@ def generate_prompts(score:str, type: str, annee_min: str, annee_max:str, contex
86
  suffix_prompt = """Présente une réponse claire et concise, en commençant par une explication des causes du problème,
87
  suivie d'une analyse des facteurs de risque, et termine par des recommandations. Utilise un ton
88
  professionnel, clair et rigoureux. Réponds en français.
89
- Fini par faire une bibliographie avec les références bibliographiques que tu as utilisé."""
90
  elif score == "3":
91
  prefix_prompt = f"""Tu es un assistant IA spécialisé en nutrition de la volaille. Ton utilisateur est un chercheur travaillant sur
92
  l'amélioration des régimes alimentaires pour optimiser la santé et la croissance des {type_description}. Voici les informations extraites des documents à utiliser avec priorité : {context}.
@@ -97,7 +97,7 @@ def generate_prompts(score:str, type: str, annee_min: str, annee_max:str, contex
97
  suivie d'une analyse des facteurs de risque, et termine par des recommandations pratiques et éprouvées. Utilise un ton
98
  professionnel, clair et rigoureux. Si possible, inclue des chiffres, des études ou des références pertinentes pour renforcer
99
  la crédibilité de la réponse. Réponds en français.
100
- Fini par faire une bibliographie avec les références bibliographiques que tu as utilisé."""
101
  else:
102
  raise ValueError("Score must be 1, 2, or 3")
103
 
 
11
  # from llama_index.core import VectorStoreIndex, SimpleDirectoryReader
12
  # from dotenv import load_dotenv
13
 
14
+ MISTRAL_API_KEY = os.getenv("api_mistral")
15
  model = "mistral-large-latest" #"ministral-8b-latest" # ancien model : 'mistral-large-latest'
16
  mistral_client = Mistral(api_key=MISTRAL_API_KEY)
17
  MAX_TOKENS = 1500
 
76
  Réponds en vulgarisant les informations.
77
  Pour fournir la réponse, tu dois te baser sur des publications/articles qui ont une date de publication entre {annee_min} et {annee_max}."""
78
  suffix_prompt = """Réponds en français et donne une réponse directe et claire.
79
+ N'inclus pas de bibliographie dans ta réponse."""
80
  elif score == "2":
81
  prefix_prompt = f"""Tu es un assistant IA spécialisé en nutrition de la volaille. Ton utilisateur est un chercheur travaillant sur
82
  l'amélioration des régimes alimentaires pour optimiser la santé et la croissance des {type_description}. Voici les informations extraites des documents à utiliser avec priorité : {context}.
 
86
  suffix_prompt = """Présente une réponse claire et concise, en commençant par une explication des causes du problème,
87
  suivie d'une analyse des facteurs de risque, et termine par des recommandations. Utilise un ton
88
  professionnel, clair et rigoureux. Réponds en français.
89
+ N'inclus pas de bibliographie dans ta réponse."""
90
  elif score == "3":
91
  prefix_prompt = f"""Tu es un assistant IA spécialisé en nutrition de la volaille. Ton utilisateur est un chercheur travaillant sur
92
  l'amélioration des régimes alimentaires pour optimiser la santé et la croissance des {type_description}. Voici les informations extraites des documents à utiliser avec priorité : {context}.
 
97
  suivie d'une analyse des facteurs de risque, et termine par des recommandations pratiques et éprouvées. Utilise un ton
98
  professionnel, clair et rigoureux. Si possible, inclue des chiffres, des études ou des références pertinentes pour renforcer
99
  la crédibilité de la réponse. Réponds en français.
100
+ N'inclus pas de bibliographie dans ta réponse."""
101
  else:
102
  raise ValueError("Score must be 1, 2, or 3")
103