leandrocarneiro commited on
Commit
3a80ab5
·
verified ·
1 Parent(s): c057cd2

Update llm.py

Browse files
Files changed (1) hide show
  1. llm.py +1 -1
llm.py CHANGED
@@ -16,7 +16,7 @@ def invoke_llm(context, task):
16
 
17
  llm=ChatOpenAI(model_name="gpt-3.5-turbo",
18
  temperature=0,
19
- openai_api_key=os.getenv('OPENAI_KEY'),
20
  max_tokens=1000)
21
  result = llm.invoke(prompt)
22
  return result.content
 
16
 
17
  llm=ChatOpenAI(model_name="gpt-3.5-turbo",
18
  temperature=0,
19
+ openai_api_key=os.environ['OPENAI_KEY'],
20
  max_tokens=1000)
21
  result = llm.invoke(prompt)
22
  return result.content