[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

.gitignore

@@ -174,4 +174,4 @@ cython_debug/
 .DS_Store
 
 # output
-stdout
+stdout

ai_feedback/__main__.py

@@ -6,9 +6,9 @@
 from pathlib import Path
 
 from . import code_processing, image_processing, text_processing
-from .models import ModelFactory
 from .helpers import arg_options
 from .helpers.constants import HELP_MESSAGES
+from .models import ModelFactory
 
 _TYPE_BY_EXTENSION = {
     '.c': 'C',

ai_feedback/models/ClaudeModel.py

@@ -1,10 +1,10 @@
 import os
 from pathlib import Path
-from ollama import Message
 from typing import Optional, Tuple
 
 import anthropic
 from dotenv import load_dotenv
+from ollama import Message
 
 from ..helpers.image_extractor import encode_image
 from ..helpers.model_options_helpers import cast_to_type, claude_option_schema

ai_feedback/models/DeepSeekModelModified.py

@@ -1,19 +1,20 @@
-import os
-import sys
 import json
+import os
 import subprocess
+import sys
 from pathlib import Path
 from typing import Optional, Tuple
 
 import ollama
 import requests
 from dotenv import load_dotenv
 
-from .Model import Model
 from ..helpers.model_options_helpers import cast_to_type, ollama_option_schema
+from .Model import Model
 
 load_dotenv()
 
+
 class DeepSeekModelModified(Model):
     """
     Backends:
@@ -99,7 +100,9 @@ def generate_response(
         if self.backend == "ollama":
             response = self._generate_with_ollama(prompt, system_instructions, model_options, schema)
         elif self.backend == "llama":
-            response = self._generate_with_llama(prompt, system_instructions, model_options, schema, effective_llama_mode)
+            response = self._generate_with_llama(
+                prompt, system_instructions, model_options, schema, effective_llama_mode
+            )
         else:
             raise ValueError(f"Unsupported backend: {self.backend}. Use 'ollama' or 'llama'.")
 

ai_feedback/models/Model.py

@@ -1,5 +1,6 @@
 from typing import Any, Tuple
-from ollama import chat, Message
+
+from ollama import Message, chat
 
 """
 Parent Class for LLMs.

ai_feedback/models/OpenAIModel.py

@@ -1,18 +1,18 @@
 import json
 import os
 from pathlib import Path
-from ollama import Message
 from typing import Optional, Tuple
-from ..helpers.image_extractor import encode_image
 
 import openai
 from dotenv import load_dotenv
+from ollama import Message
 
 from ai_feedback.helpers.model_options_helpers import (
     cast_to_type,
     openai_chat_option_schema,
 )
 
+from ..helpers.image_extractor import encode_image
 from .Model import Model
 
 load_dotenv()

ai_feedback/models/RemoteModel.py

@@ -4,8 +4,8 @@
 from typing import Any, Optional, Tuple
 
 import requests
-from ollama import Message
 from dotenv import load_dotenv
+from ollama import Message
 
 from .Model import Model
 

ai_feedback/models/__init__.py

@@ -1,12 +1,13 @@
-from typing import Type, Dict
+from typing import Dict, Type
 
-from .Model import Model
 from .ClaudeModel import ClaudeModel
-from .RemoteModel import RemoteModel
-from .OpenAIModel import OpenAIModel
 from .CodeLlamaModel import CodeLlamaModel
-from .OpenAIModelVector import OpenAIModelVector
 from .DeepSeekModelModified import DeepSeekModelModified
+from .Model import Model
+from .OpenAIModel import OpenAIModel
+from .OpenAIModelVector import OpenAIModelVector
+from .RemoteModel import RemoteModel
+
 
 class ModelFactory:
     """Factory for creating AI model instances with proper dependency injection."""
@@ -54,10 +55,7 @@ def get_model_class(cls, provider: str) -> Type[Model]:
         """
         if provider not in cls._registry:
             available = ", ".join(sorted(cls._registry.keys()))
-            raise ValueError(
-                f"Unknown model provider '{provider}'. "
-                f"Available providers: {available}"
-            )
+            raise ValueError(f"Unknown model provider '{provider}'. " f"Available providers: {available}")
         return cls._registry[provider]
 
     @classmethod