tangger/Search-R1
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add local sparse retriever, ann dense retriever and online search engine

Browse Source
This commit is contained in:
PeterGriffinJin
2025-04-07 18:20:43 +00:00
parent 0b26e614f7
commit ba152349fd
8 changed files with 470 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
example/retriever/retrieval_launch_google.sh Normal file
View File

@@ -0,0 +1,8 @@
api_key="" # put your google custom API key here (https://developers.google.com/custom-search/v1/overview)
cse_id="" # put your google cse API key here (https://developers.google.com/custom-search/v1/overview)
python search_r1/search/internal_google_server.py --api_key $api_key \
--topk 5 \
--cse_id $cse_id \
--snippet_only

7
example/retriever/retrieval_launch_serpapi.sh Normal file
View File

@@ -0,0 +1,7 @@
search_url=https://serpapi.com/search
serp_api_key="" # put your serp api key here (https://serpapi.com/)
python search_r1/search/online_search_server.py --search_url $search_url \
--topk 3 \
--serp_api_key $serp_api_key