diff --git a/03.Code/업로드용/fix_keys.py b/03.Code/업로드용/fix_keys.py new file mode 100644 index 0000000..ca9e8f0 --- /dev/null +++ b/03.Code/업로드용/fix_keys.py @@ -0,0 +1,53 @@ +import os, re + +ROOT = r"D:\for python\geulbeot-light\upload" + +replacements = [ + (r'OPENAI_API_KEY\s*=\s*"sk-proj-[^"]*"', + 'OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")'), + (r'GEMINI_API_KEY\s*=\s*"AIza[^"]*"', + 'GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")'), + (r"api_key='sk-ant-[^']*'", + 'api_key=os.environ.get("ANTHROPIC_API_KEY")'), +] + +targets = [ + "converters/pipeline/step3_domain.py", + "converters/pipeline/step4_chunk.py", + "converters/pipeline/step5_rag.py", + "converters/pipeline/step6_corpus.py", + "converters/pipeline/step7_index.py", + "converters/pipeline/step8_content.py", + "handlers/common.py", +] + +# .env 濡 肄 異媛 +dotenv_snippet = "from dotenv import load_dotenv\nload_dotenv()\n" + +for rel_path in targets: + path = os.path.join(ROOT, rel_path) + with open(path, encoding="utf-8") as f: + content = f.read() + if "load_dotenv" not in content: + content = dotenv_snippet + content + with open(path, "w", encoding="utf-8") as f: + f.write(content) + print(f" + dotenv 異媛: {rel_path}") + changed = False + for pattern, replacement in replacements: + new_content = re.sub(pattern, replacement, content) + if new_content != content: + content = new_content + changed = True + + if changed: + with open(path, "w", encoding="utf-8") as f: + f.write(content) + print(f" + : {rel_path}") + else: + print(f" 蹂寃쎌: {rel_path}") + +print("\n + 猷!")