agenthink 0.1.16__py3-none-any.whl → 0.1.18__py3-none-any.whl

agenthink/connection.py

@@ -21,13 +21,17 @@ import re
 dotenv.load_dotenv()
 
 # Create and configure logger
-logging.basicConfig(
-    filename="datastore_library.log",
-    format='%(asctime)s %(levelname)s:%(name)s:%(message)s',
-    filemode='w'
-)
+# logging.basicConfig(
+#     filename="datastore_library.log",
+#     format='%(asctime)s %(levelname)s:%(name)s:%(message)s',
+#     filemode='w'
+# )
+# logger = logging.getLogger(__name__)
+# logger.setLevel(logging.DEBUG)
+
 logger = logging.getLogger(__name__)
-logger.setLevel(logging.DEBUG)
+logger.addHandler(logging.NullHandler())
+
 
 
 class DBConnector:
@@ -53,6 +57,12 @@ 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)
             self.__container_client = blob_service.get_container_client(self.__container_name)
@@ -242,18 +252,19 @@ class DBConnector:
             return None
 
         if database_type == "mysql":
-            try:
-                cursor = database_connection.cursor()
-                cursor.execute(query)
-                results = cursor.fetchall()
-                logger.info("Executed query on MySQL database '%s': %s", db_name, query)
-                return f"{results}"
-            except Exception as e:
-                logger.exception("Failed to execute query on MySQL database '%s': %s", db_name, e)
-                return None
-            finally:
-                if cursor:
-                    cursor.close()
+            # try:
+            cursor = database_connection.cursor()
+            cursor.execute(query)
+            results = cursor.fetchall()
+            logger.info("Executed query on MySQL database '%s': %s", db_name, query)
+            # return f"{results}"
+            return f"{cursor}, {results}"
+            # except Exception as e:
+            #     logger.exception("Failed to execute query on MySQL database '%s': %s", db_name, e)
+            #     return None
+            # finally:
+            #     if cursor:
+            #         cursor.close()
         elif database_type == "mssql":
             try:
                 cursor = database_connection.cursor()
@@ -273,7 +284,7 @@ class DBConnector:
             logger.error("Unsupported database type '%s' for database '%s'", database_type, db_name)
             return None
 
-    def __debugging_function(self):
+    def _debugging_function(self):
             prefix = f"{self._workflow_id}/{self._user_id}/"
             all_blob_list = self.__container_client.list_blobs(name_starts_with=prefix)
             output_message = f"""Debugging function executed. Current connections:\n

{agenthink-0.1.16.dist-info → agenthink-0.1.18.dist-info}/METADATA

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

agenthink-0.1.18.dist-info/RECORD

@@ -0,0 +1,8 @@
+agenthink/__init__.py,sha256=6oUldrZgE76j8OhwsQgVY5vdaPTDFyChOljske-so8U,78
+agenthink/connection.py,sha256=ngnRP-5qrIssl8Xeib0aD9yHUuDiC0WWaQcKRax6mfc,13108
+agenthink/models.py,sha256=PhXEMmfU31MTt1IwXxXV9DVPl5PLroQFHjRgRDs8iTo,251
+agenthink/utils.py,sha256=r5o74RbenFhQ7E3N7naoLJ-fYEe9otz0nkcvwKHDTaU,911
+agenthink-0.1.18.dist-info/METADATA,sha256=Qj9Eg1TtdoE5JpU8zXgdFhnmy0caMtHq601rj1kXKJk,1722
+agenthink-0.1.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agenthink-0.1.18.dist-info/top_level.txt,sha256=rYw4Lx2uqOzbGCSoJEaikme7vS9NvgbVMc26QUIZoZM,10
+agenthink-0.1.18.dist-info/RECORD,,

agenthink-0.1.16.dist-info/RECORD

@@ -1,8 +0,0 @@
-agenthink/__init__.py,sha256=6oUldrZgE76j8OhwsQgVY5vdaPTDFyChOljske-so8U,78
-agenthink/connection.py,sha256=dqBbFuizCoIKDTD46F1Q6_80gjdewAyhadDs5BMOvAk,12753
-agenthink/models.py,sha256=PhXEMmfU31MTt1IwXxXV9DVPl5PLroQFHjRgRDs8iTo,251
-agenthink/utils.py,sha256=r5o74RbenFhQ7E3N7naoLJ-fYEe9otz0nkcvwKHDTaU,911
-agenthink-0.1.16.dist-info/METADATA,sha256=HOBaS-Julyhax98k0hjyVlXpapmiryAUZhOsYywLIHM,1722
-agenthink-0.1.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agenthink-0.1.16.dist-info/top_level.txt,sha256=rYw4Lx2uqOzbGCSoJEaikme7vS9NvgbVMc26QUIZoZM,10
-agenthink-0.1.16.dist-info/RECORD,,