You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
105 lines
3.8 KiB
Python
105 lines
3.8 KiB
Python
import os
|
|
import unittest
|
|
from types import SimpleNamespace
|
|
|
|
os.environ.setdefault("DEBUG", "false")
|
|
|
|
from app.services.ai.llm_service import (
|
|
INVALID_RECEIPT_WATERMARK_MESSAGE,
|
|
VALID_RECEIPT_WATERMARK_MARKER,
|
|
LLMService,
|
|
)
|
|
|
|
|
|
class LLMServiceResponseParsingTests(unittest.TestCase):
|
|
def test_extract_response_payload_supports_text_and_function_call_in_same_candidate(self):
|
|
service = LLMService.__new__(LLMService)
|
|
response = SimpleNamespace(
|
|
candidates=[
|
|
SimpleNamespace(
|
|
content=SimpleNamespace(
|
|
parts=[
|
|
SimpleNamespace(text="Legal! Buscando carros de ate 70 mil para voce.", function_call=None),
|
|
SimpleNamespace(
|
|
text=None,
|
|
function_call=SimpleNamespace(
|
|
name="consultar_estoque",
|
|
args={"preco_max": 70000.0},
|
|
),
|
|
),
|
|
]
|
|
)
|
|
)
|
|
]
|
|
)
|
|
|
|
payload = service._extract_response_payload(response)
|
|
|
|
self.assertEqual(payload["response"], "Legal! Buscando carros de ate 70 mil para voce.")
|
|
self.assertEqual(
|
|
payload["tool_call"],
|
|
{
|
|
"name": "consultar_estoque",
|
|
"arguments": {"preco_max": 70000.0},
|
|
},
|
|
)
|
|
|
|
def test_extract_response_payload_handles_text_only_candidate_without_response_text_accessor(self):
|
|
service = LLMService.__new__(LLMService)
|
|
response = SimpleNamespace(
|
|
candidates=[
|
|
SimpleNamespace(
|
|
content=SimpleNamespace(
|
|
parts=[
|
|
SimpleNamespace(text="Resposta simples", function_call=None),
|
|
]
|
|
)
|
|
)
|
|
]
|
|
)
|
|
|
|
payload = service._extract_response_payload(response)
|
|
|
|
self.assertEqual(payload, {"response": "Resposta simples", "tool_call": None})
|
|
|
|
|
|
class LLMServiceImageWorkflowPromptTests(unittest.TestCase):
|
|
def test_build_image_workflow_prompt_preserves_visible_payment_time(self):
|
|
service = LLMService.__new__(LLMService)
|
|
|
|
prompt = service._build_image_workflow_prompt(caption="Segue o comprovante")
|
|
|
|
self.assertIn(
|
|
"preserve a data e a hora no campo data_pagamento no formato DD/MM/AAAA HH:MM",
|
|
prompt,
|
|
)
|
|
self.assertIn("Nao reduza para somente a data quando a hora estiver visivel.", prompt)
|
|
self.assertIn("marca d'agua exatamente escrita como SysaltiIA", prompt)
|
|
self.assertIn(
|
|
"O comprovante enviado nao e valido. Envie um comprovante valido com a marca d'agua SysaltiIA visivel.",
|
|
prompt,
|
|
)
|
|
self.assertIn(VALID_RECEIPT_WATERMARK_MARKER, prompt)
|
|
self.assertIn("Legenda do usuario: Segue o comprovante", prompt)
|
|
|
|
def test_coerce_image_workflow_response_rejects_payment_without_marker(self):
|
|
service = LLMService.__new__(LLMService)
|
|
|
|
response = service._coerce_image_workflow_response(
|
|
"Registrar pagamento de aluguel: contrato LOC-20260319-33CD6567; valor R$ 379,80."
|
|
)
|
|
|
|
self.assertEqual(response, INVALID_RECEIPT_WATERMARK_MESSAGE)
|
|
|
|
def test_coerce_image_workflow_response_strips_valid_watermark_marker(self):
|
|
service = LLMService.__new__(LLMService)
|
|
|
|
response = service._coerce_image_workflow_response(
|
|
f"{VALID_RECEIPT_WATERMARK_MARKER} Registrar pagamento de aluguel: contrato LOC-20260319-33CD6567; valor R$ 379,80."
|
|
)
|
|
|
|
self.assertEqual(
|
|
response,
|
|
"Registrar pagamento de aluguel: contrato LOC-20260319-33CD6567; valor R$ 379,80.",
|
|
)
|