{
 "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
}