diff --git a/api/chat.py b/api/chat.py index d0e692e..1bed490 100644 --- a/api/chat.py +++ b/api/chat.py @@ -10,7 +10,7 @@ import re # OpenAI models EMBEDDING_MODEL = "text-embedding-ada-002" COMPLETIONS_MODEL = "gpt-3.5-turbo" -COMPLETIONS_MODEL = "gpt-4" +# COMPLETIONS_MODEL = "gpt-4" STANDARD_K = 20 if COMPLETIONS_MODEL == 'gpt-4' else 10 diff --git a/api/get_blocks.py b/api/get_blocks.py index a54389a..a26f46a 100644 --- a/api/get_blocks.py +++ b/api/get_blocks.py @@ -45,7 +45,7 @@ def get_embedding(text: str) -> np.ndarray: # Get the k blocks most semantically similar to the query using Pinecone. -def get_top_k_blocks(index, user_query: str, k: int = 10) -> List[Block]: +def get_top_k_blocks(index, user_query: str, k: int = 20) -> List[Block]: # print time t = time.time()