@@ -151,7 +151,7 @@ def test_handle_completion_unexpected_error(self, mock_create):
151151
152152 @patch (
153153 "app.prompt.openai.chat.completions.create" ,
154- side_effect = openai .RateLimitError (message = "rate limit" , response = dummy_response , body = None ),
154+ side_effect = openai .RateLimitError (message = "rate limit" , response = dummy_response , body = None ), # type: ignore
155155 )
156156 def test_handle_completion_rate_limit_error (self , mock_create ):
157157 """Test that rate limit errors during completion are handled."""
@@ -169,7 +169,7 @@ def test_handle_completion_api_connection_error(self, mock_create):
169169
170170 @patch (
171171 "app.prompt.openai.chat.completions.create" ,
172- side_effect = openai .AuthenticationError (message = "auth error" , response = dummy_response , body = None ),
172+ side_effect = openai .AuthenticationError (message = "auth error" , response = dummy_response , body = None ), # type: ignore
173173 )
174174 def test_handle_completion_authentication_error (self , mock_create ):
175175 """Test that authentication errors during completion are handled."""
@@ -178,15 +178,15 @@ def test_handle_completion_authentication_error(self, mock_create):
178178
179179 @patch (
180180 "app.prompt.openai.chat.completions.create" ,
181- side_effect = openai .BadRequestError (message = "bad request" , response = dummy_response , body = None ),
181+ side_effect = openai .BadRequestError (message = "bad request" , response = dummy_response , body = None ), # type: ignore
182182 )
183183 def test_handle_completion_bad_request_error (self , mock_create ):
184184 """Test that bad request errors during completion are handled."""
185185 with self .assertRaises (openai .BadRequestError ):
186186 prompt .completion ("test prompt" )
187187
188188 @patch (
189- "app.prompt.openai.chat.completions.create" , side_effect = openai .APIError (dummy_request , "api error" , body = None )
189+ "app.prompt.openai.chat.completions.create" , side_effect = openai .APIError (dummy_request , "api error" , body = None ) # type: ignore
190190 )
191191 def test_handle_completion_api_error (self , mock_create ):
192192 """Test that API errors during completion are handled."""
0 commit comments