z00mP commited on
Commit
9f3ddbf
1 Parent(s): 4f6fabd

fix topk type bug

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. backend/semantic_search.py +4 -4
app.py CHANGED
@@ -35,6 +35,7 @@ def add_text(history, text):
35
 
36
 
37
  def bot(history, api_kind, chunk_table, embedding_model, llm_model, eross_encoder, top_k_param):
 
38
  query = history[-1][0]
39
 
40
  if not query:
 
35
 
36
 
37
  def bot(history, api_kind, chunk_table, embedding_model, llm_model, eross_encoder, top_k_param):
38
+ top_k_param = int(top_k_param)
39
  query = history[-1][0]
40
 
41
  if not query:
backend/semantic_search.py CHANGED
@@ -15,10 +15,10 @@ retriever = SentenceTransformer(os.getenv("EMB_MODEL"))
15
 
16
 
17
  def retrieve(query, k, table_name, emb_name):
18
- print(table_name)
19
- print(emb_name)
20
- #TABLE = db.open_table(table_name)
21
- #retriever = SentenceTransformer(emb_name)
22
  query_vec = retriever.encode(query)
23
  try:
24
  documents = TABLE.search(query_vec, vector_column_name=VECTOR_COLUMN).limit(k).to_list()
 
15
 
16
 
17
  def retrieve(query, k, table_name, emb_name):
18
+ #print(table_name)
19
+ #print(emb_name)
20
+ TABLE = db.open_table(table_name)
21
+ retriever = SentenceTransformer(emb_name)
22
  query_vec = retriever.encode(query)
23
  try:
24
  documents = TABLE.search(query_vec, vector_column_name=VECTOR_COLUMN).limit(k).to_list()