Spaces:
Running
Running
RobertoBarrosoLuque
commited on
Commit
·
15dacd4
1
Parent(s):
385bc37
Add stage 2 embeddings
Browse files- src/app.py +2 -1
- src/constants/code_snippets.py +1 -1
- src/search/vector_search.py +1 -1
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(
|
| 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 |
+
]
|