add travel history

app.py

@@ -5,13 +5,76 @@ import pytz
 import yaml
 from typing import List, Dict
 import tkapi
-from tkapi.fractie import Fractie, FractieFilter
+from tkapi.fractie import Fractie, FractieFilter, PersoonFilter
+from tkapi.util import queries
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
 api = tkapi.TKApi()
 
+@tool
+def get_kamerlid_reizen(achternaam: str) -> List[Dict[str, str]]:
+    """
+    Retrieves all registered travels for a specific parliament member by their last name.
+    
+    Args:
+        achternaam: The last name of the parliament member to search for
+                   (case-sensitive, partial matches are supported)
+    
+    Returns:
+        List[Dict[str, str]]: A list of dictionaries containing travel information:
+            - 'bestemming': Destination of the travel
+            - 'doel': Purpose of the travel
+            - 'van': Start date of the travel
+            - 'tot': End date of the travel
+            - 'betaald_door': Who paid for the travel
+            
+    Example return value:
+        [
+            {
+                'bestemming': 'Brussels',
+                'doel': 'EU Parliament meeting',
+                'van': '2024-01-15',
+                'tot': '2024-01-17',
+                'betaald_door': 'European Parliament'
+            }
+        ]
+        
+    Note:
+        If multiple parliament members match the given last name, 
+        travels for all matching members will be returned.
+    """
+    try:
+        # Initialize API
+        api = tkapi.TKApi(verbose=False)
+        
+        # Create person filter
+        filter = PersoonFilter()
+        filter.filter_achternaam(achternaam)
+        
+        # Get matching persons
+        persons = api.get_personen(filter=filter)
+        
+        # Collect all travels
+        all_travels = []
+        for person in persons:
+            for reis in person.reizen:
+                travel_info = {
+                    'bestemming': reis.bestemming,
+                    'doel': reis.doel,
+                    'van': str(reis.van) if reis.van else None,
+                    'tot': str(reis.tot_en_met) if reis.tot_en_met else None,
+                    'betaald_door': reis.betaald_door
+                }
+                all_travels.append(travel_info)
+        
+        return all_travels
+    
+    except Exception as e:
+        print(f"Error fetching travel data: {str(e)}")
+        return []
+
 @tool
 def Get_Active_Fractions() -> list[dict[str, str | int]]:
     """
@@ -173,7 +236,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,Get_Active_Fractions,get_fraction_members], ## the given tools
+    tools=[final_answer,Get_Active_Fractions,get_fraction_members,get_kamerlid_reizen], ## the given tools
     max_steps=6,
     verbosity_level=1,
     grammar=None,