haive.core.engine.document.loaders.specific.database¶

Database Loaders for Document Engine.

This module implements database loaders for MongoDB, PostgreSQL, and other databases adapted for the document engine framework.

Classes¶

MongoDBSource

MongoDB database source implementation.

PostgreSQLSource

PostgreSQL database source implementation.

Module Contents¶

class haive.core.engine.document.loaders.specific.database.MongoDBSource(connection_string, database_name=None, collection_name=None, filter_criteria=None, **kwargs)[source]¶

Bases: haive.core.engine.document.loaders.sources.implementation.DatabaseSource

MongoDB database source implementation.

Init .

Parameters:
  • connection_string (str) – [TODO: Add description]

  • database_name (str | None) – [TODO: Add description]

  • collection_name (str | None) – [TODO: Add description]

  • filter_criteria (dict[str, Any] | None) – [TODO: Add description]

can_handle(path)[source]¶

Check if this is a MongoDB connection string.

Parameters:

path (str)

Return type:

bool

create_loader()[source]¶

Create a MongoDB loader.

Return type:

langchain_core.document_loaders.base.BaseLoader | None

get_confidence_score(path)[source]¶

Get confidence score for MongoDB connections.

Parameters:

path (str)

Return type:

float

get_credential_requirements()[source]¶

MongoDB needs connection credentials.

Return type:

list[haive.core.engine.document.loaders.sources.implementation.CredentialType]

requires_authentication()[source]¶

MongoDB typically requires authentication.

Return type:

bool

class haive.core.engine.document.loaders.specific.database.PostgreSQLSource(connection_string, query=None, table_name=None, **kwargs)[source]¶

Bases: haive.core.engine.document.loaders.sources.implementation.DatabaseSource

PostgreSQL database source implementation.

Init .

Parameters:
  • connection_string (str) – [TODO: Add description]

  • query (str | None) – [TODO: Add description]

  • table_name (str | None) – [TODO: Add description]

can_handle(path)[source]¶

Check if this is a PostgreSQL connection string.

Parameters:

path (str)

Return type:

bool

create_loader()[source]¶

Create a PostgreSQL loader.

Return type:

langchain_core.document_loaders.base.BaseLoader | None

get_confidence_score(path)[source]¶

Get confidence score for PostgreSQL connections.

Parameters:

path (str)

Return type:

float

get_credential_requirements()[source]¶

PostgreSQL needs connection credentials.

Return type:

list[haive.core.engine.document.loaders.sources.implementation.CredentialType]

requires_authentication()[source]¶

PostgreSQL typically requires authentication.

Return type:

bool