@@ -30,6 +30,8 @@ public void ItCreatesOpenAIExecutionSettingsWithCorrectDefaults()
30
30
Assert . Equal ( 1 , executionSettings . ResultsPerPrompt ) ;
31
31
Assert . Null ( executionSettings . StopSequences ) ;
32
32
Assert . Null ( executionSettings . TokenSelectionBiases ) ;
33
+ Assert . Null ( executionSettings . TopLogprobs ) ;
34
+ Assert . Null ( executionSettings . Logprobs ) ;
33
35
Assert . Equal ( 128 , executionSettings . MaxTokens ) ;
34
36
}
35
37
@@ -47,6 +49,8 @@ public void ItUsesExistingOpenAIExecutionSettings()
47
49
StopSequences = new string [ ] { "foo" , "bar" } ,
48
50
ChatSystemPrompt = "chat system prompt" ,
49
51
MaxTokens = 128 ,
52
+ Logprobs = true ,
53
+ TopLogprobs = 5 ,
50
54
TokenSelectionBiases = new Dictionary < int , int > ( ) { { 1 , 2 } , { 3 , 4 } } ,
51
55
} ;
52
56
@@ -97,6 +101,8 @@ public void ItCreatesOpenAIExecutionSettingsFromExtraPropertiesSnakeCase()
97
101
{ "max_tokens" , 128 } ,
98
102
{ "token_selection_biases" , new Dictionary < int , int > ( ) { { 1 , 2 } , { 3 , 4 } } } ,
99
103
{ "seed" , 123456 } ,
104
+ { "logprobs" , true } ,
105
+ { "top_logprobs" , 5 } ,
100
106
}
101
107
} ;
102
108
@@ -105,7 +111,6 @@ public void ItCreatesOpenAIExecutionSettingsFromExtraPropertiesSnakeCase()
105
111
106
112
// Assert
107
113
AssertExecutionSettings ( executionSettings ) ;
108
- Assert . Equal ( executionSettings . Seed , 123456 ) ;
109
114
}
110
115
111
116
[ Fact ]
@@ -124,7 +129,10 @@ public void ItCreatesOpenAIExecutionSettingsFromExtraPropertiesAsStrings()
124
129
{ "stop_sequences" , new [ ] { "foo" , "bar" } } ,
125
130
{ "chat_system_prompt" , "chat system prompt" } ,
126
131
{ "max_tokens" , "128" } ,
127
- { "token_selection_biases" , new Dictionary < string , string > ( ) { { "1" , "2" } , { "3" , "4" } } }
132
+ { "token_selection_biases" , new Dictionary < string , string > ( ) { { "1" , "2" } , { "3" , "4" } } } ,
133
+ { "seed" , 123456 } ,
134
+ { "logprobs" , true } ,
135
+ { "top_logprobs" , 5 }
128
136
}
129
137
} ;
130
138
@@ -149,7 +157,10 @@ public void ItCreatesOpenAIExecutionSettingsFromJsonSnakeCase()
149
157
"stop_sequences": [ "foo", "bar" ],
150
158
"chat_system_prompt": "chat system prompt",
151
159
"token_selection_biases": { "1": 2, "3": 4 },
152
- "max_tokens": 128
160
+ "max_tokens": 128,
161
+ "seed": 123456,
162
+ "logprobs": true,
163
+ "top_logprobs": 5
153
164
}
154
165
""" ;
155
166
var actualSettings = JsonSerializer . Deserialize < PromptExecutionSettings > ( json ) ;
@@ -255,5 +266,8 @@ private static void AssertExecutionSettings(OpenAIPromptExecutionSettings execut
255
266
Assert . Equal ( "chat system prompt" , executionSettings . ChatSystemPrompt ) ;
256
267
Assert . Equal ( new Dictionary < int , int > ( ) { { 1 , 2 } , { 3 , 4 } } , executionSettings . TokenSelectionBiases ) ;
257
268
Assert . Equal ( 128 , executionSettings . MaxTokens ) ;
269
+ Assert . Equal ( 123456 , executionSettings . Seed ) ;
270
+ Assert . Equal ( true , executionSettings . Logprobs ) ;
271
+ Assert . Equal ( 5 , executionSettings . TopLogprobs ) ;
258
272
}
259
273
}
0 commit comments