caarleexx commited on
Commit
6896126
·
verified ·
1 Parent(s): 4298dfd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -48,8 +48,8 @@ CLIENT = genai.Client(api_key=API_KEY)
48
 
49
  # --- Definição dos Modelos ---
50
  # ATUALIZADO: Agora são apenas strings com os nomes dos modelos.
51
- COUNSELOR_MODEL_NAME = "gemini-2.5-pro"
52
- SUPERVISOR_MODEL_NAME = "gemini-2.5-pro" # Pode ser diferente se desejado
53
 
54
  # --- Título da Interface ---
55
  TITLE = "# 🚀 Pipeline v10 | Nova API do Gemini Pro\n**Utilizando `Client`, `stream` e `tools` para raciocínio aprimorado.**"
@@ -157,8 +157,9 @@ def chamar_gemini_json(model_name: str, prompt: str, temperatura: float = 0.7, m
157
  # 2. Configurar o Thinking
158
  # include_thoughts=True fará com que o pensamento apareça no output (ajuda a debugar),
159
  # mas o seu regex filtrará para pegar só o JSON depois.
160
- thinking_config = types.ThinkingConfig(
161
- include_thoughts=True
 
162
  )
163
 
164
  # 3. Ferramentas (Google Search)
 
48
 
49
  # --- Definição dos Modelos ---
50
  # ATUALIZADO: Agora são apenas strings com os nomes dos modelos.
51
+ COUNSELOR_MODEL_NAME = "gemini-flash-latest"
52
+ SUPERVISOR_MODEL_NAME = "gemini-flash-latest"" # Pode ser diferente se desejado
53
 
54
  # --- Título da Interface ---
55
  TITLE = "# 🚀 Pipeline v10 | Nova API do Gemini Pro\n**Utilizando `Client`, `stream` e `tools` para raciocínio aprimorado.**"
 
157
  # 2. Configurar o Thinking
158
  # include_thoughts=True fará com que o pensamento apareça no output (ajuda a debugar),
159
  # mas o seu regex filtrará para pegar só o JSON depois.
160
+ thinking_config = types.thinkingConfig(
161
+ #include_thoughts=True
162
+ thinkingBudget: -1
163
  )
164
 
165
  # 3. Ferramentas (Google Search)