Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
| 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
|
| 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()
|