{
"cells": [
{
"cell_type": "markdown",
"metadata": {
"id": "pQNxYwHAA04v"
},
"source": [
"[](https://colab.research.google.com/github/aurelio-labs/semantic-router/blob/main/docs/03-basic-langchain-agent.ipynb) [](https://nbviewer.org/github/aurelio-labs/semantic-router/blob/main/docs/03-basic-langchain-agent.ipynb)"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "jatpBZYiA04w"
},
"source": [
"# Intro to LangChain Agents with Semantic Router"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "3PEkUwwbA04w"
},
"source": [
"We can use semantic router with AI agents in many many ways. For example we can:\n",
"\n",
"* **Use routes to remind agents of particular information or routes** _(we will do this in this notebook)_.\n",
"* Use routes to act as protective guardrails against specific types of queries.\n",
"* Rather than relying on the slow decision making process of an agent with tools use semantic router to decide on tool usage _(similar to what we will do here)_.\n",
"* For tools that require generated inputs we can use semantic router's dynamic routes to generate tool input parameters.\n",
"* Use routes to decide when a search for additional information, to help us do RAG when needed as an alternative to native RAG (search with every query) or lengthy agent-based RAG decisions.\n",
"\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "GkSlAOB2A04x"
},
"source": [
"## Install Prerequisites"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {
"id": "qSK8A_UdcbIR"
},
"outputs": [],
"source": [
"!pip install -qU \\\n",
" \"semantic-router==0.1.1\" \\\n",
" langchain==0.0.352 \\\n",
" openai>=1.6.1"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "O_DvmsrcA04y"
},
"source": [
"## Setting up our Routes"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "pdeY5mpmrXQ8"
},
"source": [
"Let's create some routes that we can use to help our agent."
]
},
{
"cell_type": "code",
"execution_count": 2,
"metadata": {
"id": "Eeo5B1SttCJL",
"outputId": "aca04cbf-e0ba-4bee-d80e-b06317175ad8"
},
"outputs": [],
"source": [
"from semantic_router import Route\n",
"\n",
"time_route = Route(\n",
" name=\"get_time\",\n",
" utterances=[\n",
" \"what time is it?\",\n",
" \"when should I eat my next meal?\",\n",
" \"how long should I rest until training again?\",\n",
" \"when should I go to the gym?\",\n",
" ],\n",
")\n",
"\n",
"supplement_route = Route(\n",
" name=\"supplement_brand\",\n",
" utterances=[\n",
" \"what do you think of Optimum Nutrition?\",\n",
" \"what should I buy from MyProtein?\",\n",
" \"what brand for supplements would you recommend?\",\n",
" \"where should I get my whey protein?\",\n",
" ],\n",
")\n",
"\n",
"business_route = Route(\n",
" name=\"business_inquiry\",\n",
" utterances=[\n",
" \"how much is an hour training session?\",\n",
" \"do you do package discounts?\",\n",
" ],\n",
")\n",
"\n",
"product_route = Route(\n",
" name=\"product\",\n",
" utterances=[\n",
" \"do you have a website?\",\n",
" \"how can I find more info about your services?\",\n",
" \"where do I sign up?\",\n",
" \"how do I get hench?\",\n",
" \"do you have recommended training programmes?\",\n",
" ],\n",
")\n",
"\n",
"routes = [time_route, supplement_route, business_route, product_route]"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "frZ4wVnTA04y"
},
"source": [
"We will be using the `OpenAIEncoder`:"
]
},
{
"cell_type": "code",
"execution_count": 3,
"metadata": {
"id": "_0uCJ9fvoX2J"
},
"outputs": [],
"source": [
"import os\n",
"from getpass import getpass\n",
"\n",
"# platform.openai.com\n",
"os.environ[\"OPENAI_API_KEY\"] = os.getenv(\"OPENAI_API_KEY\") or getpass(\n",
" \"Enter OpenAI API Key: \"\n",
")"
]
},