codebase-brain 0.2.0__tar.gz → 0.3.0__tar.gz

{codebase_brain-0.2.0 → codebase_brain-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codebase-brain
-Version: 0.2.0
+Version: 0.3.0
 Summary: An AI brain that lives inside your codebase
 Requires-Python: >=3.8
 Requires-Dist: watchdog

{codebase_brain-0.2.0 → codebase_brain-0.3.0}/brain_cli.py

@@ -65,6 +65,75 @@ def cmd_start(path):
         print(f"\nBrain: {answer}\n")
 
 
+def _explain_risk(staged_files, G, risk, brain):
+    from brain_parser.llm_router import call_llm
+
+    context = ""
+    for f in staged_files:
+        target_norm = os.path.abspath(f).replace('\\', '/')
+        matched = None
+        for node in G.nodes():
+            node_norm = os.path.abspath(node).replace('\\', '/')
+            if node_norm == target_norm:
+                matched = node
+                break
+        if not matched:
+            continue
+
+        file_risk = risk.get(matched, 'LOW')
+        if file_risk != 'HIGH':
+            continue
+
+        direct = list(G.predecessors(matched))
+        file_data = brain.get(matched, {})
+        functions = [fn['name'] for fn in file_data.get('functions', [])]
+
+        context += f"\nFile: {f}\n"
+        context += f"Risk: {file_risk}\n"
+        context += f"Functions in this file: {functions}\n"
+        context += f"Files that depend on it: {direct}\n"
+
+    prompt = f"""A developer is about to commit changes to a HIGH risk file.
+Given this context:
+{context}
+
+In 2-3 sentences, explain in plain English:
+1. Why this file is risky to change
+2. What could realistically break if the change is wrong
+3. What the developer should double check before committing
+
+Be direct and specific. No generic advice."""
+
+    try:
+        return call_llm(prompt, max_tokens=200)
+    except Exception as e:
+        return f"Could not generate explanation: {str(e)}"
+
+
+def _print_impact(G, risk, filepath, clean):
+    from brain_parser.graph_builder import get_impact
+
+    result = get_impact(G, filepath)
+
+    if not result:
+        print(f"File not found in brain: {filepath}")
+        return None
+
+    file_risk = risk.get(result['target'], 'LOW')
+
+    print(f"\n Change Impact: {filepath}")
+    print(f"   Risk Level: {file_risk}\n")
+    print(f"DIRECT IMPACT ({len(result['direct'])} files):")
+    for f in result['direct']:
+        print(f"  → {clean(f)}")
+    print(f"\nINDIRECT IMPACT ({len(result['indirect'])} files):")
+    for f in result['indirect']:
+        print(f"  → {clean(f)}")
+    print(f"\nTotal files affected: {result['total_affected']}\n")
+
+    return {'risk': file_risk, 'total': result['total_affected']}
+
+
 def cmd_impact(filepath=None, staged=False, block=False):
     from brain_parser.codebase_walker import load_brain
     from brain_parser.graph_builder import build_graph, get_impact, calculate_risk
@@ -106,10 +175,13 @@ def cmd_impact(filepath=None, staged=False, block=False):
             print("BRAIN WARNING: HIGH RISK commit detected.")
             print("This change affects critical files.")
             print("Production incidents have originated from files like these.")
+
+            print("\nAsking Brain to explain the risk...\n")
+            explanation = _explain_risk(staged_files, G, risk, brain)
+            print(explanation)
+
             print("\nType 'yes' to commit anyway, anything else to abort: ", end='', flush=True)
             try:
-                # Windows: CON is the console device
-                # Unix: /dev/tty is the terminal
                 import platform
                 if platform.system() == 'Windows':
                     with open('CON', 'r') as tty:
@@ -134,30 +206,6 @@ def cmd_impact(filepath=None, staged=False, block=False):
         return
 
     _print_impact(G, risk, filepath, clean)
-
-def _print_impact(G, risk, filepath, clean):
-    from brain_parser.graph_builder import get_impact
-
-    result = get_impact(G, filepath)
-
-    if not result:
-        print(f"File not found in brain: {filepath}")
-        return None
-
-    file_risk = risk.get(result['target'], 'LOW')
-
-    print(f"\n Change Impact: {filepath}")
-    print(f"   Risk Level: {file_risk}\n")
-    print(f"DIRECT IMPACT ({len(result['direct'])} files):")
-    for f in result['direct']:
-        print(f"  → {clean(f)}")
-    print(f"\nINDIRECT IMPACT ({len(result['indirect'])} files):")
-    for f in result['indirect']:
-        print(f"  → {clean(f)}")
-    print(f"\nTotal files affected: {result['total_affected']}\n")
-
-    return {'risk': file_risk, 'total': result['total_affected']}
-
 def install_hook(repo_path="."):
     hook_dir = os.path.join(repo_path, ".git", "hooks")
     hook_path = os.path.join(hook_dir, "pre-commit")
@@ -200,7 +248,14 @@ def install_hook(repo_path="."):
 
 
 def cmd_init():
+    import os
     print("Initializing Codebase Brain...\n")
+
+    # global config — stored once per machine
+    config_dir = os.path.expanduser("~/.codebase-brain")
+    os.makedirs(config_dir, exist_ok=True)
+    config_path = os.path.join(config_dir, "config.env")
+
     print("Choose your LLM provider:")
     print("  1. Groq        — Free, fast. llama-3.3-70b (recommended to start)")
     print("  2. OpenAI      — Reliable, trusted by enterprises. gpt-4o-mini")
@@ -237,7 +292,7 @@ def cmd_init():
         },
         "5": {
             "name": "ollama",
-            "key_prompt": None,  # no key needed
+            "key_prompt": None,
             "models": ["llama3.2", "codellama", "deepseek-coder"],
             "default_model": "llama3.2"
         }
