MindsDB 25.5.4.1__py3-none-any.whl → 25.6.2.0__py3-none-any.whl

mindsdb/interfaces/database/log.py

@@ -35,7 +35,7 @@ class LogTable(ABC):
     name: str
     deletable: bool = False
     visible: bool = True
-    kind: str = 'table'
+    kind: str = "table"
 
     @staticmethod
     @abstractmethod
@@ -59,142 +59,113 @@ class LogTable(ABC):
             BinaryOperation: statement that can be used for 'safe' comparison
         """
         return BinaryOperation(
-            op='=',
+            op="=",
             args=(
-                Function(
-                    op='coalesce',
-                    args=(
-                        Identifier(f'{table_a}.company_id'),
-                        0
-                    )
-                ),
-                Function(
-                    op='coalesce',
-                    args=(
-                        Identifier(f'{table_b}.company_id'),
-                        0
-                    )
-                )
-            )
+                Function(op="coalesce", args=(Identifier(f"{table_a}.company_id"), 0)),
+                Function(op="coalesce", args=(Identifier(f"{table_b}.company_id"), 0)),
+            ),
         )
 
 
 class LLMLogTable(LogTable):
-    name = 'llm_log'
+    name = "llm_log"
 
     columns = [
-        'API_KEY', 'MODEL_NAME', 'INPUT', 'OUTPUT', 'START_TIME', 'END_TIME',
-        'PROMPT_TOKENS', 'COMPLETION_TOKENS', 'TOTAL_TOKENS', 'SUCCESS'
+        "API_KEY",
+        "MODEL_NAME",
+        "INPUT",
+        "OUTPUT",
+        "START_TIME",
+        "END_TIME",
+        "PROMPT_TOKENS",
+        "COMPLETION_TOKENS",
+        "TOTAL_TOKENS",
+        "SUCCESS",
     ]
 
-    types_map = {
-        'SUCCESS': 'boolean',
-        'START_TIME': 'datetime64[ns]',
-        'END_TIME': 'datetime64[ns]'
-    }
+    types_map = {"SUCCESS": "boolean", "START_TIME": "datetime64[ns]", "END_TIME": "datetime64[ns]"}
 
     @staticmethod
     def _get_base_subquery() -> Select:
         query = Select(
             targets=[
-                Identifier('llm_log.api_key', alias=Identifier('api_key')),
-                Identifier('predictor.name', alias=Identifier('model_name')),
-                Identifier('llm_log.input', alias=Identifier('input')),
-                Identifier('llm_log.output', alias=Identifier('output')),
-                Identifier('llm_log.start_time', alias=Identifier('start_time')),
-                Identifier('llm_log.end_time', alias=Identifier('end_time')),
-                Identifier('llm_log.prompt_tokens', alias=Identifier('prompt_tokens')),
-                Identifier('llm_log.completion_tokens', alias=Identifier('completion_tokens')),
-                Identifier('llm_log.total_tokens', alias=Identifier('total_tokens')),
-                Identifier('llm_log.success', alias=Identifier('success'))
+                Identifier("llm_log.api_key", alias=Identifier("api_key")),
+                Identifier("predictor.name", alias=Identifier("model_name")),
+                Identifier("llm_log.input", alias=Identifier("input")),
+                Identifier("llm_log.output", alias=Identifier("output")),
+                Identifier("llm_log.start_time", alias=Identifier("start_time")),
+                Identifier("llm_log.end_time", alias=Identifier("end_time")),
+                Identifier("llm_log.prompt_tokens", alias=Identifier("prompt_tokens")),
+                Identifier("llm_log.completion_tokens", alias=Identifier("completion_tokens")),
+                Identifier("llm_log.total_tokens", alias=Identifier("total_tokens")),
+                Identifier("llm_log.success", alias=Identifier("success")),
             ],
             from_table=Join(
-                left=Identifier('llm_log'),
-                right=Identifier('predictor'),
+                left=Identifier("llm_log"),
+                right=Identifier("predictor"),
                 join_type=JoinType.LEFT_JOIN,
                 condition=BinaryOperation(
-                    op='and',
+                    op="and",
                     args=(
-                        LLMLogTable.company_id_comparison('llm_log', 'predictor'),
-                        BinaryOperation(
-                            op='=',
-                            args=(
-                                Identifier('llm_log.model_id'),
-                                Identifier('predictor.id')
-                            )
-                        )
-                    )
-                )
+                        LLMLogTable.company_id_comparison("llm_log", "predictor"),
+                        BinaryOperation(op="=", args=(Identifier("llm_log.model_id"), Identifier("predictor.id"))),
+                    ),
+                ),
             ),
             where=BinaryOperation(
-                op='is' if ctx.company_id is None else '=',
-                args=(Identifier('llm_log.company_id'), Constant(ctx.company_id))
+                op="is" if ctx.company_id is None else "=",
+                args=(Identifier("llm_log.company_id"), Constant(ctx.company_id)),
             ),
-            alias=Identifier('llm_log')
+            alias=Identifier("llm_log"),
         )
         return query
 
 
 class JobsHistoryTable(LogTable):
-    name = 'jobs_history'
+    name = "jobs_history"
 
-    columns = ['NAME', 'PROJECT', 'RUN_START', 'RUN_END', 'ERROR', 'QUERY']
-    types_map = {
-        'RUN_START': 'datetime64[ns]',
-        'RUN_END': 'datetime64[ns]'
-    }
+    columns = ["NAME", "PROJECT", "RUN_START", "RUN_END", "ERROR", "QUERY"]
+    types_map = {"RUN_START": "datetime64[ns]", "RUN_END": "datetime64[ns]"}
 
     @staticmethod
     def _get_base_subquery() -> Select:
         query = Select(
             targets=[
-                Identifier('jobs.name', alias=Identifier('name')),
-                Identifier('project.name', alias=Identifier('project')),
-                Identifier('jobs_history.start_at', alias=Identifier('run_start')),
-                Identifier('jobs_history.end_at', alias=Identifier('run_end')),
-                Identifier('jobs_history.error', alias=Identifier('error')),
-                Identifier('jobs_history.query_str', alias=Identifier('query'))
+                Identifier("jobs.name", alias=Identifier("name")),
+                Identifier("project.name", alias=Identifier("project")),
+                Identifier("jobs_history.start_at", alias=Identifier("run_start")),
+                Identifier("jobs_history.end_at", alias=Identifier("run_end")),
+                Identifier("jobs_history.error", alias=Identifier("error")),
+                Identifier("jobs_history.query_str", alias=Identifier("query")),
             ],
             from_table=Join(
                 left=Join(
-                    left=Identifier('jobs_history'),
-                    right=Identifier('jobs'),
+                    left=Identifier("jobs_history"),
+                    right=Identifier("jobs"),
                     join_type=JoinType.LEFT_JOIN,
                     condition=BinaryOperation(
-                        op='and',
+                        op="and",
                         args=(
-                            LLMLogTable.company_id_comparison('jobs_history', 'jobs'),
-                            BinaryOperation(
-                                op='=',
-                                args=(
-                                    Identifier('jobs_history.job_id'),
-                                    Identifier('jobs.id')
-                                )
-                            )
-                        )
-                    )
+                            LLMLogTable.company_id_comparison("jobs_history", "jobs"),
+                            BinaryOperation(op="=", args=(Identifier("jobs_history.job_id"), Identifier("jobs.id"))),
+                        ),
+                    ),
                 ),
-                right=Identifier('project'),
+                right=Identifier("project"),
                 join_type=JoinType.LEFT_JOIN,
                 condition=BinaryOperation(
-                    op='and',
+                    op="and",
                     args=(
-                        LLMLogTable.company_id_comparison('project', 'jobs'),
-                        BinaryOperation(
-                            op='=',
-                            args=(
-                                Identifier('project.id'),
-                                Identifier('jobs.project_id')
-                            )
-                        )
-                    )
-                )
+                        LLMLogTable.company_id_comparison("project", "jobs"),
+                        BinaryOperation(op="=", args=(Identifier("project.id"), Identifier("jobs.project_id"))),
+                    ),
+                ),
             ),
             where=BinaryOperation(
-                op='is' if ctx.company_id is None else '=',
-                args=(Identifier('jobs_history.company_id'), Constant(ctx.company_id))
+                op="is" if ctx.company_id is None else "=",
+                args=(Identifier("jobs_history.company_id"), Constant(ctx.company_id)),
             ),
-            alias=Identifier('jobs_history')
+            alias=Identifier("jobs_history"),
         )
         return query
 
@@ -202,8 +173,8 @@ class JobsHistoryTable(LogTable):
 class LogDBController:
     def __init__(self):
         self._tables = OrderedDict()
-        self._tables['llm_log'] = LLMLogTable
-        self._tables['jobs_history'] = JobsHistoryTable
+        self._tables["llm_log"] = LLMLogTable
+        self._tables["jobs_history"] = JobsHistoryTable
 
     def get_list(self) -> List[LogTable]:
         return list(self._tables.values())
@@ -212,11 +183,14 @@ class LogDBController:
         try:
             return self._tables[name]
         except KeyError:
-            raise EntityNotExistsError(f'Table log.{name} does not exists')
+            raise EntityNotExistsError(f"Table log.{name} does not exists")
 
     def get_tables(self) -> OrderedDict:
         return self._tables
 
+    def get_tree_tables(self) -> OrderedDict:
+        return self._tables
+
     def get_tables_rows(self) -> List[TablesRow]:
         return [
             TablesRow(TABLE_TYPE=TABLES_ROW_TYPE.SYSTEM_VIEW, TABLE_NAME=table_name)
@@ -237,7 +211,7 @@ class LogDBController:
         if len(tables) != 1:
             raise Exception("Only one table may be in query to log database")
         table = tables[0]
-        if table[0] is not None and table[0].lower() != 'log':
+        if table[0] is not None and table[0].lower() != "log":
             raise Exception("This is not a query to the log database")
         if table[1].lower() not in self._tables.keys():
             raise Exception(f"There is no table '{table[1]}' in the log database")
@@ -273,7 +247,7 @@ class LogDBController:
 
         render_engine = db.engine.name
         if render_engine == "postgresql":
-            'postgres'
+            "postgres"
         render = SqlalchemyRender(render_engine)
         query_str = render.get_string(query, with_failback=False)
         df = pd.read_sql_query(query_str, db.engine)
@@ -285,12 +259,6 @@ class LogDBController:
                     df[df_column_name] = df[df_column_name].astype(column_type)
         # endregion
 
-        columns_info = [{
-            'name': k,
-            'type': v
-        } for k, v in df.dtypes.items()]
+        columns_info = [{"name": k, "type": v} for k, v in df.dtypes.items()]
 
-        return DataHubResponse(
-            data_frame=df,
-            columns=columns_info
-        )
+        return DataHubResponse(data_frame=df, columns=columns_info)