fixes

.github/workflows/black.yml

@@ -1,4 +1,4 @@
-name: Lint
+name: Black
 
 on: [push, pull_request]
 

.github/workflows/pylint.yml

@@ -3,11 +3,11 @@ name: Pylint
 on: [push]
 
 jobs:
-  build:
+  lint:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.8", "3.9", "3.10"]
+        python-version: ["3.10"]
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python ${{ matrix.python-version }}

crllm_config.toml

@@ -3,5 +3,5 @@ description = "A cli tool that used large language model to generate code review
 
 [crllm]
 loader = "git_staging"
-model = "llama3.1"
-git_changed_lines=false
+model = "llama3.2"
+git_changed_lines=true

pyproject.toml

@@ -22,3 +22,7 @@ build-backend = "hatchling.build"
 
 [project.scripts]
 crllm = "crllm.cli:cli"
+
+[tool.black]
+line-length = 88
+target-version = ['py37']

src/crllm/app.py

@@ -23,5 +23,6 @@ def app(path: str, isRepo: bool = True):
 
     parser = ResultParser()
     md = Markdown(parser.parse(result))
+
     console = Console()
     console.print(md)

src/crllm/cli.py

@@ -1,7 +1,6 @@
 import argparse
 import logging
 
-# Works in build fix everywhere
 from crllm.app import app
 
 

src/crllm/code/loaders/git_stage.py

@@ -1,30 +1,27 @@
 import logging
-
-from crllm.code.ignored_files_filter import filter_ignored_files
 from git import Repo
 
 from crllm.config.config_service import config_service
 
 
-# ToDo: not staged https://stackoverflow.com/questions/33733453/get-changed-files-using-gitpython
+# ToDo: not staged
+# https://stackoverflow.com/questions/33733453/get-changed-files-using-gitpython
 def git_stage(path):
     config = config_service.getConfig()
     use_changed_lines = config["crllm"]["git_changed_lines"]
 
     repo = Repo(path)
     diff = repo.index.diff(None, create_patch=True)
 
-    print(str(diff[0]))
-
     if len(diff) == 0:
         raise Exception("No changed files found!")
 
     code = ""
 
     if not use_changed_lines:
-        changedFiles = [item.a_path for item in diff if item.a_path != None]
-        changedFiles = filter_ignored_files(changedFiles)
-        
+        changedFiles = [
+            item.a_path for item in diff if item.a_path is not None]
+
         logging.info(f"Found changed files: {changedFiles}")
 
         for file in changedFiles:
@@ -35,6 +32,4 @@ def git_stage(path):
         for line in diff:
             code += str(line) + "\n"
 
-    print(code)
-
     return code

src/crllm/config/config_service.py

@@ -6,7 +6,6 @@
 from crllm.config import CONFIG_DIR
 
 
-# config loader?
 class ConfigService:
     configPath = os.path.join(CONFIG_DIR, "config.toml")
     config = None
@@ -16,14 +15,14 @@ def getConfig(self, path="./crllm_config.toml"):
             return self.config
 
         self.config = toml.load(self.configPath)
-        
+
         if not os.path.isfile(path):
             return self.config
-        
+
         project_config = toml.load(path)
-        
+
         self.config = always_merger.merge(self.config, project_config)
-        
+
         logging.debug(self.config)
 
         return self.config

src/crllm/model/model.py

@@ -12,11 +12,11 @@ def generate(self, promptTemplate, promptArgs):
 
         llm_chain = promptTemplate | model
         result = llm_chain.invoke(promptArgs)
-        
+
         logging.debug(result)
 
         return result.content
-    
+
     @abstractmethod
     def _getModel(self, model_config):
         pass

src/crllm/model/provider/__init__.py

@@ -6,5 +6,5 @@
 providers = {
     "ollama": OllamaModel,
     "azure": AzureModel,
-    "huggingface": HuggingFaceModel
+    "huggingface": HuggingFaceModel,
 }

src/crllm/model/provider/azure.py

@@ -2,6 +2,6 @@
 from crllm.model.model import Model
 
 
-class AzureModel(Model):    
+class AzureModel(Model):
     def _getModel(self, model_config):
         return AzureChatOpenAI(**model_config)

src/crllm/model/provider/huggingface.py

@@ -1,8 +1,8 @@
-from langchain_huggingface import ChatHuggingFace,HuggingFaceEndpoint
+from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
 from crllm.model.model import Model
 
 
-class HuggingFaceModel(Model):    
+class HuggingFaceModel(Model):
     def _getModel(self, model_config):
         llm = HuggingFaceEndpoint(**model_config)
 

src/crllm/prompt_builder.py

@@ -15,15 +15,11 @@ def build(self):
 
         logging.debug(config["prompt"])
 
-        prompt = ChatPromptTemplate.from_messages([
-            (
-                "system",
-                config["prompt"]["template"]
-            ),
-            (
-                "human",
-                "Do code review for the following code: \n {code}"
-            )
-        ])
+        prompt = ChatPromptTemplate.from_messages(
+            [
+                ("system", config["prompt"]["template"]),
+                ("human", "Do code review for the following code: \n {code}"),
+            ]
+        )
 
         return prompt