fixes

.secrets.baseline

@@ -133,7 +133,7 @@
         "filename": "autogen/agentchat/contrib/vectordb/pgvectordb.py",
         "hashed_secret": "5baa61e4c9b93f3f0682250b6cf8331b7ee68fd8",
         "is_verified": false,
-        "line_number": 580,
+        "line_number": 579,
         "is_secret": false
       }
     ],
@@ -1596,5 +1596,5 @@
       }
     ]
   },
-  "generated_at": "2025-02-20T22:13:34Z"
+  "generated_at": "2025-02-21T07:43:00Z"
 }

autogen/agentchat/contrib/vectordb/mongodb.py

@@ -8,13 +8,12 @@
 from time import monotonic, sleep
 from typing import Any, Callable, Iterable, Literal, Mapping, Optional, Union
 
-import numpy as np
-
 from ....import_utils import optional_import_block, require_optional_import
 from .base import Document, ItemID, QueryResults, VectorDB
 from .utils import get_logger
 
 with optional_import_block():
+    import numpy as np
     from pymongo import MongoClient, UpdateOne, errors
     from pymongo.collection import Collection
     from pymongo.driver_info import DriverInfo
@@ -33,7 +32,7 @@ def with_id_rename(docs: Iterable) -> list[dict[str, Any]]:
     return [{**{k: v for k, v in d.items() if k != "_id"}, "id": d["_id"]} for d in docs]
 
 
-@require_optional_import(["pymongo", "sentence_transformers"], "retrievechat-mongodb")
+@require_optional_import(["pymongo", "sentence_transformers", "numpy"], "retrievechat-mongodb")
 class MongoDBAtlasVectorDB(VectorDB):
     """A Collection object for MongoDB."""
 

autogen/agentchat/contrib/vectordb/pgvectordb.py

@@ -9,13 +9,12 @@
 import urllib.parse
 from typing import Any, Callable, Optional, Union
 
-import numpy as np
-
 from ....import_utils import optional_import_block, require_optional_import
 from .base import Document, ItemID, QueryResults, VectorDB
 from .utils import get_logger
 
 with optional_import_block():
+    import numpy as np
     import pgvector  # noqa: F401
     import psycopg
     from pgvector.psycopg import register_vector
@@ -25,7 +24,7 @@
 logger = get_logger(__name__)
 
 
-@require_optional_import(["psycopg", "sentence_transformers"], "retrievechat-pgvector")
+@require_optional_import(["psycopg", "sentence_transformers", "numpy"], "retrievechat-pgvector")
 class Collection:
     """A Collection object for PGVector.
 

pyproject.toml

@@ -102,6 +102,7 @@ retrievechat-pgvector = [
 retrievechat-mongodb = [
     "pyautogen[retrievechat]",
     "pymongo>=4.0.0",
+    "numpy",
 ]
 
 retrievechat-qdrant = [