codegraphcontext 0.1.37__tar.gz → 0.1.39__tar.gz

{codegraphcontext-0.1.37/src/codegraphcontext.egg-info → codegraphcontext-0.1.39}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.1.37
+Version: 0.1.39
 Summary: An MCP server that indexes local code into a graph database to provide context to AI assistants.
 Author-email: Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 License: MIT License
@@ -76,8 +76,8 @@ Dynamic: license-file
     <img src="https://img.shields.io/github/license/CodeGraphContext/CodeGraphContext?style=flat-square" alt="License">
   </a>
   <img src="https://img.shields.io/badge/MCP-Compatible-green?style=flat-square" alt="MCP Compatible">
-  <a href="https://discord.gg/dR4QY32uYQ">
-    <img src="https://img.shields.io/discord/dR4QY32uYQ?style=flat-square&logo=discord" alt="Discord">
+  <a href="https://discord.gg/VCwUdCnn">
+    <img src="https://img.shields.io/discord/1421769154507309150?label=Discord&logo=discord&logoColor=white&style=flat-square">
   </a>
   <br><br>
   <a href="https://github.com/CodeGraphContext/CodeGraphContext/stargazers">
@@ -139,7 +139,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.1.37
+- **Version:** 0.1.39
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)
@@ -419,25 +419,6 @@ Add the following server configuration to your client's settings file (e.g., VS
         "NEO4J_USERNAME": "YOUR_NEO4J_USERNAME",
         "NEO4J_PASSWORD": "YOUR_NEO4J_PASSWORD"
       },
-      "tools": {
-        "alwaysAllow": [
-          "add_code_to_graph",
-          "add_package_to_graph",
-          "check_job_status",
-          "list_jobs",
-          "find_code",
-          "analyze_code_relationships",
-          "watch_directory",
-          "find_dead_code",
-          "execute_cypher_query",
-          "calculate_cyclomatic_complexity",
-          "find_most_complex_functions",
-          "list_indexed_repositories",
-          "delete_repository",
-          "visualize_graph_query",
-          "list_watched_paths",
-          "unwatch_directory"
-        ],
         "disabled": false
       },
       "disabled": false,

{codegraphcontext-0.1.37 → codegraphcontext-0.1.39}/README.md

@@ -14,8 +14,8 @@
     <img src="https://img.shields.io/github/license/CodeGraphContext/CodeGraphContext?style=flat-square" alt="License">
   </a>
   <img src="https://img.shields.io/badge/MCP-Compatible-green?style=flat-square" alt="MCP Compatible">
-  <a href="https://discord.gg/dR4QY32uYQ">
-    <img src="https://img.shields.io/discord/dR4QY32uYQ?style=flat-square&logo=discord" alt="Discord">
+  <a href="https://discord.gg/VCwUdCnn">
+    <img src="https://img.shields.io/discord/1421769154507309150?label=Discord&logo=discord&logoColor=white&style=flat-square">
   </a>
   <br><br>
   <a href="https://github.com/CodeGraphContext/CodeGraphContext/stargazers">
@@ -77,7 +77,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.1.37
+- **Version:** 0.1.39
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)
@@ -357,25 +357,6 @@ Add the following server configuration to your client's settings file (e.g., VS
         "NEO4J_USERNAME": "YOUR_NEO4J_USERNAME",
         "NEO4J_PASSWORD": "YOUR_NEO4J_PASSWORD"
       },
-      "tools": {
-        "alwaysAllow": [
-          "add_code_to_graph",
-          "add_package_to_graph",
-          "check_job_status",
-          "list_jobs",
-          "find_code",
-          "analyze_code_relationships",
-          "watch_directory",
-          "find_dead_code",
-          "execute_cypher_query",
-          "calculate_cyclomatic_complexity",
-          "find_most_complex_functions",
-          "list_indexed_repositories",
-          "delete_repository",
-          "visualize_graph_query",
-          "list_watched_paths",
-          "unwatch_directory"
-        ],
         "disabled": false
       },
       "disabled": false,

{codegraphcontext-0.1.37 → codegraphcontext-0.1.39}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codegraphcontext"
-version = "0.1.37"
+version = "0.1.39"
 description = "An MCP server that indexes local code into a graph database to provide context to AI assistants."
 authors = [{ name = "Shashank Shekhar Singh", email = "shashankshekharsingh1205@gmail.com" }]
 readme = "README.md"
