Kamervragen Implemented

app.py

@@ -8,12 +8,99 @@ import tkapi
 from tkapi.fractie import Fractie, FractieFilter
 from tkapi.persoon import PersoonFilter, PersoonNevenfunctie
 from tkapi.util import queries
+from tkapi.zaak import Zaak, ZaakSoort
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
 api = tkapi.TKApi()
 
+@tool 
+def get_kamervragen(
+    start_datetime: datetime.datetime,
+    end_datetime: datetime.datetime,
+    only_answered: Optional[bool] = None,
+    zaak_soort: ZaakSoort = ZaakSoort.SCHRIFTELIJKE_VRAGEN
+) -> List[Dict[str, any]]:
+    """
+    Retrieves basic information about parliamentary items (kamervragen, moties, etc.) within a specified date range.
+    
+    Args:
+        start_datetime: Start date as datetime object
+        end_datetime: End date as datetime object
+        only_answered: Optional filter for answered items (True = only answered, False = only unanswered, None = both)
+        zaak_soort: Type of parliamentary item to filter on. Options from ZaakSoort enum:
+            - AMENDEMENT: Amendments
+            - BRIEF_COMMISSIE: Committee letters
+            - BRIEF_REGERING: Government letters
+            - BRIEF_LID: Member letters
+            - INITIATIEF_NOTA: Initiative notes
+            - INITIATIEF_WETGEVING: Initiative legislation
+            - INTERPELLATIE: Interpellations
+            - LIJST_MET_VRAGEN: List of questions
+            - MEDEDELINGEN: Announcements
+            - MONDELINGE_VRAGEN: Oral questions
+            - MOTIE: Motions
+            - OVERIG: Other
+            - PARLEMENTAIR_DOCUMENT: Parliamentary documents
+            - RAPPORT: Reports
+            - SCHRIFTELIJKE_VRAGEN: Written questions (default)
+            - VERSLAG_COMMISSIE: Committee reports
+            - VOORDRACHT: Nominations
+            - WETSVOORSTEL: Bill proposals
+    
+    Returns:
+        List[Dict[str, any]]: List of dictionaries containing basic item information:
+            - 'nummer': Unique identifier of the item
+            - 'soort': Type of the item
+            - 'onderwerp': Subject/topic of the item
+            - 'datum': Date the item was started (as datetime.date)
+            - 'afgedaan': Boolean indicating if the item has been handled/answered
+            
+    Example return value:
+        [
+            {
+                'nummer': '2024Z00541',
+                'soort': 'Motie',
+                'onderwerp': 'Motie van de leden over uitingen van de schutterijcultuur',
+                'datum': datetime.date(2024, 1, 18),
+                'afgedaan': True
+            }
+        ]
+    """
+    try:
+        # Initialize API
+        api = TKApi(verbose=False)
+        
+        # Create filter
+        filter = Zaak.create_filter()
+        filter.filter_soort(zaak_soort)
+        filter.filter_date_range(start_datetime, end_datetime)
+        
+        if only_answered is not None:
+            filter.filter_afgedaan(only_answered)
+        
+        # Get items
+        zaken = api.get_zaken(filter=filter)
+        
+        # Format results
+        results = []
+        for zaak in zaken:
+            zaak_info = {
+                'nummer': zaak.nummer,
+                'soort': zaak.soort.value if zaak.soort else None,
+                'onderwerp': zaak.onderwerp,
+                'datum': zaak.gestart_op,
+                'afgedaan': zaak.afgedaan
+            }
+            results.append(zaak_info)
+        
+        return results
+        
+    except Exception as e:
+        print(f"Error fetching parliamentary items: {str(e)}")
+        return []
+
 @tool
 def get_kamerlid_reizen(achternaam: str) -> List[Dict[str, str]]:
     """
@@ -361,7 +448,13 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,Get_Active_Fractions,get_fraction_members,get_kamerlid_reizen,get_kamerlid_geschenken, get_kamerlid_nevenfuncties], ## the given tools
+    tools=[final_answer,
+           Get_Active_Fractions,
+           get_fraction_members,
+           get_kamerlid_reizen,
+           get_kamerlid_geschenken,
+           get_kamerlid_nevenfuncties,
+           get_kamervragen], ## the given tools
     max_steps=6,
     verbosity_level=1,
     grammar=None,