vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

server: tests: fix regex content matching

Browse source
This commit is contained in:
Pierrick HYMBERT 2024-03-02 18:57:57 +01:00
parent 407cc609d3
commit 178b0c693d
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/server/tests/features/steps/steps.py
View file

@ -813,7 +813,7 @@ def assert_n_tokens_predicted(completion_response, expected_predicted_n=None, re
re_content = f'^(.*)({re_content})(.*)$' re_content = f'^(.*)({re_content})(.*)$'
p = re.compile(re_content, flags=RegexFlag.IGNORECASE | RegexFlag.MULTILINE | RegexFlag.DOTALL) p = re.compile(re_content, flags=RegexFlag.IGNORECASE | RegexFlag.MULTILINE | RegexFlag.DOTALL)
match = p.match(content) match = p.match(content)
assert match and len(match.groups()) == 3, f'/{re_content}/g must match ```{content}```' assert match and len(match.groups()) >= 3, f'/{re_content}/g must match ```{content}```'
if 'DEBUG' in os.environ and os.environ['DEBUG'] == 'ON': if 'DEBUG' in os.environ and os.environ['DEBUG'] == 'ON':
highlighted = p.sub(r"\1<hi>\2</hi>\3", content).replace('<hi>', '\x1b[33m').replace('</hi>', '\x1b[0m') highlighted = p.sub(r"\1<hi>\2</hi>\3", content).replace('<hi>', '\x1b[33m').replace('</hi>', '\x1b[0m')
print(f"Checking completion response: {highlighted}\n") print(f"Checking completion response: {highlighted}\n")