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,7 +8,7 @@ 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},
         inputs={"question": "What is the capital of the France?"},
@@ -27,7 +28,7 @@ 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},
         inputs={"question": "What is the capital of the France?"},