agenthink 0.1.19__py3-none-any.whl → 0.1.20__py3-none-any.whl

agenthink/connection.py

@@ -372,7 +372,7 @@ class DBConnector:
                 cursor.execute(sql, values)
                 database_connection.commit()
                 cursor.close()
-                logger.info("Data inserted into MySQL database '%s', table '%s'", db_name, table_name)
+                logger.debug("Data inserted into MySQL database '%s', table '%s'", db_name, table_name)
             except Exception as e:
                 logger.exception("Failed to insert data into MySQL database '%s', table '%s': %s", db_name, table_name, e)
         if database_type == "mssql":
@@ -396,6 +396,109 @@ class DBConnector:
             except Exception as e:
                 logger.exception("Insert failed for %s database '%s', table '%s'", database_type, db_name, table_name)
 
+    def get_data(self, db_name: str, table_name: str, num_rows: int = 5):
+        """
+        Retrieve a limited number of rows from a table for display.
+
+        Parameters
+        ----------
+        db_name : str
+            Name of the database connection.
+        table_name : str
+            Target table name (must be trusted).
+        num_rows : int, optional
+            Number of rows to retrieve (default is 5).
+
+        Returns
+        -------
+        list[tuple] or None
+            Retrieved rows if successful, otherwise None.
+        """
+        db = self.connection_object_dict[db_name]
+        db_type = db["database_type"]
+        conn = db["connection_object"]
+
+        cursor = conn.cursor()
+        try:
+            if db_type == "mysql":
+                sql = f"SELECT * FROM {table_name} LIMIT %s"
+                cursor.execute(sql, (num_rows,))
+            elif db_type == "mssql":
+                sql = f"SELECT TOP ({num_rows}) * FROM [{table_name}]"
+                cursor.execute(sql)
+            else:
+                return None
+
+            return cursor.fetchall()
+        except Exception:
+            logger.exception(
+                "Failed to retrieve data from %s database '%s', table '%s'",
+                db_type, db_name, table_name
+            )
+            return None
+        finally:
+            cursor.close()
+
+    def get_schema(self, db_name: str, table_name: str):
+        """
+        Retrieve column metadata for a table from a database.
+
+        Parameters
+        ----------
+        db_name : str
+            Name of the database connection.
+        table_name : str
+            Target table name (must be trusted).
+
+        Returns
+        -------
+        list[tuple] or None
+            List of (column_name, data_type, is_nullable) ordered by column position,
+            or None if the schema cannot be retrieved.
+
+        Notes
+        -----
+        - Uses INFORMATION_SCHEMA for portability.
+        - Supports MySQL and MSSQL.
+        - Errors are logged and not raised.
+        """
+
+        db = self.connection_object_dict[db_name]
+        db_type = db["database_type"]
+        conn = db["connection_object"]
+
+        cursor = conn.cursor()
+        try:
+            if db_type == "mysql":
+                sql = """
+                    SELECT COLUMN_NAME, DATA_TYPE, IS_NULLABLE
+                    FROM INFORMATION_SCHEMA.COLUMNS
+                    WHERE TABLE_SCHEMA = DATABASE()
+                    AND TABLE_NAME = %s
+                    ORDER BY ORDINAL_POSITION
+                    """
+                cursor.execute(sql, (table_name,))
+                result =  cursor.fetchall()
+            if db_type == "mssql":
+                sql = """
+                    SELECT COLUMN_NAME, DATA_TYPE, IS_NULLABLE
+                    FROM INFORMATION_SCHEMA.COLUMNS
+                    WHERE TABLE_NAME = ?
+                    ORDER BY ORDINAL_POSITION
+                    """
+                cursor.execute(sql, (table_name,))
+                result = cursor.fetchall()
+        except Exception:
+            logger.exception(
+                "Failed to retrieve schema from %s database '%s', table '%s'",
+                db_type, db_name, table_name
+            )
+            return None
+        finally:
+            cursor.close()
+        return result
+
+
     def _debugging_function(self):
             prefix = f"{self._workflow_id}/{self._user_id}/"
             all_blob_list = self.__container_client.list_blobs(name_starts_with=prefix)
@@ -412,4 +515,6 @@ class DBConnector:
             Number of datastores: {self.no_of_datastores}\n
             Current connection_object_dict: {self.connection_object_dict}\n
             """
-            return output_message
+            return output_message
+    
+    

{agenthink-0.1.19.dist-info → agenthink-0.1.20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agenthink
-Version: 0.1.19
+Version: 0.1.20
 Summary: A unified agent framework for connecting workflows, databases, and agents
 Author: Ritobroto
 Requires-Python: >=3.10

agenthink-0.1.20.dist-info/RECORD

@@ -0,0 +1,8 @@
+agenthink/__init__.py,sha256=6oUldrZgE76j8OhwsQgVY5vdaPTDFyChOljske-so8U,78
+agenthink/connection.py,sha256=inG3EtYsI5aYuzLaDhJ--wJE931j0SQrx9AhG5IJG84,21081
+agenthink/models.py,sha256=PhXEMmfU31MTt1IwXxXV9DVPl5PLroQFHjRgRDs8iTo,251
+agenthink/utils.py,sha256=r5o74RbenFhQ7E3N7naoLJ-fYEe9otz0nkcvwKHDTaU,911
+agenthink-0.1.20.dist-info/METADATA,sha256=0BFyylzRM0WiZWF4xpkvw38VMSskQLGQ29sM1OzjIKo,1722
+agenthink-0.1.20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agenthink-0.1.20.dist-info/top_level.txt,sha256=rYw4Lx2uqOzbGCSoJEaikme7vS9NvgbVMc26QUIZoZM,10
+agenthink-0.1.20.dist-info/RECORD,,

agenthink-0.1.19.dist-info/RECORD

@@ -1,8 +0,0 @@
-agenthink/__init__.py,sha256=6oUldrZgE76j8OhwsQgVY5vdaPTDFyChOljske-so8U,78
-agenthink/connection.py,sha256=ZWWXsn1RiylzgOyL_-TQEuYYxNEMpqqAF0_HFedBoH4,17648
-agenthink/models.py,sha256=PhXEMmfU31MTt1IwXxXV9DVPl5PLroQFHjRgRDs8iTo,251
-agenthink/utils.py,sha256=r5o74RbenFhQ7E3N7naoLJ-fYEe9otz0nkcvwKHDTaU,911
-agenthink-0.1.19.dist-info/METADATA,sha256=-DF_4vTTRb6FF1T7rEGFnSGDYYOWfAHe2r3pplPy7-o,1722
-agenthink-0.1.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agenthink-0.1.19.dist-info/top_level.txt,sha256=rYw4Lx2uqOzbGCSoJEaikme7vS9NvgbVMc26QUIZoZM,10
-agenthink-0.1.19.dist-info/RECORD,,