highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-03-26 17:51:54 +01:00
Code Issues Packages Projects Releases Wiki Activity

Simplify query

Browse Source
This commit is contained in:
Weves 2023-11-01 22:45:08 -07:00
parent 54f5248eb1
commit 39341516ab
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/scripts/benchmark_search_isolated.py
View File

@ -5,8 +5,8 @@ import time
import nltk
from danswer.configs.app_configs import DOC_TIME_DECAY
from danswer.configs.app_configs import DOCUMENT_INDEX_NAME
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
# Download the wordlist
@ -133,7 +133,11 @@ def _measure_vespa_latency(filters: dict = {}):
# + '({grammar: "weakAnd"}userInput(@query) '
# + f'or ({{defaultIndex: "{CONTENT_SUMMARY}"}}userInput(@query)))'
# )
yql = VespaIndex.yql_base + '({grammar: "weakAnd"}userInput(@query))'
yql = (
f"select "
f"documentid, "
f"from {DOCUMENT_INDEX_NAME} where " + '({grammar: "weakAnd"}userInput(@query))'
)
query = generate_random_sentence()
query_embedding = embed_query(query)
num_to_retrieve = 50