Revert "get_neighbors get list of ids"

This reverts commit 9ae4321cc0d5622ef5917b36f11e5cf1dbc4141b.

Author: Anchel123

api/graph.py

@@ -1,7 +1,7 @@
 import os
 import time
 from .entities import *
-from typing import Dict, Optional, List, Tuple
+from typing import Dict, List, Optional, Tuple
 from falkordb import FalkorDB, Path, Node, QueryResult
 
 # Configure the logger
@@ -194,12 +194,12 @@ def get_sub_graph(self, l: int) -> dict:
         return sub_graph
 
 
-    def get_neighbors(self, node_ids: List[int], rel: Optional[str] = None, lbl: Optional[str] = None) -> Dict[str, List[dict]]:
+    def get_neighbors(self, node_id: int, rel: Optional[str] = None, lbl: Optional[str] = None) -> Dict[str, List[dict]]:
         """
-        Fetch the neighbors of a given nodes in the graph based on relationship type and/or label.
+        Fetch the neighbors of a given node in the graph based on relationship type and/or label.
 
         Args:
-            node_ids (List[int]): The IDs of the source nodes.
+            node_id (int): The ID of the source node.
             rel (str, optional): The type of relationship to filter by. Defaults to None.
             lbl (str, optional): The label of the destination node to filter by. Defaults to None.
 
@@ -208,8 +208,8 @@ def get_neighbors(self, node_ids: List[int], rel: Optional[str] = None, lbl: Opt
         """
 
         # Validate inputs
-        if not all(isinstance(node_id, int) for node_id in node_ids):
-            raise ValueError("node_ids must be an integer list")
+        if not isinstance(node_id, int):
+            raise ValueError("node_id must be an integer")
 
         # Build relationship and label query parts
         rel_query = f":{rel}" if rel else ""
@@ -218,7 +218,7 @@ def get_neighbors(self, node_ids: List[int], rel: Optional[str] = None, lbl: Opt
         # Parameterized Cypher query to find neighbors
         query = f"""
             MATCH (n)-[e{rel_query}]->(dest{lbl_query})
-            WHERE ID(n) IN $node_ids
+            WHERE ID(n) = $node_id
             RETURN e, dest
         """
 
@@ -227,7 +227,7 @@ def get_neighbors(self, node_ids: List[int], rel: Optional[str] = None, lbl: Opt
 
         try:
             # Execute the graph query with node_id parameter
-            result_set = self._query(query, {'node_ids': node_ids}).result_set
+            result_set = self._query(query, {'node_id': node_id}).result_set
 
             # Iterate over the result set and process nodes and edges
             for edge, destination_node in result_set:
@@ -237,7 +237,7 @@ def get_neighbors(self, node_ids: List[int], rel: Optional[str] = None, lbl: Opt
             return neighbors
 
         except Exception as e:
-            logging.error(f"Error fetching neighbors for node {node_ids}: {e}")
+            logging.error(f"Error fetching neighbors for node {node_id}: {e}")
             return {'nodes': [], 'edges': []}
 
 

api/index.py

@@ -79,31 +79,28 @@ def graph_entities():
         return jsonify({"status": "Internal server error"}), 500
 
 
-@app.route('/get_neighbors', methods=['POST'])
+@app.route('/get_neighbors', methods=['GET'])
 @token_required  # Apply token authentication decorator
 def get_neighbors():
     """
-    Endpoint to get neighbors of a nodes list in the graph.
-    Expects 'repo' and 'node_ids' as body parameters.
+    Endpoint to get neighbors of a specific node in the graph.
+    Expects 'repo' and 'node_id' as query parameters.
 
     Returns:
         JSON response containing neighbors or error messages.
     """
 
-    # Get JSON data from the request
-    data = request.get_json()
-
     # Get query parameters
-    repo    = data.get('repo')
-    node_ids = data.get('node_ids')
+    repo    = request.args.get('repo')
+    node_id = request.args.get('node_id')
 
     # Validate 'repo' parameter
     if not repo:
         logging.error("Repository name is missing in the request.")
         return jsonify({"status": "Repository name is required."}), 400
 
     # Validate 'node_id' parameter
-    if not node_ids:
+    if not node_id:
         logging.error("Node ID is missing in the request.")
         return jsonify({"status": "Node ID is required."}), 400
 
@@ -114,19 +111,19 @@ def get_neighbors():
 
     # Try converting node_id to an integer
     try:
-        node_ids = [int(node_id) for node_id in node_ids]
+        node_id = int(node_id)
     except ValueError:
-        logging.error(f"Invalid node ID: {node_ids}. It must be an integer.")
+        logging.error(f"Invalid node ID: {node_id}. It must be an integer.")
         return jsonify({"status": "Invalid node ID. It must be an integer."}), 400
 
     # Initialize the graph with the provided repository
     g = Graph(repo)
 
     # Fetch the neighbors of the specified node
-    neighbors = g.get_neighbors(node_ids)
+    neighbors = g.get_neighbors(node_id)
 
     # Log and return the neighbors
-    logging.info(f"Successfully retrieved neighbors for node ID {node_ids} in repo '{repo}'.")
+    logging.info(f"Successfully retrieved neighbors for node ID {node_id} in repo '{repo}'.")
 
     response = {
         'status': 'success',
@@ -173,41 +170,6 @@ def auto_complete():
 
     return jsonify(response), 200
 
-@app.route('/process_repo', methods=['POST'])
-@token_required  # Apply token authentication decorator
-def process_repo():
-     """
-     Process a GitHub repository.
-
-     Expected JSON payload:
-     {
-         "repo_url": "string",
-         "ignore": ["string"]  # optional
-     }
-
-     Returns:
-         JSON response with processing status
-     """
-
-     data = request.get_json()
-     url = data.get('repo_url')
-     if url is None:
-         return jsonify({'status': f'Missing mandatory parameter "url"'}), 400
-     logger.debug(f'Received repo_url: {url}')
-
-     ignore = data.get('ignore', [])
-
-     proj = Project.from_git_repository(url)
-     proj.analyze_sources(ignore)
-    #  proj.process_git_history(ignore)
-
-     # Create a response
-     response = {
-         'status': 'success',
-     }
-
-     return jsonify(response), 200
-
 
 @app.route('/list_repos', methods=['GET'])
 @token_required  # Apply token authentication decorator
@@ -221,7 +183,7 @@ def list_repos():
 
     # Fetch list of repositories
     repos = get_repos()
-    print(repos)
+
     # Create a success response with the list of repositories
     response = {
         'status': 'success',