Update chunk_python_code.py

chunk_python_code.py

@@ -1,8 +1,11 @@
 import ast
 from langchain.schema import Document 
 
+
+
 def chunk_python_code_with_metadata(python_code, file_path):
     """
+    Custom Python Code Splitter for KadiApy-Use Case
     Entry point method to process the Python file.
     It invokes the iterate_ast function.
     """
@@ -18,7 +21,7 @@ def chunk_python_code_with_metadata(python_code, file_path):
     else:
         usage = "undefined"
 
-    # Add metadata for usage to all documents
+    # Add metadata-type "usage" to all documents
     for doc in documents:
         doc.metadata["source"] = file_path
         doc.metadata["usage"] = usage  # Add the determined usage metadata
@@ -44,7 +47,7 @@ def _iterate_ast(python_code, documents, file_path):
     all_imports = all(isinstance(node, (ast.Import, ast.ImportFrom)) for node in first_level_nodes)
     if all_imports:
         documents.extend(
-            _chunk_import_only_python_code(documents, python_code, file_path))
+            _chunk_import_only_python_code(python_code, file_path))
     
     # Iterate over first-level nodes
     for first_level_node in ast.iter_child_nodes(tree):