Create gaia_search.py

tools/gaia_search.py

@@ -0,0 +1,37 @@
+from smolagents import tool
+
+from gaia_loader import search_examples
+
+
+@tool
+def search_gaia_examples(query: str) -> str:
+    """
+    Search similar GAIA benchmark questions.
+
+    Args:
+        query: User question
+    """
+
+    docs = search_examples(query)
+
+    result = ""
+
+    for i, doc in enumerate(docs, 1):
+
+        result += f"""
+
+===========================
+
+Example {i}
+
+Question
+
+{doc["question"]}
+
+Reference Answer
+
+{doc["answer"]}
+
+"""
+
+    return result