amauricunha commited on
Commit
9edb294
·
verified ·
1 Parent(s): 4bdb212

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -1,4 +1,4 @@
1
- #app.py
2
  import os
3
  import io
4
  import json
@@ -229,7 +229,6 @@ def get_ai_text_response(provider, model_name, system_instruction, user_prompt,
229
  if provider == 'gemini':
230
  model = genai_client.GenerativeModel(model_name, system_instruction=system_instruction)
231
 
232
- # CORREÇÃO DEFINITIVA: Usa GenerationConfig para JSON e nada para texto.
233
  config = None
234
  if json_schema:
235
  config = GenerationConfig(response_mime_type="application/json", response_schema=json_schema)
@@ -240,7 +239,7 @@ def get_ai_text_response(provider, model_name, system_instruction, user_prompt,
240
  parsed_json = json.loads(response.text)
241
  required_keys = json_schema.get("required", [])
242
  if not all(key in parsed_json and parsed_json[key] for key in required_keys):
243
- raise ValueError(f"AI response missing required keys. Required: {required_keys}, Got: {list(parsed_json.keys())}")
244
  return parsed_json
245
  else:
246
  return response.text.strip()
@@ -258,7 +257,7 @@ def get_ai_text_response(provider, model_name, system_instruction, user_prompt,
258
  parsed_json = json.loads(response.choices[0].message.content)
259
  required_keys = json_schema.get("required", [])
260
  if not all(key in parsed_json and parsed_json[key] for key in required_keys):
261
- raise ValueError(f"AI response missing required keys. Required: {required_keys}, Got: {list(parsed_json.keys())}")
262
  return parsed_json
263
  else:
264
  return response.choices[0].message.content.strip()
 
1
+ ]#app.py
2
  import os
3
  import io
4
  import json
 
229
  if provider == 'gemini':
230
  model = genai_client.GenerativeModel(model_name, system_instruction=system_instruction)
231
 
 
232
  config = None
233
  if json_schema:
234
  config = GenerationConfig(response_mime_type="application/json", response_schema=json_schema)
 
239
  parsed_json = json.loads(response.text)
240
  required_keys = json_schema.get("required", [])
241
  if not all(key in parsed_json and parsed_json[key] for key in required_keys):
242
+ raise ValueError(f"AI response missing required keys or has empty values.")
243
  return parsed_json
244
  else:
245
  return response.text.strip()
 
257
  parsed_json = json.loads(response.choices[0].message.content)
258
  required_keys = json_schema.get("required", [])
259
  if not all(key in parsed_json and parsed_json[key] for key in required_keys):
260
+ raise ValueError(f"AI response missing required keys or has empty values.")
261
  return parsed_json
262
  else:
263
  return response.choices[0].message.content.strip()