add some tools

app.py

@@ -18,7 +18,53 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
-chat_history = []  # Global variable to store conversation history
+@tool
+def search_flights(departure: str, destination: str, date: str) -> str:
+    """Finds flights from departure to destination on the given date using DuckDuckGo.
+    
+    Args:
+        departure: The city or airport code where the flight starts.
+        destination: The city or airport code where the flight ends.
+        date: The departure date in YYYY-MM-DD format.
+
+    Returns:
+        A string containing flight search results.
+    """
+    query = f"flights from {departure} to {destination} on {date}"
+    search_results = DuckDuckGoSearchTool()(query)  # Calls DuckDuckGo search
+    return f"Here are some flight options:\n{search_results}"
+
+API_KEY = "fca_live_L2xM404CxH5IQWoNotD3X645W8lv7UHOoJdswwgd"  # 🔹 Your API key
+
+@tool
+def convert_currency(amount: float, from_currency: str, to_currency: str) -> str:
+    """Converts currency from one to another using FreeCurrencyAPI.
+    
+    Args:
+        amount: The amount to convert.
+        from_currency: The original currency (e.g., "USD").
+        to_currency: The target currency (e.g., "EUR").
+
+    Returns:
+        The converted amount in the target currency.
+    """
+    try:
+        url = f"https://api.freecurrencyapi.com/v1/latest?apikey={API_KEY}&base_currency={from_currency.upper()}"
+
+        response = requests.get(url).json()
+
+        # ✅ Check if the API returned valid exchange rates
+        if "data" in response and to_currency.upper() in response["data"]:
+            rate = response["data"][to_currency.upper()]
+            converted_amount = amount * rate
+            return f"{amount} {from_currency.upper()} is approximately {converted_amount:.2f} {to_currency.upper()}."
+
+        return f"Error: Could not find exchange rate for {to_currency.upper()}."
+
+    except Exception as e:
+        return f"Error fetching exchange rates: {str(e)}"
+
+chat_history = []  # Store conversation history
 
 @tool
 def chat_with_ai(message: str) -> str:
@@ -26,25 +72,34 @@ def chat_with_ai(message: str) -> str:
     Args:
         message: The user's message.
     """
-    global chat_history  # Keep track of past messages
+    global chat_history  
 
-    # Limit conversation history to the last 5 messages
+    # Keep the last 5 messages for context
     if len(chat_history) > 5:
-        chat_history.pop(0)  # Remove the oldest message
+        chat_history.pop(0)
 
-    # Add the new message to history
+    # Add user message to history
     chat_history.append({"role": "user", "content": message})
 
-    # Format history as context for the AI
+    # Format the history as input for the AI model
     formatted_history = "\n".join(f"{msg['role']}: {msg['content']}" for msg in chat_history)
 
-    # Generate AI response
-    response = model(formatted_history)  # AI model processes chat history
-
-    # Add AI response to history
-    chat_history.append({"role": "assistant", "content": response})
+    # ✅ Ensure AI response is handled correctly
+    try:
+        response = model(formatted_history)  # Call model
+        
+        # ✅ Handle both string and dictionary responses
+        if isinstance(response, dict):  
+            response_text = response.get("text", str(response))  # Extract text if available
+        else:
+            response_text = str(response)  # Convert non-dict responses to string
+
+        # Add AI response to history
+        chat_history.append({"role": "assistant", "content": response_text})
+        return response_text
 
-    return response
+    except Exception as e:
+        return f"Error processing chat: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -68,7 +123,7 @@ final_answer = FinalAnswerTool()
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
 model = HfApiModel(
-max_tokens=512,
+max_tokens=2096,
 temperature=0.5,
 model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
 custom_role_conversions=None,
@@ -83,15 +138,20 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[my_custom_tool, chat_with_ai, get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
+    tools=[
+        final_answer,
+        get_current_time_in_timezone,
+        my_custom_tool,
+        chat_with_ai,           # Regular chat tool
+        search_flights,
+        #scrape_webpage,
+        convert_currency,
+        #get_weather,
+        #generate_ai_image
+    ],
     max_steps=6,
     verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
     prompt_templates=prompt_templates
 )
 
-
 GradioUI(agent).launch()