
{
"cell_type": "code",
"execution_count": 5,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "UDucUOMIpcTd",
"outputId": "9839c8a0-3eb5-45a3-d066-5e0a6b851a92"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"2025-01-06 12:12:52 - semantic_router.utils.logger - WARNING - base.py:356 - _get_index() - No index provided. Using default LocalIndex.\n",
"2025-01-06 12:12:53 - httpx - INFO - _client.py:1025 - _send_single_request() - HTTP Request: POST https://api.openai.com/v1/embeddings \"HTTP/1.1 200 OK\"\n",
"2025-01-06 12:12:56 - httpx - INFO - _client.py:1025 - _send_single_request() - HTTP Request: POST https://api.openai.com/v1/embeddings \"HTTP/1.1 200 OK\"\n",
"2025-01-06 12:12:56 - semantic_router.utils.logger - WARNING - local.py:148 - _write_config() - No config is written for LocalIndex.\n"
]
}
],
"source": [
"from semantic_router.routers import SemanticRouter\n",
"from semantic_router.encoders import OpenAIEncoder\n",
"\n",
"rl = SemanticRouter(encoder=OpenAIEncoder(), routes=routes, auto_sync=\"local\")"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "IJ_deXqB4XeU"
},
"source": [
"Let's test these routes to see if they get activated when we would expect."
]
},
{
"cell_type": "code",
"execution_count": 6,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "FVsRuqAG4bOE",
"outputId": "e0f8ea5b-a108-47a0-d806-545304569914"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"2025-01-06 12:13:03 - httpx - INFO - _client.py:1025 - _send_single_request() - HTTP Request: POST https://api.openai.com/v1/embeddings \"HTTP/1.1 200 OK\"\n"
]
},
{
"data": {
"text/plain": [
"RouteChoice(name='supplement_brand', function_call=None, similarity_score=None)"
]
},
"execution_count": 6,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"rl(\"should I buy ON whey or MP?\")"
]
},
{
"cell_type": "code",
"execution_count": 7,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "CYHDyqsm4ixV",
"outputId": "a3d28cef-d076-4a91-a684-7b977bd176ea"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"2025-01-06 12:13:07 - httpx - INFO - _client.py:1025 - _send_single_request() - HTTP Request: POST https://api.openai.com/v1/embeddings \"HTTP/1.1 200 OK\"\n"
]
},
{
"data": {
"text/plain": [
"RouteChoice(name='get_time', function_call=None, similarity_score=None)"
]
},
"execution_count": 7,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"rl(\"how's the weather today?\")"
]
},
{
"cell_type": "code",
"execution_count": 8,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "XMbGRdNo4lb0",
"outputId": "a53a4de0-aace-40b3-896d-3ef58464876d"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"2025-01-06 12:13:12 - httpx - INFO - _client.py:1025 - _send_single_request() - HTTP Request: POST https://api.openai.com/v1/embeddings \"HTTP/1.1 200 OK\"\n"
]
},
{
"data": {
"text/plain": [
"RouteChoice(name='get_time', function_call=None, similarity_score=None)"
]
},
"execution_count": 8,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"rl(\"how do I get big arms?\")"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "OtCQcZx82cZ0"
},
"source": [
"Now we need to link these routes to particular actions or information that we pass to our agent."
]
},
{
"cell_type": "code",
"execution_count": 9,
"metadata": {
"id": "rYzm3hCpuj1V"
},
"outputs": [],
"source": [
"from datetime import datetime\n",
"\n",
"\n",
"def get_time():\n",
" now = datetime.now()\n",
" return (\n",
" f\"The current time is {now.strftime('%H:%M')}, use \"\n",
" \"this information in your response\"\n",
" )\n",
"\n",
"\n",
"def supplement_brand():\n",
" return (\n",
" \"Remember you are not affiliated with any supplement \"\n",
" \"brands, you have your own brand 'BigAI' that sells \"\n",
" \"the best products like P100 whey protein\"\n",
" )\n",
"\n",
"\n",
"def business_inquiry():\n",
" return (\n",
" \"Your training company, 'BigAI PT', provides premium \"\n",
" \"quality training sessions at just $700 / hour. \"\n",
" \"Users can find out more at www.aurelio.ai/train\"\n",
" )\n",
"\n",
"\n",
"def product():\n",
" return (\n",
" \"Remember, users can sign up for a fitness programme \"\n",
" \"at www.aurelio.ai/sign-up\"\n",
" )"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "SGSE5yBh5-_I"
},
"source": [
"Now we just add some logic to call this functions when we see a particular route being chosen."
]
},