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