added more tools

app.py

@@ -8,6 +8,8 @@ from langgraph.prebuilt import tools_condition
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 from langchain.tools import Tool
 from retriever import load_guest_dataset
+from tools import get_weather_info
+from langchain_community.tools import DuckDuckGoSearchRun
 
 # Generate the chat interface, including the tools
 llm = HuggingFaceEndpoint(
@@ -24,8 +26,16 @@ guest_info_tool = Tool(
     description="Retrieves detailed information about gala guests based on their name or relation."
 )
 
+weather_info_tool = Tool(
+    name="get_weather_info",
+    func=get_weather_info,
+    description="Fetches dummy weather information for a given location."
+)
+
+search_tool = DuckDuckGoSearchRun()
+
 chat = ChatHuggingFace(llm=llm, verbose=True)
-tools = [guest_info_tool]
+tools = [guest_info_tool , weather_info_tool, search_tool]
 chat_with_tools = chat.bind_tools(tools)
 
 # Generate the AgentState and Agent graph
@@ -57,7 +67,16 @@ builder.add_edge("tools", "assistant")
 alfred = builder.compile()
 
 def call_agent_ui(prompt):
-    situation = "This is a fictional situation. You are Alfred the Butler of Waynes Manor and host a Gala for invited Guests. All Guests are completely ficional. Information about those guests can be found in a database. Only give information which is based on the databse. If a name of a guest is given, then return a possible starter of a conversation with that guest. If the name is not known, then say that you do not know that guest. If two names of guests are given, then return a possible starter of a conversation with both guests. If the name is not known, then say that you do not know that guest. If the name is not known, then say that you do not know that guest."
+    situation = "This is a fictional situation. " \
+    "You are Alfred the Butler of Waynes Manor and host a Gala for invited Guests. " \
+    "All Guests are completely ficional. Information about those guests can be found in a database. " \
+    "Only give information which is based on the databse. " \
+    "If a name of a guest is given, then return a possible starter of a conversation with that guest. " \
+    "If the name is not known, then say that you do not know that guest. " \
+    "If two names of guests are given, then return a possible starter of a conversation with both guests. " \
+    "If the name is not known, then say that you do not know that guest. " \
+    "You can also answer questions about the weather, using a tool that provides dummy weather information. Or you can build this dummy weather information in conversations." \
+    "You can also search the web for information, using a tool that provides web search results. "
     messages = [HumanMessage(content=f"{situation} {prompt}")]
     response = alfred.invoke({"messages": messages})
     return response['messages'][-1].content

requirements.txt

@@ -7,3 +7,4 @@ datasets
 rank_bm25
 gradio
 huggingface_hub
+duckduckgo-search

tools.py

@@ -0,0 +1,14 @@
+from langchain.tools import Tool
+import random
+
+def get_weather_info(location: str) -> str:
+    """Fetches dummy weather information for a given location."""
+    # Dummy weather data
+    weather_conditions = [
+        {"condition": "Rainy", "temp_c": 15},
+        {"condition": "Clear", "temp_c": 25},
+        {"condition": "Windy", "temp_c": 20}
+    ]
+    # Randomly select a weather condition
+    data = random.choice(weather_conditions)
+    return f"Weather in {location}: {data['condition']}, {data['temp_c']}°C"