agenthink 0.1.19__tar.gz → 0.1.21__tar.gz

{agenthink-0.1.19 → agenthink-0.1.21}/PKG-INFO

@@ -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.19 → agenthink-0.1.21}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agenthink"
-version = "0.1.19"
+version = "0.1.21"
 description = "A unified agent framework for connecting workflows, databases, and agents"
 authors = [
   { name = "Ritobroto" }

{agenthink-0.1.19 → agenthink-0.1.21}/src/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-0.1.21/src/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-0.1.21/src/agenthink/models.py

@@ -0,0 +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: 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 → agenthink-0.1.21}/src/agenthink.egg-info/PKG-INFO

@@ -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.19 → agenthink-0.1.21}/src/agenthink.egg-info/SOURCES.txt

@@ -2,6 +2,7 @@ README.md
 pyproject.toml
 src/agenthink/__init__.py
 src/agenthink/connection.py
+src/agenthink/data_source.py
 src/agenthink/models.py
 src/agenthink/utils.py
 src/agenthink.egg-info/PKG-INFO

agenthink-0.1.19/src/agenthink/models.py

@@ -1,10 +0,0 @@
-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] = []