mirror of
https://github.com/wassname/stampy-chat.git
synced 2026-09-11 12:50:34 +08:00
Changed back to using ChatGPT by default
This commit is contained in:
+1
-1
@@ -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
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user