habulaj commited on
Commit
0533635
·
verified ·
1 Parent(s): cc46673

Update routers/inference.py

Browse files
Files changed (1) hide show
  1. routers/inference.py +2 -2
routers/inference.py CHANGED
@@ -344,7 +344,7 @@ async def rewrite_news(news: NewsRequest):
344
  sources_content = ""
345
 
346
  client = genai.Client(api_key=api_key)
347
- model = "gemini-2.5-pro"
348
 
349
  # Obter data formatada
350
  date_string = get_brazilian_date_string()
@@ -466,7 +466,7 @@ The 48th annual Kennedy Center Honors, set to air on the CBS network and stream
466
  config = types.GenerateContentConfig(
467
  system_instruction=SYSTEM_INSTRUCTIONS,
468
  thinking_config=types.ThinkingConfig(
469
- thinking_budget=128,
470
  ),
471
  response_mime_type="text/plain",
472
  max_output_tokens=16000,
 
344
  sources_content = ""
345
 
346
  client = genai.Client(api_key=api_key)
347
+ model = "gemini-2.5-flash"
348
 
349
  # Obter data formatada
350
  date_string = get_brazilian_date_string()
 
466
  config = types.GenerateContentConfig(
467
  system_instruction=SYSTEM_INSTRUCTIONS,
468
  thinking_config=types.ThinkingConfig(
469
+ thinking_budget=0,
470
  ),
471
  response_mime_type="text/plain",
472
  max_output_tokens=16000,