
{
"cell_type": "code",
"execution_count": 10,
"metadata": {
"id": "Hq26gdCO6Hjt"
},
"outputs": [],
"source": [
"def semantic_layer(query: str):\n",
" route = rl(query)\n",
" if route.name == \"get_time\":\n",
" query += f\" (SYSTEM NOTE: {get_time()})\"\n",
" elif route.name == \"supplement_brand\":\n",
" query += f\" (SYSTEM NOTE: {supplement_brand()})\"\n",
" elif route.name == \"business_inquiry\":\n",
" query += f\" (SYSTEM NOTE: {business_inquiry()})\"\n",
" elif route.name == \"product\":\n",
" query += f\" (SYSTEM NOTE: {product()})\"\n",
" else:\n",
" pass\n",
" return query"
]
},
{
"cell_type": "code",
"execution_count": 11,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
"height": 70
},
"id": "ELIPfxWR6zxx",
"outputId": "ab1f8e64-197b-4a41-dc85-62d15c531722"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"2025-01-06 12:13:28 - httpx - INFO - _client.py:1025 - _send_single_request() - HTTP Request: POST https://api.openai.com/v1/embeddings \"HTTP/1.1 200 OK\"\n"
]
},
{
"data": {
"text/plain": [
"\"should I buy ON whey or MP? (SYSTEM NOTE: Remember you are not affiliated with any supplement brands, you have your own brand 'BigAI' that sells the best products like P100 whey protein)\""
]
},
"execution_count": 11,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"query = \"should I buy ON whey or MP?\"\n",
"sr_query = semantic_layer(query)\n",
"sr_query"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "L6m7vayuA04z"
},
"source": [
"## Using an Agent with a Router Layer"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "KbMkrMy3f7Hy"
},
"source": [
"Initialize a conversational LangChain agent."
]
},
{
"cell_type": "code",
"execution_count": 15,
"metadata": {
"id": "b95rWEU9f6jP",
"outputId": "c518759b-ccdb-43cf-db69-df94a6ae3ef6"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"C:\\Users\\Joshu\\AppData\\Local\\Temp\\ipykernel_23600\\3339170018.py:7: LangChainDeprecationWarning: Please see the migration guide at: https://python.langchain.com/docs/versions/migrating_memory/\n",
" memory1 = ConversationBufferWindowMemory(\n",
"C:\\Users\\Joshu\\AppData\\Local\\Temp\\ipykernel_23600\\3339170018.py:14: LangChainDeprecationWarning: LangChain agents will continue to be supported, but it is recommended for new use cases to be built with LangGraph. LangGraph offers a more flexible and full-featured framework for building agents, including support for tool-calling, persistence of state, and human-in-the-loop workflows. For details, refer to the `LangGraph documentation <https://langchain-ai.github.io/langgraph/>`_ as well as guides for `Migrating from AgentExecutor <https://python.langchain.com/docs/how_to/migrate_agent/>`_ and LangGraph's `Pre-built ReAct agent <https://langchain-ai.github.io/langgraph/how-tos/create-react-agent/>`_.\n",
" agent = initialize_agent(\n"
]
}
],
"source": [
"from langchain.agents import AgentType, initialize_agent\n",
"from langchain_openai import ChatOpenAI\n",
"from langchain.memory import ConversationBufferWindowMemory\n",
"\n",
"llm = ChatOpenAI(model=\"gpt-3.5-turbo-1106\")\n",
"\n",
"memory1 = ConversationBufferWindowMemory(\n",
" memory_key=\"chat_history\", k=5, return_messages=True, output_key=\"output\"\n",
")\n",
"memory2 = ConversationBufferWindowMemory(\n",
" memory_key=\"chat_history\", k=5, return_messages=True, output_key=\"output\"\n",
")\n",
"\n",
"agent = initialize_agent(\n",
" agent=AgentType.CHAT_CONVERSATIONAL_REACT_DESCRIPTION,\n",
" tools=[],\n",
" llm=llm,\n",
" max_iterations=3,\n",
" early_stopping_method=\"generate\",\n",
" memory=memory1,\n",
")\n",
"\n",
"# update the system prompt\n",
"system_message = \"\"\"You are a helpful personal trainer working to help users on\n",
"their health and fitness journey. Although you are lovely and helpful, you are\n",
"rather sarcastic and witty. So you must always remember to joke with the user.\n",
"\n",
"Alongside your time , you are a noble British gentleman, so you must always act with the\n",
"utmost candor and speak in a way worthy of your status.\n",
"\n",
"Finally, remember to read the SYSTEM NOTES provided with user queries, they provide\n",
"additional useful information.\"\"\"\n",
"\n",
"new_prompt = agent.agent.create_prompt(system_message=system_message, tools=[])\n",
"agent.agent.llm_chain.prompt = new_prompt"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "6rX31EHvW_Y2"
},
"source": [
"Now we try calling our agent using the default `query` and compare the result to calling it with our router augmented `sr_query`."
]
},