fix: Type checking fixes

Author: jamesbaskerville

tests/integration/conftest.py

@@ -1,13 +1,14 @@
-from contextlib import contextmanager, redirect_stdout
-from dataclasses import dataclass
-import os
-from typing import Any, ContextManager, Generator
 import io
-from typing import TextIO
+import os
 import uuid
-import pytest
+from contextlib import contextmanager, redirect_stdout
+from dataclasses import dataclass
+from typing import Any, ContextManager, Generator, TextIO
+
 import dotenv
+import pytest
 from humanloop.client import Humanloop
+from humanloop.requests.prompt_kernel_request import PromptKernelRequestParams
 
 
 @dataclass
@@ -55,7 +56,7 @@ def sdk_test_dir(humanloop_test_client: Humanloop) -> Generator[str, None, None]
 
 
 @pytest.fixture(scope="function")
-def test_prompt_config() -> dict[str, Any]:
+def test_prompt_config() -> PromptKernelRequestParams:
     return {
         "provider": "openai",
         "model": "gpt-4o-mini",

tests/integration/test_prompts.py

@@ -1,4 +1,5 @@
 from humanloop.client import Humanloop
+
 from tests.integration.conftest import TestIdentifiers
 
 
@@ -7,17 +8,17 @@ def test_prompts_call(
     prompt: TestIdentifiers,
     test_prompt_config: TestIdentifiers,
 ) -> None:
-    response = humanloop_test_client.prompts.call(
+    response = humanloop_test_client.prompts.call(  # type: ignore [attr-defined]
         path=prompt.file_path,
-        prompt={**test_prompt_config},
+        prompt={**test_prompt_config},  # type: ignore [misc, arg-type, typeddict-item, dict-item, list-item]
         inputs={"question": "What is the capital of the France?"},
     )
     assert response is not None
     assert response.log_id is not None
     assert response.logs is not None
     for log in response.logs:
         assert log is not None
-        assert log.output or log.error or log.output_message is not None
+        assert log.output is not None
         assert "Paris" in log.output
     assert response.prompt.path == prompt.file_path
 
@@ -27,16 +28,16 @@ def test_prompts_call_stream(
     prompt: TestIdentifiers,
     test_prompt_config: TestIdentifiers,
 ) -> None:
-    response = humanloop_test_client.prompts.call_stream(
+    response = humanloop_test_client.prompts.call_stream(  # type: ignore [attr-defined]
         path=prompt.file_path,
-        prompt={**test_prompt_config},
+        prompt={**test_prompt_config},  # type: ignore [misc, arg-type, typeddict-item, dict-item, list-item]
         inputs={"question": "What is the capital of the France?"},
     )
 
     output = ""
     for chunk in response:
         assert chunk is not None
-        assert chunk.output or chunk.error or chunk.output_message is not None
+        assert chunk.output is not None
         assert chunk.id is not None
         assert chunk.prompt_id is not None
         assert chunk.version_id is not None