refactor: Simplify and optimize weight calculation in RepoMap

Author: paul-gauthier

aider/repomap.py

@@ -105,6 +105,7 @@ def get_repo_map(
         mentioned_idents=None,
         force_refresh=False,
     ):
+
         if self.max_map_tokens <= 0:
             return
         if not other_files:
@@ -445,23 +446,28 @@ def get_ranked_tags(
                 progress()
 
             definers = defines[ident]
+
+            mul = 1.0
+            # if ident is snake_case or camelCase, mul * 10 ai!
             if ident in mentioned_idents:
-                mul = 10
-            elif ident.startswith("_"):
-                mul = 0.1
-            else:
-                mul = 1
+                mul *= 10
+            if ident.startswith("_"):
+                mul *= 0.1
 
             for referencer, num_refs in Counter(references[ident]).items():
                 for definer in definers:
                     # dump(referencer, definer, num_refs, mul)
                     # if referencer == definer:
                     #    continue
 
+                    use_mul = mul
+                    if referencer in chat_rel_fnames:
+                        use_mul *= 50
+
                     # scale down so high freq (low value) mentions don't dominate
                     num_refs = math.sqrt(num_refs)
 
-                    G.add_edge(referencer, definer, weight=mul * num_refs, ident=ident)
+                    G.add_edge(referencer, definer, weight=use_mul * num_refs, ident=ident)
 
         if not references:
             pass