fix: use match_phrase_prefix on full_name field for table suggestion (#…

…1433) * fix: use match_phrase_prefix on full_name field for table suggestion * add gpt-4-turbo contextsize
pinterest · Apr 11, 2024 · cd11be9 · cd11be9
1 parent 067d96d
commit cd11be9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/querybook/server/lib/ai_assistant/assistants/openai_assistant.py b/querybook/server/lib/ai_assistant/assistants/openai_assistant.py
@@ -13,6 +13,7 @@
     "gpt-3.5-turbo-16k": 16385,
     "gpt-4": 8192,
     "gpt-4-32k": 32768,
+    "gpt-4-turbo": 128000,
 }
 DEFAULT_MODEL_NAME = "gpt-3.5-turbo"
 

diff --git a/querybook/server/lib/elasticsearch/suggest_table.py b/querybook/server/lib/elasticsearch/suggest_table.py
@@ -8,9 +8,7 @@ def construct_suggest_table_query(
         "size": limit,
         "query": {
             "bool": {
-                "must": {
-                    "match": {"full_name_ngram": {"query": keyword, "operator": "and"}}
-                },
+                "must": [{"match_phrase_prefix": {"full_name": {"query": keyword}}}],
                 "filter": {"match": {"metastore_id": metastore_id}},
             }
         },