Public Member Functions
	__init__ (self, str sql_database)

	run (self, List[str] queries)

	__str__ (self)

Public Attributes
	engine = create_engine(f"sqlite:///{os.path.abspath(sql_database)}")

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.engine = create_engine(f"sqlite:///{os.path.abspath(sql_database)}")
 

Member Function Documentation

◆ str()

ollama_manager.SQLQuery.__str__ ( self )

Definition at line 124 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 on the SQLite database and returns the results.
Args:
    queries (List[str]): A list of SQL query strings 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 on the SQLite database and returns the results.
        Args:
            queries (List[str]): A list of SQL query strings to be executed.
        Returns:
            dict: A dictionary containing the results of the executed queries and the original queries.
        """
        results = []
        for query in queries:
            try:
                # pandas kann direkt mit der SQLAlchemy-Engine arbeiten
                df = pd.read_sql(query, self.engine)
                result = df.to_json(orient="records")
                results.append(result)
            except Exception as e:
                logger.error(f"Error executing SQL query: {e}")
                return {"results": ["error"], "queries": queries}
 
        return {"results": results, "queries": queries}
 

Member Data Documentation

◆ engine

ollama_manager.SQLQuery.engine = create_engine(f"sqlite:///{os.path.abspath(sql_database)}")

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