Spaces:

Oldmangrizzz
/

Agent-Node

Paused

Agent-Node / agent_zero_framework /tests /chunk_parser_test.py

Upload folder using huggingface_hub

dff1e71 verified 10 months ago

631 Bytes

import sys
import 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)

	import sys
	import 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)