AstraDBChatMessageHistory#
- class langchain_astradb.chat_message_histories.AstraDBChatMessageHistory(
- *,
- session_id: str,
- collection_name: str = 'langchain_message_store',
- token: str | TokenProvider | None = None,
- api_endpoint: str | None = None,
- namespace: str | None = None,
- environment: str | None = None,
- setup_mode: SetupMode = SetupMode.SYNC,
- pre_delete_collection: bool = False,
- ext_callers: list[tuple[str | None, str | None] | str | None] | None = None,
- api_options: APIOptions | None = None,
Chat message history that stores history in Astra DB.
- Parameters:
session_id (str) – arbitrary key that is used to store the messages of a single chat session.
collection_name (str) – name of the Astra DB collection to create/use.
token (str | TokenProvider | None) – API token for Astra DB usage, either in the form of a string or a subclass of astrapy.authentication.TokenProvider. If not provided, the environment variable ASTRA_DB_APPLICATION_TOKEN is inspected.
api_endpoint (str | None) – full URL to the API endpoint, such as https://<DB-ID>-us-east1.apps.astra.datastax.com. If not provided, the environment variable ASTRA_DB_API_ENDPOINT is inspected.
namespace (str | None) – namespace (aka keyspace) where the collection is created. If not provided, the environment variable ASTRA_DB_KEYSPACE is inspected. Defaults to the database’s “default namespace”.
environment (str | None) – a string specifying the environment of the target Data API. If omitted, defaults to “prod” (Astra DB production). Other values are in astrapy.constants.Environment enum class.
setup_mode (SetupMode) – mode used to create the Astra DB collection (SYNC, ASYNC or OFF).
pre_delete_collection (bool) – whether to delete the collection before creating it. If False and the collection already exists, the collection will be used as is.
ext_callers (list[tuple[str | None, str | None] | str | None] | None) – one or more caller identities to identify Data API calls in the User-Agent header. This is a list of (name, version) pairs, or just strings if no version info is provided, which, if supplied, becomes the leading part of the User-Agent string in all API requests related to this component.
api_options (APIOptions | None) – an instance of
astrapy.utils.api_options.APIOptions
that can be supplied to customize the interaction with the Data API regarding serialization/deserialization, timeouts, custom headers and so on. The provided options are applied on top of settings already tailored to this library, and if specified will take precedence. Passing None (default) means no customization is requested. Refer to the astrapy documentation for details.
Attributes
messages
Retrieve all session messages from DB.
Methods
__init__
(*, session_id[, collection_name, ...])Chat message history that stores history in Astra DB.
aadd_messages
(messages)Async add a list of messages.
aclear
()Async remove all messages from the store.
add_ai_message
(message)Convenience method for adding an AI message string to the store.
add_message
(message)Add a Message object to the store.
add_messages
(messages)Add a list of messages.
add_user_message
(message)Convenience method for adding a human message string to the store.
Async version of getting messages.
clear
()Remove all messages from the store.
- __init__(
- *,
- session_id: str,
- collection_name: str = 'langchain_message_store',
- token: str | TokenProvider | None = None,
- api_endpoint: str | None = None,
- namespace: str | None = None,
- environment: str | None = None,
- setup_mode: SetupMode = SetupMode.SYNC,
- pre_delete_collection: bool = False,
- ext_callers: list[tuple[str | None, str | None] | str | None] | None = None,
- api_options: APIOptions | None = None,
Chat message history that stores history in Astra DB.
- Parameters:
session_id (str) – arbitrary key that is used to store the messages of a single chat session.
collection_name (str) – name of the Astra DB collection to create/use.
token (str | TokenProvider | None) – API token for Astra DB usage, either in the form of a string or a subclass of astrapy.authentication.TokenProvider. If not provided, the environment variable ASTRA_DB_APPLICATION_TOKEN is inspected.
api_endpoint (str | None) – full URL to the API endpoint, such as https://<DB-ID>-us-east1.apps.astra.datastax.com. If not provided, the environment variable ASTRA_DB_API_ENDPOINT is inspected.
namespace (str | None) – namespace (aka keyspace) where the collection is created. If not provided, the environment variable ASTRA_DB_KEYSPACE is inspected. Defaults to the database’s “default namespace”.
environment (str | None) – a string specifying the environment of the target Data API. If omitted, defaults to “prod” (Astra DB production). Other values are in astrapy.constants.Environment enum class.
setup_mode (SetupMode) – mode used to create the Astra DB collection (SYNC, ASYNC or OFF).
pre_delete_collection (bool) – whether to delete the collection before creating it. If False and the collection already exists, the collection will be used as is.
ext_callers (list[tuple[str | None, str | None] | str | None] | None) – one or more caller identities to identify Data API calls in the User-Agent header. This is a list of (name, version) pairs, or just strings if no version info is provided, which, if supplied, becomes the leading part of the User-Agent string in all API requests related to this component.
api_options (APIOptions | None) – an instance of
astrapy.utils.api_options.APIOptions
that can be supplied to customize the interaction with the Data API regarding serialization/deserialization, timeouts, custom headers and so on. The provided options are applied on top of settings already tailored to this library, and if specified will take precedence. Passing None (default) means no customization is requested. Refer to the astrapy documentation for details.
- Return type:
None
- async aadd_messages(
- messages: Sequence[BaseMessage],
Async add a list of messages.
- Parameters:
messages (Sequence[BaseMessage]) – A sequence of BaseMessage objects to store.
- Return type:
None
- add_ai_message(
- message: AIMessage | str,
Convenience method for adding an AI message string to the store.
Please note that this is a convenience method. Code should favor the bulk add_messages interface instead to save on round-trips to the underlying persistence layer.
This method may be deprecated in a future release.
- Parameters:
message (AIMessage | str) – The AI message to add.
- Return type:
None
- add_message(
- message: BaseMessage,
Add a Message object to the store.
- Parameters:
message (BaseMessage) – A BaseMessage object to store.
- Raises:
NotImplementedError – If the sub-class has not implemented an efficient add_messages method.
- Return type:
None
- add_messages(
- messages: Sequence[BaseMessage],
Add a list of messages.
Implementations should over-ride this method to handle bulk addition of messages in an efficient manner to avoid unnecessary round-trips to the underlying store.
- Parameters:
messages (Sequence[BaseMessage]) – A sequence of BaseMessage objects to store.
- Return type:
None
- add_user_message(
- message: HumanMessage | str,
Convenience method for adding a human message string to the store.
Please note that this is a convenience method. Code should favor the bulk add_messages interface instead to save on round-trips to the underlying persistence layer.
This method may be deprecated in a future release.
- Parameters:
message (HumanMessage | str) – The human message to add to the store.
- Return type:
None
- async aget_messages() list[BaseMessage] [source]#
Async version of getting messages.
Can over-ride this method to provide an efficient async implementation.
In general, fetching messages may involve IO to the underlying persistence layer.
- Return type:
list[BaseMessage]
Examples using AstraDBChatMessageHistory