Adding a few pytest filterwarnings for clean test logs

.pre-commit-config.yaml

@@ -75,7 +75,7 @@ repos:
           - fhaviary>=0.6 # Match pyproject.toml
           - httpx
           - litellm>=1.42.1 # Match pyproject.toml
-          - numpy
+          - numpy>=1.20 # Match pyproject.toml
           - pydantic~=2.0 # Match pyproject.toml
           - tenacity
           - torch

ldp/graph/memory.py

@@ -2,10 +2,11 @@
 from abc import ABC, abstractmethod
 from collections.abc import AsyncIterator
 from contextlib import asynccontextmanager
-from typing import Any, ClassVar, Generic, TypeVar
+from typing import Any, ClassVar, Generic, TypeVar, cast
 from uuid import UUID
 
 import numpy as np
+import numpy.typing as npt
 from pydantic import (
     BaseModel,
     ConfigDict,
@@ -134,7 +135,10 @@ def __init__(self, **kwargs):
 
     async def _add_to_index(self, embedding: np.ndarray) -> int:
         async with self.safe_access_index() as index:
-            return int(index.add(len(self.memories), embedding))
+            added_value = cast(
+                npt.NDArray[np.int_], index.add(len(self.memories), embedding)
+            )
+            return added_value.item()
 
     async def _search_index(
         self, embedding: np.ndarray, matches: int = MemoryModel.DEFAULT_MEMORY_MATCHES

pyproject.toml

@@ -21,7 +21,7 @@ dependencies = [
     "httpx",
     "litellm",
     "networkx[default]~=3.0",  # Pin just to keep recent
-    "numpy",
+    "numpy>=1.20",  # For numpy.typing
     "openai>=1",
     "pydantic~=2.0",
     "tenacity",
@@ -228,6 +228,14 @@ min-similarity-lines = 12
 # Add the specified OPTS to the set of command line arguments as if they had been
 # specified by the user.
 addopts = "--doctest-modules"
+# Sets a list of filters and actions that should be taken for matched warnings.
+# By default all warnings emitted during the test session will be displayed in
+# a summary at the end of the test session.
+filterwarnings = [
+    "ignore:'imghdr' is deprecated and slated for removal in Python 3.13:DeprecationWarning",  # SEE: https://github.com/BerriAI/litellm/issues/2929#issuecomment-2345224633
+    "ignore:Support for class-based `config` is deprecated, use ConfigDict instead",  # SEE: https://github.com/BerriAI/litellm/issues/5648
+    'ignore:open_text is deprecated. Use files\(\) instead:DeprecationWarning',  # SEE: https://github.com/BerriAI/litellm/issues/5647
+]
 # List of directories that should be searched for tests when no specific directories,
 # files or test ids are given in the command line when executing pytest from the rootdir
 # directory. File system paths may use shell-style wildcards, including the recursive **