updating the dependencies.

.pre-commit-config.yaml

@@ -49,6 +49,7 @@ repos:
             language_version: python3.11
           - id: poetry-export
             args: ["-f", "requirements.txt", "-o", "requirements.txt"]
+
 # -   repo: https://github.com/Lucas-C/pre-commit-hooks-safety
 #     rev: v1.3.1
 #     hooks:

agent/utils/vdb.py

@@ -35,9 +35,35 @@ def init_vdb(cfg: DictConfig, collection_name: str, embedding: Embeddings) -> Qd
 
 
 @load_config("config/main.yml")
-def load_vec_db_conn(cfg: DictConfig) -> QdrantClient:
-    """Load the Vector Database Connection."""
-    return QdrantClient(cfg.qdrant.url, port=cfg.qdrant.port, api_key=os.getenv("QDRANT_API_KEY"), prefer_grpc=cfg.qdrant.prefer_grpc), cfg
+def load_vec_db_conn(cfg: DictConfig) -> Tuple[QdrantClient, DictConfig]:
+    """
+    Load the Vector Database Connection.
+
+    This function creates a new QdrantClient instance using the configuration
+    parameters provided in the 'cfg' argument. The QdrantClient is used to
+    interact with the Qdrant vector database.
+
+    Args:
+    -----
+        cfg (DictConfig): A configuration object that contains the settings
+                          for the QdrantClient. It should have 'url', 'port',
+                          and 'prefer_grpc' fields under the 'qdrant' key.
+
+    Returns:
+    --------
+        Tuple[QdrantClient, DictConfig]: A tuple containing the created
+                                         QdrantClient instance and the
+                                         original configuration object.
+    """
+    return (
+        QdrantClient(
+            cfg.qdrant.url,
+            port=cfg.qdrant.port,
+            api_key=os.getenv("QDRANT_API_KEY"),
+            prefer_grpc=cfg.qdrant.prefer_grpc
+        ),
+        cfg
+    )
 
 
 def initialize_vector_db(collection_name: str, embeddings_size: int) -> None: