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)