Handle google genai update to import

cpsievert · cpsievert · commit 33d6a3b5bbe2 · 2025-01-28T13:30:47.000-06:00
diff --git a/tests/pytest/test_chat.py b/tests/pytest/test_chat.py
@@ -206,7 +206,7 @@ def test_google_normalization():
     if sys.version_info < (3, 9):
         return
 
-    from google.generativeai import (  # pyright: ignore[reportMissingTypeStubs]
+    from google.generativeai.generative_models import (  # pyright: ignore[reportMissingTypeStubs]
         GenerativeModel,
     )
 
@@ -382,7 +382,7 @@ def test_as_google_message():
     if sys.version_info < (3, 9):
         return
 
-    from google.generativeai import (  # pyright: ignore[reportMissingTypeStubs]
+    from google.generativeai.generative_models import (  # pyright: ignore[reportMissingTypeStubs]
         GenerativeModel,
     )
 

Original file line number	Diff line number	Diff line change
`@@ -206,7 +206,7 @@ def test_google_normalization():`
`206`	`206`	`if sys.version_info < (3, 9):`
`207`	`207`	`return`
`208`	`208`
`209`		`- from google.generativeai import ( # pyright: ignore[reportMissingTypeStubs]`
	`209`	`+ from google.generativeai.generative_models import ( # pyright: ignore[reportMissingTypeStubs]`
`210`	`210`	`GenerativeModel,`
`211`	`211`	`)`
`212`	`212`
`@@ -382,7 +382,7 @@ def test_as_google_message():`
`382`	`382`	`if sys.version_info < (3, 9):`
`383`	`383`	`return`
`384`	`384`
`385`		`- from google.generativeai import ( # pyright: ignore[reportMissingTypeStubs]`
	`385`	`+ from google.generativeai.generative_models import ( # pyright: ignore[reportMissingTypeStubs]`
`386`	`386`	`GenerativeModel,`
`387`	`387`	`)`
`388`	`388`