from langchain.agents import AgentExecutor
from langchain.agents.openai_functions_agent.base import OpenAIFunctionsAgent
from langchain.callbacks.stdout import StdOutCallbackHandler
from langchain.chat_models import ChatOpenAI
from config import Config
from modules.ask.prompt import system_message
from modules.ask.tools import tools
llm = ChatOpenAI(
model='gpt-3.5-turbo-0613',
temperature=0,
openai_api_key=Config('openai_api_key'),
model_kwargs={
'frequency_penalty': 0.0,
'presence_penalty': 0.0})
agent = OpenAIFunctionsAgent.from_llm_and_tools(llm=llm, tools=tools,
callback_manager=StdOutCallbackHandler, system_message=system_message)
agent_executor = AgentExecutor.from_agent_and_tools(agent=agent, tools=tools, verbose=True)
