Tpayne101 commited on
Commit
1a92328
·
verified ·
1 Parent(s): 7466bc6

Update agentos_core.py

Browse files
Files changed (1) hide show
  1. agentos_core.py +3 -3
agentos_core.py CHANGED
@@ -1,12 +1,12 @@
1
  from langchain.memory import ConversationBufferMemory
2
- from langchainchat_openai import ChatOpenAI
3
  from langchain.chains import ConversationChain
4
 
5
  class AgentCore:
6
- def __init__(self, model="gpt-4-turbo"):
7
  self.memory = ConversationBufferMemory()
8
  self.agent = ConversationChain(
9
- llm=ChatOpenAI(model_name=model),
10
  memory=self.memory
11
  )
12
 
 
1
  from langchain.memory import ConversationBufferMemory
2
+ from langchain_openai import ChatOpenAI
3
  from langchain.chains import ConversationChain
4
 
5
  class AgentCore:
6
+ def __init__(self, model="gpt-4o-mini"):
7
  self.memory = ConversationBufferMemory()
8
  self.agent = ConversationChain(
9
+ llm=ChatOpenAI(model=model),
10
  memory=self.memory
11
  )
12