mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-18 11:51:41 +03:00
9766befb53
# Description - Refactor knowledge to a module - This PR breaks the Github Processor function -> need to comment brain and file imports as it creates a circular dependency issue. Should be fixed and reverted in next PR.
98 lines
2.8 KiB
Python
98 lines
2.8 KiB
Python
from fastapi import HTTPException
|
|
from models.settings import get_supabase_client
|
|
from modules.knowledge.dto.outputs import DeleteKnowledgeResponse
|
|
from modules.knowledge.entity.knowledge import Knowledge
|
|
from modules.knowledge.repository.knowledge_interface import KnowledgeInterface
|
|
|
|
|
|
class Knowledges(KnowledgeInterface):
|
|
def __init__(self):
|
|
supabase_client = get_supabase_client()
|
|
self.db = supabase_client
|
|
|
|
def insert_knowledge(self, knowledge):
|
|
"""
|
|
Add a knowledge
|
|
"""
|
|
response = (self.db.from_("knowledge").insert(knowledge.dict()).execute()).data
|
|
return Knowledge(**response[0])
|
|
|
|
def remove_knowledge_by_id(
|
|
# todo: update remove brain endpoints to first delete the knowledge
|
|
self,
|
|
knowledge_id,
|
|
):
|
|
"""
|
|
Args:
|
|
knowledge_id (UUID): The id of the knowledge
|
|
|
|
Returns:
|
|
str: Status message
|
|
"""
|
|
response = (
|
|
self.db.from_("knowledge")
|
|
.delete()
|
|
.filter("id", "eq", knowledge_id)
|
|
.execute()
|
|
.data
|
|
)
|
|
|
|
if response == []:
|
|
raise HTTPException(404, "Knowledge not found")
|
|
|
|
return DeleteKnowledgeResponse(
|
|
# change to response[0].brain_id and knowledge_id[0].brain_id
|
|
status="deleted",
|
|
knowledge_id=knowledge_id,
|
|
)
|
|
|
|
def get_knowledge_by_id(self, knowledge_id):
|
|
"""
|
|
Get a knowledge by its id
|
|
Args:
|
|
brain_id (UUID): The id of the brain
|
|
"""
|
|
knowledge = (
|
|
self.db.from_("knowledge")
|
|
.select("*")
|
|
.filter("id", "eq", str(knowledge_id))
|
|
.execute()
|
|
).data
|
|
|
|
return Knowledge(**knowledge[0])
|
|
|
|
def get_all_knowledge_in_brain(self, brain_id):
|
|
"""
|
|
Get all the knowledge in a brain
|
|
Args:
|
|
brain_id (UUID): The id of the brain
|
|
"""
|
|
all_knowledge = (
|
|
self.db.from_("knowledge")
|
|
.select("*")
|
|
.filter("brain_id", "eq", str(brain_id))
|
|
.execute()
|
|
).data
|
|
|
|
return [Knowledge(**knowledge) for knowledge in all_knowledge]
|
|
|
|
def remove_brain_all_knowledge(self, brain_id):
|
|
"""
|
|
Remove all knowledge in a brain
|
|
Args:
|
|
brain_id (UUID): The id of the brain
|
|
"""
|
|
all_knowledge = self.get_all_knowledge_in_brain(brain_id)
|
|
knowledge_to_delete_list = []
|
|
|
|
for knowledge in all_knowledge:
|
|
if knowledge.file_name:
|
|
knowledge_to_delete_list.append(f"{brain_id}/{knowledge.file_name}")
|
|
|
|
if knowledge_to_delete_list:
|
|
self.db.storage.from_("quivr").remove(knowledge_to_delete_list)
|
|
|
|
self.db.from_("knowledge").delete().filter(
|
|
"brain_id", "eq", str(brain_id)
|
|
).execute()
|