YvesP commited on
Commit
b1a81ef
1 Parent(s): 5de1276

Update src/tools/llms.py

Browse files

changed the langchain llm import

Files changed (1) hide show
  1. src/tools/llms.py +1 -1
src/tools/llms.py CHANGED
@@ -1,7 +1,7 @@
1
 
2
  import os
3
 
4
- from langchain.llms import OpenAI
5
 
6
  if not "OPENAI_API_KEY" in os.environ:
7
  os.environ['OPENAI_API_KEY'] = OPENAI_API_KEY
 
1
 
2
  import os
3
 
4
+ from langchain_openai import OpenAI
5
 
6
  if not "OPENAI_API_KEY" in os.environ:
7
  os.environ['OPENAI_API_KEY'] = OPENAI_API_KEY