Leonydis137 commited on
Commit
80e76bd
·
verified ·
1 Parent(s): 5d7c1d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -1
app.py CHANGED
@@ -7,7 +7,20 @@ import numpy as np
7
  import faiss
8
  import os
9
  import pickle
10
- from openai.embeddings_utils import get_embedding, cosine_similarity
 
 
 
 
 
 
 
 
 
 
 
 
 
11
 
12
  # === CONFIG ===
13
  openai.api_key = os.environ.get("OPENAI_API_KEY")
 
7
  import faiss
8
  import os
9
  import pickle
10
+ import openai
11
+ import numpy as np
12
+
13
+ def get_embedding(text, model="text-embedding-3-small"):
14
+ response = openai.embeddings.create(
15
+ input=[text],
16
+ model=model
17
+ )
18
+ return response.data[0].embedding
19
+
20
+ def cosine_similarity(vec1, vec2):
21
+ vec1 = np.array(vec1)
22
+ vec2 = np.array(vec2)
23
+ return np.dot(vec1, vec2) / (np.linalg.norm(vec1) * np.linalg.norm(vec2))
24
 
25
  # === CONFIG ===
26
  openai.api_key = os.environ.get("OPENAI_API_KEY")