linting issues

Author: Andrei Bratu

tests/integration/test_decorators.py

@@ -30,6 +30,7 @@ def my_prompt(question: str) -> str:
                 messages=[{"role": "user", "content": question}],
             )
 
+            assert response.choices[0].message.content is not None
             return response.choices[0].message.content
 
         assert "paris" in my_prompt("What is the capital of the France?").lower()
@@ -38,7 +39,8 @@ def my_prompt(question: str) -> str:
         prompt_versions_response = humanloop_test_client.prompts.list_versions(id=prompt_response.id)
         assert len(prompt_versions_response.records) == 2
 
-        logs = humanloop_test_client.logs.list(file_id=prompt_response.id, page=1, size=50)
-        assert len(logs.items) == 1
+        logs_response = humanloop_test_client.logs.list(file_id=prompt_response.id, page=1, size=50)
+
+        assert logs_response.items is not None and len(logs_response.items) == 1
     finally:
         humanloop_test_client.prompts.delete(id=prompt_response.id)