Docfile commited on
Commit
6a2e943
·
verified ·
1 Parent(s): 4533086

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -2
app.py CHANGED
@@ -120,7 +120,7 @@ def perform_web_search_with_gemini(query, client, model_id):
120
  """Effectue une recherche web via l'API Google Search intégrée à Gemini."""
121
  try:
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,
@@ -129,6 +129,14 @@ def perform_web_search_with_gemini(query, client, model_id):
129
  tools=[{"google_search": {}}]
130
  )
131
  )
 
 
 
 
 
 
 
 
132
 
133
  print("--- LOG WEBSEARCH: Résultats de recherche Google obtenus.")
134
 
@@ -387,10 +395,16 @@ def chat_api():
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}...")
 
120
  """Effectue une recherche web via l'API Google Search intégrée à Gemini."""
121
  try:
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,
 
129
  tools=[{"google_search": {}}]
130
  )
131
  )
132
+ """
133
+ response = client.models.generate_content(
134
+ model=model_id,
135
+ contents=[{"role": "user", "parts": [{"text": query}]}],
136
+ config=types.GenerateContentConfig(
137
+ tools=[types.Tool(google_search=types.GoogleSearchToolConfig())]
138
+ )
139
+ )
140
 
141
  print("--- LOG WEBSEARCH: Résultats de recherche Google obtenus.")
142
 
 
395
  system_instruction=SYSTEM_INSTRUCTION,
396
  safety_settings=SAFETY_SETTINGS
397
  )
398
+ """
399
  # Ajouter la configuration de recherche web si demandée
400
  if use_web_search:
401
  generate_config.tools = [{"google_search": {}}]
402
+ """
403
+ if use_web_search:
404
+ generate_config.tools = [types.Tool(google_search=types.GoogleSearchToolConfig())]
405
+
406
+
407
+
408
 
409
  # Appel à l'API avec le nouveau SDK
410
  print(f"--- LOG [/api/chat]: Envoi de la requête à {selected_model_name}...")