fmt

Author: donald-pinckney

temporalio/contrib/openai_agents/testing.py

@@ -24,14 +24,14 @@
 
 class ResponseBuilders:
     """Builders for creating model responses for testing.
-    
+
     .. warning::
         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."""
         return ModelResponse(
@@ -43,7 +43,7 @@ def model_response(output: TResponseOutputItem) -> ModelResponse:
     @staticmethod
     def response_output_message(text: str) -> ResponseOutputMessage:
         """Create a ResponseOutputMessage with text content.
-    
+
         .. warning::
            This API is experimental and may change in the future."""
         return ResponseOutputMessage(
@@ -63,7 +63,7 @@ def response_output_message(text: str) -> ResponseOutputMessage:
     @staticmethod
     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."""
         return ResponseBuilders.model_response(
@@ -80,7 +80,7 @@ def tool_call(arguments: str, name: str) -> ModelResponse:
     @staticmethod
     def output_message(text: str) -> ModelResponse:
         """Create a ModelResponse with an output message.
-    
+
         .. warning::
            This API is experimental and may change in the future."""
         return ResponseBuilders.model_response(
@@ -90,38 +90,38 @@ def output_message(text: str) -> ModelResponse:
 
 class TestModelProvider(ModelProvider):
     """Test model provider which simply returns the given module.
-    
+
     .. warning::
         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."""
         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."""
         return self._model
 
 
 class TestModel(Model):
     """Test model for use mocking model responses.
-    
+
     .. warning::
         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."""
         self.fn = fn
@@ -157,7 +157,7 @@ def stream_response(
     @staticmethod
     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."""
         i = iter(responses)