tueniuu commited on
Commit
b446d57
·
verified ·
1 Parent(s): b286843

Update app/logic/calc_e.py

Browse files
Files changed (1) hide show
  1. app/logic/calc_e.py +6 -2
app/logic/calc_e.py CHANGED
@@ -26,8 +26,12 @@ df_base = pd.read_csv(csv_path)
26
  def query_faiss_index(query_text,
27
  faiss_index_path=None,
28
  metadata_path=None,
29
- openai_api_key="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",
30
  top_k=5):
 
 
 
 
 
31
  client = OpenAI(api_key=openai_api_key)
32
 
33
  BASE_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -45,7 +49,7 @@ def query_faiss_index(query_text,
45
  # Get embedding for query
46
  response = client.embeddings.create(
47
  input=query_text.lower(),
48
- model="text-embedding-3-large"
49
  )
50
  query_embedding = response.data[0].embedding
51
  query_embedding_np = np.array([query_embedding]).astype("float32")
 
26
  def query_faiss_index(query_text,
27
  faiss_index_path=None,
28
  metadata_path=None,
 
29
  top_k=5):
30
+
31
+ openai_api_key = os.getenv("OPENAI_API_KEY")
32
+ if not openai_api_key:
33
+ raise ValueError("OPENAI_API_KEY is not set")
34
+
35
  client = OpenAI(api_key=openai_api_key)
36
 
37
  BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 
49
  # Get embedding for query
50
  response = client.embeddings.create(
51
  input=query_text.lower(),
52
+ model="text-embedding-3-small"
53
  )
54
  query_embedding = response.data[0].embedding
55
  query_embedding_np = np.array([query_embedding]).astype("float32")