gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: fixes modal deprecation warnings

Browse source
This commit is contained in:
hajdul88 2025-03-18 15:22:16 +01:00
parent 24e0805f50
commit 334506b298
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
modal_deployment.py
View file

@ -12,15 +12,16 @@ app = modal.App("cognee-runner")
image = ( image = (
modal.Image.from_dockerfile(path="Dockerfile_modal", force_build=False) modal.Image.from_dockerfile(path="Dockerfile_modal", force_build=False)
.copy_local_file("pyproject.toml", "pyproject.toml") .add_local_file("pyproject.toml", remote_path="/root/pyproject.toml", copy=True)
.copy_local_file("poetry.lock", "poetry.lock") .add_local_file("poetry.lock", remote_path="/root/poetry.lock", copy=True)
.env({"ENV": os.getenv("ENV"), "LLM_API_KEY": os.getenv("LLM_API_KEY")}) .env({"ENV": os.getenv("ENV"), "LLM_API_KEY": os.getenv("LLM_API_KEY")})
.poetry_install_from_file(poetry_pyproject_toml="pyproject.toml") .poetry_install_from_file(poetry_pyproject_toml="pyproject.toml")
.pip_install("protobuf", "h2") .pip_install("protobuf", "h2")
.add_local_python_source("cognee")
) )
@app.function(image=image, concurrency_limit=10) @app.function(image=image, max_containers=4)
async def entry(text: str, query: str): async def entry(text: str, query: str):
setup_logging(logging.ERROR) setup_logging(logging.ERROR)
await cognee.prune.prune_data() await cognee.prune.prune_data()