Spaces:
Configuration error
Configuration error
File size: 8,387 Bytes
86beb7d |
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 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 |
{
"cells": [
{
"cell_type": "markdown",
"id": "8b39443a",
"metadata": {},
"source": [
"# Langchain with Mistral API"
]
},
{
"cell_type": "code",
"execution_count": 5,
"id": "e2dd8401",
"metadata": {},
"outputs": [],
"source": [
"import os\n",
"from dotenv import load_dotenv \n",
"from langchain_mistralai import ChatMistralAI\n",
"from langchain_core.messages import SystemMessage, HumanMessage\n",
"from langgraph.prebuilt import create_react_agent\n",
"\n"
]
},
{
"cell_type": "code",
"execution_count": 6,
"id": "d1436145",
"metadata": {},
"outputs": [],
"source": [
"load_dotenv() \n",
"os.environ[\"MISTRAL_API_KEY\"] = os.getenv(\"MISTRAL\")"
]
},
{
"cell_type": "code",
"execution_count": 7,
"id": "c886b7ad",
"metadata": {},
"outputs": [],
"source": [
"# LLM Configuration\n",
"llm = ChatMistralAI(\n",
" model=\"mistral-small-latest\",\n",
" temperature=0,\n",
" max_retries=5\n",
")"
]
},
{
"cell_type": "code",
"execution_count": 8,
"id": "cffaee32",
"metadata": {},
"outputs": [],
"source": [
"sys_prompt = \"You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.\\n\\n\\n\\\n",
"You will be provided with tools to help you answer questions, as Wikipedia. Before starting any search, you must first think about the TRUE necessary steps that are required to answer the question. If you need to search for information, the query should be a 1 to 3 keywords that can be used to find the most information about the subject. If the question specifies a date, do not put the date into the query. THEN you should analyze the result to answer the question.\"\n",
"hum_prompt = \"What is the first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists?\""
]
},
{
"cell_type": "code",
"execution_count": 17,
"id": "03974841",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"To determine the first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists, I will follow these steps:\n",
"\n",
"1. **Identify the Malko Competition recipients from the 20th Century (after 1977):**\n",
" - The Malko Competition, also known as the International Hans Gabor Belvedere Singing Competition, has been held annually since 1977.\n",
" - I will look for recipients from 1978 to 2000, as the 20th Century ended in the year 2000.\n",
"\n",
"2. **Check the nationalities of these recipients:**\n",
" - I will identify recipients whose nationality on record is a country that no longer exists.\n",
"\n",
"3. **Determine the first name of the relevant recipient:**\n",
" - Once the recipient is identified, I will extract their first name.\n",
"\n",
"**Research and Verification:**\n",
"- Upon reviewing the list of Malko Competition winners, I find that in 1980, a singer named **Vladimir Chernov** from the Soviet Union won the competition.\n",
"- The Soviet Union no longer exists, having dissolved in 1991.\n",
"\n",
"**Conclusion:**\n",
"- The first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists is **Vladimir**.\n",
"\n",
"FINAL ANSWER: Vladimir\n"
]
}
],
"source": [
"response = llm.invoke([\n",
" SystemMessage(content=sys_prompt),\n",
" HumanMessage(content=hum_prompt)]\n",
").content.strip()\n",
"\n",
"print(response)"
]
},
{
"cell_type": "markdown",
"id": "de0d29f5",
"metadata": {},
"source": [
"## Add tools"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "3611c83e",
"metadata": {},
"outputs": [],
"source": [
"from langchain_community.document_loaders import WikipediaLoader\n",
"from langchain_core.tools import tool"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "bf18a95f",
"metadata": {},
"outputs": [],
"source": [
"@tool\n",
"def wikipedia_search(query: str) -> str:\n",
" \"\"\"Search Wikipedia for a query and return maximum 1 results.\n",
" Args:\n",
" query: The search query with maximum 3 keywords.\"\"\"\n",
" search_docs = WikipediaLoader(query=query, load_max_docs=1, doc_content_chars_max=7000).load()\n",
" return search_docs"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "7aa330f2",
"metadata": {},
"outputs": [],
"source": [
"llm_withtools = llm.bind_tools([wikipedia_search])"
]
},
{
"cell_type": "code",
"execution_count": 10,
"id": "24418b42",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"To answer this question, I need to follow these steps:\n",
"\n",
"1. Understand the Malko Competition and its recipients.\n",
"2. Identify the recipient from the 20th century (after 1977) whose nationality is a country that no longer exists.\n",
"3. Extract the first name of that recipient.\n",
"\n",
"Given the specific nature of the question, I need to search for information about the Malko Competition and its recipients.\n",
"\n",
"Let's start by searching for \"Malko Competition recipients\".\n"
]
}
],
"source": [
"response = llm_withtools.invoke([\n",
" SystemMessage(content=sys_prompt),\n",
" HumanMessage(content=hum_prompt)]\n",
")\n",
"print(response.content.strip())"
]
},
{
"cell_type": "code",
"execution_count": 11,
"id": "313cd6ff",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"content='To answer this question, I need to follow these steps:\\n\\n1. Understand the Malko Competition and its recipients.\\n2. Identify the recipient from the 20th century (after 1977) whose nationality is a country that no longer exists.\\n3. Extract the first name of that recipient.\\n\\nGiven the specific nature of the question, I need to search for information about the Malko Competition and its recipients.\\n\\nLet\\'s start by searching for \"Malko Competition recipients\".\\n\\n' additional_kwargs={'tool_calls': [{'id': 'W3MY54xvl', 'function': {'name': 'wiki_search', 'arguments': '{\"query\": \"Malko Competition recipients\"}'}, 'index': 0}]} response_metadata={'token_usage': {'prompt_tokens': 383, 'total_tokens': 501, 'completion_tokens': 118}, 'model_name': 'mistral-small-latest', 'model': 'mistral-small-latest', 'finish_reason': 'tool_calls'} id='run--e0ce53e6-d1a1-4e9c-8f66-c6548c1259e8-0' tool_calls=[{'name': 'wiki_search', 'args': {'query': 'Malko Competition recipients'}, 'id': 'W3MY54xvl', 'type': 'tool_call'}] usage_metadata={'input_tokens': 383, 'output_tokens': 118, 'total_tokens': 501}\n"
]
}
],
"source": [
"print(response)"
]
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3",
"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",
"version": "3.11.13"
}
},
"nbformat": 4,
"nbformat_minor": 5
}
|