fix mypy stubs

cookbook/advanced_rag/image_search/01_download_images.py

@@ -24,7 +24,12 @@ def generate_and_download_image(prompt, filename):
  print(f"Generated image URL: {image_url}")
 
  # Download image
- image_response = httpx.get(image_url)
+ if image_url is not None:
+ image_response = httpx.get(image_url)
+ else:
+ # Handle the case where image_url is None
+ return "No image URL available"
+
  if image_response.status_code == 200:
  file_path = data_dir.joinpath(filename)
  with open(file_path, "wb") as file:

cookbook/advanced_rag/image_search/02_upsert_pinecone.py

@@ -2,12 +2,10 @@
 from pathlib import Path
 from typing import List, Tuple, Dict, Any
 
-import torch # type: ignore
-from PIL import Image # type: ignore
-from pinecone import Pinecone, ServerlessSpec, Index # type: ignore
-
-# Type ignore for third-party libraries without type stubs
-import clip # type: ignore
+import torch
+from PIL import Image
+from pinecone import Pinecone, ServerlessSpec, Index
+import clip
 
 # Load the CLIP model
 device = "cuda" if torch.cuda.is_available() else "cpu"

phi/document/reader/docx.py

@@ -4,7 +4,7 @@
 from phi.document.reader.base import Reader
 from phi.utils.log import logger
 import io
-from docx import Document as DocxDocument # type: ignore
+from docx import Document as DocxDocument
 
 
 class DocxReader(Reader):

pyproject.toml

@@ -96,9 +96,11 @@ module = [
  "botocore.*",
  "bs4.*",
  "chromadb.*",
+ "clip.*",
  "cohere.*",
  "crawl4ai.*",
  "docker.*",
+ "docx.*",
  "duckdb.*",
  "duckduckgo_search.*",
  "exa_py.*",
@@ -122,6 +124,7 @@ module = [
  "pandas.*",
  "pgvector.*",
  "pinecone.*",
+ "PIL.*",
  "psycopg.*",
  "psycopg2.*",
  "pyarrow.*",
@@ -139,6 +142,7 @@ module = [
  "streamlit.*",
  "tavily.*",
  "textract.*",
+ "torch.*",
  "timeout_decorator.*",
  "vertexai.*",
  "voyageai.*",