Skip to content

Commit bba274c

Browse files
committed
perf: openai alignment
1 parent 71840fe commit bba274c

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

controller.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,13 @@
1212
import zlib
1313
import gc
1414
import os
15-
import openai
1615
import pandas as pd
1716
import shutil
17+
from openai import APIConnectionError
1818

1919
from src.embedders import Transformer, util
20+
21+
# Embedder imports are used by eval(Embedder) in __setup_tmp_embedder
2022
from src.embedders.classification.contextual import (
2123
OpenAISentenceEmbedder,
2224
HuggingFaceSentenceEmbedder,
@@ -350,7 +352,7 @@ def run_encoding(
350352
enums.EmbeddingState.ENCODING.value,
351353
initial_count,
352354
)
353-
except openai.error.APIConnectionError as e:
355+
except APIConnectionError as e:
354356
embedding.update_embedding_state_failed(
355357
project_id,
356358
embedding_id,

0 commit comments

Comments
 (0)