add model_url to openai model

langextract/io.py

@@ -81,7 +81,7 @@ def load(self, delimiter: str = ',') -> Iterator[data.Document]:
 
 def save_annotated_documents(
     annotated_documents: Iterator[data.AnnotatedDocument],
-    output_dir: pathlib.Path | None = None,
+    output_dir: pathlib.Path | str | None = None,
     output_name: str = 'data.jsonl',
     show_progress: bool = True,
 ) -> None:
@@ -90,7 +90,7 @@ def save_annotated_documents(
   Args:
     annotated_documents: Iterator over AnnotatedDocument objects to save.
     output_dir: The directory to which the JSONL file should be written.
-      Defaults to 'test_output/' if None.
+      Can be a Path object or a string. Defaults to 'test_output/' if None.
     output_name: File name for the JSONL file.
     show_progress: Whether to show a progress bar during saving.
 
@@ -100,6 +100,8 @@ def save_annotated_documents(
   """
   if output_dir is None:
     output_dir = pathlib.Path('test_output')
+  else:
+    output_dir = pathlib.Path(output_dir)
 
   output_dir.mkdir(parents=True, exist_ok=True)