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

Chore: Remove Ontology file size limit. Code duplications

Browse source
This commit is contained in:
Pavel Zorin 2025-12-09 17:55:43 +01:00
parent 49f7c5188c
commit d0b914acaa
1 changed files with 8 additions and 42 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
cognee/api/v1/ontologies/ontologies.py
View file

@ -5,7 +5,7 @@ from pathlib import Path
from datetime import datetime, timezone from datetime import datetime, timezone
from typing import Optional, List from typing import Optional, List
from dataclasses import dataclass from dataclasses import dataclass
from fastapi import UploadFile
@dataclass @dataclass
class OntologyMetadata: class OntologyMetadata:
@ -45,7 +45,7 @@ class OntologyService:
json.dump(metadata, f, indent=2) json.dump(metadata, f, indent=2)
async def upload_ontology( async def upload_ontology(
self, ontology_key: str, file, user, description: Optional[str] = None self, ontology_key: str, file: UploadFile, user, description: Optional[str] = None
) -> OntologyMetadata: ) -> OntologyMetadata:
if not file.filename.lower().endswith(".owl"): if not file.filename.lower().endswith(".owl"):
raise ValueError("File must be in .owl format") raise ValueError("File must be in .owl format")
@ -57,8 +57,6 @@ class OntologyService:
raise ValueError(f"Ontology key '{ontology_key}' already exists") raise ValueError(f"Ontology key '{ontology_key}' already exists")
content = await file.read() content = await file.read()
if len(content) > 10 * 1024 * 1024:
raise ValueError("File size exceeds 10MB limit")
file_path = user_dir / f"{ontology_key}.owl" file_path = user_dir / f"{ontology_key}.owl"
with open(file_path, "wb") as f: with open(file_path, "wb") as f:
@ -105,47 +103,15 @@ class OntologyService:
if len(set(ontology_key)) != len(ontology_key): if len(set(ontology_key)) != len(ontology_key):
raise ValueError("Duplicate ontology keys not allowed") raise ValueError("Duplicate ontology keys not allowed")
if descriptions and len(descriptions) != len(files):
raise ValueError("Number of descriptions must match number of files")
results = [] results = []
user_dir = self._get_user_dir(str(user.id))
metadata = self._load_metadata(user_dir)
for i, (key, file) in enumerate(zip(ontology_key, files)): for i, (key, file) in enumerate(zip(ontology_key, files)):
if key in metadata: results.append(await self.upload_ontology(
raise ValueError(f"Ontology key '{key}' already exists") ontology_key=key,
file=file,
if not file.filename.lower().endswith(".owl"): user=user,
raise ValueError(f"File '{file.filename}' must be in .owl format") description=descriptions[i] if descriptions else None,
))
content = await file.read()
if len(content) > 10 * 1024 * 1024:
raise ValueError(f"File '{file.filename}' exceeds 10MB limit")
file_path = user_dir / f"{key}.owl"
with open(file_path, "wb") as f:
f.write(content)
ontology_metadata = {
"filename": file.filename,
"size_bytes": len(content),
"uploaded_at": datetime.now(timezone.utc).isoformat(),
"description": descriptions[i] if descriptions else None,
}
metadata[key] = ontology_metadata
results.append(
OntologyMetadata(
ontology_key=key,
filename=file.filename,
size_bytes=len(content),
uploaded_at=ontology_metadata["uploaded_at"],
description=descriptions[i] if descriptions else None,
)
)
self._save_metadata(user_dir, metadata)
return results return results
def get_ontology_contents(self, ontology_key: List[str], user) -> List[str]: def get_ontology_contents(self, ontology_key: List[str], user) -> List[str]: