taskswithcode commited on
Commit
3ce8ea6
·
1 Parent(s): c2ca3f3
Files changed (1) hide show
  1. twc_openai_embeddings.py +3 -2
twc_openai_embeddings.py CHANGED
@@ -14,6 +14,7 @@ class OpenAIModel:
14
  def __init__(self):
15
  self.debug = False
16
  self.model_name = None
 
17
  print("In OpenAI API constructor")
18
 
19
 
@@ -30,7 +31,7 @@ class OpenAIModel:
30
 
31
 
32
  def compute_embeddings(self,input_file_name,input_data,is_file):
33
- if (len(openai.api_key) == 0):
34
  print("Open API key not set")
35
  return [],[]
36
  in_file = self.model_name + '.'.join(input_file_name.split('.')[:-1]) + "_embed.json"
@@ -59,7 +60,7 @@ class OpenAIModel:
59
  return texts,embeddings
60
 
61
  def output_results(self,output_file,texts,embeddings,main_index = 0):
62
- if (len(openai.api_key) == 0):
63
  print("Open API key not set")
64
  return {}
65
  # Calculate cosine similarities
 
14
  def __init__(self):
15
  self.debug = False
16
  self.model_name = None
17
+ self.skip_key = True
18
  print("In OpenAI API constructor")
19
 
20
 
 
31
 
32
 
33
  def compute_embeddings(self,input_file_name,input_data,is_file):
34
+ if (len(openai.api_key) == 0 and not self.skip_key):
35
  print("Open API key not set")
36
  return [],[]
37
  in_file = self.model_name + '.'.join(input_file_name.split('.')[:-1]) + "_embed.json"
 
60
  return texts,embeddings
61
 
62
  def output_results(self,output_file,texts,embeddings,main_index = 0):
63
+ if (len(openai.api_key) == 0 and not self.skip_key):
64
  print("Open API key not set")
65
  return {}
66
  # Calculate cosine similarities