Ankerkraut commited on
Commit
7e7599b
·
1 Parent(s): bda6b57

init system message

Browse files
Files changed (1) hide show
  1. app.py +11 -11
app.py CHANGED
@@ -121,12 +121,13 @@ def generate_response(query, context, prompts, max_tokens, temperature, top_p, g
121
  <|im_end|>
122
  <|im_start|>assistant"""
123
 
 
 
124
  if "rezept" in query.lower() or "gericht" in query.lower():
125
  system_message = system_message_recipes
126
  elif "bestellung" in query.lower() or "order" in query.lower():
127
  system_message = system_message_support
128
- else:
129
- system_message = system_message_products
130
 
131
  print("Prompt: ", system_message)
132
 
@@ -164,15 +165,14 @@ def respond(
164
  top_p,
165
  ):
166
  generator = get_model()
167
- if system_message == None:
168
- system_message = f"""<|im_start|>system Rolle: Du bist ein KI-Assistent der die vom Kunden formuliert Frage in Stichworte verwandelt die für eine Vektorsuche verwendet werden.
169
- Oberstes Ziel: Suche Schlüsselbegriffe aus der Frage heraus und gebe diese als Hauptbegriff aus. Suche zusätzlich ähnliche Begriffe aus.
170
- Meta-Anweisung: Wenn nach Produkten beziehungsweise Gewürzen gefragt wird, suche ähnliche Eigenschaften. Wenn nach einem Rezept gefragt ist, versuche die Küche beziehungsweise regionale Abstammung herauszufinden und als Schlüsselbegriff ausgeben. Gebe die vermutete Abstammung wie folgt aus: "Küche: ''". Du bekommst maximal 5 vorherige Fragen und Antworten aus dem Gespräch als Kontext. Wenn du keine exakten antworten geben kannst, geb nur Schlüsselbegriffe aus der Frage und den vorherigen wieder. Antworte in maximal 3 Stichpunkten und gebe keine Beschreibung.
171
- <|im_end|>
172
- <|im_start|>user
173
- Frage: {query}
174
- <|im_end|>
175
- <|im_start|>assistant"""
176
  refined_context = generator[1](system_message, do_sample=True, padding=True, truncation=True, top_p=0.95, max_new_tokens=150)
177
  # Retrieve relevant context from Qdrant
178
  if "rezept" in query.lower() or "gericht" in query.lower():
 
121
  <|im_end|>
122
  <|im_start|>assistant"""
123
 
124
+ system_message = system_message_products
125
+
126
  if "rezept" in query.lower() or "gericht" in query.lower():
127
  system_message = system_message_recipes
128
  elif "bestellung" in query.lower() or "order" in query.lower():
129
  system_message = system_message_support
130
+
 
131
 
132
  print("Prompt: ", system_message)
133
 
 
165
  top_p,
166
  ):
167
  generator = get_model()
168
+ system_message = f"""<|im_start|>system Rolle: Du bist ein KI-Assistent der die vom Kunden formuliert Frage in Stichworte verwandelt die für eine Vektorsuche verwendet werden.
169
+ Oberstes Ziel: Suche Schlüsselbegriffe aus der Frage heraus und gebe diese als Hauptbegriff aus. Suche zusätzlich ähnliche Begriffe aus.
170
+ Meta-Anweisung: Wenn nach Produkten beziehungsweise Gewürzen gefragt wird, suche ähnliche Eigenschaften. Wenn nach einem Rezept gefragt ist, versuche die Küche beziehungsweise regionale Abstammung herauszufinden und als Schlüsselbegriff ausgeben. Gebe die vermutete Abstammung wie folgt aus: "Küche: ''". Du bekommst maximal 5 vorherige Fragen und Antworten aus dem Gespräch als Kontext. Wenn du keine exakten antworten geben kannst, geb nur Schlüsselbegriffe aus der Frage und den vorherigen wieder. Antworte in maximal 3 Stichpunkten und gebe keine Beschreibung.
171
+ <|im_end|>
172
+ <|im_start|>user
173
+ Frage: {query}
174
+ <|im_end|>
175
+ <|im_start|>assistant"""
 
176
  refined_context = generator[1](system_message, do_sample=True, padding=True, truncation=True, top_p=0.95, max_new_tokens=150)
177
  # Retrieve relevant context from Qdrant
178
  if "rezept" in query.lower() or "gericht" in query.lower():