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

fix format

Browse source
This commit is contained in:
Pavel Zorin 2025-12-09 18:22:44 +01:00
parent d932ee4bd9
commit 2ca194c28f
1 changed files with 14 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -7,6 +7,7 @@ from typing import Optional, List
from dataclasses import dataclass from dataclasses import dataclass
from fastapi import UploadFile from fastapi import UploadFile
@dataclass @dataclass
class OntologyMetadata: class OntologyMetadata:
ontology_key: str ontology_key: str
@ -80,7 +81,11 @@ class OntologyService:
) )
async def upload_ontologies( async def upload_ontologies(
self, ontology_key: List[str], files: List[UploadFile], user, descriptions: Optional[List[str]] = None self,
ontology_key: List[str],
files: List[UploadFile],
user,
descriptions: Optional[List[str]] = None,
) -> List[OntologyMetadata]: ) -> List[OntologyMetadata]:
""" """
Upload ontology files with their respective keys. Upload ontology files with their respective keys.
@ -106,12 +111,14 @@ class OntologyService:
results = [] results = []
for i, (key, file) in enumerate(zip(ontology_key, files)): for i, (key, file) in enumerate(zip(ontology_key, files)):
results.append(await self.upload_ontology( results.append(
ontology_key=key, await self.upload_ontology(
file=file, ontology_key=key,
user=user, file=file,
description=descriptions[i] if descriptions else None, user=user,
)) description=descriptions[i] if descriptions else None,
)
)
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]: