mirror of
https://github.com/hwchase17/langchain.git
synced 2025-07-07 13:40:46 +00:00
parent
175a248506
commit
45d6de177e
@ -20,7 +20,6 @@ from langchain.chains import (
|
|||||||
from langchain.docstore import InMemoryDocstore, Wikipedia
|
from langchain.docstore import InMemoryDocstore, Wikipedia
|
||||||
from langchain.llms import Cohere, HuggingFaceHub, OpenAI
|
from langchain.llms import Cohere, HuggingFaceHub, OpenAI
|
||||||
from langchain.llms.huggingface_pipeline import HuggingFacePipeline
|
from langchain.llms.huggingface_pipeline import HuggingFacePipeline
|
||||||
from langchain.logger import BaseLogger, StdOutLogger
|
|
||||||
from langchain.prompts import (
|
from langchain.prompts import (
|
||||||
BasePromptTemplate,
|
BasePromptTemplate,
|
||||||
FewShotPromptTemplate,
|
FewShotPromptTemplate,
|
||||||
@ -31,7 +30,6 @@ from langchain.serpapi import SerpAPIChain, SerpAPIWrapper
|
|||||||
from langchain.sql_database import SQLDatabase
|
from langchain.sql_database import SQLDatabase
|
||||||
from langchain.vectorstores import FAISS, ElasticVectorSearch
|
from langchain.vectorstores import FAISS, ElasticVectorSearch
|
||||||
|
|
||||||
logger: BaseLogger = StdOutLogger()
|
|
||||||
verbose: bool = False
|
verbose: bool = False
|
||||||
llm_cache: Optional[BaseCache] = None
|
llm_cache: Optional[BaseCache] = None
|
||||||
set_default_callback_manager()
|
set_default_callback_manager()
|
||||||
|
@ -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)
|
|
Loading…
Reference in New Issue
Block a user