chore: cleans the modal entrypoint

modal_wrapper.py

@@ -3,8 +3,6 @@ import os
 import logging
 import asyncio
 import cognee
-import sentry_sdk
-import concurrent.futures
 import signal
 
 from cognee.api.v1.search import SearchType
@@ -22,44 +20,64 @@ image = (
 )
 
 
-@app.function(image=image, concurrency_limit=5)
+@app.function(image=image, concurrency_limit=10)
 async def entry(text: str, query: str):
-    try:
+    setup_logging(logging.ERROR)
-        setup_logging(logging.ERROR)
+    await cognee.prune.prune_data()
-        sentry_sdk.init(dsn=None)
+    await cognee.prune.prune_system(metadata=True)
-        await cognee.prune.prune_data()
+    await cognee.add(text)
-        await cognee.prune.prune_system(metadata=True)
+    await cognee.cognify()
-        await cognee.add(text)
+    search_results = await cognee.search(SearchType.GRAPH_COMPLETION, query_text=query)
-        await cognee.cognify()
-        search_results = await cognee.search(SearchType.GRAPH_COMPLETION, query_text=query)
 
-        return {
+    return {
-            "text": text,
+        "text": text,
-            "query": query,
+        "query": query,
-            "answer": search_results[0] if search_results else None,
+        "answer": search_results[0] if search_results else None,
-        }
+    }
-    finally:
-        await asyncio.sleep(1)
 
 
 @app.local_entrypoint()
 async def main():
     text_queries = [
         {
-            "text": "The Mars 2023 mission discovered 4.3% water content in soil samples from Jezero Crater.",
+            "text": "NASA's Artemis program aims to return humans to the Moon by 2026, focusing on sustainable exploration and preparing for future Mars missions.",
-            "query": "What percentage of water was found in Jezero Crater's soil based on the provided context?",
+            "query": "When does NASA plan to return humans to the Moon under the Artemis program?",
         },
         {
-            "text": "Bluefin tuna populations decreased by 72% in the Mediterranean between 2010-2022 according to WWF.",
+            "text": "According to a 2022 UN report, global food waste amounts to approximately 931 million tons annually, with households contributing 61% of the total.",
-            "query": "What percentage of water was found in Jezero Crater's soil based on the provided context?",
+            "query": "How much food waste do households contribute annually according to the 2022 UN report?",
         },
         {
-            "text": "Tesla's Q2 2024 report shows 412,000 Model Y vehicles produced with new 4680 battery cells.",
+            "text": "The 2021 census data revealed that Tokyo's population reached 14 million, reflecting a 2.1% increase compared to the previous census conducted in 2015.",
-            "query": "How many Model Y cars used the 4680 batteries in Q2 2024?",
+            "query": "What was Tokyo's population according to the 2021 census data?",
         },
         {
-            "text": "A 2023 Johns Hopkins study found 23-minute daily naps improve cognitive performance by 18% in adults.",
+            "text": "A recent study published in the Journal of Nutrition found that consuming 30 grams of almonds daily can lower LDL cholesterol levels by 7% over a 12-week period.",
-            "query": "What duration of daily naps boosts cognition according to the 2023 study?",
+            "query": "How much can daily almond consumption lower LDL cholesterol according to the study?",
+        },
+        {
+            "text": "Amazon's Prime membership grew to 200 million subscribers in 2023, marking a 10% increase from the previous year, driven by exclusive content and faster delivery options.",
+            "query": "How many Prime members did Amazon have in 2023?",
+        },
+        {
+            "text": "A new report by the International Energy Agency states that global renewable energy capacity increased by 295 gigawatts in 2022, primarily driven by solar and wind power expansion.",
+            "query": "By how much did global renewable energy capacity increase in 2022 according to the report?",
+        },
+        {
+            "text": "The World Health Organization reported in 2023 that the global life expectancy has risen to 73.4 years, an increase of 5.5 years since the year 2000.",
+            "query": "What is the current global life expectancy according to the WHO's 2023 report?",
+        },
+        {
+            "text": "The FIFA World Cup 2022 held in Qatar attracted a record-breaking audience of 5 billion people across various digital and traditional broadcasting platforms.",
+            "query": "How many people watched the FIFA World Cup 2022?",
+        },
+        {
+            "text": "The European Space Agency's JUICE mission, launched in 2023, aims to explore Jupiter's icy moons, including Ganymede, Europa, and Callisto, over the next decade.",
+            "query": "Which moons is the JUICE mission set to explore?",
+        },
+        {
+            "text": "According to a report by the International Labour Organization, the global unemployment rate in 2023 was estimated at 5.4%, reflecting a slight decrease compared to the previous year.",
+            "query": "What was the global unemployment rate in 2023 according to the ILO?",
         },
     ]
 
@@ -73,6 +91,4 @@ async def main():
         print(result)
         print("----")
 
-    os.kill(os.getpid(), signal.SIGKILL)
+    os.kill(os.getpid(), signal.SIGTERM)
-
-    return 0