Add pydantic.mypy plugin to mypy configuration

pyproject.toml

@@ -76,6 +76,7 @@ markers = ["integrate: integration test"]
 [tool.mypy]
 warn_return_any = true
 warn_unused_configs = true
+plugins = "pydantic.mypy"
 
 [[tool.mypy.overrides]]
 module = "sshkeyboard"

simple_typing_application/sentence_generator/openai_sentence_generator.py

@@ -22,17 +22,20 @@ def __init__(
         self,
         model: str = 'gpt-3.5-turbo-16k',
         temperature: float = .7,
-        openai_api_key: SecretStr | None = None,
+        openai_api_key: SecretStr | str | None = None,
         memory_size: int = 1,
         max_retry: int = 5,
         logger: Logger = getLogger(__name__),
     ):
         self._max_retry = max_retry
-
         self._llm = ChatOpenAI(
             model=model,
             temperature=temperature,
-            api_key=openai_api_key,
+            openai_api_key=(
+                openai_api_key.get_secret_value()  # type: ignore
+                if isinstance(openai_api_key, SecretStr) else
+                openai_api_key
+            ),
         )
         self._memory = ConversationBufferMemory(k=memory_size)  # type: ignore  # noqa
         self._chain = ConversationChain(llm=self._llm, memory=self._memory)  # type: ignore  # noqa