try gemini

react_agent_test.ipynb

@@ -10,25 +10,27 @@
     "import os\n",
     "from dotenv import load_dotenv \n",
     "from langchain_mistralai import ChatMistralAI\n",
+    "from langchain_google_genai import ChatGoogleGenerativeAI\n",
     "from langchain_core.messages import SystemMessage, HumanMessage\n",
     "from langgraph.prebuilt import create_react_agent\n",
     "\n",
     "from custom_tools import custom_tools\n",
     "\n",
     "load_dotenv() \n",
-    "os.environ[\"MISTRAL_API_KEY\"] = os.getenv(\"MISTRAL\")"
+    "os.environ[\"MISTRAL_API_KEY\"] = os.getenv(\"MISTRAL\")\n",
+    "os.environ[\"GOOGLE_API_KEY\"] = os.getenv(\"GOOGLE\")"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 2,
    "id": "7df8bade",
    "metadata": {},
    "outputs": [],
    "source": [
     "# LLM Configuration\n",
-    "llm = ChatMistralAI(\n",
-    "    model=\"mistral-small-latest\",\n",
+    "llm = ChatGoogleGenerativeAI( # ChatMistralAI\n",
+    "    model=\"gemini-2.5-flash\", # mistral-small-latest\n",
     "    temperature=0,\n",
     "    max_retries=5\n",
     ")\n",
@@ -47,7 +49,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 15,
+   "execution_count": 3,
    "id": "0ce2dd73",
    "metadata": {},
    "outputs": [
@@ -55,7 +57,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "a, b, e\n"
+      "b,e\n"
      ]
     }
    ],
