output format typing fixes

src/agent_interfaces.py

@@ -41,7 +41,7 @@ class BaseAgentInterface:
     def respond_to_formatted(
             self,
             message: Message,
-            output_format: OutputFormat,
+            output_format: Type[OutputFormatModel],
             max_retries=3,
             **kwargs
     ) -> OutputFormatModel:

src/output_formats.py

@@ -23,9 +23,6 @@ class OutputFormatModel(BaseModel):
         return FORMAT_INSTRUCTIONS.format(schema=json_format)
 
 
-OutputFormat = NewType("OutputFormat", OutputFormatModel)
-
-
 class AnimalDescriptionFormat(OutputFormatModel):
     # Define fields of our class here
     description: str = Field(description="A brief description of the animal")