From 57f370cde97fb3f2c836cbc0dc764be55a571d5a Mon Sep 17 00:00:00 2001 From: Eduard van Valkenburg Date: Thu, 29 Jun 2023 03:16:41 +0200 Subject: [PATCH] PowerBI Toolkit additional logs (#6881) Added some additional logs to better be able to troubleshoot and understand the performance of the call to PBI vs the rest of the work. --- langchain/tools/powerbi/tool.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/langchain/tools/powerbi/tool.py b/langchain/tools/powerbi/tool.py index 85b181f906e..2113618b3d4 100644 --- a/langchain/tools/powerbi/tool.py +++ b/langchain/tools/powerbi/tool.py @@ -1,5 +1,6 @@ """Tools for interacting with a Power BI dataset.""" import logging +from time import perf_counter from typing import Any, Dict, Optional, Tuple from pydantic import Field, validator @@ -93,8 +94,12 @@ class QueryPowerBITool(BaseTool): if query == "I cannot answer this": self.session_cache[tool_input] = query return self.session_cache[tool_input] - logger.info("Query: %s", query) + logger.info("PBI Query: %s", query) + start_time = perf_counter() pbi_result = self.powerbi.run(command=query) + end_time = perf_counter() + logger.debug("PBI Result: %s", pbi_result) + logger.debug(f"PBI Query duration: {end_time - start_time:0.6f}") result, error = self._parse_output(pbi_result) if error is not None and "TokenExpired" in error: self.session_cache[ @@ -142,8 +147,12 @@ class QueryPowerBITool(BaseTool): if query == "I cannot answer this": self.session_cache[tool_input] = query return self.session_cache[tool_input] - logger.info("Query: %s", query) + logger.info("PBI Query: %s", query) + start_time = perf_counter() pbi_result = await self.powerbi.arun(command=query) + end_time = perf_counter() + logger.debug("PBI Result: %s", pbi_result) + logger.debug(f"PBI Query duration: {end_time - start_time:0.6f}") result, error = self._parse_output(pbi_result) if error is not None and "TokenExpired" in error: self.session_cache[