AFischer1985 commited on
Commit
c912628
·
verified ·
1 Parent(s): 972aa6d

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +2 -2
run.py CHANGED
@@ -172,7 +172,7 @@ def extend_prompt(message="", history=None, system=None, RAGAddon=None, system2=
172
  for user_message, bot_response in history[-historylimit:]:
173
  if user_message is None: user_message = ""
174
  if bot_response is None: bot_response = ""
175
- bot_response = re.sub("\n\n<details>.*?</details>","", bot_response, flags=re.DOTALL) # remove RAG-compontents
176
  if removeHTML==True: bot_response = re.sub("<(.*?)>","\n", bot_response) # remove HTML-components in general (may cause bugs with markdown-rendering)
177
  if user_message is not None: prompt += template1.format(message=user_message[:zeichenlimit])
178
  if bot_response is not None: prompt += template2.format(response=bot_response[:zeichenlimit])
@@ -205,7 +205,7 @@ def response(
205
  system="Du bist ein deutschsprachiges KI-basiertes Assistenzsystem, das zu jedem Anliegen möglichst geeignete KI-Tools empfiehlt." #+addon #+"\n\nUser-Anliegen:"
206
  #body={"prompt":system+"### Instruktion:\n"+message+"\n\n### Antwort:","max_tokens":500, "echo":"False","stream":"True"} #e.g. SauerkrautLM
207
  #formatted_prompt = extend_prompt(system+"\n"+prompt, None) #history)
208
- removeHTML=True
209
  prompt=extend_prompt(
210
  message, # current message of the user
211
  history, # complete history
 
172
  for user_message, bot_response in history[-historylimit:]:
173
  if user_message is None: user_message = ""
174
  if bot_response is None: bot_response = ""
175
+ bot_response = re.sub("\n\n<details(| open)>.*?</details>","", bot_response, flags=re.DOTALL) # remove RAG-compontents
176
  if removeHTML==True: bot_response = re.sub("<(.*?)>","\n", bot_response) # remove HTML-components in general (may cause bugs with markdown-rendering)
177
  if user_message is not None: prompt += template1.format(message=user_message[:zeichenlimit])
178
  if bot_response is not None: prompt += template2.format(response=bot_response[:zeichenlimit])
 
205
  system="Du bist ein deutschsprachiges KI-basiertes Assistenzsystem, das zu jedem Anliegen möglichst geeignete KI-Tools empfiehlt." #+addon #+"\n\nUser-Anliegen:"
206
  #body={"prompt":system+"### Instruktion:\n"+message+"\n\n### Antwort:","max_tokens":500, "echo":"False","stream":"True"} #e.g. SauerkrautLM
207
  #formatted_prompt = extend_prompt(system+"\n"+prompt, None) #history)
208
+ removeHTML=False
209
  prompt=extend_prompt(
210
  message, # current message of the user
211
  history, # complete history