Spaces:
Sleeping
Sleeping
Commit
·
62f4f30
1
Parent(s):
ca3b1c9
fix bugs
Browse files- models/llms.py +1 -1
models/llms.py
CHANGED
|
@@ -397,7 +397,7 @@ class OpenAIModel(BaseModel):
|
|
| 397 |
load_dotenv()
|
| 398 |
self.model_endpoint = 'https://api.302.ai/v1/chat/completions'
|
| 399 |
|
| 400 |
-
|
| 401 |
self.headers = {
|
| 402 |
'Content-Type': 'application/json',
|
| 403 |
'Authorization': f'Bearer {self.api_key}'
|
|
|
|
| 397 |
load_dotenv()
|
| 398 |
self.model_endpoint = 'https://api.302.ai/v1/chat/completions'
|
| 399 |
|
| 400 |
+
self.api_key = os.environ.get('OPENAI_API_KEY')
|
| 401 |
self.headers = {
|
| 402 |
'Content-Type': 'application/json',
|
| 403 |
'Authorization': f'Bearer {self.api_key}'
|