Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -17,13 +17,14 @@ import faiss
|
|
| 17 |
client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
|
| 18 |
|
| 19 |
def get_embedding(text: str, model="text-embedding-ada-002"):
|
| 20 |
-
"""
|
| 21 |
try:
|
|
|
|
| 22 |
response = client.embeddings.create(
|
| 23 |
input=[text],
|
| 24 |
model=model
|
| 25 |
)
|
| 26 |
-
return response.data[0]
|
| 27 |
except Exception as e:
|
| 28 |
print(f"Error getting embedding: {e}")
|
| 29 |
return None
|
|
@@ -103,6 +104,8 @@ def analyze_query_and_find_products(query: str) -> str:
|
|
| 103 |
)
|
| 104 |
|
| 105 |
analysis = analysis_response.choices[0].message.content
|
|
|
|
|
|
|
| 106 |
|
| 107 |
# Get embedding for the query
|
| 108 |
query_embedding = get_embedding(query + " " + analysis)
|
|
|
|
| 17 |
client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
|
| 18 |
|
| 19 |
def get_embedding(text: str, model="text-embedding-ada-002"):
|
| 20 |
+
"""Get embeddings for a text using OpenAI's API"""
|
| 21 |
try:
|
| 22 |
+
text = text.replace("\n", " ")
|
| 23 |
response = client.embeddings.create(
|
| 24 |
input=[text],
|
| 25 |
model=model
|
| 26 |
)
|
| 27 |
+
return response.data[0].embedding
|
| 28 |
except Exception as e:
|
| 29 |
print(f"Error getting embedding: {e}")
|
| 30 |
return None
|
|
|
|
| 104 |
)
|
| 105 |
|
| 106 |
analysis = analysis_response.choices[0].message.content
|
| 107 |
+
|
| 108 |
+
print("現在要比對向量的是<analysis>":analysis)
|
| 109 |
|
| 110 |
# Get embedding for the query
|
| 111 |
query_embedding = get_embedding(query + " " + analysis)
|