From e08d6939256d40e901ce9b25414f773993d56277 Mon Sep 17 00:00:00 2001
From: Robert Craigie <robert@craigie.dev>
Date: Mon, 22 Jul 2024 11:04:28 +0100
Subject: [PATCH] handle the string case in audio directly

---
 src/openai/cli/_api/audio.py | 11 +++++++++--
 src/openai/cli/_utils.py     |  5 +----
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/src/openai/cli/_api/audio.py b/src/openai/cli/_api/audio.py
index 90d21b9932..fab452b440 100644
--- a/src/openai/cli/_api/audio.py
+++ b/src/openai/cli/_api/audio.py
@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import sys
 from typing import TYPE_CHECKING, Any, Optional, cast
 from argparse import ArgumentParser
 
@@ -75,7 +76,10 @@ def transcribe(args: CLITranscribeArgs) -> None:
             # but we don't want to validate that here for forwards-compat
             response_format=cast(Any, args.response_format),
         )
-        print_model(model)
+        if isinstance(model, str):
+            sys.stdout.write(model + "\n")
+        else:
+            print_model(model)
 
     @staticmethod
     def translate(args: CLITranslationArgs) -> None:
@@ -91,4 +95,7 @@ def translate(args: CLITranslationArgs) -> None:
             # but we don't want to validate that here for forwards-compat
             response_format=cast(Any, args.response_format),
         )
-        print_model(model)
+        if isinstance(model, str):
+            sys.stdout.write(model + "\n")
+        else:
+            print_model(model)
diff --git a/src/openai/cli/_utils.py b/src/openai/cli/_utils.py
index d53e6acc6c..673eed613c 100644
--- a/src/openai/cli/_utils.py
+++ b/src/openai/cli/_utils.py
@@ -33,10 +33,7 @@ def organization_info() -> str:
 
 
 def print_model(model: BaseModel) -> None:
-    if isinstance(model, BaseModel):
-        sys.stdout.write(model_json(model, indent=2) + "\n")
-    elif isinstance(model, str):
-        sys.stdout.write(model)
+    sys.stdout.write(model_json(model, indent=2) + "\n")
 
 
 def can_use_http2() -> bool: