amauricunha commited on
Commit
1f7c160
verified
1 Parent(s): 48381a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -52,7 +52,6 @@ def list_models():
52
  "openai/gpt-oss-120b",
53
  "openai/gpt-oss-20b"
54
  ]
55
-
56
  try:
57
  if genai_client:
58
  for m in genai_client.list_models():
@@ -240,12 +239,15 @@ def get_ai_text_response(provider, model_name, system_instruction, user_prompt,
240
 
241
  response = model.generate_content(user_prompt, generation_config=config)
242
 
243
- parsed_json = json.loads(response.text)
244
  if json_schema:
 
245
  required_keys = json_schema.get("required", [])
246
  if not all(key in parsed_json and parsed_json[key] for key in required_keys):
247
  raise ValueError(f"AI response missing required keys. Required: {required_keys}, Got: {list(parsed_json.keys())}")
248
- return parsed_json if json_schema else response.text.strip()
 
 
249
 
250
  elif provider == 'groq':
251
  # ... (c贸digo existente sem altera莽玫es) ...
 
52
  "openai/gpt-oss-120b",
53
  "openai/gpt-oss-20b"
54
  ]
 
55
  try:
56
  if genai_client:
57
  for m in genai_client.list_models():
 
239
 
240
  response = model.generate_content(user_prompt, generation_config=config)
241
 
242
+ # CORRE脟脙O CR脥TICA: S贸 faz parse de JSON se foi solicitado.
243
  if json_schema:
244
+ parsed_json = json.loads(response.text)
245
  required_keys = json_schema.get("required", [])
246
  if not all(key in parsed_json and parsed_json[key] for key in required_keys):
247
  raise ValueError(f"AI response missing required keys. Required: {required_keys}, Got: {list(parsed_json.keys())}")
248
+ return parsed_json
249
+ else:
250
+ return response.text.strip()
251
 
252
  elif provider == 'groq':
253
  # ... (c贸digo existente sem altera莽玫es) ...