Skip to content

Commit ab3fcd9

Browse files
fix(#120): Update environment variable names to use API_KEY instead of TOKEN (#122)
1 parent 9c234e2 commit ab3fcd9

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

internal/env/env.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,9 @@ import (
1212
func Token(path, provider string) string {
1313
switch provider {
1414
case "deepseek":
15-
return find(path, "DEEPSEEK_TOKEN")
15+
return find(path, "DEEPSEEK_API_KEY")
1616
case "openai":
17-
return find(path, "OPENAI_TOKEN")
17+
return find(path, "OPENAI_API_KEY")
1818
case "mock":
1919
return find(path, "MOCK_TOKEN")
2020
default:

internal/env/env_test.go

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ func TestProviderToken_DeepseekTokenPresent(t *testing.T) {
6767
tmp := t.TempDir()
6868
deepseek := "deepseek-token-value"
6969
env := filepath.Join(tmp, ".env")
70-
err := os.WriteFile(env, fmt.Appendf(nil, "DEEPSEEK_TOKEN=%s", deepseek), 0o600)
70+
err := os.WriteFile(env, fmt.Appendf(nil, "DEEPSEEK_API_KEY=%s", deepseek), 0o600)
7171
require.NoError(t, err)
7272

7373
result := Token(env, "deepseek")
@@ -79,7 +79,7 @@ func TestProviderToken_OpenAITokenPresent(t *testing.T) {
7979
tmp := t.TempDir()
8080
openai := "openai-token-value"
8181
env := filepath.Join(tmp, ".env")
82-
err := os.WriteFile(env, fmt.Appendf(nil, "OPENAI_TOKEN=%s", openai), 0o600)
82+
err := os.WriteFile(env, fmt.Appendf(nil, "OPENAI_API_KEY=%s", openai), 0o600)
8383
require.NoError(t, err)
8484

8585
result := Token(env, "openai")
@@ -124,7 +124,7 @@ func TestProviderToken_DefaultTokenIgnoredWhenDeepseekTokenExists(t *testing.T)
124124
deepseek := "deepseek-token-val"
125125
token := "default-ignored-deepseek-token-val"
126126
env := filepath.Join(tmp, ".env")
127-
err := os.WriteFile(env, fmt.Appendf(nil, "DEEPSEEK_TOKEN=%s\nTOKEN=%s\n", deepseek, token), 0o600)
127+
err := os.WriteFile(env, fmt.Appendf(nil, "DEEPSEEK_API_KEY=%s\nTOKEN=%s\n", deepseek, token), 0o600)
128128
require.NoError(t, err)
129129

130130
result := Token(env, "deepseek")
@@ -137,7 +137,7 @@ func TestProviderToken_DefaultTokenIgnoredWhenOpenAITokenExists(t *testing.T) {
137137
openai := "openai-token-val"
138138
token := "default-openai-token-val"
139139
env := filepath.Join(tmp, ".env")
140-
err := os.WriteFile(env, fmt.Appendf(nil, "OPENAI_TOKEN=%s\nTOKEN=%s\n", openai, token), 0o600)
140+
err := os.WriteFile(env, fmt.Appendf(nil, "OPENAI_API_KEY=%s\nTOKEN=%s\n", openai, token), 0o600)
141141
require.NoError(t, err)
142142

143143
result := Token(env, "openai")
@@ -150,7 +150,7 @@ func TestProviderToken_DefaultTokenUsedWhenDeepseekNotRequested(t *testing.T) {
150150
deepseek := "deepseek-token"
151151
token := "default-used-deepseek-token"
152152
env := filepath.Join(tmp, ".env")
153-
err := os.WriteFile(env, fmt.Appendf(nil, "DEEPSEEK_TOKEN=%s\nTOKEN=%s\n", deepseek, token), 0o600)
153+
err := os.WriteFile(env, fmt.Appendf(nil, "DEEPSEEK_API_KEY=%s\nTOKEN=%s\n", deepseek, token), 0o600)
154154
require.NoError(t, err)
155155

156156
result := Token(env, "otherprovider")
@@ -163,7 +163,7 @@ func TestProviderToken_DefaultTokenUsedWhenOpenAINotRequested(t *testing.T) {
163163
openai := "openai-token"
164164
token := "default-openai-token"
165165
env := filepath.Join(tmp, ".env")
166-
err := os.WriteFile(env, fmt.Appendf(nil, "OPENAI_TOKEN=%s\nTOKEN=%s\n", openai, token), 0o600)
166+
err := os.WriteFile(env, fmt.Appendf(nil, "OPENAI_API_KEY=%s\nTOKEN=%s\n", openai, token), 0o600)
167167
require.NoError(t, err)
168168

169169
result := Token(env, "otherprovider")

0 commit comments

Comments
 (0)