Update treesitter analysis to use global declerations of parser and language

rahlk · rahlk · commit 3bd115dead5d · 2025-05-06T15:13:52.000-04:00
Signed-off-by: Rahul Krishna &lt;i.m.ralk@gmail.com&gt;
diff --git a/cldk/analysis/python/python_analysis.py b/cldk/analysis/python/python_analysis.py
@@ -30,17 +30,11 @@ class PythonAnalysis:
 
     def __init__(
         self,
-        eager_analysis: bool,
         project_dir: str | Path | None,
         source_code: str | None,
-        analysis_backend_path: str | None,
-        analysis_json_path: str | Path | None,
     ) -> None:
         self.project_dir = project_dir
         self.source_code = source_code
-        self.analysis_json_path = analysis_json_path
-        self.analysis_backend_path = analysis_backend_path
-        self.eager_analysis = eager_analysis
         self.analysis_backend: TreesitterPython = TreesitterPython()
 
     def get_methods(self) -> List[PyMethod]:
diff --git a/cldk/core.py b/cldk/core.py
@@ -27,6 +27,7 @@
 from cldk.analysis.c import CAnalysis
 from cldk.analysis.java import JavaAnalysis
 from cldk.analysis.commons.treesitter import TreesitterJava
+from cldk.analysis.python.python_analysis import PythonAnalysis
 from cldk.utils.exceptions import CldkInitializationException
 from cldk.utils.sanitization.java import TreesitterSanitizer
 
@@ -118,6 +119,14 @@ def analysis(
                 target_files=target_files,
                 eager_analysis=eager,
             )
+        elif self.language == "python":
+            return PythonAnalysis(
+                project_dir=project_path,
+                source_code=source_code,
+                analysis_backend_path=analysis_backend_path,
+                analysis_json_path=analysis_json_path,
+                eager_analysis=eager,
+            )
         elif self.language == "c":
             return CAnalysis(project_dir=project_path)
         else:
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cldk"
-version = "1.0.0"
+version = "1.0.1"
 description = "The official Python SDK for Codellm-Devkit."
 authors = ["Rahul Krishna <i.m.ralk@gmail.com>", "Rangeet Pan <rangeet.pan@ibm.com>", "Saurabh Sinhas <sinhas@us.ibm.com>",
             "Raju Pavuluri <pavuluri@us.ibm.com>"]