@@ -57,3 +57,13 @@ package-dir = { "" = "src" }
 [tool.setuptools.packages.find]
 where = ["src"]
 include = ["codegraphcontext*"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+python_files = ["test_*.py"]
+norecursedirs = ["tests/fixtures", "tests/e2e/sample_projects", "venv", ".git", "__pycache__"]
+markers = [
+    "integration: mark a test as an integration test.",
+    "e2e: mark a test as an end-to-end test.",
+    "slow: mark test as slow."
+]

{codegraphcontext-0.1.37 → codegraphcontext-0.1.39}/src/codegraphcontext/cli/cli_helpers.py

@@ -542,29 +542,32 @@ def stats_helper(path: str = None):
                     return
                 
                 # Get stats
-                stats_query = """
-                MATCH (r:Repository {path: $path})-[:CONTAINS]->(f:File)
-                WITH r, count(f) as file_count, f
-                OPTIONAL MATCH (f)-[:CONTAINS]->(func:Function)
-                OPTIONAL MATCH (f)-[:CONTAINS]->(cls:Class)
-                OPTIONAL MATCH (f)-[:IMPORTS]->(m:Module)
-                RETURN 
-                    file_count,
-                    count(DISTINCT func) as function_count,
-                    count(DISTINCT cls) as class_count,
-                    count(DISTINCT m) as module_count
-                """
-                result = session.run(stats_query, path=str(path_obj))
-                record = result.single()
+                # Get stats using separate queries to handle depth and avoid Cartesian products
+                # 1. Files
+                file_query = "MATCH (r:Repository {path: $path})-[:CONTAINS*]->(f:File) RETURN count(f) as c"
+                file_count = session.run(file_query, path=str(path_obj)).single()["c"]
+                
+                # 2. Functions (including methods in classes)
+                func_query = "MATCH (r:Repository {path: $path})-[:CONTAINS*]->(func:Function) RETURN count(func) as c"
+                func_count = session.run(func_query, path=str(path_obj)).single()["c"]
+                
+                # 3. Classes
+                class_query = "MATCH (r:Repository {path: $path})-[:CONTAINS*]->(c:Class) RETURN count(c) as c"
+                class_count = session.run(class_query, path=str(path_obj)).single()["c"]
                 
+                # 4. Modules (imported) - Note: Module nodes are outside the repo structure usually, connected via IMPORTS
+                # We need to traverse from files to modules
+                module_query = "MATCH (r:Repository {path: $path})-[:CONTAINS*]->(f:File)-[:IMPORTS]->(m:Module) RETURN count(DISTINCT m) as c"
+                module_count = session.run(module_query, path=str(path_obj)).single()["c"]
+
                 table = Table(show_header=True, header_style="bold magenta")
                 table.add_column("Metric", style="cyan")
                 table.add_column("Count", style="green", justify="right")
                 
-                table.add_row("Files", str(record["file_count"] if record else 0))
-                table.add_row("Functions", str(record["function_count"] if record else 0))
-                table.add_row("Classes", str(record["class_count"] if record else 0))
-                table.add_row("Imported Modules", str(record["module_count"] if record else 0))
+                table.add_row("Files", str(file_count))
+                table.add_row("Functions", str(func_count))
+                table.add_row("Classes", str(class_count))
+                table.add_row("Imported Modules", str(module_count))
                 
                 console.print(table)
         else:
@@ -572,34 +575,24 @@ def stats_helper(path: str = None):
             console.print("[cyan]📊 Overall Database Statistics[/cyan]\n")
             
             with db_manager.get_driver().session() as session:
-                # Get overall counts using separate queries to avoid empty results
-                # when some node types don't exist
-                stats_query = """
-                MATCH (r:Repository)
-                OPTIONAL MATCH (f:File)
-                OPTIONAL MATCH (func:Function)
-                OPTIONAL MATCH (cls:Class)
-                OPTIONAL MATCH (m:Module)
-                RETURN 
-                    count(DISTINCT r) as repo_count,
-                    count(DISTINCT f) as file_count,
-                    count(DISTINCT func) as function_count,
-                    count(DISTINCT cls) as class_count,
-                    count(DISTINCT m) as module_count
-                """
-                result = session.run(stats_query)
-                record = result.single()
+                # Get overall counts using separate O(1) queries
+                repo_count = session.run("MATCH (r:Repository) RETURN count(r) as c").single()["c"]
                 
-                if record and record["repo_count"] > 0:
+                if repo_count > 0:
+                    file_count = session.run("MATCH (f:File) RETURN count(f) as c").single()["c"]
+                    func_count = session.run("MATCH (f:Function) RETURN count(f) as c").single()["c"]
+                    class_count = session.run("MATCH (c:Class) RETURN count(c) as c").single()["c"]
+                    module_count = session.run("MATCH (m:Module) RETURN count(m) as c").single()["c"]
+                    
                     table = Table(show_header=True, header_style="bold magenta")
                     table.add_column("Metric", style="cyan")
                     table.add_column("Count", style="green", justify="right")
                     
-                    table.add_row("Repositories", str(record["repo_count"]))
-                    table.add_row("Files", str(record["file_count"]))
-                    table.add_row("Functions", str(record["function_count"]))
-                    table.add_row("Classes", str(record["class_count"]))
-                    table.add_row("Modules", str(record["module_count"]))
+                    table.add_row("Repositories", str(repo_count))
+                    table.add_row("Files", str(file_count))
+                    table.add_row("Functions", str(func_count))
+                    table.add_row("Classes", str(class_count))
+                    table.add_row("Modules", str(module_count))
                     
                     console.print(table)
                 else:

{codegraphcontext-0.1.37 → codegraphcontext-0.1.39}/src/codegraphcontext/cli/config_manager.py

@@ -38,6 +38,7 @@ DEFAULT_CONFIG = {
     "PARALLEL_WORKERS": "4",
     "CACHE_ENABLED": "true",
     "IGNORE_DIRS": "node_modules,venv,.venv,env,.env,dist,build,target,out,.git,.idea,.vscode,__pycache__",
+    "INDEX_SOURCE": "false",
 }
 
 # Configuration key descriptions
@@ -60,6 +61,7 @@ CONFIG_DESCRIPTIONS = {
     "PARALLEL_WORKERS": "Number of parallel indexing workers",
     "CACHE_ENABLED": "Enable caching for faster re-indexing",
     "IGNORE_DIRS": "Comma-separated list of directory names to ignore during indexing",
+    "INDEX_SOURCE": "Store full source code in graph database (recommended false)",
 }
 
 # Valid values for each config key
@@ -73,6 +75,7 @@ CONFIG_VALIDATORS = {
     "IGNORE_HIDDEN_FILES": ["true", "false"],
     "ENABLE_AUTO_WATCH": ["true", "false"],
     "CACHE_ENABLED": ["true", "false"],
+    "INDEX_SOURCE": ["true", "false"],
 }
 
 

{codegraphcontext-0.1.37 → codegraphcontext-0.1.39}/src/codegraphcontext/tools/graph_builder.py

@@ -15,6 +15,7 @@ from tree_sitter import Language, Parser
 from ..utils.tree_sitter_manager import get_tree_sitter_manager
 from ..cli.config_manager import get_config_value
 
+
 class TreeSitterParser:
     """A generic parser wrapper for a specific language using tree-sitter."""
 
@@ -151,7 +152,7 @@ class GraphBuilder:
                 session.run("""
                     CREATE FULLTEXT INDEX code_search_index IF NOT EXISTS 
                     FOR (n:Function|Class|Variable) 
-                    ON EACH [n.name, n.source, n.docstring]
+                    ON EACH [n.name, coalesce(n.source, ''), coalesce(n.docstring, '')]
                 """ )
                 
                 info_logger("Database schema verified/created successfully")
@@ -333,6 +334,7 @@ class GraphBuilder:
                         SET n += $props
                         MERGE (f)-[:CONTAINS]->(n)
                     """
+
                     session.run(query, file_path=file_path_str, name=item['name'], line_number=item['line_number'], props=item)
                     
                     if label == 'Function':
@@ -802,15 +804,23 @@ class GraphBuilder:
 
         debug_log(f"[parse_file] Starting parsing for: {file_path} with {parser.language_name} parser")
         try:
+            index_source = (get_config_value("INDEX_SOURCE") or "false").lower() == "true"
             if parser.language_name == 'python':
                 is_notebook = file_path.suffix == '.ipynb'
-                file_data = parser.parse(file_path, is_dependency, is_notebook=is_notebook)
+                file_data = parser.parse(
+                    file_path,
+                    is_dependency,
+                    is_notebook=is_notebook,
+                    index_source=index_source
+                )
             else:
-                file_data = parser.parse(file_path, is_dependency)
+                file_data = parser.parse(
+                    file_path,
+                    is_dependency,
+                    index_source=index_source
+                )
             file_data['repo_path'] = str(repo_path)
-            debug_log(f"[parse_file] Successfully parsed: {file_path}")
             return file_data
-            
         except Exception as e:
             error_logger(f"Error parsing {file_path} with {parser.language_name} parser: {e}")
             debug_log(f"[parse_file] Error parsing {file_path}: {e}")

{codegraphcontext-0.1.37 → codegraphcontext-0.1.39}/src/codegraphcontext/tools/languages/c.py

@@ -96,8 +96,9 @@ class CTreeSitterParser:
     def _get_node_text(self, node: Any) -> str:
         return node.text.decode("utf-8")
 
-    def parse(self, file_path: Path, is_dependency: bool = False) -> Dict[str, Any]:
+    def parse(self, file_path: Path, is_dependency: bool = False, index_source: bool = False) -> Dict[str, Any]:
         """Parses a C file and returns its structure."""
+        self.index_source = index_source
         with open(file_path, "r", encoding="utf-8", errors="ignore") as f:
             source_code = f.read()
 
@@ -238,13 +239,11 @@ class CTreeSitterParser:
                 args = self._parse_function_args(params_node) if params_node else []
                 context, context_type, _ = self._get_parent_context(func_node)
 
-                functions.append({
+                func_data = {
                     "name": name,
                     "line_number": node.start_point[0] + 1,
                     "end_line": func_node.end_point[0] + 1,
                     "args": [arg["name"] for arg in args if arg["name"]],  # Simplified args for compatibility
-                    "source": self._get_node_text(func_node),
-
                     "docstring": self._get_docstring(func_node),
                     "cyclomatic_complexity": self._calculate_complexity(func_node),
                     "context": context,
@@ -254,7 +253,12 @@ class CTreeSitterParser:
                     "lang": self.language_name,
                     "is_dependency": False,
                     "detailed_args": args,  # Keep detailed args for future use
-                })
+                }
+
+                if self.index_source:
+                    func_data["source"] = self._get_node_text(func_node)
+                
+                functions.append(func_data)
         return functions
 
     def _find_structs_unions_enums(self, root_node: Any) -> list[Dict[str, Any]]:
@@ -271,19 +275,23 @@ class CTreeSitterParser:
                 name = self._get_node_text(node)
                 context, context_type, _ = self._get_parent_context(struct_node)
                 
-                classes.append({
+                struct_data = {
                     "name": name,
                     "line_number": node.start_point[0] + 1,
                     "end_line": struct_node.end_point[0] + 1,
                     "bases": [],  # C doesn't have inheritance
-                    "source": self._get_node_text(struct_node),
                     "docstring": self._get_docstring(struct_node),
                     "context": context,
                     "decorators": [],
                     "lang": self.language_name,
                     "is_dependency": False,
                     "type": "struct",
-                })
+                }
+
+                if self.index_source:
+                    struct_data["source"] = self._get_node_text(struct_node)
+                
+                classes.append(struct_data)
 
         # Find unions
         query_str = C_QUERIES["unions"]
@@ -295,19 +303,23 @@ class CTreeSitterParser:
                 name = self._get_node_text(node)
                 context, context_type, _ = self._get_parent_context(union_node)
                 
-                classes.append({
+                union_data = {
                     "name": name,
                     "line_number": node.start_point[0] + 1,
                     "end_line": union_node.end_point[0] + 1,
                     "bases": [],
-                    "source": self._get_node_text(union_node),
                     "docstring": self._get_docstring(union_node),
                     "context": context,
                     "decorators": [],
                     "lang": self.language_name,
                     "is_dependency": False,
                     "type": "union",
-                })
+                }
+
+                if self.index_source:
+                    union_data["source"] = self._get_node_text(union_node)
+                
+                classes.append(union_data)
 
         # Find enums
         query_str = C_QUERIES["enums"]
@@ -319,19 +331,23 @@ class CTreeSitterParser:
                 name = self._get_node_text(node)
                 context, context_type, _ = self._get_parent_context(enum_node)
                 
-                classes.append({
+                enum_data = {
                     "name": name,
                     "line_number": node.start_point[0] + 1,
                     "end_line": enum_node.end_point[0] + 1,
                     "bases": [],
-                    "source": self._get_node_text(enum_node),
                     "docstring": self._get_docstring(enum_node),
                     "context": context,
                     "decorators": [],
                     "lang": self.language_name,
                     "is_dependency": False,
                     "type": "enum",
-                })
+                }
+
+                if self.index_source:
+                    enum_data["source"] = self._get_node_text(enum_node)
+                
+                classes.append(enum_data)
 
         return classes
 
@@ -474,17 +490,21 @@ class CTreeSitterParser:
                 
                 context, context_type, _ = self._get_parent_context(macro_node)
                 
-                macros.append({
+                macro_data = {
                     "name": name,
                     "line_number": node.start_point[0] + 1,
                     "end_line": macro_node.end_point[0] + 1,
-                    "source": self._get_node_text(macro_node),
                     "value": value,
                     "params": params,
                     "context": context,
                     "lang": self.language_name,
                     "is_dependency": False,
-                })
+                }
+
+                if self.index_source:
+                    macro_data["source"] = self._get_node_text(macro_node)
+                
+                macros.append(macro_data)
         return macros
 
 

{codegraphcontext-0.1.37 → codegraphcontext-0.1.39}/src/codegraphcontext/tools/languages/cpp.py

@@ -116,8 +116,9 @@ class CppTreeSitterParser:
     def _get_node_text(self, node) -> str:
         return node.text.decode('utf-8')
 
-    def parse(self, file_path: Path, is_dependency: bool = False, **kwargs) -> Dict:
+    def parse(self, file_path: Path, is_dependency: bool = False, index_source: bool = False, **kwargs) -> Dict:
         """Parses a C++ file and returns its structure."""
+        self.index_source = index_source
         with open(file_path, "r", encoding="utf-8", errors="ignore") as f:
             source_code = f.read()
 
@@ -177,13 +178,17 @@ class CppTreeSitterParser:
                 
                 params = self._extract_function_params(func_node)
                 
-                functions.append({
+                func_data = {
                     "name": name,
                     "line_number": node.start_point[0] + 1,
                     "end_line": func_node.end_point[0] + 1,
-                    "source": self._get_node_text(func_node),
                     "args": params,
-                })
+                }
+                
+                if self.index_source:
+                    func_data["source"] = self._get_node_text(func_node)
+                
+                functions.append(func_data)
         return functions
 
     def _extract_function_params(self, func_node) -> list[str]:
@@ -233,13 +238,15 @@ class CppTreeSitterParser:
             if capture_name == 'name':
                 class_node = node.parent
                 name = self._get_node_text(node)
-                classes.append({
+                class_data = {
                     "name": name,
                     "line_number": node.start_point[0] + 1,
                     "end_line": class_node.end_point[0] + 1,
-                    "source": self._get_node_text(class_node),
                     "bases": [], # Placeholder
-                })
+                }
+                if self.index_source:
+                    class_data["source"] = self._get_node_text(class_node)
+                classes.append(class_data)
         return classes
 
     def _find_imports(self, root_node):
@@ -263,14 +270,15 @@ class CppTreeSitterParser:
         query_str = CPP_QUERIES['enums']
         for node, capture_name in execute_query(self.language, query_str, root_node):
             if capture_name == 'name':
-                name = self._get_node_text(node)
                 enum_node = node.parent
-                enums.append({
+                enum_data = {
                     "name": name,
                     "line_number": node.start_point[0] + 1,
                     "end_line": enum_node.end_point[0] + 1,
-                    "source": self._get_node_text(enum_node),
-                })
+                }
+                if self.index_source:
+                    enum_data["source"] = self._get_node_text(enum_node)
+                enums.append(enum_data)
         return enums
  
     def _find_structs(self, root_node):
@@ -280,12 +288,14 @@ class CppTreeSitterParser:
             if capture_name == 'name':
                 name = self._get_node_text(node)
                 struct_node = node.parent
-                structs.append({
+                struct_data = {
                     "name": name,
                     "line_number": node.start_point[0] + 1,
                     "end_line": struct_node.end_point[0] + 1,
-                    "source": self._get_node_text(struct_node),
-                })
+                }
+                if self.index_source:
+                    struct_data["source"] = self._get_node_text(struct_node)
+                structs.append(struct_data)
         return structs
 
     def _find_unions(self, root_node):
@@ -295,12 +305,14 @@ class CppTreeSitterParser:
             if capture_name == 'name':
                 name = self._get_node_text(node)
                 union_node = node.parent
-                unions.append({
+                union_data = {
                     "name": name,
                     "line_number": node.start_point[0] + 1,
                     "end_line": union_node.end_point[0] + 1,
-                    "source": self._get_node_text(union_node),
-                })
+                }
+                if self.index_source:
+                    union_data["source"] = self._get_node_text(union_node)
+                unions.append(union_data)
         return unions
 
     def _find_macros(self, root_node):
@@ -312,12 +324,14 @@ class CppTreeSitterParser:
             if capture_name == 'name':
                 macro_node = node.parent
                 name = self._get_node_text(node)
-                macros.append({
+                macro_data = {
                     "name": name,
                     "line_number": node.start_point[0] + 1,
                     "end_line": macro_node.end_point[0] + 1,
-                    "source": self._get_node_text(macro_node),
-                })
+                }
+                if self.index_source:
+                    macro_data["source"] = self._get_node_text(macro_node)
+                macros.append(macro_data)
         return macros
     
     def _find_lambda_assignments(self, root_node):
@@ -349,8 +363,7 @@ class CppTreeSitterParser:
                     "line_number": node.start_point[0] + 1,
                     "end_line": assignment_node.end_point[0] + 1,
                     "args": [p for p in [self._get_node_text(p) for p in params_node.children if p.type == 'identifier'] if p] if params_node else [],
-                    "source": self._get_node_text(assignment_node),
-
+                    
                     "docstring": None,
                     "cyclomatic_complexity": 1,
                     "context": context,
@@ -360,6 +373,10 @@ class CppTreeSitterParser:
                     "lang": self.language_name,
                     "is_dependency": False,
                 }
+
+                if self.index_source:
+                    func_data["source"] = self._get_node_text(assignment_node)
+
                 functions.append(func_data)
         return functions
     

{codegraphcontext-0.1.37 → codegraphcontext-0.1.39}/src/codegraphcontext/tools/languages/csharp.py

@@ -84,8 +84,9 @@ class CSharpTreeSitterParser:
         self.language = generic_parser_wrapper.language
         self.parser = generic_parser_wrapper.parser
 
-    def parse(self, file_path: Path, is_dependency: bool = False) -> Dict[str, Any]:
+    def parse(self, file_path: Path, is_dependency: bool = False, index_source: bool = False) -> Dict[str, Any]:
         try:
+            self.index_source = index_source
             with open(file_path, 'r', encoding='utf-8', errors='ignore') as f:
                 source_code = f.read()
 
@@ -221,7 +222,6 @@ class CSharpTreeSitterParser:
                             "attributes": attributes,
                             "line_number": start_line,
                             "end_line": end_line,
-                            "source": source_text,
                             "file_path": str(file_path),
                             "lang": self.language_name,
                         }
@@ -229,6 +229,9 @@ class CSharpTreeSitterParser:
                         # Add class context if found
                         if class_context:
                             func_data["class_context"] = class_context
+
+                        if self.index_source:
+                            func_data["source"] = source_text
                         
                         functions.append(func_data)
                         
@@ -288,7 +291,6 @@ class CSharpTreeSitterParser:
                             "name": type_name,
                             "line_number": start_line,
                             "end_line": end_line,
-                            "source": source_text,
                             "file_path": str(file_path),
                             "lang": self.language_name,
                         }
@@ -296,6 +298,9 @@ class CSharpTreeSitterParser:
                         # Add bases if found
                         if bases:
                             type_data["bases"] = bases
+
+                        if self.index_source:
+                            type_data["source"] = source_text
                         
                         types.append(type_data)
                         
@@ -467,13 +472,15 @@ class CSharpTreeSitterParser:
                             "type": prop_type,
                             "line_number": start_line,
                             "end_line": end_line,
-                            "source": source_text,
                             "file_path": str(file_path),
                             "lang": self.language_name,
                         }
                         
                         if class_context:
                             prop_data["class_context"] = class_context
+                            
+                        if self.index_source:
+                            prop_data["source"] = source_text
                         
                         properties.append(prop_data)