graphiti/graphiti_core/llm_client
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
..
__init__.py Add support for falkordb (#575) 2025-06-13 12:06:57 -04:00
anthropic_client.py Anthropic cleanup (#431) 2025-05-03 09:15:03 -04:00
azure_openai_client.py Azure OpenAI improvements and fixes; Improve Graphiti Azure OpenAI config (#620) 2025-06-25 14:48:12 -04:00
client.py small model fix (#432) 2025-05-02 10:08:25 -04:00
config.py small model fix (#432) 2025-05-02 10:08:25 -04:00
errors.py Anthropic client (#361) 2025-04-16 12:35:07 -07:00
gemini_client.py Gemini client improvements; Gemini reranker (#645) 2025-06-30 12:55:17 -07:00
groq_client.py small model fix (#432) 2025-05-02 10:08:25 -04:00
openai_base_client.py Azure OpenAI improvements and fixes; Improve Graphiti Azure OpenAI config (#620) 2025-06-25 14:48:12 -04:00
openai_client.py Azure OpenAI improvements and fixes; Improve Graphiti Azure OpenAI config (#620) 2025-06-25 14:48:12 -04:00
openai_generic_client.py small model fix (#432) 2025-05-02 10:08:25 -04:00
utils.py update new names with input_data (#204) 2024-10-29 11:03:31 -04:00