tueniuu commited on
Commit
3f396dd
·
verified ·
1 Parent(s): a893c27

Update app/logic/calc_a.py

Browse files
Files changed (1) hide show
  1. app/logic/calc_a.py +6 -2
app/logic/calc_a.py CHANGED
@@ -39,8 +39,12 @@ df_base = df_base.rename(columns={
39
  def query_faiss_index(query_text,
40
  faiss_index_path=None,
41
  metadata_path=None,
42
- openai_api_key="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",
43
  top_k=5):
 
 
 
 
 
44
  client = OpenAI(api_key=openai_api_key)
45
 
46
  BASE_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -58,7 +62,7 @@ def query_faiss_index(query_text,
58
  # Get embedding for query
59
  response = client.embeddings.create(
60
  input=query_text.lower(),
61
- model="text-embedding-3-large"
62
  )
63
  query_embedding = response.data[0].embedding
64
  query_embedding_np = np.array([query_embedding]).astype("float32")
 
39
  def query_faiss_index(query_text,
40
  faiss_index_path=None,
41
  metadata_path=None,
 
42
  top_k=5):
43
+
44
+ openai_api_key = os.getenv("OPENAI_API_KEY")
45
+ if not openai_api_key:
46
+ raise ValueError("OPENAI_API_KEY is not set")
47
+
48
  client = OpenAI(api_key=openai_api_key)
49
 
50
  BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 
62
  # Get embedding for query
63
  response = client.embeddings.create(
64
  input=query_text.lower(),
65
+ model="text-embedding-3-small"
66
  )
67
  query_embedding = response.data[0].embedding
68
  query_embedding_np = np.array([query_embedding]).astype("float32")