File tree Expand file tree Collapse file tree 1 file changed +20
-1
lines changed Expand file tree Collapse file tree 1 file changed +20
-1
lines changed Original file line number Diff line number Diff line change @@ -80,7 +80,7 @@ def test_stop_sequence():
80
80
"messages" : [
81
81
{
82
82
"role" : "user" ,
83
- "content" : "你要严格按照我接下来的话输出,输出冒号后面的内容,请输出:这是第一段。这是第二段啦啦啦啦啦 。" ,
83
+ "content" : "你要严格按照我接下来的话输出,输出冒号后面的内容,请输出:这是第一段。果冻这是第二段啦啦啦啦啦 。" ,
84
84
},
85
85
],
86
86
"max_tokens" : 20 ,
@@ -96,6 +96,25 @@ def test_stop_sequence():
96
96
assert "。" in token_list , "没有找到。符号"
97
97
98
98
99
+ def test_stop_sequence1 ():
100
+ data = {
101
+ "stream" : False ,
102
+ "messages" : [
103
+ {
104
+ "role" : "user" ,
105
+ "content" : "你要严格按照我接下来的话输出,输出冒号后面的内容,请输出:这是第一段。果冻这是第二段啦啦啦啦啦。" ,
106
+ },
107
+ ],
108
+ "max_tokens" : 20 ,
109
+ "top_p" : 0 ,
110
+ }
111
+ payload = build_request_payload (TEMPLATE , data )
112
+ resp = send_request (URL , payload ).json ()
113
+ content = resp ["choices" ][0 ]["message" ]["content" ]
114
+ print ("截断输出:" , content )
115
+ assert "第二段" in content
116
+
117
+
99
118
def test_sampling_parameters ():
100
119
data = {
101
120
"stream" : False ,
You can’t perform that action at this time.
0 commit comments