Updates to functions
This commit is contained in:
parent
797e7baba3
commit
66749fadc2
2 changed files with 20 additions and 5 deletions
|
|
@ -86,7 +86,7 @@ async def add_files(file_paths: List[str], dataset_name: str, user_id: str = "d
|
|||
)
|
||||
|
||||
@dlt.resource(standalone = True, merge_key = "id")
|
||||
def data_resources(file_paths: str, user_id: str = user_id):
|
||||
async def data_resources(file_paths: str, user_id: str = user_id):
|
||||
for file_path in file_paths:
|
||||
with open(file_path.replace("file://", ""), mode = "rb") as file:
|
||||
classified_data = ingestion.classify(file)
|
||||
|
|
@ -94,9 +94,9 @@ async def add_files(file_paths: List[str], dataset_name: str, user_id: str = "d
|
|||
data_id = ingestion.identify(classified_data)
|
||||
async with get_async_session_context() as session:
|
||||
if user_id is None:
|
||||
current_active_user = create_default_user()
|
||||
current_active_user = await create_default_user()
|
||||
|
||||
give_permission_document(current_active_user, data_id, "write", session= session)
|
||||
await give_permission_document(current_active_user, data_id, "write", session= session)
|
||||
|
||||
file_metadata = classified_data.get_metadata()
|
||||
|
||||
|
|
|
|||
|
|
@ -1,9 +1,24 @@
|
|||
from cognee.infrastructure.databases.relational.user_authentication.users import authenticate_user_method
|
||||
|
||||
|
||||
async def authenticate_user():
|
||||
async def authenticate_user(email: str, password: str):
|
||||
"""
|
||||
This function is used to authenticate a user.
|
||||
"""
|
||||
output = await authenticate_user_method()
|
||||
output = await authenticate_user_method(email=email, password=password)
|
||||
return output
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
import asyncio
|
||||
# Define an example user
|
||||
example_email = "example@example.com"
|
||||
example_password = "securepassword123"
|
||||
example_is_superuser = False
|
||||
|
||||
# Create an event loop and run the create_user function
|
||||
loop = asyncio.get_event_loop()
|
||||
result = loop.run_until_complete(authenticate_user(example_email, example_password))
|
||||
|
||||
# Print the result
|
||||
print(result)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue