Phoenix21 commited on
Commit
ae82839
·
verified ·
1 Parent(s): dea73d3

added StdOutCallBackhandler

Browse files
Files changed (1) hide show
  1. pipeline.py +3 -2
pipeline.py CHANGED
@@ -28,7 +28,8 @@ from langchain.prompts import PromptTemplate
28
  from langchain.docstore.document import Document
29
 
30
  from langchain.callbacks.base import BaseCallbacks # Updated import
31
- from langchain.callbacks.manager import CallbackManager
 
32
  # Custom chain imports
33
  from classification_chain import get_classification_chain
34
  from refusal_chain import get_refusal_chain
@@ -203,7 +204,7 @@ class NoOpCallbacks(BaseCallbacks):
203
  def on_chain_error(self, *args, **kwargs): pass
204
 
205
  # Create a callback manager with no-op callbacks
206
- callback_manager = CallbackManager([NoOpCallbacks()])
207
 
208
  # Initialize ChatGroq with the callback manager
209
  try:
 
28
  from langchain.docstore.document import Document
29
 
30
  from langchain.callbacks.base import BaseCallbacks # Updated import
31
+ # from langchain.callbacks.manager import CallbackManager
32
+ from langchain.callbacks import StdOutCallbackHandler
33
  # Custom chain imports
34
  from classification_chain import get_classification_chain
35
  from refusal_chain import get_refusal_chain
 
204
  def on_chain_error(self, *args, **kwargs): pass
205
 
206
  # Create a callback manager with no-op callbacks
207
+ callback_manager = CallbackManager([StdOutCallbackHandler()])
208
 
209
  # Initialize ChatGroq with the callback manager
210
  try: