| import os | |
| os.environ['TOKENIZERS_PARALLELISM'] = 'true' | |
| os.environ['MISTRAL_API_KEY'] = "i5jSJkCFNGKfgIztloxTMjfckiFbYBj4" | |
| from langchain import hub | |
| from langchain.agents import AgentExecutor, create_json_chat_agent | |
| from langchain_mistralai.chat_models import ChatMistralAI | |
| prompt = hub.pull("hwchase17/react-chat-json") | |
| from langchain_community.tools import DuckDuckGoSearchRun | |
| tools = [DuckDuckGoSearchRun()] | |
| llm = ChatMistralAI(model='mistral-large-latest') | |
| agent = create_json_chat_agent( | |
| llm=llm, | |
| tools=tools, | |
| prompt=prompt, | |
| ) | |
| agent_executor = AgentExecutor( | |
| agent=agent, | |
| tools=tools, | |
| verbose=True, | |
| handle_parsing_errors=True | |
| ) | |
| agent_executor.invoke({"input":"How many proteins should I eat per day? Search mainly on wikipedia"}) | |