From adc05da1f8d87ce0ba0255170eccd1edbe215cc0 Mon Sep 17 00:00:00 2001 From: "Jun Siang Cheah (aider)" Date: Sun, 1 Sep 2024 23:23:26 +0100 Subject: [PATCH] refactor: make path to 1-second-of-silence.mp3 agnostic --- src/pull_available_models.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/pull_available_models.py b/src/pull_available_models.py index 877b297..1e66ba6 100644 --- a/src/pull_available_models.py +++ b/src/pull_available_models.py @@ -112,7 +112,7 @@ def get_model_name(id): return id -def get_groq_limits_for_stt_model(model_id): +def get_groq_limits_for_stt_model(model_id, script_dir): print(f"Getting limits for STT model {model_id}...") r = requests.post( "https://api.groq.com/openai/v1/audio/transcriptions", @@ -135,9 +135,9 @@ def get_groq_limits_for_stt_model(model_id): } -def get_groq_limits_for_model(model_id): +def get_groq_limits_for_model(model_id, script_dir): if "whisper" in model_id: - return get_groq_limits_for_stt_model(model_id) + return get_groq_limits_for_stt_model(model_id, script_dir) print(f"Getting limits for chat model {model_id}...") r = requests.post( "https://api.groq.com/openai/v1/chat/completions", @@ -174,11 +174,12 @@ def fetch_groq_models(): }, ) r.raise_for_status() + script_dir = os.path.dirname(os.path.abspath(__file__)) models = r.json()["data"] pprint(models) ret_models = [] for model in models: - limits = get_groq_limits_for_model(model["id"]) + limits = get_groq_limits_for_model(model["id"], script_dir) ret_models.append( { "id": model["id"],