app/styles/components/code.scss

.actions .shiki {
  background-color: var(--bolt-elements-actions-code-background) !important;
}

.shiki {
  &:not(:has(.actions), .actions *) {
    background-color: var(--bolt-elements-messages-code-background) !important;
  }
}