Merge pull request #158 from Cloud-Code-AI/156-update-how-we-pass-model-configuration-with-extra-params

fix: flake8 issues fixed

Author: sauravpanda

kaizen/llms/provider.py

@@ -31,9 +31,7 @@ def chat_completion(self, prompt, user: str = None):
             {"role": "user", "content": prompt},
         ]
 
-        response = litellm.completion(
-            messages=messages, user=user, **self.model_config
-        )
+        response = litellm.completion(messages=messages, user=user, **self.model_config)
         return response["choices"][0]["message"]["content"], response["usage"]
 
     def is_inside_token_limit(self, PROMPT, percentage=0.7):

tests/llms/test_provider.py

@@ -1,5 +1,5 @@
 import pytest
-from unittest.mock import patch, Mock
+from unittest.mock import patch
 from kaizen.llms.provider import LLMProvider
 
 
@@ -33,7 +33,8 @@ def test_chat_completion(mock_completion, llm_provider):
         "usage": {"prompt_tokens": 10, "completion_tokens": 10},
     }
     response, usage = llm_provider.chat_completion("test prompt")
-    assert response != None
+    assert response is not None
+    assert usage is not None
 
 
 @patch("kaizen.llms.provider.litellm.token_counter")