Spaces:
Sleeping
Sleeping
Commit ·
d597cd8
1
Parent(s): 069a92b
llm gen modified
Browse files- src/llm_generator.py +2 -1
src/llm_generator.py
CHANGED
|
@@ -8,7 +8,8 @@ from openai import OpenAI
|
|
| 8 |
|
| 9 |
# load_dotenv()
|
| 10 |
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
| 11 |
-
|
|
|
|
| 12 |
|
| 13 |
TMP_DIR = Path("/tmp/llm_attachments")
|
| 14 |
TMP_DIR.mkdir(parents=True, exist_ok=True)
|
|
|
|
| 8 |
|
| 9 |
# load_dotenv()
|
| 10 |
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
| 11 |
+
OPENAI_BASE_URL = os.getenv("OPENAI_BASE_URL", "https://api.openai.com/v1")
|
| 12 |
+
client = OpenAI(api_key=OPENAI_API_KEY, base_url=OPENAI_BASE_URL)
|
| 13 |
|
| 14 |
TMP_DIR = Path("/tmp/llm_attachments")
|
| 15 |
TMP_DIR.mkdir(parents=True, exist_ok=True)
|