84ad99b98b
Some checks failed
CI / skinny-install (aco) (push) Successful in 1m10s
CI / skinny-install (api) (push) Successful in 32s
CI / skinny-install (bcda) (push) Successful in 37s
CI / skinny-install (bib) (push) Successful in 35s
CI / skinny-install (bls) (push) Successful in 27s
CI / skinny-install (ccw) (push) Successful in 27s
CI / skinny-install (cli) (push) Successful in 39s
CI / skinny-install (cms) (push) Successful in 30s
CI / skinny-install (conf) (push) Successful in 27s
CI / skinny-install (opps) (push) Successful in 31s
CI / skinny-install (perf) (push) Successful in 40s
CI / skinny-install (pfs) (push) Successful in 36s
CI / skinny-install (rex) (push) Successful in 36s
CI / lint-test (push) Failing after 10m34s
Deploy / build-scan-report (push) Failing after 5m27s
Covers iom, oig, cli/{mail,prisma,zot}, mail/{cloudflare,postmark,
droplet,resend}, prisma/{vpn,screen,llm,project,export,eligibility,
extract,flow,ingest,tools}. Import + smoke tests with mocked deps.
Tracks #353.
20 lines
520 B
Python
20 lines
520 B
Python
"""Tests for prisma.llm — provider abstraction."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from prisma.llm import LLMCall, LLMMessage, LLMResult
|
|
|
|
|
|
class TestLLMCall:
|
|
def test_create(self):
|
|
call = LLMCall(messages=[LLMMessage(role="user", content="hi")], max_tokens=10)
|
|
assert len(call.messages) == 1
|
|
|
|
|
|
class TestLLMResult:
|
|
def test_create(self):
|
|
r = LLMResult(
|
|
text="pong", tool_calls=[], usage={"input_tokens": 1, "output_tokens": 1}
|
|
)
|
|
assert r.text == "pong"
|