Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
| 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
|
| 387 |
-
|
| 388 |
-
|
| 389 |
-
|
| 390 |
-
|
| 391 |
|
| 392 |
-
#
|
| 393 |
if use_web_search:
|
| 394 |
-
|
| 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 |
-
|
| 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) ---
|