Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add VoyageAI Reranker module #809

Merged
merged 15 commits into from
Oct 8, 2024
Merged

Add VoyageAI Reranker module #809

Changes from 1 commit
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
4cdf1f9
Baseline
bwook00 Oct 6, 2024
8506c1c
Add test code
bwook00 Oct 6, 2024
276b080
Add mock api key at node test
bwook00 Oct 6, 2024
5c8bfeb
Add rst full.yaml support.py
bwook00 Oct 6, 2024
b06e285
delete batch parameters
bwook00 Oct 6, 2024
b533848
Add docs
bwook00 Oct 6, 2024
0e41efb
Add requirements.txt
bwook00 Oct 6, 2024
fbc2e25
Set api_key parameter
bwook00 Oct 6, 2024
ab10693
Merge branch 'refs/heads/main' into Feature/#793
bwook00 Oct 7, 2024
ad48a92
apply async
bwook00 Oct 7, 2024
7d992f6
apply voyage_client
bwook00 Oct 7, 2024
d808fe2
Add mock def
bwook00 Oct 7, 2024
1c94952
Merge branch 'main' into Feature/#793
bwook00 Oct 8, 2024
8b1fb6b
Merge branch 'main' into Feature/#793
bwook00 Oct 8, 2024
c407ef9
Merge branch 'main' into Feature/#793
vkehfdl1 Oct 8, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
apply voyage_client
  • Loading branch information
@bwook00
bwook00 committed Oct 7, 2024
commit 7d992f6d59c0ffd3cd4f75e7aa1136582ae5fe91
4 changes: 2 additions & 2 deletions autorag/nodes/passagereranker/voyageai.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@ def _pure(
"""
tasks = [
voyageai_rerank_pure(
self.client, query, contents, ids, top_k=top_k, model=model
self.voyage_client, model, query, contents, ids, top_k, truncation
)
for query, contents, ids in zip(queries, contents_list, ids_list)
]
Expand All @@ -73,7 +73,7 @@ def _pure(


async def voyageai_rerank_pure(
voyage_client: voyageai.Client,
voyage_client: voyageai.AsyncClient,
model: str,
query: str,
documents: List[str],
Expand Down
Loading