Public Access
20 lines
632 B
Python
20 lines
632 B
Python
import re
|
|
|
|
with open("/opt/litellm/config.yaml") as f:
|
|
c = f.read()
|
|
|
|
keys = re.findall(r'api_key:\s*"([^"]+)"', c)
|
|
gemini = [k for k in keys if k.startswith("AIza")]
|
|
github = [k for k in keys if k.startswith("ghp_")]
|
|
groq = [k for k in keys if k.startswith("gsk_")]
|
|
|
|
with open("/etc/litellm/.env", "a") as f:
|
|
for i, k in enumerate(gemini):
|
|
f.write(f"GEMINI_API_KEY_{i+1}={k}\n")
|
|
for i, k in enumerate(github):
|
|
f.write(f"GITHUB_TOKEN_{i+1}={k}\n")
|
|
for i, k in enumerate(groq):
|
|
f.write(f"GROQ_API_KEY_{i+1}={k}\n")
|
|
|
|
print(f"Added: {len(gemini)} Gemini, {len(github)} GitHub, {len(groq)} Groq")
|