fix: coderabbit suggestions

topoteretes · Oct 14, 2024 · acd7f10 · acd7f10
1 parent 9bf2cc9
commit acd7f10
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cognee/api/client.py b/cognee/api/client.py
@@ -165,7 +165,7 @@ async def get_datasets(user: User = Depends(get_authenticated_user)):
 async def delete_dataset(dataset_id: str, user: User = Depends(get_authenticated_user)):
     from cognee.modules.data.methods import get_dataset, delete_dataset
 
-    dataset = get_dataset(user.id, dataset_id)
+    dataset = await get_dataset(user.id, dataset_id)
 
     if dataset is None:
         raise HTTPException(

diff --git a/cognee/modules/settings/get_settings.py b/cognee/modules/settings/get_settings.py
@@ -16,7 +16,7 @@ class LLMConfig(BaseModel):
     api_key: str
     model: ConfigChoice
     provider: ConfigChoice
-    models: dict[ModelName, list[ConfigChoice]]
+    models: dict[str, list[ConfigChoice]]
     providers: list[ConfigChoice]
 
 class VectorDBConfig(BaseModel):