adds new file-supported-batch flag for tests

core/internal/testutil/account.go

@@ -51,6 +51,7 @@ type TestScenarios struct {
 	FileRetrieve          bool // File API retrieve functionality
 	FileDelete            bool // File API delete functionality
 	FileContent           bool // File API content download functionality
+	FileBatchInput        bool // Whether batch create supports file-based input (InputFileID)
 }
 
 // ComprehensiveTestConfig extends TestConfig with additional scenarios

core/internal/testutil/batch.go

@@ -627,10 +627,10 @@ func RunFileUnsupportedTest(t *testing.T, client *bifrost.Bifrost, ctx context.C
 
 // RunFileAndBatchIntegrationTest tests the integration between file upload and batch create
 func RunFileAndBatchIntegrationTest(t *testing.T, client *bifrost.Bifrost, ctx context.Context, testConfig ComprehensiveTestConfig) {
-	// Skip if either file upload or batch create is not supported
-	if !testConfig.Scenarios.FileUpload || !testConfig.Scenarios.BatchCreate {
-		t.Logf("[SKIPPED] File and Batch Integration: FileUpload=%v, BatchCreate=%v for provider %s",
-			testConfig.Scenarios.FileUpload, testConfig.Scenarios.BatchCreate, testConfig.Provider)
+	// Skip if file-based batch input is not supported
+	if !testConfig.Scenarios.FileBatchInput {
+		t.Logf("[SKIPPED] File and Batch Integration: FileBatchInput=%v for provider %s",
+			testConfig.Scenarios.FileBatchInput, testConfig.Provider)
 		return
 	}
 

core/internal/testutil/tests.go

@@ -115,7 +115,7 @@ func printTestSummary(t *testing.T, testConfig ComprehensiveTestConfig) {
 		{"FileDelete", testConfig.Scenarios.FileDelete},
 		{"FileContent", testConfig.Scenarios.FileContent},
 		{"FileUnsupported", !testConfig.Scenarios.FileUpload && !testConfig.Scenarios.FileList && !testConfig.Scenarios.FileRetrieve && !testConfig.Scenarios.FileDelete && !testConfig.Scenarios.FileContent},
-		{"FileAndBatchIntegration", testConfig.Scenarios.FileUpload && testConfig.Scenarios.BatchCreate},
+		{"FileAndBatchIntegration", testConfig.Scenarios.FileBatchInput},
 	}
 
 	supported := 0

core/providers/anthropic/anthropic_test.go

@@ -57,7 +57,8 @@ func TestAnthropic(t *testing.T) {
 			FileList:              true,
 			FileRetrieve:          true,
 			FileDelete:            true,
-			FileContent:           true,
+			FileContent:           false,
+			FileBatchInput:        false, // Anthropic batch API only supports inline requests, not file-based input
 		},
 	}
 

core/providers/bedrock/bedrock_test.go

@@ -79,6 +79,7 @@ func TestBedrock(t *testing.T) {
 			FileRetrieve:          true,
 			FileDelete:            true,
 			FileContent:           true,
+			FileBatchInput:        true,
 		},
 	}
 

core/providers/gemini/batch.go

@@ -64,34 +64,55 @@ func buildBatchRequestItems(requests []schemas.BatchRequestItem) []GeminiBatchRe
 			requestData = req.Params
 		}
 
-		// Extract messages from the request data
+		// Extract messages from the request data - handle multiple possible types
+		// Go type assertions don't work across slice types, so we handle each case
 		if requestData != nil {
-			if msgs, ok := requestData["messages"].([]interface{}); ok {
-				for _, msg := range msgs {
-					if msgMap, ok := msg.(map[string]interface{}); ok {
-						role := "user"
-						if r, ok := msgMap["role"].(string); ok {
-							if r == "assistant" {
-								role = "model"
-							} else if r == "system" {
-								// System messages are handled separately in Gemini
-								continue
-							} else {
-								role = r
-							}
-						}
+			var messages []map[string]interface{}
 
-						parts := []*Part{}
-						if c, ok := msgMap["content"].(string); ok {
-							parts = append(parts, &Part{Text: c})
-						}
+			// Try []interface{} first (generic JSON unmarshaling)
+			if msgsInterface, ok := requestData["messages"].([]interface{}); ok {
+				for _, m := range msgsInterface {
+					if msgMap, ok := m.(map[string]interface{}); ok {
+						messages = append(messages, msgMap)
+					}
+				}
+			} else if msgsTyped, ok := requestData["messages"].([]map[string]interface{}); ok {
+				// Try []map[string]interface{} (typed maps)
+				messages = msgsTyped
+			} else if msgsString, ok := requestData["messages"].([]map[string]string); ok {
+				// Try []map[string]string (test case format)
+				for _, m := range msgsString {
+					msgMap := make(map[string]interface{})
+					for k, v := range m {
+						msgMap[k] = v
+					}
+					messages = append(messages, msgMap)
+				}
+			}
 
-						contents = append(contents, Content{
-							Role:  role,
-							Parts: parts,
-						})
+			// Process extracted messages
+			for _, msgMap := range messages {
+				role := "user"
+				if r, ok := msgMap["role"].(string); ok {
+					if r == "assistant" {
+						role = "model"
+					} else if r == "system" {
+						// System messages are handled separately in Gemini
+						continue
+					} else {
+						role = r
 					}
 				}
+
+				parts := []*Part{}
+				if c, ok := msgMap["content"].(string); ok {
+					parts = append(parts, &Part{Text: c})
+				}
+
+				contents = append(contents, Content{
+					Role:  role,
+					Parts: parts,
+				})
 			}
 		}
 

core/providers/gemini/gemini_test.go

@@ -63,7 +63,8 @@ func TestGemini(t *testing.T) {
 			FileList:              true,
 			FileRetrieve:          true,
 			FileDelete:            true,
-			FileContent:           true,
+			FileContent:           false,
+			FileBatchInput:        true,
 		},
 	}
 

core/providers/openai/openai_test.go

@@ -70,6 +70,7 @@ func TestOpenAI(t *testing.T) {
 			FileRetrieve:          true,
 			FileDelete:            true,
 			FileContent:           true,
+			FileBatchInput:        true,
 		},
 	}