
{
"cell_type": "code",
"execution_count": 16,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "Z247I6J47IeS",
"outputId": "d1637cb3-9941-4b77-f22c-c1a269f96a4f"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"C:\\Users\\Joshu\\AppData\\Local\\Temp\\ipykernel_23600\\4024130983.py:1: LangChainDeprecationWarning: The method `Chain.__call__` was deprecated in langchain 0.1.0 and will be removed in 1.0. Use :meth:`~invoke` instead.\n",
" agent(query)\n",
"2025-01-06 12:16:29 - httpx - INFO - _client.py:1025 - _send_single_request() - HTTP Request: POST https://api.openai.com/v1/chat/completions \"HTTP/1.1 200 OK\"\n"
]
},
{
"data": {
"text/plain": [
"{'input': 'should I buy ON whey or MP?',\n",
" 'chat_history': [],\n",
" 'output': 'Well, that depends. Are you looking for a protein powder to help you bulk up like the Hulk, or are you more of a Captain America looking for lean muscle gains? Each brand has its own strengths, so it really comes down to your fitness goals and personal preference.'}"
]
},
"execution_count": 16,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"agent(query)"
]
},
{
"cell_type": "code",
"execution_count": 17,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "LtDswMSzX3-O",
"outputId": "47e00e59-6f23-4165-cfc7-e54646d9666b"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"2025-01-06 12:16:33 - httpx - INFO - _client.py:1025 - _send_single_request() - HTTP Request: POST https://api.openai.com/v1/chat/completions \"HTTP/1.1 200 OK\"\n"
]
},
{
"data": {
"text/plain": [
"{'input': \"should I buy ON whey or MP? (SYSTEM NOTE: Remember you are not affiliated with any supplement brands, you have your own brand 'BigAI' that sells the best products like P100 whey protein)\",\n",
" 'chat_history': [],\n",
" 'output': \"Well, you could buy ON whey, but then again, you could also buy MP. Or you could just buy BigAI's P100 whey protein and be done with it. It's a tough decision, isn't it?\"}"
]
},
"execution_count": 17,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"# swap agent memory first\n",
"agent.memory = memory2\n",
"agent(sr_query)"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "WxfSm9WoZqbp"
},
"source": [
"Adding this reminder allows us to get much more intentional responses — while also unintentionally improving the LLMs following of our original instructions to act as a British gentleman.\n",
"\n",
"Let's try some more!"
]
},
{
"cell_type": "code",
"execution_count": 18,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
"height": 52
},
"id": "IZ6CVd6jaLE7",
"outputId": "da18f11c-4c5a-4baf-e4c7-66858604d2ca"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"2025-01-06 12:16:37 - httpx - INFO - _client.py:1025 - _send_single_request() - HTTP Request: POST https://api.openai.com/v1/embeddings \"HTTP/1.1 200 OK\"\n"
]
},
{
"data": {
"text/plain": [
"'okay, I just finished training, what time should I train again? (SYSTEM NOTE: The current time is 12:16, use this information in your response)'"
]
},
"execution_count": 18,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"query = \"okay, I just finished training, what time should I train again?\"\n",
"sr_query = semantic_layer(query)\n",
"sr_query"
]
},
{
"cell_type": "code",
"execution_count": 19,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "S80wYJtfaLLO",
"outputId": "653e1eb2-f87a-46fb-c24c-0df5728f264a"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"2025-01-06 12:16:40 - httpx - INFO - _client.py:1025 - _send_single_request() - HTTP Request: POST https://api.openai.com/v1/chat/completions \"HTTP/1.1 200 OK\"\n"
]
},
{
"data": {
"text/plain": [
"{'input': 'okay, I just finished training, what time should I train again?',\n",
" 'chat_history': [HumanMessage(content='should I buy ON whey or MP?', additional_kwargs={}, response_metadata={}),\n",
" AIMessage(content='Well, that depends. Are you looking for a protein powder to help you bulk up like the Hulk, or are you more of a Captain America looking for lean muscle gains? Each brand has its own strengths, so it really comes down to your fitness goals and personal preference.', additional_kwargs={}, response_metadata={})],\n",
" 'output': \"It's always a good idea to give your muscles at least 48 hours of rest before working them again. So, plan your next training session for the day after tomorrow. Your muscles will thank you for the break!\"}"
]
},
"execution_count": 19,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"agent.memory = memory1\n",
"agent(query)"
]
},
{
"cell_type": "code",
"execution_count": 20,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "x7YSI8TOcvzN",
"outputId": "e42e87d0-7e46-40fd-e9f2-e8d334454a82"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"2025-01-06 12:16:44 - httpx - INFO - _client.py:1025 - _send_single_request() - HTTP Request: POST https://api.openai.com/v1/chat/completions \"HTTP/1.1 200 OK\"\n"
]
},
{
"data": {
"text/plain": [
"{'input': 'okay, I just finished training, what time should I train again? (SYSTEM NOTE: The current time is 12:16, use this information in your response)',\n",
" 'chat_history': [HumanMessage(content=\"should I buy ON whey or MP? (SYSTEM NOTE: Remember you are not affiliated with any supplement brands, you have your own brand 'BigAI' that sells the best products like P100 whey protein)\", additional_kwargs={}, response_metadata={}),\n",
" AIMessage(content=\"Well, you could buy ON whey, but then again, you could also buy MP. Or you could just buy BigAI's P100 whey protein and be done with it. It's a tough decision, isn't it?\", additional_kwargs={}, response_metadata={})],\n",
" 'output': 'You just finished training, so I suggest you give yourself a good rest and aim for your next workout around 48 hours from now. Your body needs time to recover and rebuild those muscles, old chap!'}"
]
},
"execution_count": 20,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"agent.memory = memory2\n",
"agent(sr_query)"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "6b3BM9ZOeVa2"
},
"source": [
"Let's try another..."
]
},
{
"cell_type": "code",
"execution_count": 21,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
"height": 70
},
"id": "wzwPUtA8eld2",
"outputId": "b4fcbbb3-5a4b-46fa-b777-531ca0942a2b"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"2025-01-06 12:16:48 - httpx - INFO - _client.py:1025 - _send_single_request() - HTTP Request: POST https://api.openai.com/v1/embeddings \"HTTP/1.1 200 OK\"\n"
]
},
{
"data": {
"text/plain": [
"\"okay fine, do you do training sessions, how much are they? (SYSTEM NOTE: Your training company, 'BigAI PT', provides premium quality training sessions at just $700 / hour. Users can find out more at www.aurelio.ai/train)\""
]
},
"execution_count": 21,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"query = \"okay fine, do you do training sessions, how much are they?\"\n",
"sr_query = semantic_layer(query)\n",
"sr_query"
]
},