Merge pull request #25 from Azure-Samples/langchainv1

Update examples

Author: pamelafox

examples/images/langchainv1_supervisor_graph.png

@@ -82,7 +82,7 @@ def get_current_date() -> str:
     return datetime.now().strftime("%Y-%m-%d")
 
 
-activity_agent = create_agent(
+weekend_agent = create_agent(
     model=base_model,
     prompt=(
         "You help users plan their weekends and choose the best activities for the given weather."
@@ -94,10 +94,10 @@ def get_current_date() -> str:
 
 
 @tool
-def activity_agent_tool(query: str) -> str:
+def weekend_agent_tool(query: str) -> str:
     """Invoke the activity planning agent and return its final response as plain text."""
-    logger.info("Tool:activity_agent invoked")
-    response = activity_agent.invoke({"messages": [HumanMessage(content=query)]})
+    logger.info("Tool:weekend_agent invoked")
+    response = weekend_agent.invoke({"messages": [HumanMessage(content=query)]})
     final = response["messages"][-1].content
     return final
 
@@ -147,7 +147,7 @@ def check_fridge() -> list[str]:
         return ["tofu", "soy sauce", "broccoli", "carrots"]
 
 
-recipe_agent = create_agent(
+meal_agent = create_agent(
     model=base_model,
     prompt=(
         "You help users plan meals and choose the best recipes."
@@ -159,10 +159,10 @@ def check_fridge() -> list[str]:
 
 
 @tool
-def recipe_agent_tool(query: str) -> str:
+def meal_agent_tool(query: str) -> str:
     """Invoke the recipe planning agent and return its final response as plain text."""
-    logger.info("Tool:recipe_agent invoked")
-    response = recipe_agent.invoke({"messages": [HumanMessage(content=query)]})
+    logger.info("Tool:meal_agent invoked")
+    response = meal_agent.invoke({"messages": [HumanMessage(content=query)]})
     final = response["messages"][-1].content
     return final
 
@@ -176,7 +176,7 @@ def recipe_agent_tool(query: str) -> str:
         "You are a supervisor, managing an activity planning agent and recipe planning agent."
         "Assign work to them as needed in order to answer user's question."
     ),
-    tools=[activity_agent_tool, recipe_agent_tool],
+    tools=[weekend_agent_tool, meal_agent_tool],
 )
 
 

examples/langchainv1_supervisor.py

@@ -12,7 +12,9 @@
     client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
     MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
 elif API_HOST == "azure":
-    token_provider = azure.identity.get_bearer_token_provider(azure.identity.DefaultAzureCredential(), "https://cognitiveservices.azure.com/.default")
+    token_provider = azure.identity.get_bearer_token_provider(
+        azure.identity.DefaultAzureCredential(), "https://cognitiveservices.azure.com/.default"
+    )
     client = openai.AzureOpenAI(
         api_version=os.environ["AZURE_OPENAI_VERSION"],
         azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
@@ -41,38 +43,16 @@
                 "additionalProperties": False,
             },
         },
-    },
-    {
-        "type": "function",
-        "function": {
-            "name": "lookup_movies",
-            "description": "Lookup movies playing in a given city name or zip code.",
-            "parameters": {
-                "type": "object",
-                "properties": {
-                    "city_name": {
-                        "type": "string",
-                        "description": "The city name",
-                    },
-                    "zip_code": {
-                        "type": "string",
-                        "description": "The zip code",
-                    },
-                },
-                "additionalProperties": False,
-            },
-        },
-    },
+    }
 ]
 
 response = client.chat.completions.create(
     model=MODEL_NAME,
     messages=[
-        {"role": "system", "content": "You are a tourism chatbot."},
-        {"role": "user", "content": "is it rainy enough in sydney to watch movies and which ones are on?"},
+        {"role": "system", "content": "You're a weather chatbot"},
+        {"role": "user", "content": "whats the weather in NYC?"},
     ],
     tools=tools,
-    tool_choice="auto",
 )
 
 print(f"Response from {MODEL_NAME} on {API_HOST}: \n")