Spaces:
Running
Running
JarvisChan630
commited on
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}'
|