graphiti/tests/cross_encoder
Daniel Chalef 689d669559
Gemini client improvements; Gemini reranker (#645)
* add support for Gemini 2.5 model thinking budget

* allow adding thinking config to support current and future gemini models

* merge

* improve client; add reranker

* refactor: change type hint for gemini_messages to Any for flexibility

* refactor: update GeminiRerankerClient to use direct relevance scoring and improve ranking logic. Add tests

* fix fixtures

---------

Co-authored-by: realugbun <github.disorder751@passmail.net>
2025-06-30 12:55:17 -07:00
..
test_bge_reranker_client.py chore: Set up cross encoder client (#201) 2024-10-24 11:36:10 -04:00
test_gemini_reranker_client.py Gemini client improvements; Gemini reranker (#645) 2025-06-30 12:55:17 -07:00