elfsong
commited on
Commit
·
780de62
1
Parent(s):
314b49d
Update Key
Browse files- src/caller.py +1 -1
src/caller.py
CHANGED
@@ -13,7 +13,7 @@ class OpenAI_Caller:
|
|
13 |
def __init__(self, model_name: str) -> None:
|
14 |
super().__init__()
|
15 |
self.model_name = model_name
|
16 |
-
self.client = OpenAI(api_key=os.
|
17 |
|
18 |
def generate(self, model_inputs: List[object], json_mode=True) -> str:
|
19 |
model_input_prompt = self.prompt_generate(model_inputs, json_mode)
|
|
|
13 |
def __init__(self, model_name: str) -> None:
|
14 |
super().__init__()
|
15 |
self.model_name = model_name
|
16 |
+
self.client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'))
|
17 |
|
18 |
def generate(self, model_inputs: List[object], json_mode=True) -> str:
|
19 |
model_input_prompt = self.prompt_generate(model_inputs, json_mode)
|