Changed name of arango_ev_class

arango_ev_class.py

@@ -1,70 +0,0 @@
-from arango import ArangoClient
-from dotenv import load_dotenv
-import os
-
-# Load environment variables from .env file
-load_dotenv() # Install with pip install python-dotenv
-
-
-class ArangoDB:
-    def __init__(self):
-        """
-        Initializes an instance of the ArangoEVClass.
-
-        Args:
-            db_name (str): The name of the database.
-            username (str): The username for authentication.
-            password (str): The password for authentication.
-        """
-        password = os.getenv("PASSWORD_ARANGO")
-        self.client = ArangoClient(hosts='https://arango.lasseedfast.se')
-        self.db = self.client.db('ev_dataharvest', username='dataharvest', password=password)
-
-
-    def all_ev_speeches(self):
-            """
-            Retrieves all speeches from the 'ev_speeches' collection.
-
-            Returns:
-                A list of all speeches in the 'ev_speeches' collection.
-            """
-            return list(self.db.collection('ev_speeches').all())
-
-    def update_ev_document(self, document):
-        """
-        Updates an EV document in the 'ev_speeches' collection.
-
-        Args:
-            document: The document to be updated.
-
-        Returns:
-            None
-        """
-        self.db.collection('ev_speeches').update(document, merge=False)
-
-    def get_document_by_id(self, document_id):
-            """
-            Retrieves a document from the 'ev_speeches' collection by its ID.
-
-            Args:
-                document_id (str): The ID of the document to retrieve.
-
-            Returns:
-                dict: The retrieved document.
-
-            """
-            if '/' in document_id:
-                document_id = document_id.split('/')[-1]
-            return self.db.collection('ev_speeches').get(document_id)
-
-
-if __name__ == "__main__":
-    arango = ArangoDB()
-    
-    # Example usage
-    speeches = arango.all_ev_speeches()
-    print(type(speeches))
-    for speech in speeches:
-        print(speech['_key']) # Print the id of the speech
-        print(speech.keys()) # Print the keys of the speech
-        exit()

example_fetch_docs.py

@@ -1,6 +1,6 @@
 # Example of fetching documents and go though them one by one using Ollama:
 
-from arango_ev_class import ArangoDB
+from arango_class import ArangoDB
 import ollama