draft: Add an option to limit maximum commits used for frecency

Refs #711
kantord · Nov 18, 2024 · b614d29 · b614d29
1 parent 587938a
commit b614d29
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/docs/configuration.md b/docs/configuration.md
@@ -25,6 +25,9 @@ This is an example of a configuration file:
 
 server:
   port: 31134  # A port number to run the server on
+  # Increase number of commits used for computing frecency score
+  # Default is `1000`, set to `null` to read all history
+  readMaxCommits: 5000
 
   # globs to ignore in addition to .gitignore
   ignorePatterns:

diff --git a/seagoat/repository.py b/seagoat/repository.py
@@ -92,6 +92,9 @@ def analyze_files(self):
             "--no-merges",
         ]
 
+        if (max_commits := self.config["server"]["readMaxCommits"]) is not None:
+            cmd.append(f"--max-count={max_commits}")
+
         self.file_changes.clear()
 
         files = set(

diff --git a/seagoat/utils/config.py b/seagoat/utils/config.py
@@ -13,6 +13,7 @@
     "server": {
         "port": None,
         "ignorePatterns": [],
+        "readMaxCommits": 1000,
         "chroma": {
             "embeddingFunction": {
                 "name": "DefaultEmbeddingFunction",
@@ -35,6 +36,7 @@
             "additionalProperties": False,
             "properties": {
                 "port": {"type": "integer", "minimum": 1, "maximum": 65535},
+                "readMaxCommits": {"type": ["integer", "null"], "minimum": 1, "maximum": 65535},
                 "ignorePatterns": {
                     "type": "array",
                     "items": {"type": "string"},