Update app/logic/calc_b.py

app/logic/calc_b.py

@@ -21,10 +21,14 @@ df_base = pd.read_csv(csv_path)
 def query_faiss_index(query_text, 
                       faiss_index_path=None,
                       metadata_path=None,
-                      openai_api_key="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",
                       top_k=5):
-    client = OpenAI(api_key=openai_api_key)
+    
+    openai_api_key = os.getenv("OPENAI_API_KEY")
+    if not openai_api_key:
+        raise ValueError("OPENAI_API_KEY is not set")
 
+    client = OpenAI(api_key=openai_api_key)
+    
     BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 
     if faiss_index_path is None:
@@ -40,7 +44,7 @@ def query_faiss_index(query_text,
     # Get embedding for query
     response = client.embeddings.create(
         input=query_text.lower(),
-        model="text-embedding-3-large"
+        model="text-embedding-3-small"
     )
     query_embedding = response.data[0].embedding
     query_embedding_np = np.array([query_embedding]).astype("float32")