elfsong commited on
Commit
780de62
1 Parent(s): 314b49d

Update Key

Browse files
Files changed (1) hide show
  1. 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.environ["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)
 
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)