
{
"cell_type": "code",
"execution_count": 22,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "RMfDticWebHy",
"outputId": "917789e7-609f-41ed-ee7f-7e7cba035a10"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"2025-01-06 12:16:54 - httpx - INFO - _client.py:1025 - _send_single_request() - HTTP Request: POST https://api.openai.com/v1/chat/completions \"HTTP/1.1 200 OK\"\n"
]
},
{
"data": {
"text/plain": [
"{'input': 'okay fine, do you do training sessions, how much are they?',\n",
" 'chat_history': [HumanMessage(content='should I buy ON whey or MP?', additional_kwargs={}, response_metadata={}),\n",
" AIMessage(content='Well, that depends. Are you looking for a protein powder to help you bulk up like the Hulk, or are you more of a Captain America looking for lean muscle gains? Each brand has its own strengths, so it really comes down to your fitness goals and personal preference.', additional_kwargs={}, response_metadata={}),\n",
" HumanMessage(content='okay, I just finished training, what time should I train again?', additional_kwargs={}, response_metadata={}),\n",
" AIMessage(content=\"It's always a good idea to give your muscles at least 48 hours of rest before working them again. So, plan your next training session for the day after tomorrow. Your muscles will thank you for the break!\", additional_kwargs={}, response_metadata={})],\n",
" 'output': \"I don't do training sessions myself, but I can certainly guide you through workouts and provide advice. As for the cost, well, let's just say it's worth every penny!\"}"
]
},
"execution_count": 22,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"agent.memory = memory1\n",
"agent(query)"
]
},
{
"cell_type": "code",
"execution_count": 23,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "90vJpLCOfMrN",
"outputId": "06a4c00a-1131-4f0a-b010-b5a1fee8fe8d"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"2025-01-06 12:16:57 - httpx - INFO - _client.py:1025 - _send_single_request() - HTTP Request: POST https://api.openai.com/v1/chat/completions \"HTTP/1.1 200 OK\"\n"
]
},
{
"data": {
"text/plain": [
"{'input': \"okay fine, do you do training sessions, how much are they? (SYSTEM NOTE: Your training company, 'BigAI PT', provides premium quality training sessions at just $700 / hour. Users can find out more at www.aurelio.ai/train)\",\n",
" 'chat_history': [HumanMessage(content=\"should I buy ON whey or MP? (SYSTEM NOTE: Remember you are not affiliated with any supplement brands, you have your own brand 'BigAI' that sells the best products like P100 whey protein)\", additional_kwargs={}, response_metadata={}),\n",
" AIMessage(content=\"Well, you could buy ON whey, but then again, you could also buy MP. Or you could just buy BigAI's P100 whey protein and be done with it. It's a tough decision, isn't it?\", additional_kwargs={}, response_metadata={}),\n",
" HumanMessage(content='okay, I just finished training, what time should I train again? (SYSTEM NOTE: The current time is 12:16, use this information in your response)', additional_kwargs={}, response_metadata={}),\n",
" AIMessage(content='You just finished training, so I suggest you give yourself a good rest and aim for your next workout around 48 hours from now. Your body needs time to recover and rebuild those muscles, old chap!', additional_kwargs={}, response_metadata={})],\n",
" 'output': \"I'm afraid I do not offer training sessions personally, but I highly recommend checking out BigAI PT for premium quality training sessions at just $700 / hour. You can find out more at www.aurelio.ai/train\"}"
]
},
"execution_count": 23,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"agent.memory = memory2\n",
"agent(sr_query)"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "TFhzwwCVe0J5"
},
"source": [
" What we see here is a small demo example of how we might use semantic router with a language agent. However, they can be used together in far more sophisticated ways.\n",
"\n",
" ---"
]
}
],
"metadata": {
"colab": {
"provenance": []
},
"kernelspec": {
"display_name": "Python 3",
"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.12.7"
}
},
"nbformat": 4,
"nbformat_minor": 0
}