leandrocarneiro commited on
Commit
57315ef
·
verified ·
1 Parent(s): f7aa9bd

Update llm.py

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