This is an automated email from the ASF dual-hosted git repository.
jin pushed a change to branch graphspace
in repository https://gitbox.apache.org/repos/asf/incubator-hugegraph-ai.git
from c399b3b feat(llm): support record QPS by decorator (#76)
add b7d9d22 feat(llm): add reranker
add e692cf4 fix graph context_head
add 204bea2 fix ci
add 42bc9d6 Add near neighbor first and custom related information
add 28e6ab2 Merge branch 'main' of
https://github.com/apache/incubator-hugegraph-ai into rerank
add 029f4bf tiny fix
add 28fed50 fix format
add 46c072f fix api
add fd9d83d Merge branch 'apache:main' into rerank
add 2f75822 reserve priority
add 86090e5 fix prompt
add de55754 Merge branch 'main' into rerank
add f14cbb0 fix ui
add e31d788 Merge branch 'main' into pr/73
add cefe11a answer button primary
add 3100c2b auto switch to bleu
add 611eb5e tiny format
No new revisions were added by this update.
Summary of changes:
.../src/hugegraph_llm/api/models/rag_requests.py | 7 +
hugegraph-llm/src/hugegraph_llm/api/rag_api.py | 30 ++-
hugegraph-llm/src/hugegraph_llm/config/config.py | 17 +-
.../src/hugegraph_llm/demo/rag_web_demo.py | 201 +++++++++++++++++----
.../{ => models/rerankers}/__init__.py | 0
.../src/hugegraph_llm/models/rerankers/cohere.py | 58 ++++++
.../rerankers/init_reranker.py} | 24 ++-
.../hugegraph_llm/models/rerankers/siliconflow.py | 58 ++++++
.../operators/common_op/merge_dedup_rerank.py | 127 ++++++++++---
.../src/hugegraph_llm/operators/graph_rag_task.py | 41 +++--
.../operators/hugegraph_op/graph_rag_query.py | 67 ++++---
.../operators/llm_op/answer_synthesize.py | 12 +-
12 files changed, 517 insertions(+), 125 deletions(-)
copy hugegraph-llm/src/hugegraph_llm/{ => models/rerankers}/__init__.py (100%)
create mode 100644 hugegraph-llm/src/hugegraph_llm/models/rerankers/cohere.py
copy hugegraph-llm/src/hugegraph_llm/{config/generate.py =>
models/rerankers/init_reranker.py} (55%)
create mode 100644
hugegraph-llm/src/hugegraph_llm/models/rerankers/siliconflow.py
