highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-04-07 11:28:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add embedding

Browse Source
This commit is contained in:
Weves 2023-10-31 15:10:45 -07:00
parent a9e0967771
commit 1a58ad4276
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/scripts/benchmark_search_isolated.py
View File

@ -1,8 +1,10 @@
import random
import time
from danswer.configs.app_configs import DOC_TIME_DECAY
from danswer.document_index.vespa.index import _query_vespa
from danswer.document_index.vespa.index import VespaIndex
from danswer.search.search_runner import embed_query
question_bank = [
"Who was the first president of the United States?",
@ -119,10 +121,13 @@ def _measure_vespa_latency(filters: dict = {}):
# )
yql = VespaIndex.yql_base + '({grammar: "weakAnd"}userInput(@query))'
query = random.choice(additional_questions)
query_embedding = embed_query(query)
num_to_retrieve = 50
params: dict[str, str | int] = {
"yql": yql,
"query": query,
"input.query(query_embedding)": str(query_embedding),
"input.query(decay_factor)": str(DOC_TIME_DECAY),
"hits": num_to_retrieve,
"offset": 0,
"ranking.profile": "hybrid_search",