From a561404befae7ccc874a69c52fc0bbc578dc4b27 Mon Sep 17 00:00:00 2001 From: henri123lemoine Date: Wed, 5 Apr 2023 13:17:07 -0400 Subject: [PATCH] Changed back to using ChatGPT by default --- api/chat.py | 2 +- api/get_blocks.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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()