database.db_movies

This file contains the Database class for IAI MovieBot.

The classes mentioned handle the functionality of query processing for IAI MovieBot. Currently, the database class is implemented for SQL DB.

Module Contents

Classes

DataBase

DataBase class for SQL databases.
class database.db_movies.DataBase(path: str)

DataBase class for SQL databases.

It provides the functionality to search the database according to user preferences.

get_sql_condition(dialogue_state: moviebot.dialogue_manager.dialogue_state.DialogueState, domain: moviebot.domain.movie_domain.MovieDomain) → str | None

Returns the condition for a SQL query based on dialogue state.

Parameters:

  • dialogue_state – Dialogue state.

  • domain – Domain to check specific parameters.

Returns:

SQL condition if there is any.

database_lookup(dialogue_state: moviebot.dialogue_manager.dialogue_state.DialogueState, domain: moviebot.domain.movie_domain.MovieDomain) → List[Dict[str, Any]]

Performs an SQL query to answer a user requirement.

Parameters:

  • dialogue_state – The current dialogue state.

  • domain – Domain to check specific parameters.

Returns:

The results of the SQL query.