Spaces:
Runtime error
Runtime error
taskswithcode
commited on
Commit
·
3ce8ea6
1
Parent(s):
c2ca3f3
Fixes
Browse files- 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
|