fix modal

Dockerfile.modal

@@ -25,17 +25,11 @@ COPY pyproject.toml poetry.lock /app/
 
 
 RUN pip install poetry
-
-# Don't create virtualenv since docker is already isolated
-RUN poetry config virtualenvs.create false
-
-# Install the dependencies
-RUN poetry install --all-extras --no-root --without dev
-
 RUN pip install python-dotenv
 RUN pip install transformers
 RUN pip install swebench
 RUN pip install sentencepiece
+RUN pip install protobuf
 
 
 COPY  cognee/ /app/cognee

cognee-mcp/cognee_mcp/server.py

@@ -173,7 +173,7 @@ async def handle_call_tool(
 
                 search_query = arguments.get("query")
 
-                search_results = await cognee.search(SearchType.INSIGHTS, query_text=search_query)
+                search_results = await cognee.search(SearchType.COMPLETION, query_text=search_query)
 
                 results = retrieved_edges_to_string(search_results)
 

eval_with_modal.py

@@ -42,7 +42,8 @@
 @app.function(
     image=image,
     gpu="T4",
-    concurrency_limit=5
+    concurrency_limit=5,
+    timeout=9000000
 )
 async def run_single_repo(instance_data: dict, disable_cognee: bool = False):
     import os