Changed back to using ChatGPT by default

This commit is contained in:
henri123lemoine
2023-04-05 13:17:07 -04:00
parent bfe01889ab
commit a561404bef
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -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
+1 -1
View File
@@ -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()