encode auto complete entities

Author: swilly22

app.py

@@ -281,7 +281,7 @@ def auto_complete():
         return jsonify({'status': f'Missing mandatory parameter "prefix"'}), 400
 
     # Fetch auto-completion results
-    completions = auto_complete(repo, prefix)
+    completions = prefix_search(repo, prefix)
 
     # Create a success response
     response = {

code_graph/__init__.py

@@ -4,4 +4,4 @@
 from .git_utils import *
 from .code_coverage import *
 from .analyzers.source_analyzer import *
-from. auto_complete import auto_complete
+from. auto_complete import prefix_search

code_graph/auto_complete.py

@@ -2,7 +2,7 @@
 from .graph import Graph
 from .entities import Function
 
-def auto_complete(repo: str, prefix: str) -> str:
+def prefix_search(repo: str, prefix: str) -> str:
     g = Graph(repo)
     return g.prefix_search(prefix)
 

code_graph/graph.py

@@ -4,6 +4,12 @@
 from typing import Dict, List, Optional
 from falkordb import FalkorDB, Path, Node, QueryResult
 
+# Configure the logger
+import logging
+logging.basicConfig(level=logging.DEBUG,
+                    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
 def list_repos() -> List[str]:
     """
         List processed repositories
@@ -329,27 +335,16 @@ def prefix_search(self, prefix: str) -> str:
             CALL db.idx.fulltext.queryNodes('Searchable', $prefix)
             YIELD node
             WITH node
+            RETURN node
             LIMIT 10
-            RETURN collect(node)
         """
 
-        try:
-            # Execute the query using the provided graph database connection.
-            completions = self._query(query, {'prefix': search_prefix}).result_set[0][0]
-
-            # Remove label Searchable from each node
-            for node in completions:
-                node.labels.remove('Searchable')
+        # Execute the query using the provided graph database connection.
+        result_set = self._query(query, {'prefix': search_prefix}).result_set
 
-            # Sort the results by the label for better organization.
-            completions = sorted(completions, key=lambda x: x.labels)
+        completions = [encode_node(row[0]) for row in result_set]
 
-            return completions
-
-        except Exception as e:
-            # Log any errors encountered during the search and return an empty list.
-            logging.error(f"Error while searching for entities with prefix '{prefix}': {e}")
-            return []
+        return completions
 
 
     def get_function(self, func_id: int) -> Optional[Function]: