agenthink 0.1.26__py3-none-any.whl → 0.1.28__py3-none-any.whl

agenthink/connection.py

@@ -55,8 +55,7 @@ class DBConnector:
             blob_client = self.__container_client.get_blob_client(blob_path)
             data = blob_client.download_blob().readall()
         except Exception as e:
-            logger.exception("Failed to initialize BlobServiceClient or container client: %s", e)
-            container_client = None
+            logger.exception("Failed to read datastore blob. Datastore cannot be connected in local setup: %s", e)
 
         try:
             # Fallback: if 'data' not present, log and set json_data to empty list
@@ -367,6 +366,7 @@ class DBConnector:
         - Uses `INFORMATION_SCHEMA.TABLES` for MSSQL.
         - Errors are logged and not raised.
         """
+        print(f"{self.connection_object_dict}")
         database_dict = self.connection_object_dict[db_name]
         database_type = database_dict.get("database_type")
         database_connection = database_dict.get("connection_object")

agenthink/models.py

@@ -29,4 +29,4 @@ class ConnectionRequest(BaseModel):
     datastore: Optional[List[Dict[str, Any]]] = None
     tools: Optional[List[Dict[str, Any]]] = None
     d365request: Optional[List[Dict[str, Any]]] = None
-    context: Optional[Dict[str, str]] = {}
+    context: Optional[Any] = {}

agenthink-0.1.28.dist-info/METADATA

@@ -0,0 +1,31 @@
+Metadata-Version: 2.4
+Name: agenthink
+Version: 0.1.28
+Summary: A unified agent framework for connecting workflows, databases, and agents
+Author: Ritobroto
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: fastapi
+Requires-Dist: mysql-connector-python
+Requires-Dist: azure-identity
+Requires-Dist: azure-keyvault-secrets
+Requires-Dist: azure-storage-blob
+Requires-Dist: pyodbc
+Requires-Dist: python-dotenv
+
+DBConnector Library DocumentationThe DBConnector library is a robust Python-based utility designed to manage multi-database connections dynamically. It integrates with Azure Blob Storage for configuration management and Azure Key Vault for secure credential retrieval. The library supports MySQL, Microsoft SQL Server (MSSQL), and PostgreSQL.Table of ContentsCore ArchitectureInitialization & AuthenticationCore MethodsSecurity FeaturesUsage ExampleCore ArchitectureThe library follows a Singleton-like pattern using a class-level cache to ensure that only one instance of the connector exists per session, optimizing resource usage and connection pooling.Key Components:Azure Blob Storage: Used to store a JSON manifest (datastores_output.json) that defines which databases a specific user/workflow has access to.Azure Key Vault: Stores the sensitive credentials (host, port, username, password) for each database.Connection Dictionary: An internal mapping (connection_object_dict) that stores active connection objects keyed by the database name.Initialization & AuthenticationThe get MethodInstead of standard instantiation, use the @classmethod get to retrieve an instance.Pythonconnector = DBConnector.get(session_id="123", user_id="user_01", workflow_id="wf_abc")
+
+Authentication FlowIdentity: Uses ClientSecretCredential (Service Principal) to authenticate with Azure services.Manifest Fetching: Downloads the datastore list from Azure Blob Storage based on the workflow_id and user_id path.Secret Retrieval: Iterates through the manifest, cleanses the "key" to match Azure naming conventions, and fetches the JSON secret from the Key Vault.Automatic Connection: Based on the datastore_type, it automatically initializes the appropriate driver (mysql-connector, pyodbc, or psycopg2).Core Methods1. Database ExplorationMethodDescriptiondisplay_connections()Returns a string representation of all active DB connections.display_tables(db_name)Returns a list of all base tables in the specified database.get_schema(db_name, table_name)Returns column names, data types, and nullability for a specific table.2. Data Operationsexecute_query(db_name, query): Executes a read-only SQL statement. It returns the result set as a list of tuples.insert_data(db_name, table_name, data): Performs an INSERT operation.data: A dictionary where keys are column names and values are the data to insert.get_data(db_name, table_name, num_rows=5): A convenience method to quickly preview the first few rows of a table.Security FeaturesRead-Only EnforcementThe execute_query method contains a strict safeguard. It checks the beginning of every SQL string against a list of approved Read-Only Prefixes:SELECT, WITH, SHOW, DESCRIBE, DESC, EXPLAINIf a query starts with DELETE, DROP, or UPDATE, the method will return None and refuse execution.Secret SanitizationAzure Key Vault has strict naming requirements (alphanumerics and hyphens only). The library includes a __sanitize_secret_name helper that:Converts spaces and underscores to hyphens.Removes any non-compliant special characters.Usage ExamplePython# 1. Initialize the connector
+db_manager = DBConnector.get("session_45", "john_doe", "marketing_workflow")
+
+# 2. List available tables in the 'CustomerDB'
+tables = db_manager.display_tables("CustomerDB")
+print(f"Available tables: {tables}")
+
+# 3. Fetch data safely
+query = "SELECT name, email FROM Users WHERE status = 'active'"
+results = db_manager.execute_query("CustomerDB", query)
+
+# 4. Insert a log entry
+log_data = {"event": "login", "user_id": 101, "status": "success"}
+db_manager.insert_data("LogDB", "activity_logs", log_data)

agenthink-0.1.28.dist-info/RECORD

@@ -0,0 +1,9 @@
+agenthink/__init__.py,sha256=6oUldrZgE76j8OhwsQgVY5vdaPTDFyChOljske-so8U,78
+agenthink/connection.py,sha256=x5-2l_V7o9YJmvnpPVoKNtwjeUYWzApazjt_YNcD7Vk,25506
+agenthink/data_source.py,sha256=hSlpKcS_I6ws9vSKeMnt7EwSHcfO-fFKF8jfXNUE7DI,1196
+agenthink/models.py,sha256=yI3yMV3JTWYHi5c70oVQTr4uHLJTJmf7ep4PzC273xI,1024
+agenthink/utils.py,sha256=r5o74RbenFhQ7E3N7naoLJ-fYEe9otz0nkcvwKHDTaU,911
+agenthink-0.1.28.dist-info/METADATA,sha256=kSgKs5U-X9-pP2qhentlQWm-UBJRzkfS5DK1kVA3FlE,4091
+agenthink-0.1.28.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+agenthink-0.1.28.dist-info/top_level.txt,sha256=rYw4Lx2uqOzbGCSoJEaikme7vS9NvgbVMc26QUIZoZM,10
+agenthink-0.1.28.dist-info/RECORD,,

{agenthink-0.1.26.dist-info → agenthink-0.1.28.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

agenthink-0.1.26.dist-info/METADATA

@@ -1,49 +0,0 @@
-Metadata-Version: 2.4
-Name: agenthink
-Version: 0.1.26
-Summary: A unified agent framework for connecting workflows, databases, and agents
-Author: Ritobroto
-Requires-Python: >=3.10
-Description-Content-Type: text/markdown
-Requires-Dist: fastapi
-Requires-Dist: mysql-connector-python
-Requires-Dist: azure-identity
-Requires-Dist: azure-keyvault-secrets
-Requires-Dist: azure-storage-blob
-Requires-Dist: pyodbc
-Requires-Dist: python-dotenv
-
-# DBConnector (Agenthink)
-
-`DBConnector` is a database connectivity utility for the **Agenthink platform**.  
-It enables agents running on Agenthink to securely connect to external **MySQL** and **MSSQL** databases using credentials managed by **Azure Blob Storage** and **Azure Key Vault**.
-
-Connections are cached per session and are designed for **read-only access**, allowing agents to safely query data without modifying source databases.
-
-## Purpose
-This library allows **Agenthink agents** to:
-- Connect to user-registered databases on the Agenthink platform
-- Execute safe, read-only SQL queries
-- Reuse database connections across agent sessions
-
-## Features
-- Session-based connection caching for agents  
-- Supports **MySQL** and **MS SQL Server**  
-- Secure secret management via **Azure Key Vault**  
-- Datastore metadata loaded from **Azure Blob Storage**  
-- Read-only SQL enforcement (`SELECT`, `SHOW`, `EXPLAIN`, etc.)  
-- Built-in logging for observability and debugging  
-
-## Basic Usage (Agenthink Agent)
-
-```python
-db = DBConnector.get(
-    session_id="session_123",
-    user_id="user1",
-    workflow_id="workflow_A"
-)
-
-result = db.execute_query(
-    db_name="database_name",
-    query="SELECT * FROM item LIMIT 10"
-)

agenthink-0.1.26.dist-info/RECORD

@@ -1,9 +0,0 @@
-agenthink/__init__.py,sha256=6oUldrZgE76j8OhwsQgVY5vdaPTDFyChOljske-so8U,78
-agenthink/connection.py,sha256=sh4sbc7y-BVyCTMMjTSMMEJYeedB6_XBOKpdBb2jKEY,25477
-agenthink/data_source.py,sha256=hSlpKcS_I6ws9vSKeMnt7EwSHcfO-fFKF8jfXNUE7DI,1196
-agenthink/models.py,sha256=PumlTg-5IYkAqx1LJCfeJvHE8XeiOgw6kExSClhbGTU,1035
-agenthink/utils.py,sha256=r5o74RbenFhQ7E3N7naoLJ-fYEe9otz0nkcvwKHDTaU,911
-agenthink-0.1.26.dist-info/METADATA,sha256=8cWbzCDMRSJLHn8fwaAwuqbMpHD1I9OAvKFhV4UgXPc,1722
-agenthink-0.1.26.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agenthink-0.1.26.dist-info/top_level.txt,sha256=rYw4Lx2uqOzbGCSoJEaikme7vS9NvgbVMc26QUIZoZM,10
-agenthink-0.1.26.dist-info/RECORD,,