Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
open-webui
Commits
a02139ba
"...composable_kernel.git" did not exist on "00c1016e4979b2c121bf76ff20d4c2b7c29845a1"
Unverified
Commit
a02139ba
authored
Jun 17, 2024
by
Que Nguyen
Committed by
GitHub
Jun 17, 2024
Browse files
Set filter_list as optional param in brave.py
parent
b3d136b3
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
5 deletions
+7
-5
backend/apps/rag/search/brave.py
backend/apps/rag/search/brave.py
+7
-5
No files found.
backend/apps/rag/search/brave.py
View file @
a02139ba
import
logging
import
logging
from
typing
import
List
from
typing
import
List
,
Optional
import
requests
import
requests
from
apps.rag.search.main
import
SearchResult
,
filter
_by_whitelist
from
apps.rag.search.main
import
SearchResult
,
get_
filter
ed_results
from
config
import
SRC_LOG_LEVELS
from
config
import
SRC_LOG_LEVELS
log
=
logging
.
getLogger
(
__name__
)
log
=
logging
.
getLogger
(
__name__
)
log
.
setLevel
(
SRC_LOG_LEVELS
[
"RAG"
])
log
.
setLevel
(
SRC_LOG_LEVELS
[
"RAG"
])
def
search_brave
(
api_key
:
str
,
query
:
str
,
whitelist
:
List
[
str
],
count
:
int
)
->
list
[
SearchResult
]:
def
search_brave
(
api_key
:
str
,
query
:
str
,
count
:
int
,
filter_list
:
Optional
[
List
[
str
]]
=
None
)
->
list
[
SearchResult
]:
"""Search using Brave's Search API and return the results as a list of SearchResult objects.
"""Search using Brave's Search API and return the results as a list of SearchResult objects.
Args:
Args:
...
@@ -29,10 +29,12 @@ def search_brave(api_key: str, query: str, whitelist:List[str], count: int) -> l
...
@@ -29,10 +29,12 @@ def search_brave(api_key: str, query: str, whitelist:List[str], count: int) -> l
json_response
=
response
.
json
()
json_response
=
response
.
json
()
results
=
json_response
.
get
(
"web"
,
{}).
get
(
"results"
,
[])
results
=
json_response
.
get
(
"web"
,
{}).
get
(
"results"
,
[])
filtered_results
=
filter_by_whitelist
(
results
,
whitelist
)
if
filter_list
:
results
=
get_filtered_results
(
results
,
filter_list
)
return
[
return
[
SearchResult
(
SearchResult
(
link
=
result
[
"url"
],
title
=
result
.
get
(
"title"
),
snippet
=
result
.
get
(
"snippet"
)
link
=
result
[
"url"
],
title
=
result
.
get
(
"title"
),
snippet
=
result
.
get
(
"snippet"
)
)
)
for
result
in
filtered_
results
[:
count
]
for
result
in
results
[:
count
]
]
]
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment