73db1a5a53
<!-- .github/pull_request_template.md --> ## Description Introducing scructlog. ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin
96 lines
4.4 KiB
Python
96 lines
4.4 KiB
Python
import modal
|
|
import os
|
|
from cognee.shared.logging_utils import get_logger
|
|
import asyncio
|
|
import cognee
|
|
import signal
|
|
|
|
|
|
from cognee.modules.search.types import SearchType
|
|
|
|
app = modal.App("cognee-runner")
|
|
|
|
image = (
|
|
modal.Image.from_dockerfile(path="Dockerfile_modal", force_build=False)
|
|
.copy_local_file("pyproject.toml", "pyproject.toml")
|
|
.copy_local_file("poetry.lock", "poetry.lock")
|
|
.env({"ENV": os.getenv("ENV"), "LLM_API_KEY": os.getenv("LLM_API_KEY")})
|
|
.poetry_install_from_file(poetry_pyproject_toml="pyproject.toml")
|
|
.pip_install("protobuf", "h2")
|
|
)
|
|
|
|
|
|
@app.function(image=image, concurrency_limit=10)
|
|
async def entry(text: str, query: str):
|
|
logger = get_logger()
|
|
logger.info("Initializing Cognee")
|
|
await cognee.prune.prune_data()
|
|
await cognee.prune.prune_system(metadata=True)
|
|
await cognee.add(text)
|
|
await cognee.cognify()
|
|
search_results = await cognee.search(query_type=SearchType.GRAPH_COMPLETION, query_text=query)
|
|
|
|
return {
|
|
"text": text,
|
|
"query": query,
|
|
"answer": search_results[0] if search_results else None,
|
|
}
|
|
|
|
|
|
@app.local_entrypoint()
|
|
async def main():
|
|
logger = get_logger()
|
|
text_queries = [
|
|
{
|
|
"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": "When does NASA plan to return humans to the Moon under the Artemis program?",
|
|
},
|
|
{
|
|
"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": "How much food waste do households contribute annually according to the 2022 UN report?",
|
|
},
|
|
{
|
|
"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": "What was Tokyo's population according to the 2021 census data?",
|
|
},
|
|
{
|
|
"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": "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?",
|
|
},
|
|
]
|
|
|
|
tasks = [entry.remote.aio(item["text"], item["query"]) for item in text_queries]
|
|
|
|
results = await asyncio.gather(*tasks)
|
|
|
|
logger.info("Final Results:")
|
|
|
|
for result in results:
|
|
logger.info(result)
|
|
logger.info("----")
|
|
|
|
os.kill(os.getpid(), signal.SIGTERM)
|