Refactor: Removido min doc e rules do cógido e colocado no arquivo yaml, além de mudar a forma da saída

This commit is contained in:
2026-03-12 17:49:52 -03:00
parent 63a6d1b308
commit b2469ccd0e
2 changed files with 26 additions and 435 deletions

View File

@@ -26,6 +26,11 @@ async def evaluate_servico(servico: dict, guia: dict, file_content: str) -> dict
query = json.dumps(query_data, indent=2, ensure_ascii=False)
t0 = time.time()
agent_output = ""
input_tokens = 0
output_tokens = 0
resultado = "Reprovado"
try:
result = await run_agent(query, code, file_content)
agent_output = result["response"]
@@ -33,12 +38,21 @@ async def evaluate_servico(servico: dict, guia: dict, file_content: str) -> dict
output_tokens = result["output_tokens"]
except Exception as e:
agent_output = f"ERROR: {str(e)}"
input_tokens = 0
output_tokens = 0
try:
clean = agent_output.strip()
if clean.startswith("```"):
clean = clean.split("```", 2)[1]
if clean.startswith("json"):
clean = clean[4:]
parsed = json.loads(clean.strip())
resultado = parsed.get("status", "Reprovado")
except Exception:
pass
return {
"codigoServico": codigo_servico_raw,
"resultado": "Aprovado" if "".join(c for c in agent_output.lower() if c.isalpha()).startswith("aprov") else "Reprovado",
"resultado": resultado,
"agentOutput": agent_output,
"input_tokens": input_tokens,
"output_tokens": output_tokens,