simcop2387/text-generation-webui-mirror
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-06-07 06:06:20 -04:00
Code Issues Projects Releases Packages Wiki Activity

Remove quotes from LLM-generated websearch query (closes #7045).

Browse source 
Fix by @Quiet-Joker
This commit is contained in:
oobabooga 2025-06-05 06:56:24 -07:00
parent 977ec801b7
commit d47c8eb956
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
modules/chat.py
View file

@ -604,7 +604,12 @@ def generate_search_query(user_message, state):
query = "" query = ""
for reply in generate_reply(formatted_prompt, search_state, stopping_strings=[], is_chat=True): for reply in generate_reply(formatted_prompt, search_state, stopping_strings=[], is_chat=True):
query = reply.strip() query = reply
# Strip and remove surrounding quotes if present
query = query.strip()
if len(query) >= 2 and query.startswith('"') and query.endswith('"'):
query = query[1:-1]
return query return query