fix: remove manual lancedb folder creation

This commit is contained in:
Boris Arzentar 2024-05-27 08:49:22 +02:00
parent e1876e2590
commit ebabae0308
5 changed files with 10 additions and 11 deletions

View file

@ -10,7 +10,7 @@
"dependencies": { "dependencies": {
"classnames": "^2.5.1", "classnames": "^2.5.1",
"next": "14.2.3", "next": "14.2.3",
"ohmy-ui": "^0.0.3", "ohmy-ui": "^0.0.4",
"react": "^18", "react": "^18",
"react-dom": "^18", "react-dom": "^18",
"uuid": "^9.0.1" "uuid": "^9.0.1"
@ -25,7 +25,7 @@
"typescript": "^5" "typescript": "^5"
} }
}, },
"../../../Guerrilla/ohmy-ui": { "../Guerrilla/ohmy-ui": {
"version": "0.0.3", "version": "0.0.3",
"extraneous": true, "extraneous": true,
"license": "MIT", "license": "MIT",
@ -2892,9 +2892,9 @@
} }
}, },
"node_modules/ohmy-ui": { "node_modules/ohmy-ui": {
"version": "0.0.3", "version": "0.0.4",
"resolved": "https://registry.npmjs.org/ohmy-ui/-/ohmy-ui-0.0.3.tgz", "resolved": "https://registry.npmjs.org/ohmy-ui/-/ohmy-ui-0.0.4.tgz",
"integrity": "sha512-GXtTZxbL+sXdlT26+mcSrqvUPUIWNUiZOtKT+0lxK6DglGezB0Y/NVQHBcsoPCzmXHY3wi4UBaCH6//8DjHKXA==", "integrity": "sha512-E4xgoDutgXptXGmtSFnRwMcvpQ1/r6N5Vzua0a4sq8koFzt2FJuhuf/+o4WjABEi3UPvOpaLdy3/jKn9cFG9yg==",
"dependencies": { "dependencies": {
"classnames": "^2.3.2" "classnames": "^2.3.2"
}, },

View file

@ -11,7 +11,7 @@
"dependencies": { "dependencies": {
"classnames": "^2.5.1", "classnames": "^2.5.1",
"next": "14.2.3", "next": "14.2.3",
"ohmy-ui": "^0.0.3", "ohmy-ui": "^0.0.4",
"react": "^18", "react": "^18",
"react-dom": "^18", "react-dom": "^18",
"uuid": "^9.0.1" "uuid": "^9.0.1"

View file

@ -85,7 +85,8 @@ async def cognify(datasets: Union[str, List[str]] = None):
logger.error(f"Dataset {dataset_name} is already being processed.") logger.error(f"Dataset {dataset_name} is already being processed.")
return return
update_task_status(dataset_name, "DATASET_PROCESSING_STARTED") update_task_status(dataset_name, "DATASET_PROCESSING_STARTED")
await cognify(dataset_name) await cognify(dataset_name)
update_task_status(dataset_name, "DATASET_PROCESSING_FINISHED") update_task_status(dataset_name, "DATASET_PROCESSING_FINISHED")

View file

@ -16,8 +16,6 @@ class prune():
@staticmethod @staticmethod
async def prune_system(graph = True, vector = True): async def prune_system(graph = True, vector = True):
infra_config = infrastructure_config.get_config()
if graph: if graph:
graph_client = await get_graph_client(graph_config.graph_engine) graph_client = await get_graph_client(graph_config.graph_engine)
await graph_client.delete_graph() await graph_client.delete_graph()

View file

@ -28,9 +28,9 @@ def create_vector_engine(config: VectorConfig, embedding_engine):
) )
else: else:
from .lancedb.LanceDBAdapter import LanceDBAdapter from .lancedb.LanceDBAdapter import LanceDBAdapter
from cognee.infrastructure.files.storage import LocalStorage # from cognee.infrastructure.files.storage import LocalStorage
LocalStorage.ensure_directory_exists(config["vector_db_url"]) # LocalStorage.ensure_directory_exists(config["vector_db_url"])
return LanceDBAdapter( return LanceDBAdapter(
url = config["vector_db_url"], url = config["vector_db_url"],