codegraphcontext 0.2.11__py3-none-any.whl → 0.2.12__py3-none-any.whl

codegraphcontext/tools/handlers/management_handlers.py

@@ -276,59 +276,50 @@ def get_repository_stats(code_finder: CodeFinder, **args) -> Dict[str, Any]:
                         "error": f"Repository not found: {repo_path_obj}"
                     }
                 
-                # Get stats for specific repo
-                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=repo_path_obj)
-                record = result.single()
+                # 1. Files
+                file_query = "MATCH (r:Repository {path: $path})-[:CONTAINS*]->(f:File) RETURN count(f) as c"
+                file_count = session.run(file_query, path=repo_path_obj).single()["c"]
+                
+                # 2. Functions
+                func_query = "MATCH (r:Repository {path: $path})-[:CONTAINS*]->(func:Function) RETURN count(func) as c"
+                func_count = session.run(func_query, path=repo_path_obj).single()["c"]
+                
+                # 3. Classes
+                class_query = "MATCH (r:Repository {path: $path})-[:CONTAINS*]->(cls:Class) RETURN count(cls) as c"
+                class_count = session.run(class_query, path=repo_path_obj).single()["c"]
+                
+                # 4. Modules (imported)
+                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=repo_path_obj).single()["c"]
                 
                 return {
                     "success": True,
                     "repository": repo_path_obj,
                     "stats": {
-                        "files": record["file_count"] if record else 0,
-                        "functions": record["function_count"] if record else 0,
-                        "classes": record["class_count"] if record else 0,
-                        "modules": record["module_count"] if record else 0
+                        "files": file_count,
+                        "functions": func_count,
+                        "classes": class_count,
+                        "modules": module_count
                     }
                 }
             else:
                 # Overall database stats
-                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()
+                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 (func:Function) RETURN count(func) as c").single()["c"]
+                    class_count = session.run("MATCH (cls:Class) RETURN count(cls) as c").single()["c"]
+                    module_count = session.run("MATCH (m:Module) RETURN count(m) as c").single()["c"]
+                    
                     return {
                         "success": True,
                         "stats": {
-                            "repositories": record["repo_count"],
-                            "files": record["file_count"],
-                            "functions": record["function_count"],
-                            "classes": record["class_count"],
-                            "modules": record["module_count"]
+                            "repositories": repo_count,
+                            "files": file_count,
+                            "functions": func_count,
+                            "classes": class_count,
+                            "modules": module_count
                         }
                     }
                 else:

{codegraphcontext-0.2.11.dist-info → codegraphcontext-0.2.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.2.11
+Version: 0.2.12
 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
@@ -147,7 +147,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.2.11
+- **Version:** 0.2.12
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)

{codegraphcontext-0.2.11.dist-info → codegraphcontext-0.2.12.dist-info}/RECORD

@@ -31,7 +31,7 @@ codegraphcontext/tools/scip_pb2.py,sha256=dwOMNKlu6VyLq5h8kTPZRDqxrwfVL8yw7I7zio
 codegraphcontext/tools/system.py,sha256=DGeavZoPxzV78wwApV4f7fdBFQRa8oeOSv5wprHSjRE,5733
 codegraphcontext/tools/handlers/analysis_handlers.py,sha256=lYECkMC2NKvnBWI4nQsAeCL5-e_JfJSVvQAwV7IbNJk,4749
 codegraphcontext/tools/handlers/indexing_handlers.py,sha256=B1Op0wZ8MZ2pt170irc558T7rBYyuBTxWJsgRaiQRtw,5217
-codegraphcontext/tools/handlers/management_handlers.py,sha256=EL6lTFcIGmHmlzVpIsKpvzRaJMmB5KiH2UWlpRqIi5Y,13565
+codegraphcontext/tools/handlers/management_handlers.py,sha256=jCosISjq4QD7oBa8ZaaX9xVzHO6W-6IC4nitlG9pZdk,13537
 codegraphcontext/tools/handlers/query_handlers.py,sha256=KdZ1yH4yFj3RrXT4BTnyO7OZe6OKuNLW7tCAEPmjOy4,7629
 codegraphcontext/tools/handlers/watcher_handlers.py,sha256=9hBkMdI8mwQyXj77WvgyUWCtNeItV3DiCLnmNI7Xjp8,3623
 codegraphcontext/tools/languages/c.py,sha256=mrXQWEvMtzLVElATYNZ7VThNk1ZNZdw2UV8i6r2y2y4,22291
@@ -71,9 +71,9 @@ codegraphcontext/tools/query_tool_languages/typescript_toolkit.py,sha256=3S4hpmO
 codegraphcontext/utils/debug_log.py,sha256=Qg7jwyeg7x2h3Ur_2S34bdMCkHdlk_ngHfPwa97A9vE,2836
 codegraphcontext/utils/tree_sitter_manager.py,sha256=bIuKYN1aj1Zi6BnksGjZSLZzgBwuFRselZzyUpbToAU,9180
 codegraphcontext/utils/visualize_graph.py,sha256=Ntq8l8SvAOvsOp19QKByjEwU7-5rPa_XfcGLOrUehq4,5003
-codegraphcontext-0.2.11.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
-codegraphcontext-0.2.11.dist-info/METADATA,sha256=DSR3Rl__danmwQ3QC1rfTf09radrvOaons1hhTCv3QY,21095
-codegraphcontext-0.2.11.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-codegraphcontext-0.2.11.dist-info/entry_points.txt,sha256=LCxWCWMshdvYGoHBPuQZ8C-e4CiNSHCLXofrNSGHkoE,103
-codegraphcontext-0.2.11.dist-info/top_level.txt,sha256=CBgc6LAPZIO5FS0nSYYkylDifHsZTIqw3Gf5UwDxeGI,17
-codegraphcontext-0.2.11.dist-info/RECORD,,
+codegraphcontext-0.2.12.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
+codegraphcontext-0.2.12.dist-info/METADATA,sha256=EgsTEDqJ5yLjKXbLI4GQ2gTgI3xOEP2s3Qi5TGbeIVg,21095
+codegraphcontext-0.2.12.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+codegraphcontext-0.2.12.dist-info/entry_points.txt,sha256=LCxWCWMshdvYGoHBPuQZ8C-e4CiNSHCLXofrNSGHkoE,103
+codegraphcontext-0.2.12.dist-info/top_level.txt,sha256=CBgc6LAPZIO5FS0nSYYkylDifHsZTIqw3Gf5UwDxeGI,17
+codegraphcontext-0.2.12.dist-info/RECORD,,