Public Member Functions
	__init__ (self, str sql_database)

	run (self, List[str] queries)

	__str__ (self)

Public Attributes
	connection = sqlite3.connect(f"file:{sql_database}?mode=ro", uri=True)

Detailed Description

A component to execute SQL queries on the SQLite database.

Attributes:
    connection (sqlite3.Connection): The connection to the SQLite database.

Methods:
    run(queries: List[str]) -> dict: Executes the provided SQL queries and returns the results.

Definition at line 88 of file ollama_manager.py.

Constructor & Destructor Documentation

◆ init()

ollama_manager.SQLQuery.__init__	(		self,
		str	sql_database )

Definition at line 99 of file ollama_manager.py.

    def __init__(self, sql_database: str):
        self.connection = sqlite3.connect(f"file:{sql_database}?mode=ro", uri=True)
 

Member Function Documentation

◆ str()

ollama_manager.SQLQuery.__str__ ( self )

Definition at line 125 of file ollama_manager.py.

    def __str__(self):
        return "<SQLQuery Object>"

◆ run()

ollama_manager.SQLQuery.run	(		self,
		List[str]	queries )

Executes the provided SQL queries and returns the results. Tests if the SQL queries are valid before execution.

Args:
    queries (List[str]): A list of SQL queries to be executed.

Returns:
    dict: A dictionary containing the results of the executed queries and the original queries.

Definition at line 103 of file ollama_manager.py.

    def run(self, queries: List[str]):
        """
        Executes the provided SQL queries and returns the results. Tests if the SQL queries are valid before execution.
 
        Args:
            queries (List[str]): A list of SQL queries to be executed.
 
        Returns:
            dict: A dictionary containing the results of the executed queries and the original queries.
        """
        results = []
        for query in queries:
            try:
                print(f"query: {query}")
                result = pd.read_sql(query, self.connection).to_json(orient="records")
                results.append(result)
            except ValueError as e:
                logger.error(f"Error parsing SQL query: {e}")
                return {"results": ["error"], "queries": queries}
 
        return {"results": results, "queries": queries}
 

Member Data Documentation

◆ connection

ollama_manager.SQLQuery.connection = sqlite3.connect(f"file:{sql_database}?mode=ro", uri=True)

Definition at line 100 of file ollama_manager.py.

The documentation for this class was generated from the following file:

app/ollama_manager.py

Public Member Functions

Public Attributes