diff --git a/Demo_graph.ipynb b/Demo_graph.ipynb index b6b053977..a2356a53a 100644 --- a/Demo_graph.ipynb +++ b/Demo_graph.ipynb @@ -4556,8 +4556,6 @@ "def find_relevant_chunks(query,unique_layer_uuids):\n", " out = []\n", " for id in unique_layer_uuids:\n", - " print(id)\n", - " print(query)\n", " result = qdrant_search(id, query)\n", "\n", " if result:\n", diff --git a/cognitive_architecture/infrastructure/databases/graph/__init__.py b/cognitive_architecture/infrastructure/databases/graph/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/cognitive_architecture/infrastructure/databases/graph/graph_db_interface.py b/cognitive_architecture/infrastructure/databases/graph/graph_db_interface.py new file mode 100644 index 000000000..a7be7acad --- /dev/null +++ b/cognitive_architecture/infrastructure/databases/graph/graph_db_interface.py @@ -0,0 +1,69 @@ +from typing import List +from abc import abstractmethod +from typing import Protocol + +class GraphDBInterface(Protocol): + """ Graphs """ + @abstractmethod + async def create_graph( + self, + graph_name: str, + graph_config: object + ): raise NotImplementedError + + @abstractmethod + async def update_graph( + self, + collection_name: str, + collection_config: object + ): raise NotImplementedError + + # @abstractmethod + # async def delete_collection( + # self, + # collection_name: str + # ): raise NotImplementedError + + # @abstractmethod + # async def create_vector_index( + # self, + # collection_name: str, + # vector_index_config: object + # ): raise NotImplementedError + + # @abstractmethod + # async def create_data_index( + # self, + # collection_name: str, + # vector_index_config: object + # ): raise NotImplementedError + + """ Data points """ + @abstractmethod + async def create_data_points( + self, + collection_name: str, + data_points: List[any] + ): raise NotImplementedError + + # @abstractmethod + # async def get_data_point( + # self, + # collection_name: str, + # data_point_id: str + # ): raise NotImplementedError + + # @abstractmethod + # async def update_data_point( + # self, + # collection_name: str, + # data_point_id: str, + # payload: object + # ): raise NotImplementedError + + # @abstractmethod + # async def delete_data_point( + # self, + # collection_name: str, + # data_point_id: str + # ): raise NotImplementedError diff --git a/cognitive_architecture/infrastructure/databases/graph/neo4j/__init__.py b/cognitive_architecture/infrastructure/databases/graph/neo4j/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/cognitive_architecture/infrastructure/databases/graph/neo4j/adapter.py b/cognitive_architecture/infrastructure/databases/graph/neo4j/adapter.py new file mode 100644 index 000000000..878939a58 --- /dev/null +++ b/cognitive_architecture/infrastructure/databases/graph/neo4j/adapter.py @@ -0,0 +1,7 @@ +from databases.graph.graph_db_interface import GraphDBInterface + + + + +class Neo4jDB(GraphDBInterface): + pass \ No newline at end of file diff --git a/cognitive_architecture/infrastructure/databases/graph/networkx/__init__.py b/cognitive_architecture/infrastructure/databases/graph/networkx/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/cognitive_architecture/infrastructure/databases/graph/networkx/adapter.py b/cognitive_architecture/infrastructure/databases/graph/networkx/adapter.py new file mode 100644 index 000000000..f0d8c6428 --- /dev/null +++ b/cognitive_architecture/infrastructure/databases/graph/networkx/adapter.py @@ -0,0 +1,5 @@ +from databases.graph.graph_db_interface import GraphDBInterface + + +class NetworXDB(GraphDBInterface): + pass \ No newline at end of file