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:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user