Fix cosine similarity threshold and prompt requirements

talkdai · May 23, 2024 · 833ae63 · 833ae63
1 parent ccbaeb0
commit 833ae63
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/sample_data/prompt.toml b/sample_data/prompt.toml
@@ -4,10 +4,20 @@ temperature = 0.1
 
 [prompt]
 prompt = """
-You are a nice bot, say something nice to the user and try to help him with his question, but also say to the user that you don't know totally about the content he asked for.
+Answer to any question that the user may have
+"""
+
+header = """
+You are a nice bot, say something nice to the user and try to help him with his question,
+but also say to the user that you don't know totally about the content he asked for.
+"""
+
+suggested = """
+Here are some contents that you need to use in order to help you with the user's question:
 """
 
 [fallback]
 prompt = """
-Answer the following message to the user, but also say that you are not certain about the content he asked for.
-"""
+Answer the following message to the user, but also say that you are not certain about
+the content he asked for.
+"""
diff --git a/src/dialog/settings.py b/src/dialog/settings.py
@@ -73,7 +73,7 @@ def LLM_MEMORY_SIZE(self):
 
     @property
     def COSINE_SIMILARITY_THRESHOLD(self):
-        return config.get("COSINE_SIMILARITY_THRESHOLD", default=0.2, cast=float)
+        return config.get("COSINE_SIMILARITY_THRESHOLD", default=0.5, cast=float)
 
     # Cors
     @property