File size: 946 Bytes
7c01a62 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
from langchain import hub
from langchain.agents import AgentExecutor, create_openai_functions_agent
from langchain_openai import ChatOpenAI
from langchain.prompts import PromptTemplate
from reasoning_tools import animal_tools, extract_vote
# LLM Configuration for each role
llm_parameters = {
"chameleon": {
'model': 'gpt-4-turbo-preview',
'temperature': 1
},
"herd": {
'model': 'gpt-3.5-turbo',
'temperature': 1
},
"judge": {
'model': 'gpt-3.5-turbo',
'temperature': 1
}
}
prompt = hub.pull("hwchase17/openai-functions-agent")
class PlayerAgent(AgentExecutor):
def __init__(self, role):
llm = ChatOpenAI(**llm_parameters[role])
agent = create_openai_functions_agent(llm, animal_tools, prompt)
super().__init__(agent=agent, tools=animal_tools, verbose=True, return_intermediate_steps=True)
|