lints

Author: donald-pinckney

temporalio/contrib/openai_agents/testing.py

@@ -26,14 +26,16 @@ class ResponseBuilders:
     """Builders for creating model responses for testing.
 
     .. warning::
-        This API is experimental and may change in the future."""
+        This API is experimental and may change in the future.
+    """
 
     @staticmethod
     def model_response(output: TResponseOutputItem) -> ModelResponse:
         """Create a ModelResponse with the given output.
 
         .. warning::
-           This API is experimental and may change in the future."""
+           This API is experimental and may change in the future.
+        """
         return ModelResponse(
             output=[output],
             usage=Usage(),
@@ -45,7 +47,8 @@ def response_output_message(text: str) -> ResponseOutputMessage:
         """Create a ResponseOutputMessage with text content.
 
         .. warning::
-           This API is experimental and may change in the future."""
+           This API is experimental and may change in the future.
+        """
         return ResponseOutputMessage(
             id="",
             content=[
@@ -65,7 +68,8 @@ def tool_call(arguments: str, name: str) -> ModelResponse:
         """Create a ModelResponse with a function tool call.
 
         .. warning::
-           This API is experimental and may change in the future."""
+           This API is experimental and may change in the future.
+        """
         return ResponseBuilders.model_response(
             ResponseFunctionToolCall(
                 arguments=arguments,
@@ -82,7 +86,8 @@ def output_message(text: str) -> ModelResponse:
         """Create a ModelResponse with an output message.
 
         .. warning::
-           This API is experimental and may change in the future."""
+           This API is experimental and may change in the future.
+        """
         return ResponseBuilders.model_response(
             ResponseBuilders.response_output_message(text)
         )
@@ -92,38 +97,43 @@ class TestModelProvider(ModelProvider):
     """Test model provider which simply returns the given module.
 
     .. warning::
-        This API is experimental and may change in the future."""
+        This API is experimental and may change in the future.
+    """
 
     __test__ = False
 
     def __init__(self, model: Model):
         """Initialize a test model provider with a model.
 
         .. warning::
-           This API is experimental and may change in the future."""
+           This API is experimental and may change in the future.
+        """
         self._model = model
 
     def get_model(self, model_name: Union[str, None]) -> Model:
         """Get a model from the model provider.
 
         .. warning::
-           This API is experimental and may change in the future."""
+           This API is experimental and may change in the future.
+        """
         return self._model
 
 
 class TestModel(Model):
     """Test model for use mocking model responses.
 
     .. warning::
-        This API is experimental and may change in the future."""
+        This API is experimental and may change in the future.
+    """
 
     __test__ = False
 
     def __init__(self, fn: Callable[[], ModelResponse]) -> None:
         """Initialize a test model with a callable.
 
         .. warning::
-           This API is experimental and may change in the future."""
+           This API is experimental and may change in the future.
+        """
         self.fn = fn
 
     async def get_response(
@@ -159,6 +169,7 @@ def returning_responses(responses: list[ModelResponse]) -> "TestModel":
         """Create a mock model which sequentially returns responses from a list.
 
         .. warning::
-           This API is experimental and may change in the future."""
+           This API is experimental and may change in the future.
+        """
         i = iter(responses)
         return TestModel(lambda: next(i))