Skip to content

Commit dfe5ae2

Browse files
committed
Update tests to use ChatMessageFile type
Replaces usage of ChatMessagePartFile with ChatMessageFile in chat_test.go to reflect updated type naming in the openai package. Also renames related test function for consistency.
1 parent d970292 commit dfe5ae2

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

chat_test.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -679,7 +679,7 @@ func TestMultipartChatCompletions(t *testing.T) {
679679
},
680680
{
681681
Type: openai.ChatMessagePartTypeFile,
682-
File: &openai.ChatMessagePartFile{
682+
File: &openai.ChatMessageFile{
683683
FileID: "file-123",
684684
FileName: "test.txt",
685685
FileData: "dGVzdCBmaWxlIGNvbnRlbnQ=", // base64 encoded "test file content"
@@ -768,11 +768,11 @@ func TestMultipartChatMessageSerialization(t *testing.T) {
768768
}
769769
}
770770

771-
func TestChatMessagePartFile(t *testing.T) {
771+
func TestChatMessageFile(t *testing.T) {
772772
// Test file part with FileID
773773
filePart := openai.ChatMessagePart{
774774
Type: openai.ChatMessagePartTypeFile,
775-
File: &openai.ChatMessagePartFile{
775+
File: &openai.ChatMessageFile{
776776
FileID: "file-abc123",
777777
},
778778
}
@@ -809,7 +809,7 @@ func TestChatMessagePartFile(t *testing.T) {
809809
// Test file part with all fields
810810
filePartComplete := openai.ChatMessagePart{
811811
Type: openai.ChatMessagePartTypeFile,
812-
File: &openai.ChatMessagePartFile{
812+
File: &openai.ChatMessageFile{
813813
FileID: "file-xyz789",
814814
FileName: "document.pdf",
815815
FileData: "JVBERi0xLjQK", // base64 for "%PDF-1.4\n"

0 commit comments

Comments
 (0)