20 lines
482 B
Python
20 lines
482 B
Python
"""Test de los modelos base del proveedor LLM."""
|
|
|
|
from forja_core.llm.base import CompletionResult, Message
|
|
|
|
|
|
def test_message_serializable() -> None:
|
|
m = Message(role="user", content="hola")
|
|
assert m.model_dump()["role"] == "user"
|
|
|
|
|
|
def test_completion_result_campos_basicos() -> None:
|
|
r = CompletionResult(
|
|
content='{"a": 1}',
|
|
model="gpt-4o-mock",
|
|
tokens_in=10,
|
|
tokens_out=20,
|
|
latency_ms=42,
|
|
)
|
|
assert r.tokens_in == 10
|