File size: 683 Bytes
b380004
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from agno.utils.log import log_info
from agno.vectordb.qdrant import Qdrant

from chattr.app.settings import VectorDatabaseSettings


def setup_vector_database(vectordb: VectorDatabaseSettings) -> Qdrant:
    """
    Initialize a vector database connection.

    Args:
        vectordb (VectorDatabaseSettings): The settings required for
                                           connecting to the vector database.

    Returns:
        Qdrant: An instance of Qdrant configured with the specified vector database
                settings.
    """
    log_info("Setting up vector database.")
    return Qdrant(collection=vectordb.name, url=vectordb.url.host, port=vectordb.url.port)