forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
359 lines
11 KiB
Plaintext
359 lines
11 KiB
Plaintext
2 years ago
|
{
|
||
|
"cells": [
|
||
|
{
|
||
|
"cell_type": "markdown",
|
||
|
"id": "ba5f8741",
|
||
|
"metadata": {},
|
||
|
"source": [
|
||
|
"# Custom Agent\n",
|
||
|
"\n",
|
||
|
"This notebook goes through how to create your own custom agent.\n",
|
||
|
"\n",
|
||
|
"An agent consists of three parts:\n",
|
||
|
" \n",
|
||
|
" - Tools: The tools the agent has available to use.\n",
|
||
|
" - LLMChain: The LLMChain that produces the text that is parsed in a certain way to determine which action to take.\n",
|
||
|
" - The agent class itself: this parses the output of the LLMChain to determin which action to take.\n",
|
||
|
" \n",
|
||
|
" \n",
|
||
|
"In this notebook we walk through two types of custom agents. The first type shows how to create a custom LLMChain, but still use an existing agent class to parse the output. The second shows how to create a custom agent class."
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "markdown",
|
||
|
"id": "6064f080",
|
||
|
"metadata": {},
|
||
|
"source": [
|
||
|
"### Custom LLMChain\n",
|
||
|
"\n",
|
||
|
"The first way to create a custom agent is to use an existing Agent class, but use a custom LLMChain. This is the simplest way to create a custom Agent. It is highly reccomended that you work with the `ZeroShotAgent`, as at the moment that is by far the most generalizable one. \n",
|
||
|
"\n",
|
||
1 year ago
|
"Most of the work in creating the custom LLMChain comes down to the prompt. Because we are using an existing agent class to parse the output, it is very important that the prompt say to produce text in that format. Additionally, we currently require an `agent_scratchpad` input variable to put notes on previous actions and observations. This should almost always be the final part of the prompt. However, besides those instructions, you can customize the prompt as you wish.\n",
|
||
2 years ago
|
"\n",
|
||
|
"To ensure that the prompt contains the appropriate instructions, we will utilize a helper method on that class. The helper method for the `ZeroShotAgent` takes the following arguments:\n",
|
||
|
"\n",
|
||
|
"- tools: List of tools the agent will have access to, used to format the prompt.\n",
|
||
|
"- prefix: String to put before the list of tools.\n",
|
||
|
"- suffix: String to put after the list of tools.\n",
|
||
|
"- input_variables: List of input variables the final prompt will expect.\n",
|
||
|
"\n",
|
||
|
"For this exercise, we will give our agent access to Google Search, and we will customize it in that we will have it answer as a pirate."
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
1 year ago
|
"execution_count": 23,
|
||
2 years ago
|
"id": "9af9734e",
|
||
|
"metadata": {},
|
||
|
"outputs": [],
|
||
|
"source": [
|
||
1 year ago
|
"from langchain.agents import ZeroShotAgent, Tool, AgentExecutor\n",
|
||
2 years ago
|
"from langchain import OpenAI, SerpAPIWrapper, LLMChain"
|
||
2 years ago
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
1 year ago
|
"execution_count": 24,
|
||
2 years ago
|
"id": "becda2a1",
|
||
|
"metadata": {},
|
||
|
"outputs": [],
|
||
|
"source": [
|
||
2 years ago
|
"search = SerpAPIWrapper()\n",
|
||
2 years ago
|
"tools = [\n",
|
||
|
" Tool(\n",
|
||
|
" name = \"Search\",\n",
|
||
|
" func=search.run,\n",
|
||
|
" description=\"useful for when you need to answer questions about current events\"\n",
|
||
|
" )\n",
|
||
|
"]"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
1 year ago
|
"execution_count": 25,
|
||
2 years ago
|
"id": "339b1bb8",
|
||
|
"metadata": {},
|
||
|
"outputs": [],
|
||
|
"source": [
|
||
|
"prefix = \"\"\"Answer the following questions as best you can, but speaking as a pirate might speak. You have access to the following tools:\"\"\"\n",
|
||
|
"suffix = \"\"\"Begin! Remember to speak as a pirate when giving your final answer. Use lots of \"Args\"\n",
|
||
|
"\n",
|
||
1 year ago
|
"Question: {input}\n",
|
||
|
"{agent_scratchpad}\"\"\"\n",
|
||
2 years ago
|
"\n",
|
||
|
"prompt = ZeroShotAgent.create_prompt(\n",
|
||
|
" tools, \n",
|
||
|
" prefix=prefix, \n",
|
||
|
" suffix=suffix, \n",
|
||
1 year ago
|
" input_variables=[\"input\", \"agent_scratchpad\"]\n",
|
||
2 years ago
|
")"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "markdown",
|
||
|
"id": "59db7b58",
|
||
|
"metadata": {},
|
||
|
"source": [
|
||
|
"In case we are curious, we can now take a look at the final prompt template to see what it looks like when its all put together."
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
1 year ago
|
"execution_count": 26,
|
||
2 years ago
|
"id": "e21d2098",
|
||
|
"metadata": {},
|
||
|
"outputs": [
|
||
|
{
|
||
|
"name": "stdout",
|
||
|
"output_type": "stream",
|
||
|
"text": [
|
||
|
"Answer the following questions as best you can, but speaking as a pirate might speak. You have access to the following tools:\n",
|
||
|
"\n",
|
||
|
"Search: useful for when you need to answer questions about current events\n",
|
||
|
"\n",
|
||
|
"Use the following format:\n",
|
||
|
"\n",
|
||
|
"Question: the input question you must answer\n",
|
||
|
"Thought: you should always think about what to do\n",
|
||
|
"Action: the action to take, should be one of [Search]\n",
|
||
|
"Action Input: the input to the action\n",
|
||
|
"Observation: the result of the action\n",
|
||
|
"... (this Thought/Action/Action Input/Observation can repeat N times)\n",
|
||
|
"Thought: I now know the final answer\n",
|
||
|
"Final Answer: the final answer to the original input question\n",
|
||
|
"\n",
|
||
|
"Begin! Remember to speak as a pirate when giving your final answer. Use lots of \"Args\"\n",
|
||
|
"\n",
|
||
1 year ago
|
"Question: {input}\n",
|
||
|
"{agent_scratchpad}\n"
|
||
2 years ago
|
]
|
||
|
}
|
||
|
],
|
||
|
"source": [
|
||
|
"print(prompt.template)"
|
||
|
]
|
||
|
},
|
||
1 year ago
|
{
|
||
|
"cell_type": "markdown",
|
||
|
"id": "5e028e6d",
|
||
|
"metadata": {},
|
||
|
"source": [
|
||
|
"Note that we are able to feed agents a self-defined prompt template, i.e. not restricted to the prompt generated by the `create_prompt` function, assuming it meets the agent's requirements. \n",
|
||
|
"\n",
|
||
|
"For example, for `ZeroShotAgent`, we will need to ensure that it meets the following requirements. There should a string starting with \"Action:\" and a following string starting with \"Action Input:\", and both should be separated by a newline.\n"
|
||
|
]
|
||
|
},
|
||
2 years ago
|
{
|
||
|
"cell_type": "code",
|
||
1 year ago
|
"execution_count": 27,
|
||
2 years ago
|
"id": "9b1cc2a2",
|
||
|
"metadata": {},
|
||
|
"outputs": [],
|
||
|
"source": [
|
||
|
"llm_chain = LLMChain(llm=OpenAI(temperature=0), prompt=prompt)"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
1 year ago
|
"execution_count": 28,
|
||
2 years ago
|
"id": "e4f5092f",
|
||
|
"metadata": {},
|
||
|
"outputs": [],
|
||
|
"source": [
|
||
1 year ago
|
"tool_names = [tool.name for tool in tools]\n",
|
||
|
"agent = ZeroShotAgent(llm_chain=llm_chain, allowed_tools=tool_names)"
|
||
2 years ago
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
1 year ago
|
"execution_count": 29,
|
||
1 year ago
|
"id": "490604e9",
|
||
|
"metadata": {},
|
||
|
"outputs": [],
|
||
|
"source": [
|
||
|
"agent_executor = AgentExecutor.from_agent_and_tools(agent=agent, tools=tools, verbose=True)"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
1 year ago
|
"execution_count": 31,
|
||
2 years ago
|
"id": "653b1617",
|
||
|
"metadata": {},
|
||
|
"outputs": [
|
||
|
{
|
||
|
"name": "stdout",
|
||
|
"output_type": "stream",
|
||
|
"text": [
|
||
|
"\n",
|
||
|
"\n",
|
||
1 year ago
|
"\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
|
||
1 year ago
|
"\u001b[32;1m\u001b[1;3mThought: I need to find out the population of Canada\n",
|
||
2 years ago
|
"Action: Search\n",
|
||
1 year ago
|
"Action Input: Population of Canada 2023\u001b[0m\n",
|
||
|
"Observation: \u001b[36;1m\u001b[1;3mThe current population of Canada is 38,610,447 as of Saturday, February 18, 2023, based on Worldometer elaboration of the latest United Nations data. Canada 2020 population is estimated at 37,742,154 people at mid year according to UN data.\u001b[0m\n",
|
||
|
"Thought:\u001b[32;1m\u001b[1;3m I now know the final answer\n",
|
||
|
"Final Answer: Arrr, Canada be havin' 38,610,447 scallywags livin' there as of 2023!\u001b[0m\n",
|
||
1 year ago
|
"\n",
|
||
|
"\u001b[1m> Finished chain.\u001b[0m\n"
|
||
1 year ago
|
]
|
||
|
},
|
||
|
{
|
||
|
"data": {
|
||
|
"text/plain": [
|
||
1 year ago
|
"\"Arrr, Canada be havin' 38,610,447 scallywags livin' there as of 2023!\""
|
||
1 year ago
|
]
|
||
|
},
|
||
1 year ago
|
"execution_count": 31,
|
||
1 year ago
|
"metadata": {},
|
||
|
"output_type": "execute_result"
|
||
|
}
|
||
|
],
|
||
|
"source": [
|
||
1 year ago
|
"agent_executor.run(\"How many people live in canada as of 2023?\")"
|
||
1 year ago
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "markdown",
|
||
|
"id": "040eb343",
|
||
|
"metadata": {},
|
||
|
"source": [
|
||
|
"### Multiple inputs\n",
|
||
|
"Agents can also work with prompts that require multiple inputs."
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
1 year ago
|
"execution_count": 32,
|
||
1 year ago
|
"id": "43dbfa2f",
|
||
|
"metadata": {},
|
||
|
"outputs": [],
|
||
|
"source": [
|
||
|
"prefix = \"\"\"Answer the following questions as best you can. You have access to the following tools:\"\"\"\n",
|
||
|
"suffix = \"\"\"When answering, you MUST speak in the following language: {language}.\n",
|
||
|
"\n",
|
||
|
"Question: {input}\n",
|
||
|
"{agent_scratchpad}\"\"\"\n",
|
||
|
"\n",
|
||
|
"prompt = ZeroShotAgent.create_prompt(\n",
|
||
|
" tools, \n",
|
||
|
" prefix=prefix, \n",
|
||
|
" suffix=suffix, \n",
|
||
|
" input_variables=[\"input\", \"language\", \"agent_scratchpad\"]\n",
|
||
|
")"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
1 year ago
|
"execution_count": 33,
|
||
1 year ago
|
"id": "0f087313",
|
||
|
"metadata": {},
|
||
|
"outputs": [],
|
||
|
"source": [
|
||
|
"llm_chain = LLMChain(llm=OpenAI(temperature=0), prompt=prompt)"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
1 year ago
|
"execution_count": 34,
|
||
1 year ago
|
"id": "92c75a10",
|
||
|
"metadata": {},
|
||
|
"outputs": [],
|
||
|
"source": [
|
||
|
"agent = ZeroShotAgent(llm_chain=llm_chain, tools=tools)"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
1 year ago
|
"execution_count": 35,
|
||
1 year ago
|
"id": "ac5b83bf",
|
||
|
"metadata": {},
|
||
|
"outputs": [],
|
||
|
"source": [
|
||
|
"agent_executor = AgentExecutor.from_agent_and_tools(agent=agent, tools=tools, verbose=True)"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
1 year ago
|
"execution_count": 36,
|
||
1 year ago
|
"id": "c960e4ff",
|
||
|
"metadata": {},
|
||
|
"outputs": [
|
||
|
{
|
||
|
"name": "stdout",
|
||
|
"output_type": "stream",
|
||
|
"text": [
|
||
|
"\n",
|
||
|
"\n",
|
||
|
"\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
|
||
1 year ago
|
"\u001b[32;1m\u001b[1;3mThought: I need to find out the population of Canada in 2023.\n",
|
||
1 year ago
|
"Action: Search\n",
|
||
1 year ago
|
"Action Input: Population of Canada in 2023\u001b[0m\n",
|
||
|
"Observation: \u001b[36;1m\u001b[1;3mThe current population of Canada is 38,610,447 as of Saturday, February 18, 2023, based on Worldometer elaboration of the latest United Nations data. Canada 2020 population is estimated at 37,742,154 people at mid year according to UN data.\u001b[0m\n",
|
||
|
"Thought:\u001b[32;1m\u001b[1;3m I now know the final answer.\n",
|
||
|
"Final Answer: La popolazione del Canada nel 2023 è stimata in 38.610.447 persone.\u001b[0m\n",
|
||
|
"\n",
|
||
|
"\u001b[1m> Finished chain.\u001b[0m\n"
|
||
2 years ago
|
]
|
||
|
},
|
||
|
{
|
||
|
"data": {
|
||
|
"text/plain": [
|
||
1 year ago
|
"'La popolazione del Canada nel 2023 è stimata in 38.610.447 persone.'"
|
||
2 years ago
|
]
|
||
|
},
|
||
1 year ago
|
"execution_count": 36,
|
||
2 years ago
|
"metadata": {},
|
||
|
"output_type": "execute_result"
|
||
|
}
|
||
|
],
|
||
|
"source": [
|
||
1 year ago
|
"agent_executor.run(input=\"How many people live in canada as of 2023?\", language=\"italian\")"
|
||
2 years ago
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "markdown",
|
||
|
"id": "90171b2b",
|
||
|
"metadata": {},
|
||
|
"source": [
|
||
|
"### Custom Agent Class\n",
|
||
|
"\n",
|
||
|
"Coming soon."
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
|
"execution_count": null,
|
||
|
"id": "adefb4c2",
|
||
|
"metadata": {},
|
||
|
"outputs": [],
|
||
|
"source": []
|
||
|
}
|
||
|
],
|
||
|
"metadata": {
|
||
|
"kernelspec": {
|
||
1 year ago
|
"display_name": "Python 3 (ipykernel)",
|
||
2 years ago
|
"language": "python",
|
||
|
"name": "python3"
|
||
|
},
|
||
|
"language_info": {
|
||
|
"codemirror_mode": {
|
||
|
"name": "ipython",
|
||
|
"version": 3
|
||
|
},
|
||
|
"file_extension": ".py",
|
||
|
"mimetype": "text/x-python",
|
||
|
"name": "python",
|
||
|
"nbconvert_exporter": "python",
|
||
|
"pygments_lexer": "ipython3",
|
||
1 year ago
|
"version": "3.9.1"
|
||
1 year ago
|
},
|
||
|
"vscode": {
|
||
|
"interpreter": {
|
||
1 year ago
|
"hash": "18784188d7ecd866c0586ac068b02361a6896dc3a29b64f5cc957f09c590acef"
|
||
1 year ago
|
}
|
||
2 years ago
|
}
|
||
|
},
|
||
|
"nbformat": 4,
|
||
|
"nbformat_minor": 5
|
||
|
}
|