21
22
SIMPLE_REASONING = {
23
"output": "This is a reasoning section</think>This is the rest",
24
- "reasoning_content": "This is a reasoning section",
24
+ "reasoning": "This is a reasoning section",
25
"content": "This is the rest",
26
"is_reasoning_end": True,
27
}
28
# need to get into parser again to remove newline after </think>
29
COMPLETE_REASONING = {
30
"output": "This is a reasoning section</think>",
31
- "reasoning_content": "This is a reasoning section",
31
+ "reasoning": "This is a reasoning section",
32
"content": None,
33
"is_reasoning_end": False,
34
}
35
NO_CONTENT = {
36
"output": "This is content",
37
- "reasoning_content": "This is content",
37
+ "reasoning": "This is content",
38
"content": None,
39
"is_reasoning_end": False,
40
}
41
NO_REASONING_STREAMING = {
42
"output": "This is a reasoning section",
43
- "reasoning_content": "This is a reasoning section",
43
+ "reasoning": "This is a reasoning section",
44
"content": None,
45
"is_reasoning_end": False,
46
}
47
MULTIPLE_LINES = {
48
"output": "This\nThat</think>This is the rest\nThat",
49
- "reasoning_content": "This\nThat",
49
+ "reasoning": "This\nThat",
50
"content": "This is the rest\nThat",
51
"is_reasoning_end": True,
52
}
53
SHORTEST_REASONING_NO_STREAMING = {
54
"output": "</think>This is the rest",
55
- "reasoning_content": None,
55
+ "reasoning": None,
56
"content": "This is the rest",
57
"is_reasoning_end": True,
58
}
59
SHORTEST_REASONING = {
60
"output": "</think>This is the rest",
61
- "reasoning_content": None,
61
+ "reasoning": None,
62
"content": "This is the rest",
63
"is_reasoning_end": True,
64
}
65
REASONING_WITH_THINK = {
66
"output": "<think>This is a reasoning section</think>This is the rest",
67
- "reasoning_content": "This is a reasoning section",
67
+ "reasoning": "This is a reasoning section",
68
"content": "This is the rest",
69
"is_reasoning_end": True,
70
}
71
COMPLETE_REASONING_WITH_THINK = {
72
"output": "<think>This is a reasoning section</think>",
73
- "reasoning_content": "This is a reasoning section",
73
+ "reasoning": "This is a reasoning section",
74
"content": None,
75
"is_reasoning_end": False,
76
}
77
MULTIPLE_LINES_WITH_THINK = {
78
"output": "<think>This\nThat</think>This is the rest\nThat",
79
- "reasoning_content": "This\nThat",
79
+ "reasoning": "This\nThat",
80
"content": "This is the rest\nThat",
81
"is_reasoning_end": True,
82
}
83
SHORTEST_REASONING_NO_STREAMING_WITH_THINK = {
84
"output": "</think>This is the rest",
85
- "reasoning_content": None,
85
+ "reasoning": None,
86
"content": "This is the rest",
87
"is_reasoning_end": True,
88
}
89
SHORTEST_REASONING_WITH_THINK = {
90
"output": "</think>This is the rest",
91
- "reasoning_content": None,
91
+ "reasoning": None,
92
"content": "This is the rest",
93
"is_reasoning_end": True,
94
}
95
THINK_NO_END = {
96
"output": "<think>This is a reasoning section",
97
- "reasoning_content": "This is a reasoning section",
97
+ "reasoning": "This is a reasoning section",
98
"content": None,
99
"is_reasoning_end": False,
100
}
101
EMPTY = {
102
"output": "",
103
- "reasoning_content": None,
103
+ "reasoning": None,
104
"content": None,
105
"is_reasoning_end": False,
106
}
107
EMPTY_STREAMING = {
108
"output": "",
109
- "reasoning_content": None,
109
+ "reasoning": None,
110
"content": None,
111
"is_reasoning_end": False,
112
}
113
NEW_LINE = {
114
"output": "\n<think>This is a reasoning section</think>\nThis is the rest",
115
- "reasoning_content": "This is a reasoning section",
115
+ "reasoning": "This is a reasoning section",
116
"content": "This is the rest",
117
"is_reasoning_end": True,
118
}
119
120
NEW_LINE_STREAMING = {
121
"output": "\n<think>This is a reasoning section\n</think>\nThis is the rest",
122
- "reasoning_content": "\nThis is a reasoning section",
122
+ "reasoning": "\nThis is a reasoning section",
123
"content": "This is the rest",
124
"is_reasoning_end": True,
125
}
126
127
NEW_LINE_STREAMING_COMPLEX_CONTENT = {
128
"output": "\n This is a \n reasoning section\n\n\n</think>\n\nThis is the rest",
129
- "reasoning_content": "\n This is a \n reasoning section\n\n",
129
+ "reasoning": "\n This is a \n reasoning section\n\n",
130
"content": "\nThis is the rest",
131
"is_reasoning_end": True,
132
}
133
134
MULTI_TURN_PROMPT_CONTENT = {
135
"output": "<think> This is last turn's reasoning section </think> hello <think>",
136
- "reasoning_content": "",
136
+ "reasoning": "",
137
"content": "",
138
"is_reasoning_end": False,
139
}
296
print(f"content: {content}")
297
test_id = request.node.callspec.id if hasattr(request.node, "callspec") else None
298
if request.node.callspec.id != "multi_turn_prompt_content":
299
- assert reasoning == param_dict["reasoning_content"]
299
+ assert reasoning == param_dict["reasoning"]
300
assert content == param_dict["content"]
301
302
# Test is_reasoning_end