Shri commited on
Commit
ddf4755
·
1 Parent(s): 2cad77d

fix: chatbot vector error

Browse files
Files changed (2) hide show
  1. README.md +0 -13
  2. src/chatbot/router.py +9 -3
README.md CHANGED
@@ -1,14 +1 @@
1
  # YB's Wellness App FastAPI Backend
2
- ---
3
- title: Yuvabe App Backend
4
- emoji: 🔥
5
- colorFrom: gray
6
- colorTo: gray
7
- sdk: docker
8
- pinned: false
9
- short_description: Yuvabe App Backend
10
- ---
11
-
12
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
13
-
14
- # YB's Wellness App FastAPI Backend
 
1
  # YB's Wellness App FastAPI Backend
 
 
 
 
 
 
 
 
 
 
 
 
 
src/chatbot/router.py CHANGED
@@ -88,17 +88,22 @@ async def semantic_search(
88
  q_vector = payload.embedding
89
  top_k = payload.top_k or 3
90
 
 
 
 
91
  sql = text(
92
  """
93
  SELECT id, kb_id, chunk_text, embedding <=> :query_vec AS score
94
  FROM knowledge_chunk
95
  ORDER BY embedding <=> :query_vec
96
  LIMIT :top_k
97
- """
98
  )
99
 
100
- rows = await session.exec(sql, {"query_vec": q_vector, "top_k": top_k})
101
- rows = rows.fetchall()
 
 
102
 
103
  return [
104
  SemanticSearchResult(
@@ -109,3 +114,4 @@ async def semantic_search(
109
  )
110
  for r in rows
111
  ]
 
 
88
  q_vector = payload.embedding
89
  top_k = payload.top_k or 3
90
 
91
+ # Convert Python list → pgvector string format
92
+ q_vector_str = "[" + ",".join(str(x) for x in q_vector) + "]"
93
+
94
  sql = text(
95
  """
96
  SELECT id, kb_id, chunk_text, embedding <=> :query_vec AS score
97
  FROM knowledge_chunk
98
  ORDER BY embedding <=> :query_vec
99
  LIMIT :top_k
100
+ """
101
  )
102
 
103
+ result = await session.execute(
104
+ sql, {"query_vec": q_vector_str, "top_k": top_k}
105
+ )
106
+ rows = result.fetchall()
107
 
108
  return [
109
  SemanticSearchResult(
 
114
  )
115
  for r in rows
116
  ]
117
+