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

agenthink/connection.py

@@ -48,11 +48,6 @@ class DBConnector:
             f"EndpointSuffix=core.windows.net"
         )
 
-        logger.debug("DEBUG TEST: debug level log")
-        logger.info("INFO TEST: info level log")
-        logger.warning("WARNING TEST: warning level log")
-        logger.error("ERROR TEST: error level log")
-
 
         try:
             blob_service = BlobServiceClient.from_connection_string(connection_str)
@@ -372,7 +367,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 +391,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)
@@ -411,5 +509,8 @@ class DBConnector:
             Json data: {self.json_data}\n
             Number of datastores: {self.no_of_datastores}\n
             Current connection_object_dict: {self.connection_object_dict}\n
+            D365 connection: 
             """
-            return output_message
+            return output_message
+    
+    

agenthink/data_source.py

@@ -0,0 +1,24 @@
+# ...existing code...
+import json
+from fastapi import FastAPI
+from pydantic import BaseModel
+import mysql.connector
+from fastapi import APIRouter
+import mysql.connector
+from azure.identity import ClientSecretCredential
+from azure.keyvault.secrets import SecretClient
+from mysql.connector import pooling
+import json
+from azure.storage.blob import BlobServiceClient
+from azure.storage.blob import BlobPrefix
+import pyodbc
+import agenthink.utils as utils
+import os
+import json
+import dotenv
+import logging
+import re
+dotenv.load_dotenv()
+
+
+

agenthink/models.py

@@ -1,10 +1,19 @@
 from pydantic import BaseModel
 from typing import List, Optional
 
+# class ConnectionRequest(BaseModel):
+#     query: str
+#     user_id: str
+#     session_id: str
+#     workflow_id: str
+#     datastore: Optional[List] = []
+#     tools: Optional[List] = []
+
 class ConnectionRequest(BaseModel):
-    query: str
-    user_id: str
-    session_id: str
-    workflow_id: str
-    datastore: Optional[List] = []
-    tools: Optional[List] = []
+    query: Optional[str] = None
+    user_id: Optional[str] = None
+    session_id: Optional[str] = None
+    workflow_id: Optional[str] = None
+    datastore: Optional[List] = None
+    tools: Optional[List] = None
+    d365request: Optional[dict] = None

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

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

agenthink-0.1.21.dist-info/RECORD

@@ -0,0 +1,9 @@
+agenthink/__init__.py,sha256=6oUldrZgE76j8OhwsQgVY5vdaPTDFyChOljske-so8U,78
+agenthink/connection.py,sha256=5EIZkthtAf_bCICCcEHaLRxP_tEflpJNA1hPHMGDf44,20895
+agenthink/data_source.py,sha256=HDi-UDAphTCKiFqWEB_-MfHwXpJKqfn_i05-LdID3M0,561
+agenthink/models.py,sha256=O5lWUNP1soAxU1X0iuE_lxYoyuXiKI0LXI6S_6zmxiE,559
+agenthink/utils.py,sha256=r5o74RbenFhQ7E3N7naoLJ-fYEe9otz0nkcvwKHDTaU,911
+agenthink-0.1.21.dist-info/METADATA,sha256=KLsNFvne6GMA0zrxB77X7TRgsC02Gy26HRdaGPnKa70,1722
+agenthink-0.1.21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agenthink-0.1.21.dist-info/top_level.txt,sha256=rYw4Lx2uqOzbGCSoJEaikme7vS9NvgbVMc26QUIZoZM,10
+agenthink-0.1.21.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,,