Docfile commited on
Commit
1715899
·
verified ·
1 Parent(s): eea4d76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -12
app.py CHANGED
@@ -122,13 +122,12 @@ def perform_web_search_with_gemini(query, client, model_id):
122
  print(f"--- LOG WEBSEARCH: Recherche Google avec Gemini pour: '{query}'")
123
 
124
  # Utiliser l'outil de recherche Google intégré à Gemini
125
- # Création de la configuration pour la recherche web
126
- config = {"tools": [{"google_search": {}}]}
127
-
128
  response = client.models.generate_content(
129
  model=model_id,
130
  contents=[{"role": "user", "parts": [{"text": query}]}],
131
- generation_config=config
 
 
132
  )
133
 
134
  print("--- LOG WEBSEARCH: Résultats de recherche Google obtenus.")
@@ -383,22 +382,22 @@ def chat_api():
383
 
384
  print(f" Message {i} (role: {role}): {', '.join(parts_info)}")
385
 
386
- # Configuration de base
387
- generation_config = {
388
- "system_instruction": SYSTEM_INSTRUCTION,
389
- "safety_settings": SAFETY_SETTINGS
390
- }
391
 
392
- # Ajout de la configuration de recherche web si nécessaire
393
  if use_web_search:
394
- generation_config["tools"] = [{"google_search": {}}]
395
 
396
  # Appel à l'API avec le nouveau SDK
397
  print(f"--- LOG [/api/chat]: Envoi de la requête à {selected_model_name}...")
398
  response = genai_client.models.generate_content(
399
  model=selected_model_name,
400
  contents=contents,
401
- generation_config=generation_config
402
  )
403
 
404
  # --- Traitement de la réponse (avec logs) ---
 
122
  print(f"--- LOG WEBSEARCH: Recherche Google avec Gemini pour: '{query}'")
123
 
124
  # Utiliser l'outil de recherche Google intégré à Gemini
 
 
 
125
  response = client.models.generate_content(
126
  model=model_id,
127
  contents=[{"role": "user", "parts": [{"text": query}]}],
128
+ config=types.GenerateContentConfig(
129
+ tools=[{"google_search": {}}]
130
+ )
131
  )
132
 
133
  print("--- LOG WEBSEARCH: Résultats de recherche Google obtenus.")
 
382
 
383
  print(f" Message {i} (role: {role}): {', '.join(parts_info)}")
384
 
385
+ # Configuration de l'appel à l'API
386
+ generate_config = types.GenerateContentConfig(
387
+ system_instruction=SYSTEM_INSTRUCTION,
388
+ safety_settings=SAFETY_SETTINGS
389
+ )
390
 
391
+ # Ajouter la configuration de recherche web si demandée
392
  if use_web_search:
393
+ generate_config.tools = [{"google_search": {}}]
394
 
395
  # Appel à l'API avec le nouveau SDK
396
  print(f"--- LOG [/api/chat]: Envoi de la requête à {selected_model_name}...")
397
  response = genai_client.models.generate_content(
398
  model=selected_model_name,
399
  contents=contents,
400
+ config=generate_config
401
  )
402
 
403
  # --- Traitement de la réponse (avec logs) ---