mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-21 15:49:31 +08:00
Add stable ci (#3460)
* add stable ci * fix * update * fix * rename tests dir;fix stable ci bug * add timeout limit * update
This commit is contained in:
48
tests/ce/server/demo.py
Normal file
48
tests/ce/server/demo.py
Normal file
@@ -0,0 +1,48 @@
|
||||
#!/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
# @author DDDivano
|
||||
# encoding=utf-8 vi:ts=4:sw=4:expandtab:ft=python
|
||||
|
||||
from core import TEMPLATE, URL, build_request_payload, send_request
|
||||
|
||||
|
||||
def demo():
|
||||
data = {
|
||||
"stream": False,
|
||||
"messages": [
|
||||
{"role": "system", "content": "You are a helpful assistant."},
|
||||
{"role": "user", "content": "牛顿的三大运动定律是什么?"},
|
||||
],
|
||||
"max_tokens": 3,
|
||||
}
|
||||
payload = build_request_payload(TEMPLATE, data)
|
||||
req = send_request(URL, payload)
|
||||
print(req.json())
|
||||
req = req.json()
|
||||
|
||||
assert req["usage"]["prompt_tokens"] == 22
|
||||
assert req["usage"]["total_tokens"] == 25
|
||||
assert req["usage"]["completion_tokens"] == 3
|
||||
|
||||
|
||||
def test_demo():
|
||||
data = {
|
||||
"stream": False,
|
||||
"messages": [
|
||||
{"role": "system", "content": "You are a helpful assistant."},
|
||||
{"role": "user", "content": "牛顿的三大运动定律是什么?"},
|
||||
],
|
||||
"max_tokens": 3,
|
||||
}
|
||||
payload = build_request_payload(TEMPLATE, data)
|
||||
req = send_request(URL, payload)
|
||||
print(req.json())
|
||||
req = req.json()
|
||||
|
||||
assert req["usage"]["prompt_tokens"] == 22
|
||||
assert req["usage"]["total_tokens"] == 25
|
||||
assert req["usage"]["completion_tokens"] == 5
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
demo()
|
Reference in New Issue
Block a user