Update app/logic/calc_a.py
Browse files- 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-
|
| 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")
|