wt002 commited on
Commit
1e58dde
·
verified ·
1 Parent(s): d036780

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -0
agent.py CHANGED
@@ -45,6 +45,7 @@ import torch
45
  import torch.nn.functional as F
46
  from langchain.tools import Tool
47
  from langchain.agents import initialize_agent, AgentType
 
48
  import time
49
 
50
  load_dotenv()
 
45
  import torch.nn.functional as F
46
  from langchain.tools import Tool
47
  from langchain.agents import initialize_agent, AgentType
48
+ from langchain_community.chat_models import ChatOpenAI
49
  import time
50
 
51
  load_dotenv()