pvanand commited on
Commit
3522c9e
·
verified ·
1 Parent(s): 6e10974

Update helper_functions_api.py

Browse files
Files changed (1) hide show
  1. helper_functions_api.py +2 -2
helper_functions_api.py CHANGED
@@ -250,9 +250,9 @@ def fetch_and_extract_content(
250
 
251
  @retry(tries=3, delay=0.25)
252
  def search_brave(query, num_results=5):
253
- cleaned_query = re.sub(r'[^a-zA-Z0-9]+', '', query)
254
  search_query = together_response(cleaned_query, model=llm_default_small, SysPrompt=SysPromptSearch, max_tokens = 25).strip()
255
- cleaned_search_query = re.sub(r'[^a-zA-Z0-9*]+', '', search_query)
256
  brave = Brave(BRAVE_API_KEY)
257
  search_results = brave.search(q=cleaned_search_query, count=num_results)
258
  return [url.__str__() for url in search_results.urls],cleaned_search_query
 
250
 
251
  @retry(tries=3, delay=0.25)
252
  def search_brave(query, num_results=5):
253
+ cleaned_query = query #re.sub(r'[^a-zA-Z0-9]+', '', query)
254
  search_query = together_response(cleaned_query, model=llm_default_small, SysPrompt=SysPromptSearch, max_tokens = 25).strip()
255
+ cleaned_search_query = search_query #re.sub(r'[^a-zA-Z0-9*]+', '', search_query)
256
  brave = Brave(BRAVE_API_KEY)
257
  search_results = brave.search(q=cleaned_search_query, count=num_results)
258
  return [url.__str__() for url in search_results.urls],cleaned_search_query