Fixes

twc_openai_embeddings.py

@@ -19,9 +19,13 @@ class OpenAIModel:
 
 
     def init_model(self,model_name = None):
-        #print("Init model",model_name)
-        openai.api_key = os.getenv("OPENAI_API_KEY")
-        if (len(openai.api_key) == 0):
+        print("OpanAI: Init model",model_name)
+        try:
+            openai.api_key = os.getenv("OPENAI_API_KEY")
+        except:
+            print("API key not set")
+
+        if (len(openai.api_key) == 0 and not self.skip_key):
                 print("Open API key not set")
         
         if (model_name is None):
@@ -34,6 +38,7 @@ class OpenAIModel:
         if (len(openai.api_key) == 0 and not self.skip_key):
                 print("Open API key not set")
                 return [],[]
+        print("In compute embeddings after key check")
         in_file = self.model_name + '.'.join(input_file_name.split('.')[:-1]) + "_embed.json"
         cached = False
         try:
@@ -63,6 +68,7 @@ class OpenAIModel:
         if (len(openai.api_key) == 0 and not self.skip_key):
                 print("Open API key not set")
                 return {}
+        print("In output results after key check")
         # Calculate cosine similarities
         # Cosine similarities are in [-1, 1]. Higher means more similar
         cosine_dict = {}