From 45d6de177eaf1c4899be7fc434f97fddf231f54b Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Fri, 30 Dec 2022 13:57:47 -0500 Subject: [PATCH] remove logger (#491) remove old logging class (no longer used anyways) --- langchain/__init__.py | 2 -- langchain/logger.py | 71 ------------------------------------------- 2 files changed, 73 deletions(-) delete mode 100644 langchain/logger.py diff --git a/langchain/__init__.py b/langchain/__init__.py index 0a607890d6a..4a0c81555fd 100644 --- a/langchain/__init__.py +++ b/langchain/__init__.py @@ -20,7 +20,6 @@ from langchain.chains import ( from langchain.docstore import InMemoryDocstore, Wikipedia from langchain.llms import Cohere, HuggingFaceHub, OpenAI from langchain.llms.huggingface_pipeline import HuggingFacePipeline -from langchain.logger import BaseLogger, StdOutLogger from langchain.prompts import ( BasePromptTemplate, FewShotPromptTemplate, @@ -31,7 +30,6 @@ from langchain.serpapi import SerpAPIChain, SerpAPIWrapper from langchain.sql_database import SQLDatabase from langchain.vectorstores import FAISS, ElasticVectorSearch -logger: BaseLogger = StdOutLogger() verbose: bool = False llm_cache: Optional[BaseCache] = None set_default_callback_manager() diff --git a/langchain/logger.py b/langchain/logger.py deleted file mode 100644 index d264da9048b..00000000000 --- a/langchain/logger.py +++ /dev/null @@ -1,71 +0,0 @@ -"""BETA: everything in here is highly experimental, do not rely on.""" -from typing import Any, Optional - -from langchain.input import print_text -from langchain.schema import AgentAction, AgentFinish - - -class BaseLogger: - """Base logging interface.""" - - def log_agent_start(self, text: str, **kwargs: Any) -> None: - """Log the start of an agent interaction.""" - pass - - def log_agent_end(self, finish: AgentFinish, **kwargs: Any) -> None: - """Log the end of an agent interaction.""" - pass - - def log_agent_action(self, action: AgentAction, **kwargs: Any) -> None: - """Log agent action decision.""" - pass - - def log_agent_observation(self, observation: str, **kwargs: Any) -> None: - """Log agent observation.""" - pass - - def log_llm_inputs(self, inputs: dict, prompt: str, **kwargs: Any) -> None: - """Log LLM inputs.""" - pass - - def log_llm_response(self, output: str, **kwargs: Any) -> None: - """Log LLM response.""" - pass - - -class StdOutLogger(BaseLogger): - """Interface for printing things to stdout.""" - - def log_agent_start(self, text: str, **kwargs: Any) -> None: - """Print the text to start the agent.""" - print_text(text) - - def log_agent_action( - self, action: AgentAction, color: Optional[str] = None, **kwargs: Any - ) -> None: - """Print the log of the action in a certain color.""" - print_text(action.log, color=color) - - def log_agent_observation( - self, - observation: str, - color: Optional[str] = None, - observation_prefix: Optional[str] = None, - llm_prefix: Optional[str] = None, - **kwargs: Any, - ) -> None: - """Print the observation in a special color.""" - print_text(f"\n{observation_prefix}") - print_text(observation, color=color) - print_text(f"\n{llm_prefix}") - - def log_llm_inputs(self, inputs: dict, prompt: str, **kwargs: Any) -> None: - """Print the prompt in green.""" - print("Prompt after formatting:") - print_text(prompt, color="green", end="\n") - - def log_agent_end( - self, finish: AgentFinish, color: Optional[str] = None, **kwargs: Any - ) -> None: - """Log the end of an agent interaction.""" - print_text(finish.log, color=color)