fix: integration test cases fixes

Author: TejasGhatte

core/changelog.md

@@ -1,4 +1,6 @@
 feat: send back raw request in extra fields
 feat: added support for reasoning in chat completions
 feat: enhanced reasoning support in responses api
-enhancement: improved internal inter provider conversions for integrations
+enhancement: improved internal inter provider conversions for integrations
+feat: switched to gemini native api
+feat: fallback to supported request type for custom models used in integration

core/providers/cohere/chat.go

@@ -688,5 +688,11 @@ func (cm *CohereMessage) ToBifrostChatMessage() *schemas.ChatMessage {
 		Content:              messageContent,
 		ChatAssistantMessage: assistantMessage,
 	}
+
+	if cm.Role == "tool" {
+		bifrostMessage.ChatToolMessage = &schemas.ChatToolMessage{
+			ToolCallID: cm.ToolCallID,
+		}
+	}
 	return bifrostMessage
 }

core/providers/gemini/chat.go

@@ -256,15 +256,6 @@ func (response *GenerateContentResponse) ToBifrostChatCompletionStream() (*schem
 					},
 				}
 
-				// Preserve thought signature if present (required for Gemini 3 Pro)
-				if len(part.ThoughtSignature) > 0 {
-					toolCall.ExtraContent = map[string]interface{}{
-						"google": map[string]interface{}{
-							"thought_signature": string(part.ThoughtSignature),
-						},
-					}
-				}
-
 				toolCalls = append(toolCalls, toolCall)
 
 			case part.FunctionResponse != nil:

core/providers/gemini/gemini.go

@@ -245,7 +245,7 @@ func (provider *GeminiProvider) ChatCompletion(ctx context.Context, key schemas.
 
 	// Set raw request if enabled
 	if providerUtils.ShouldSendBackRawRequest(ctx, provider.sendBackRawRequest) {
-		bifrostResponse.ExtraFields.RawRequest = jsonData
+		providerUtils.ParseAndSetRawRequest(&bifrostResponse.ExtraFields, jsonData)
 	}
 
 	// Set raw response if enabled