Skip to content

fix: handle key recreation when team_id changes or key is cascade deleted #24

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
shin-bot-litellm wants to merge 1 commit into
base: main
Choose a base branch
from
Open

fix: handle key recreation when team_id changes or key is cascade deleted #24

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 20 additions & 2 deletions litellm/resource_key.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@ package litellm
import (
"context"
"fmt"
"strings"

"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
Expand Down Expand Up @@ -34,8 +35,10 @@ func resourceKey() *schema.Resource {
Optional: true,
},
"team_id": {
Type: schema.TypeString,
Optional: true,
Type: schema.TypeString,
Optional: true,
ForceNew: true,
Description: "Team ID that owns this key. Changing this will force key recreation.",
},
"max_parallel_requests": {
Type: schema.TypeInt,
Expand Down Expand Up @@ -171,6 +174,21 @@ func resourceKeyUpdate(ctx context.Context, d *schema.ResourceData, m interface{

_, err := c.UpdateKey(key)
if err != nil {
errStr := err.Error()
// Check if the key no longer exists (e.g., cascade deleted when team was deleted)
// In this case, clear the ID so Terraform will recreate it
if strings.Contains(errStr, "does not exist") ||
strings.Contains(errStr, "not found") ||
strings.Contains(errStr, "invalid user key") {
d.SetId("")
return diag.Diagnostics{
diag.Diagnostic{
Severity: diag.Warning,
Summary: "Key no longer exists, will be recreated",
Detail: fmt.Sprintf("The key %s no longer exists in LiteLLM (possibly cascade deleted with team). Terraform will recreate it.", key.Key),
},
}
}
return diag.FromErr(fmt.Errorf("error updating key: %s", err))
}

Expand Down