Update content_generation.py
Browse files- content_generation.py +5 -0
content_generation.py
CHANGED
|
@@ -1,9 +1,14 @@
|
|
| 1 |
# content_generation.py
|
| 2 |
from groq import Groq
|
|
|
|
| 3 |
import os
|
| 4 |
|
| 5 |
# Lấy API key từ biến môi trường
|
| 6 |
GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
|
| 8 |
# Danh sách loại nội dung và hướng dẫn mặc định cho từng loại
|
| 9 |
CONTENT_TYPES = ["triết lý sống", "Theo yêu cầu"]
|
|
|
|
| 1 |
# content_generation.py
|
| 2 |
from groq import Groq
|
| 3 |
+
from openai import OpenAI
|
| 4 |
import os
|
| 5 |
|
| 6 |
# Lấy API key từ biến môi trường
|
| 7 |
GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
|
| 8 |
+
OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
|
| 9 |
+
|
| 10 |
+
# Khởi tạo client OpenAI
|
| 11 |
+
openai_client = OpenAI(api_key=OPENAI_API_KEY)
|
| 12 |
|
| 13 |
# Danh sách loại nội dung và hướng dẫn mặc định cho từng loại
|
| 14 |
CONTENT_TYPES = ["triết lý sống", "Theo yêu cầu"]
|