testing with preview

deepset-ai · Aug 30, 2024 · 0f54da8 · 0f54da8
1 parent 95c091f
commit 0f54da8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...ons/google_vertex/src/haystack_integrations/components/generators/google_vertex/gemini.py b/...ons/google_vertex/src/haystack_integrations/components/generators/google_vertex/gemini.py
@@ -7,7 +7,7 @@
 from haystack.dataclasses import ByteStream, StreamingChunk
 from haystack.utils import deserialize_callable, serialize_callable
 from vertexai import init as vertexai_init
-from vertexai.generative_models import (
+from vertexai.preview.generative_models import (
     Content,
     GenerationConfig,
     GenerationResponse,

diff --git a/integrations/google_vertex/tests/test_gemini.py b/integrations/google_vertex/tests/test_gemini.py
@@ -1,7 +1,7 @@
 from unittest.mock import MagicMock, Mock, patch
 
 from haystack.dataclasses import StreamingChunk
-from vertexai.generative_models import (
+from vertexai.preview.generative_models import (
     FunctionDeclaration,
     GenerationConfig,
     HarmBlockThreshold,