chore(lint): add linting task using Black for code formatting

Signed-off-by: Eden Reich <[email protected]>

Author: edenreich

Taskfile.yml

@@ -7,6 +7,11 @@ tasks:
     cmds:
       - curl -o openapi.yaml https://raw.githubusercontent.com/inference-gateway/inference-gateway/refs/heads/main/openapi.yaml
 
+  lint:
+    desc: Lint the code
+    cmds:
+      - black inference_gateway/ tests/
+
   test:
     desc: Run tests
     cmds:

inference_gateway/client.py

@@ -6,6 +6,7 @@
 
 class Provider(str, Enum):
     """Supported LLM providers"""
+
     OLLAMA = "ollama"
     GROQ = "groq"
     OPENAI = "openai"
@@ -16,6 +17,7 @@ class Provider(str, Enum):
 
 class Role(str, Enum):
     """Message role types"""
+
     SYSTEM = "system"
     USER = "user"
     ASSISTANT = "assistant"
@@ -28,10 +30,7 @@ class Message:
 
     def to_dict(self) -> Dict[str, str]:
         """Convert message to dictionary format with string values"""
-        return {
-            "role": self.role.value,
-            "content": self.content
-        }
+        return {"role": self.role.value, "content": self.content}
 
 
 class Model:
@@ -57,7 +56,7 @@ class InferenceGatewayClient:
 
     def __init__(self, base_url: str, token: Optional[str] = None):
         """Initialize the client with base URL and optional auth token"""
-        self.base_url = base_url.rstrip('/')
+        self.base_url = base_url.rstrip("/")
         self.session = requests.Session()
         if token:
             self.session.headers.update({"Authorization": f"Bearer {token}"})
@@ -68,20 +67,11 @@ def list_models(self) -> List[ProviderModels]:
         response.raise_for_status()
         return response.json()
 
-    def generate_content(
-        self,
-        provider: Provider,
-        model: str,
-        messages: List[Message]
-    ) -> Dict:
-        payload = {
-            "model": model,
-            "messages": [msg.to_dict() for msg in messages]
-        }
+    def generate_content(self, provider: Provider, model: str, messages: List[Message]) -> Dict:
+        payload = {"model": model, "messages": [msg.to_dict() for msg in messages]}
 
         response = self.session.post(
-            f"{self.base_url}/llms/{provider.value}/generate",
-            json=payload
+            f"{self.base_url}/llms/{provider.value}/generate", json=payload
         )
         response.raise_for_status()
         return response.json()

tests/test_client.py

@@ -24,8 +24,7 @@ def test_client_initialization():
     assert client.base_url == "http://test-api"
     assert "Authorization" not in client.session.headers
 
-    client_with_token = InferenceGatewayClient(
-        "http://test-api", token="test-token")
+    client_with_token = InferenceGatewayClient("http://test-api", token="test-token")
     assert "Authorization" in client_with_token.session.headers
     assert client_with_token.session.headers["Authorization"] == "Bearer test-token"
 
@@ -43,10 +42,7 @@ def test_list_models(mock_get, client, mock_response):
 @patch("requests.Session.post")
 def test_generate_content(mock_post, client, mock_response):
     """Test content generation"""
-    messages = [
-        Message(Role.SYSTEM, "You are a helpful assistant"),
-        Message(Role.USER, "Hello!")
-    ]
+    messages = [Message(Role.SYSTEM, "You are a helpful assistant"), Message(Role.USER, "Hello!")]
 
     mock_post.return_value = mock_response
     response = client.generate_content(Provider.OPENAI, "gpt-4", messages)
@@ -57,9 +53,9 @@ def test_generate_content(mock_post, client, mock_response):
             "model": "gpt-4",
             "messages": [
                 {"role": "system", "content": "You are a helpful assistant"},
-                {"role": "user", "content": "Hello!"}
-            ]
-        }
+                {"role": "user", "content": "Hello!"},
+            ],
+        },
     )
     assert response == {"response": "test"}