adds ability to pass in api key for every transcriber agent and synthesizer

This commit is contained in:
Ajay Raj 2023-03-28 23:33:36 -07:00
commit ecebe4c1a5
11 changed files with 71 additions and 17 deletions

View file

@ -1,5 +1,6 @@
import os
import asyncio
from typing import Optional
import openai
import numpy as np
import requests
@ -26,8 +27,11 @@ class GoodbyeModel:
embeddings_cache_path=os.path.join(
os.path.dirname(__file__), "goodbye_embeddings"
),
openai_api_key: Optional[str] = None,
):
openai.api_key = getenv("OPENAI_API_KEY")
openai.api_key = openai_api_key or getenv("OPENAI_API_KEY")
if not openai.api_key:
raise ValueError("OPENAI_API_KEY must be set in environment or passed in")
self.goodbye_embeddings = self.load_or_create_embeddings(
f"{embeddings_cache_path}/goodbye_embeddings.npy"
)