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[