diff --git a/integrations/google_ai/pyproject.toml b/integrations/google_ai/pyproject.toml index 1c9b1b4aa..33829abd0 100644 --- a/integrations/google_ai/pyproject.toml +++ b/integrations/google_ai/pyproject.toml @@ -24,7 +24,7 @@ classifiers = [ ] dependencies = [ "haystack-ai", - "google-generativeai>=0.3.1, <0.7.1" # https://github.com/deepset-ai/haystack-core-integrations/issues/855 + "google-generativeai>=0.3.1, <0.7.0" # https://github.com/deepset-ai/haystack-core-integrations/issues/855 ] [project.urls] diff --git a/integrations/google_ai/src/haystack_integrations/components/generators/google_ai/chat/gemini.py b/integrations/google_ai/src/haystack_integrations/components/generators/google_ai/chat/gemini.py index 5a6e0081f..d3a8299fd 100644 --- a/integrations/google_ai/src/haystack_integrations/components/generators/google_ai/chat/gemini.py +++ b/integrations/google_ai/src/haystack_integrations/components/generators/google_ai/chat/gemini.py @@ -95,7 +95,7 @@ def __init__( self, *, api_key: Secret = Secret.from_env_var("GOOGLE_API_KEY"), # noqa: B008 - model: str = "gemini-1.5-flash", + model: str = "gemini-pro-vision", generation_config: Optional[Union[GenerationConfig, Dict[str, Any]]] = None, safety_settings: Optional[Dict[HarmCategory, HarmBlockThreshold]] = None, tools: Optional[List[Tool]] = None,