Merge pull request #28 from acezxn/python_tsanalyzer_bug_fix

Fixed faulty python function parameter parsing

Author: chengpeng-wang

src/llmtool/dfbscan/intra_dataflow_analyzer.py

@@ -1,7 +1,7 @@
 from os import path
 import json
 import time
-from typing import List, Set, Optional, Dict
+from typing import List, Set, Optional, Dict, Any
 from llmtool.LLM_utils import *
 from llmtool.LLM_tool import *
 from memory.syntactic.function import *
@@ -138,7 +138,7 @@ def _parse_response(
             r"Line:\s*([^;]+);"
         )
 
-        current_path = None
+        current_path: dict[str, Any] | None = None
         for line in response.splitlines():
             line = line.strip().lstrip("-").strip()
             if not line:

src/tstool/analyzer/Python_TS_analyzer.py

@@ -159,6 +159,10 @@ def get_parameters_in_single_function(
             parameter_name = ""
             for sub_node in parameter_node.children:
                 for sub_sub_node in find_nodes_by_type(sub_node, "identifier"):
+                    if sub_sub_node.parent and sub_sub_node.parent.type == "type":
+                        # Disregard type annotations
+                        continue
+
                     parameter_name = file_content[
                         sub_sub_node.start_byte : sub_sub_node.end_byte
                     ]