WIP

- Added different AIs for ASR and for structure builder
- Forcing models to use JSON from now on

structure_builder.py

@@ -79,16 +79,15 @@ class ImageElement(_StrictModel):
     event_id: int
 
 
-StructureElement = Annotated[
+StructureElement = (
     HeadingElement
     | ParagraphElement
     | UnorderedListElement
     | OrderedListElement
     | DefinitionElement
     | ImportantElement
-    | ImageElement,
-    Field(discriminator="type"),
-]
+    | ImageElement
+)
 
 
 class Structure(_StrictModel):
@@ -204,7 +203,7 @@ class StructureBuilder:
         retries_left = self.MAX_RETRIES
         while retries_left > 0:
             retries_left -= 1
-            response = self._agent.completion(messages=messages)
+            response = self._agent.completion(messages=messages, response_format=_BuildResult)
             if debug_dir:
                 with open(
                     f"{debug_dir}/{retries_left}-retries-left.txt",