mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-03 01:40:21 +08:00
20 lines
723 B
Python
20 lines
723 B
Python
|
from src.api.handlers.gemini.stream_parser import GeminiStreamParser
|
||
|
|
|
||
|
|
|
||
|
|
def test_is_done_event_true_for_new_finish_reason_values() -> None:
|
||
|
|
parser = GeminiStreamParser()
|
||
|
|
event = {"candidates": [{"finishReason": "MALFORMED_FUNCTION_CALL"}]}
|
||
|
|
assert parser.is_done_event(event) is True
|
||
|
|
|
||
|
|
|
||
|
|
def test_is_done_event_false_for_unspecified() -> None:
|
||
|
|
parser = GeminiStreamParser()
|
||
|
|
event = {"candidates": [{"finishReason": "FINISH_REASON_UNSPECIFIED"}]}
|
||
|
|
assert parser.is_done_event(event) is False
|
||
|
|
|
||
|
|
|
||
|
|
def test_is_done_event_false_when_no_candidates_or_reason() -> None:
|
||
|
|
parser = GeminiStreamParser()
|
||
|
|
assert parser.is_done_event({}) is False
|
||
|
|
assert parser.is_done_event({"candidates": [{}]}) is False
|