fix: openai use responses column migration added and minor fixes

Author: Pratham-Mishra04

core/schemas/mux.go

@@ -809,6 +809,7 @@ func (br *BifrostResponse) ToResponsesOnly() {
 
 	// Clear Chat fields after conversion
 	br.Choices = nil
+	br.Object = "response"
 	br.ExtraFields.RequestType = ResponsesRequest
 }
 

core/schemas/providers/openai/text.go

@@ -29,9 +29,11 @@ func (r *OpenAITextCompletionRequest) ToBifrostRequest() *schemas.BifrostTextCom
 		return nil
 	}
 
+	provider, model := schemas.ParseModelString(r.Model, schemas.OpenAI)
+
 	return &schemas.BifrostTextCompletionRequest{
-		Provider: schemas.OpenAI,
-		Model:    r.Model,
+		Provider: provider,
+		Model:    model,
 		Input:    r.Prompt,
 		Params:   &r.TextCompletionParameters,
 	}

framework/configstore/migrations.go

@@ -22,6 +22,9 @@ func triggerMigrations(ctx context.Context, db *gorm.DB) error {
 	if err := migrationAddVirtualKeyProviderConfigTable(ctx, db); err != nil {
 		return err
 	}
+	if err := migrationAddOpenAIUseResponsesAPIColumn(ctx, db); err != nil {
+		return err
+	}
 	return nil
 }
 
@@ -282,3 +285,25 @@ func migrationAddVirtualKeyProviderConfigTable(ctx context.Context, db *gorm.DB)
 	}
 	return nil
 }
+
+func migrationAddOpenAIUseResponsesAPIColumn(ctx context.Context, db *gorm.DB) error {
+	m := migration.New(db, migration.DefaultOptions, []*migration.Migration{{
+		ID: "add_open_ai_use_responses_api_column",
+		Migrate: func(tx *gorm.DB) error {
+			tx = tx.WithContext(ctx)
+			migrator := tx.Migrator()
+
+			if !migrator.HasColumn(&TableKey{}, "open_ai_use_responses_api") {
+				if err := migrator.AddColumn(&TableKey{}, "open_ai_use_responses_api"); err != nil {
+					return err
+				}
+			}
+			return nil
+		},
+	}})
+	err := m.Migrate()
+	if err != nil {
+		return fmt.Errorf("error while running db migration: %s", err.Error())
+	}
+	return nil
+}

plugins/logging/main.go

@@ -396,6 +396,22 @@ func (p *LoggerPlugin) PostHook(ctx *context.Context, result *schemas.BifrostRes
 					}
 				}
 			}
+			if result.ResponsesResponse != nil {
+				outputMessages := result.ResponsesResponse.Output
+				if len(outputMessages) > 0 {
+					chatMessages := schemas.ToChatMessages(outputMessages)
+					if len(chatMessages) > 0 {
+						lastMessage := chatMessages[len(chatMessages)-1]
+						updateData.OutputMessage = &lastMessage
+
+						// Extract tool calls if present
+						if lastMessage.ChatAssistantMessage != nil &&
+							lastMessage.ChatAssistantMessage.ToolCalls != nil {
+							updateData.ToolCalls = lastMessage.ChatAssistantMessage.ToolCalls
+						}
+					}
+				}
+			}
 			if result.Data != nil {
 				updateData.EmbeddingOutput = result.Data
 			}