@@ -84,18 +86,18 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 16,
+   "execution_count": 4,
    "id": "17dc251b",
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/plain": [
-       "{'messages': [HumanMessage(content='\\n\"task_id\": \"6f37996b-2ac7-44b0-8e68-6d28256631b4\",\\n        \"question\": \"Given this table defining * on the set S = {a, b, c, d, e}\\n\\n|*|a|b|c|d|e|\\n|---|---|---|---|---|---|\\n|a|a|b|c|b|d|\\n|b|b|c|a|e|c|\\n|c|c|a|b|b|a|\\n|d|b|e|b|e|d|\\n|e|d|b|a|d|c|\\n\\nprovide the subset of S involved in any possible counter-examples that prove * is not commutative. Provide your answer as a comma separated list of the elements in the set in alphabetical order.\",\\n        \"Level\": \"1\",\\n        \"file_name\": \"\"\\n        ', additional_kwargs={}, response_metadata={}, id='456b861d-e60e-403b-be65-3e043565a7e6'),\n",
-       "  AIMessage(content=\"To determine if the operation \\\\( * \\\\) is commutative, we need to check if \\\\( x * y = y * x \\\\) for all \\\\( x, y \\\\in S \\\\). If we find any pair \\\\( (x, y) \\\\) such that \\\\( x * y \\\\neq y * x \\\\), then \\\\( * \\\\) is not commutative, and the subset \\\\( \\\\{x, y\\\\} \\\\) will be part of the counter-example.\\n\\nLet's analyze the table step by step:\\n\\n1. \\\\( a * b = b \\\\) and \\\\( b * a = c \\\\). Since \\\\( b \\\\neq c \\\\), \\\\( * \\\\) is not commutative.\\n2. \\\\( a * c = c \\\\) and \\\\( c * a = c \\\\). Here \\\\( a * c = c * a \\\\), so no counter-example here.\\n3. \\\\( a * d = b \\\\) and \\\\( d * a = b \\\\). Here \\\\( a * d = d * a \\\\), so no counter-example here.\\n4. \\\\( a * e = d \\\\) and \\\\( e * a = d \\\\). Here \\\\( a * e = e * a \\\\), so no counter-example here.\\n5. \\\\( b * c = a \\\\) and \\\\( c * b = a \\\\). Here \\\\( b * c = c * b \\\\), so no counter-example here.\\n6. \\\\( b * d = e \\\\) and \\\\( d * b = e \\\\). Here \\\\( b * d = d * b \\\\), so no counter-example here.\\n7. \\\\( b * e = c \\\\) and \\\\( e * b = b \\\\). Since \\\\( c \\\\neq b \\\\), \\\\( * \\\\) is not commutative.\\n8. \\\\( c * d = b \\\\) and \\\\( d * c = b \\\\). Here \\\\( c * d = d * c \\\\), so no counter-example here.\\n9. \\\\( c * e = a \\\\) and \\\\( e * c = a \\\\). Here \\\\( c * e = e * c \\\\), so no counter-example here.\\n10. \\\\( d * e = d \\\\) and \\\\( e * d = d \\\\). Here \\\\( d * e = e * d \\\\), so no counter-example here.\\n\\nFrom the analysis, the pairs \\\\( (a, b) \\\\) and \\\\( (b, e) \\\\) are counter-examples that prove \\\\( * \\\\) is not commutative.\\n\\nFINAL ANSWER: a, b, e\", additional_kwargs={}, response_metadata={'token_usage': {'prompt_tokens': 1467, 'total_tokens': 1981, 'completion_tokens': 514}, 'model_name': 'mistral-small-latest', 'model': 'mistral-small-latest', 'finish_reason': 'stop'}, id='run--54102f8a-ae50-4a89-a96f-e2b1b1178a24-0', usage_metadata={'input_tokens': 1467, 'output_tokens': 514, 'total_tokens': 1981})]}"
+       "{'messages': [HumanMessage(content='\\n\"task_id\": \"6f37996b-2ac7-44b0-8e68-6d28256631b4\",\\n        \"question\": \"Given this table defining * on the set S = {a, b, c, d, e}\\n\\n|*|a|b|c|d|e|\\n|---|---|---|---|---|---|\\n|a|a|b|c|b|d|\\n|b|b|c|a|e|c|\\n|c|c|a|b|b|a|\\n|d|b|e|b|e|d|\\n|e|d|b|a|d|c|\\n\\nprovide the subset of S involved in any possible counter-examples that prove * is not commutative. Provide your answer as a comma separated list of the elements in the set in alphabetical order.\",\\n        \"Level\": \"1\",\\n        \"file_name\": \"\"\\n        ', additional_kwargs={}, response_metadata={}, id='17e37204-e33b-4a13-8a35-5b843358f3a4'),\n",
+       "  AIMessage(content='FINAL ANSWER: b,e', additional_kwargs={}, response_metadata={'prompt_feedback': {'block_reason': 0, 'safety_ratings': []}, 'finish_reason': 'STOP', 'model_name': 'gemini-2.5-flash', 'safety_ratings': []}, id='run--30a88cdb-6076-4b29-822e-c5ded1b678e9-0', usage_metadata={'input_tokens': 1436, 'output_tokens': 6, 'total_tokens': 1864, 'input_token_details': {'cache_read': 0}, 'output_token_details': {'reasoning': 422}})]}"
       ]
      },
-     "execution_count": 16,
+     "execution_count": 4,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -114,7 +116,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 6,
    "id": "4b7ca6fa",
    "metadata": {},
    "outputs": [
@@ -122,7 +124,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Qd1#\n"
+      "Qh4+\n"
      ]
     }
    ],
@@ -181,10 +183,18 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 7,
    "id": "69cf16b8",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Claus\n"
+     ]
+    }
+   ],
    "source": [
     "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?\"\n",
     "\n",
@@ -233,10 +243,18 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 8,
    "id": "8eb59a66",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n"
+     ]
+    }
+   ],
    "source": [
     "hum_prompt = \"\"\"{'task_id': '7bd855d8-463d-4ed5-93ca-5fe35145f733', 'question': 'The attached Excel file contains the sales of menu items for a local fast-food chain. What were the total sales that the chain made from food (not including drinks)? Express your answer in USD with two decimal places.', 'Level': '1', 'file_name': '7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx'}\"\"\"\n",
     "\n",
@@ -250,27 +268,24 @@
     "    {\"messages\": HumanMessage(content=hum_prompt)}\n",
     ")\n",
     "\n",
-    "print(response[\"messages\"][-1].content.split(\"FINAL ANSWER: \")[-1])"
+    "print(response[\"messages\"][-1].content.split(\"FINAL ANSWER: \")[-1])\n",
+    "# 89 706"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": 9,
    "id": "2ca6f955",
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/plain": [
-       "{'messages': [HumanMessage(content=\"{'task_id': '7bd855d8-463d-4ed5-93ca-5fe35145f733', 'question': 'The attached Excel file contains the sales of menu items for a local fast-food chain. What were the total sales that the chain made from food (not including drinks)? Express your answer in USD with two decimal places.', 'Level': '1', 'file_name': '7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx'}\", additional_kwargs={}, response_metadata={}, id='b30a3914-ea49-43d1-be67-dc9fb8ae72ba'),\n",
-       "  AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'zlVnoOJVW', 'function': {'name': 'read_excel', 'arguments': '{\"task_id\": \"7bd855d8-463d-4ed5-93ca-5fe35145f733\", \"file_path\": \"7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx\"}'}, 'index': 0}]}, response_metadata={'token_usage': {'prompt_tokens': 1120, 'total_tokens': 1219, 'completion_tokens': 99}, 'model_name': 'mistral-large-latest', 'model': 'mistral-large-latest', 'finish_reason': 'tool_calls'}, id='run--fef0ff1c-7680-4fc1-9c2e-05d702f7ce54-0', tool_calls=[{'name': 'read_excel', 'args': {'task_id': '7bd855d8-463d-4ed5-93ca-5fe35145f733', 'file_path': '7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx'}, 'id': 'zlVnoOJVW', 'type': 'tool_call'}], usage_metadata={'input_tokens': 1120, 'output_tokens': 99, 'total_tokens': 1219}),\n",
-       "  ToolMessage(content='   Location  Burgers  Hot Dogs  Salads  Fries  Ice Cream  Soda\\n  Pinebrook     1594      1999    2002   2005       1977  1980\\n   Wharvton     1983      2008    2014   2015       2017  2018\\n    Sagrada     2019      2022    2022   2023       2021  2019\\n  Algrimand     1958      1971    1982   1989       1998  2009\\n    Marztep     2015      2016    2018   2019       2021  2022\\nSan Cecelia     2011      2010    2012   2013       2015  2016\\n    Pimento     2017      1999    2001   2003       1969  2967\\n   Tinseles     1967      1969    1982   1994       2005  2006\\n    Rosdale     2007      2009    2021   1989       2005  2011', name='read_excel', id='276bb0ab-b6d8-40a8-8c95-8b19c53bdf86', tool_call_id='zlVnoOJVW'),\n",
-       "  AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'DDJAAo0N4', 'function': {'name': 'sum_pandas_df', 'arguments': '{\"task_id\": \"7bd855d8-463d-4ed5-93ca-5fe35145f733\", \"file_name\": \"7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx\", \"column_names\": [\"Burgers\", \"Hot Dogs\", \"Salads\", \"Fries\", \"Ice Cream\"]}'}, 'index': 0}]}, response_metadata={'token_usage': {'prompt_tokens': 1579, 'total_tokens': 1710, 'completion_tokens': 131}, 'model_name': 'mistral-large-latest', 'model': 'mistral-large-latest', 'finish_reason': 'tool_calls'}, id='run--5f838a6e-0ad5-4667-80a4-f15a71da05a8-0', tool_calls=[{'name': 'sum_pandas_df', 'args': {'task_id': '7bd855d8-463d-4ed5-93ca-5fe35145f733', 'file_name': '7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx', 'column_names': ['Burgers', 'Hot Dogs', 'Salads', 'Fries', 'Ice Cream']}, 'id': 'DDJAAo0N4', 'type': 'tool_call'}], usage_metadata={'input_tokens': 1579, 'output_tokens': 131, 'total_tokens': 1710}),\n",
-       "  ToolMessage(content='89706', name='sum_pandas_df', id='a7e43a29-7c6b-45c4-93be-78757094c520', tool_call_id='DDJAAo0N4'),\n",
-       "  AIMessage(content='To find the total sales made from food (excluding drinks), we need to sum the sales of Burgers, Hot Dogs, Salads, Fries, and Ice Cream across all locations.\\n\\nThe total sales from food items are calculated as follows:\\n\\n- Burgers: 1594 + 1983 + 2019 + 1958 + 2015 + 2011 + 2017 + 1967 + 2007 = 16571\\n- Hot Dogs: 1999 + 2008 + 2022 + 1971 + 2016 + 2010 + 1999 + 1969 + 2009 = 16993\\n- Salads: 2002 + 2014 + 2022 + 1982 + 2018 + 2012 + 2001 + 1982 + 2021 = 17054\\n- Fries: 2005 + 2015 + 2023 + 1989 + 2019 + 2013 + 2003 + 1994 + 1989 = 17060\\n- Ice Cream: 1977 + 2017 + 2021 + 1998 + 2021 + 2015 + 1969 + 2005 + 2005 = 17028\\n\\nAdding these totals together:\\n\\nTotal food sales = 16571 + 16993 + 17054 + 17060 + 17028 = 89706\\n\\nFINAL ANSWER: 89706.00 USD', additional_kwargs={}, response_metadata={'token_usage': {'prompt_tokens': 1739, 'total_tokens': 2197, 'completion_tokens': 458}, 'model_name': 'mistral-large-latest', 'model': 'mistral-large-latest', 'finish_reason': 'stop'}, id='run--4431fdb9-25e6-4e67-9924-38b61e37ed92-0', usage_metadata={'input_tokens': 1739, 'output_tokens': 458, 'total_tokens': 2197})]}"
+       "{'messages': [HumanMessage(content=\"{'task_id': '7bd855d8-463d-4ed5-93ca-5fe35145f733', 'question': 'The attached Excel file contains the sales of menu items for a local fast-food chain. What were the total sales that the chain made from food (not including drinks)? Express your answer in USD with two decimal places.', 'Level': '1', 'file_name': '7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx'}\", additional_kwargs={}, response_metadata={}, id='62ca610d-ba20-4584-81bd-232fe6b6ab66'),\n",
+       "  AIMessage(content='', additional_kwargs={}, response_metadata={'prompt_feedback': {'block_reason': 0, 'safety_ratings': []}, 'finish_reason': 'MAX_TOKENS', 'model_name': 'gemini-2.5-flash', 'safety_ratings': []}, id='run--50d067b9-d6cc-44bf-a52c-1ac15bdd42dd-0', usage_metadata={'input_tokens': 1352, 'output_tokens': 0, 'total_tokens': 66887, 'input_token_details': {'cache_read': 0}, 'output_token_details': {'reasoning': 65535}})]}"
       ]
      },
-     "execution_count": 14,
+     "execution_count": 9,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -289,10 +304,18 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 10,
    "id": "f80a471d",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "extremely\n"
+     ]
+    }
+   ],
    "source": [
     "hum_prompt = \"\"\"{'task_id': '9d191bce-651d-4746-be2d-7ef8ecadb9c2', 'question': 'Examine the video at https://www.youtube.com/watch?v=1htKBjuUWec.\\n\\nWhat does Teal\\'c say in response to the question \"Isn\\'t that hot?\"', 'Level': '1', 'file_name': ''}\"\"\"\n",
     "\n",

requirements.txt

@@ -15,4 +15,5 @@ opencv
 transformers
 # pytorch==2.6.0
 # torchvision==0.22
-groq
+groq
+langchain-google-genai