Added missing attributes in compilation unit model

sinha108 · sinha108 · commit c2fa8bf3ba58 · 2025-05-21T07:08:31.000-04:00
Signed-off-by: Saurabh Sinha &lt;sinha108@gmail.com&gt;
diff --git a/cldk/models/java/models.py b/cldk/models/java/models.py
@@ -363,11 +363,15 @@ class JCompilationUnit(BaseModel):
     """Represents a compilation unit in Java.
 
     Attributes:
+        file_path (str): The path to the source file.
+        package_name (str): The name of the package for the comppilation unit.
         comments (List[JComment]): A list of comments in the compilation unit.
         imports (List[str]): A list of import statements in the compilation unit.
         type_declarations (Dict[str, JType]): A dictionary mapping type names to their corresponding JType representations.
     """
 
+    file_path: str
+    package_name: str
     comments: List[JComment]
     imports: List[str]
     type_declarations: Dict[str, JType]