@@ -248,14 +303,12 @@ def cmd_init():
 
     provider = providers[choice]
 
-    # get API key
     if provider["key_prompt"]:
         api_key = input(provider["key_prompt"]).strip()
     else:
         api_key = "ollama"
         print("Ollama selected — make sure Ollama is running at http://localhost:11434")
 
-    # pick model
     print(f"\nAvailable models for {provider['name']}:")
     for i, m in enumerate(provider["models"], 1):
         default_tag = " (default)" if m == provider["default_model"] else ""
@@ -271,16 +324,16 @@ def cmd_init():
     else:
         model = provider["default_model"]
 
-    with open(".env", "w") as f:
+    # save globally — one time setup
+    with open(config_path, "w") as f:
         f.write(f"LLM_PROVIDER={provider['name']}\n")
         f.write(f"LLM_API_KEY={api_key}\n")
         f.write(f"LLM_MODEL={model}\n")
-        # backwards compatibility
         f.write(f"GROQ_API_KEY={api_key}\n")
         f.write(f"GROQ_MODEL={model}\n")
 
-    print(f"\nBrain initialized with {provider['name']} — {model}")
-    print("Run 'brain start' to begin.\n")
+    print(f"\nBrain initialized globally — {provider['name']} / {model}")
+    print("Run 'brain start' in any project to begin.\n")
 
 def cmd_rootcause(error_text=None):
     from brain_parser.root_cause import find_root_cause
@@ -297,6 +350,41 @@ def cmd_rootcause(error_text=None):
         error_text = "\n".join(lines)
 
     print(find_root_cause(error_text))
+def cmd_uninstall():
+    import shutil
+
+    print("Uninstalling Codebase Brain...\n")
+
+    # remove global config
+    config_dir = os.path.expanduser("~/.codebase-brain")
+    if os.path.exists(config_dir):
+        shutil.rmtree(config_dir)
+        print("Removed global config.")
+
+    # remove local brain.json
+    brain_json = os.path.join(os.getcwd(), 'brain.json')
+    if os.path.exists(brain_json):
+        os.remove(brain_json)
+        print("Removed brain.json.")
+
+    # remove local brain_map.html
+    brain_map = os.path.join(os.getcwd(), 'brain_map.html')
+    if os.path.exists(brain_map):
+        os.remove(brain_map)
+        print("Removed brain_map.html.")
+
+    # remove git hook
+    hook_path = os.path.join(os.getcwd(), '.git', 'hooks', 'pre-commit')
+    if os.path.exists(hook_path):
+        # only remove if it's a Brain hook
+        with open(hook_path, 'r') as f:
+            content = f.read()
+        if 'Codebase Brain' in content:
+            os.remove(hook_path)
+            print("Removed git hook.")
+
+    print("\nBrain removed from this project.")
+    print("To fully uninstall: pip uninstall codebase-brain")
 
 
 def main():
