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.
74 lines
2.6 KiB
Python
74 lines
2.6 KiB
Python
import os
|
|
import unittest
|
|
from types import SimpleNamespace
|
|
|
|
os.environ.setdefault("DEBUG", "false")
|
|
|
|
from app.services.ai.llm_service import 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("Legenda do usuario: Segue o comprovante", prompt)
|