RobertoBarrosoLuque commited on
Commit
15dacd4
·
1 Parent(s): 385bc37

Add stage 2 embeddings

Browse files
src/app.py CHANGED
@@ -98,10 +98,11 @@ def format_results(results: List[Dict], stage_name: str, metrics: Dict) -> str:
98
  </div>
99
  """
100
  )
101
- html_parts.append('</div>')
102
 
103
  return "".join(html_parts)
104
 
 
105
  def run_search_function_and_time(query: str, func: Callable):
106
  start = time.time()
107
  results = func(query)
 
98
  </div>
99
  """
100
  )
101
+ html_parts.append("</div>")
102
 
103
  return "".join(html_parts)
104
 
105
+
106
  def run_search_function_and_time(query: str, func: Callable):
107
  start = time.time()
108
  results = func(query)
src/constants/code_snippets.py CHANGED
@@ -117,4 +117,4 @@ final_results = [
117
  for r in rerank_response.json()["results"]
118
  ]
119
  ```
120
- """
 
117
  for r in rerank_response.json()["results"]
118
  ]
119
  ```
120
+ """
src/search/vector_search.py CHANGED
@@ -48,4 +48,4 @@ def search_vector(query: str, top_k: int = 10) -> List[Dict[str, any]]:
48
  "score": float(score),
49
  }
50
  for idx, score in zip(indices[0], similarity_scores)
51
- ]
 
48
  "score": float(score),
49
  }
50
  for idx, score in zip(indices[0], similarity_scores)
51
+ ]