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

server: tests: removing debug print

Browse source
This commit is contained in:
Pierrick HYMBERT 2024-02-24 08:23:19 +01:00
parent 2d107babc4
commit 124ca773c6
1 changed files with 0 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -480,8 +480,6 @@ async def oai_chat_completions(user_prompt,
json=payload, json=payload,
headers=headers) as response: headers=headers) as response:
if enable_streaming: if enable_streaming:
# FIXME: does not work; the server is generating only one token
print("DEBUG payload", payload)
assert response.status == 200 assert response.status == 200
assert response.headers['Access-Control-Allow-Origin'] == origin assert response.headers['Access-Control-Allow-Origin'] == origin
assert response.headers['Content-Type'] == "text/event-stream" assert response.headers['Content-Type'] == "text/event-stream"
@ -503,7 +501,6 @@ async def oai_chat_completions(user_prompt,
if 'content' in delta: if 'content' in delta:
completion_response['content'] += delta['content'] completion_response['content'] += delta['content']
completion_response['timings']['predicted_n'] += 1 completion_response['timings']['predicted_n'] += 1
print(f"DEBUG completion_response: {completion_response}")
else: else:
if expect_api_error is None or not expect_api_error: if expect_api_error is None or not expect_api_error:
assert response.status == 200 assert response.status == 200