@@ -307,9 +395,10 @@ def main():
     parser.add_argument("--path", default=".", help="Path to codebase")
     parser.add_argument("--file", default="", help="File to analyze impact")
     parser.add_argument("--staged", action="store_true", help="Analyze all staged files")
-    parser.add_argument("command", choices=["start", "init", "impact", "install-hook", "rootcause"])
+    parser.add_argument("command", choices=["start", "init", "impact", "install-hook", "rootcause", "uninstall"])
     parser.add_argument("--error", default="", help="Stacktrace to analyze")
     parser.add_argument("--block", action="store_true", help="Block HIGH risk commits")
+ 
     # ← new
     args = parser.parse_args()
 
@@ -324,8 +413,12 @@ def main():
         cmd_impact(filepath=args.file or None, staged=args.staged, block=args.block)
     elif args.command == "install-hook":
         install_hook()
+    elif args.command == "uninstall":
+        cmd_uninstall()
+
 
 
 
 if __name__ == "__main__":
-    main()
+    main()
+   

{codebase_brain-0.2.0 → codebase_brain-0.3.0}/brain_parser/graph_builder.py

@@ -106,7 +106,7 @@ def visualize_interactive(G):
     node_map = {}
     for node in G.nodes():
         node_map[node] = str(node)
-
+# demo test
     for node, node_id in node_map.items():
         risk_level = risk.get(node, 'LOW')
         color = color_map[risk_level]

{codebase_brain-0.2.0 → codebase_brain-0.3.0}/brain_parser/llm_router.py

@@ -1,13 +1,31 @@
 import os
-from dotenv import load_dotenv
 
-load_dotenv()
+def get_config():
+    """Load from local .env first, then global ~/.codebase-brain/config.env"""
+    from dotenv import load_dotenv
+
+    # try local .env first
+    local_env = os.path.join(os.getcwd(), '.env')
+    if os.path.exists(local_env):
+        load_dotenv(local_env)
+    else:
+        # fallback to global config
+        global_config = os.path.expanduser("~/.codebase-brain/config.env")
+        if os.path.exists(global_config):
+            load_dotenv(global_config)
+
+    return {
+        'provider': os.getenv("LLM_PROVIDER", "groq").lower(),
+        'api_key': os.getenv("LLM_API_KEY") or os.getenv("GROQ_API_KEY"),
+        'model': os.getenv("LLM_MODEL") or os.getenv("GROQ_MODEL", "llama-3.3-70b-versatile")
+    }
 
 
 def call_llm(prompt, max_tokens=500):
-    provider = os.getenv("LLM_PROVIDER", "groq").lower()
-    api_key = os.getenv("LLM_API_KEY") or os.getenv("GROQ_API_KEY")
-    model = os.getenv("LLM_MODEL") or os.getenv("GROQ_MODEL", "llama-3.3-70b-versatile")
+    config = get_config()
+    provider = config['provider']
+    api_key = config['api_key']
+    model = config['model']
 
     if provider == "groq":
         return _call_groq(api_key, model, prompt, max_tokens)

{codebase_brain-0.2.0 → codebase_brain-0.3.0}/codebase_brain.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codebase-brain
-Version: 0.2.0
+Version: 0.3.0
 Summary: An AI brain that lives inside your codebase
 Requires-Python: >=3.8
 Requires-Dist: watchdog

{codebase_brain-0.2.0 → codebase_brain-0.3.0}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="codebase-brain",
-    version="0.2.0",
+    version="0.3.0",
     description="An AI brain that lives inside your codebase",
     package_dir={"": "."},
     packages=find_packages(where="."),