idk

Author: NicolasIRAGNE

src/gitingest/output_formatter.py

@@ -167,7 +167,6 @@ def _format_token_count(text: str) -> str | None:
 
     return str(total_tokens)
 
-# Rename JinjaFormatter to DefaultFormatter throughout the file
 class DefaultFormatter:
     def __init__(self):
         self.env = Environment(loader=BaseLoader())
@@ -207,7 +206,8 @@ def _(self, node: FileSystemDirectory, query):
 
     @summary.register
     def _(self, node: FileSystemDirectory, query):
-        template = """
+        template = \
+"""
 Directory structure:
 {{ node.tree }}
 """

src/server/query_processor.py

@@ -86,11 +86,6 @@ async def process_query(
         context = ingest_query(query)
         digest = context.generate_digest()
 
-        summary, tree, content = context.generate_digest()
-
-        # Prepare the digest content (tree + content)
-        digest_content = tree + "\n" + content
-
         # Store digest based on S3 configuration
         if is_s3_enabled():
             # Upload to S3 instead of storing locally
@@ -102,14 +97,14 @@ async def process_query(
                 include_patterns=query.include_patterns,
                 ignore_patterns=query.ignore_patterns,
             )
-            s3_url = upload_to_s3(content=digest_content, s3_file_path=s3_file_path, ingest_id=query.id)
+            s3_url = upload_to_s3(content=context.digest, s3_file_path=s3_file_path, ingest_id=query.id)
             # Store S3 URL in query for later use
             query.s3_url = s3_url
         else:
             # Store locally
             local_txt_file = Path(clone_config.local_path).with_suffix(".txt")
             with local_txt_file.open("w", encoding="utf-8") as f:
-                f.write(digest_content)
+                f.write(digest)
 
     except Exception as exc:
         _print_error(query.url, exc, max_file_size, pattern_type, pattern)