diff --git a/requirements.txt b/requirements.txt index 91666960..25f31a3a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -15,7 +15,7 @@ tiktoken ollama openai anthropic -google-generativeai +google-genai sqlmodel keybert GitPython diff --git a/src/llm/gemini_client.py b/src/llm/gemini_client.py index 0d566673..3130406d 100644 --- a/src/llm/gemini_client.py +++ b/src/llm/gemini_client.py @@ -1,5 +1,5 @@ -import google.generativeai as genai -from google.generativeai.types import HarmCategory, HarmBlockThreshold +import google.genai as genai +from google.genai.types import HarmCategory, HarmBlockThreshold from src.config import Config