Spaces:
Paused
Paused
File size: 625 Bytes
8d1819a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import sys, os
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
import models
ex1 = "<think>reasoning goes here</think>response goes here"
ex2 = "<think>reasoning goes here</thi"
import pytest
@pytest.mark.parametrize("example", [ex1, ex2])
def test_example(example: str):
res = models.ChatGenerationResult()
for i in range(len(example)):
char = example[i]
chunk = res.add_chunk({"response_delta": char, "reasoning_delta": ""})
print(i, ":", chunk)
print("output", res.output())
if __name__ == "__main__":
# test_example(ex1)
test_example(ex2)
|