Zaious commited on
Commit
bc9c8f5
·
verified ·
1 Parent(s): cb039f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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
- """ OpenAI API 獲取文本嵌入"""
21
  try:
 
22
  response = client.embeddings.create(
23
  input=[text],
24
  model=model
25
  )
26
- return response.data[0]['embedding']
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)