Fix regression

agno-agi · Dec 11, 2024 · 5b07b70 · 5b07b70
1 parent d10bc9c
commit 5b07b70
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/cookbook/assistants/llms/vertexai/samples/multimodal.py b/cookbook/assistants/llms/vertexai/samples/multimodal.py
@@ -11,7 +11,7 @@ def multimodal_example(project: Optional[str], location: Optional[str]) -> str:
     # Load the model
     multimodal_model = GenerativeModel("gemini-1.0-pro-vision")
     # Query the model
-    response = multimodal_model.generate_media(
+    response = multimodal_model.generate_content(
         [
             # Add an example image
             Part.from_uri("gs://generativeai-downloads/images/scones.jpg", mime_type="image/jpeg"),

diff --git a/cookbook/assistants/llms/vertexai/samples/text_stream.py b/cookbook/assistants/llms/vertexai/samples/text_stream.py
@@ -11,7 +11,7 @@ def generate(project: Optional[str], location: Optional[str]) -> None:
     # Load the model
     model = GenerativeModel("gemini-1.0-pro-vision")
     # Query the model
-    responses: Iterable[GenerationResponse] = model.generate_media("Who are you?", stream=True)
+    responses: Iterable[GenerationResponse] = model.generate_content("Who are you?", stream=True)
     # Process the response
     for response in responses:
         print(response.text, end="")

diff --git a/pyproject.toml b/pyproject.toml
@@ -89,7 +89,6 @@ exclude = ["phienv*", "aienv*"]
 check_untyped_defs = true
 no_implicit_optional = true
 warn_unused_configs = true
-ignore_missing_imports = true
 plugins = ["pydantic.mypy"]
 exclude = ["phienv*", "aienv*", "scratch*", "wip*", "tmp*", "cookbook/assistants/examples/*", "phi/assistant/openai/*"]