formatted

Author: hrshdhgd

src/llm_change_agent/config/llm_config.py

@@ -13,7 +13,9 @@ class LLMConfig(BaseModel):
 
 class OpenAIConfig(LLMConfig):
     """Configuration for OpenAI LLM model."""
+
     def __init__(self, **data):
+        """Initialize the OpenAIConfig class."""
         super().__init__(**data)
         if self.model.startswith("o1"):
             self.temperature = 1.0

src/llm_change_agent/utils/llm_utils.py

@@ -77,9 +77,9 @@ def get_lbl_cborg_models():
         "lbl/cborg-chat:latest",  # LBL-hosted model
         "lbl/cborg-chat-nano:latest",  # LBL-hosted model
         "lbl/cborg-coder:latest",  # LBL-hosted model
-        "openai/chatgpt:latest", # LBL cloud-hosted model
-        "anthropic/claude:latest", # LBL cloud-hosted model
-        "google/gemini:latest", # LBL cloud-hosted model
+        "openai/chatgpt:latest",  # LBL cloud-hosted model
+        "anthropic/claude:latest",  # LBL cloud-hosted model
+        "google/gemini:latest",  # LBL cloud-hosted model
     ]