Spaces:
Running
Running
added StdOutCallBackhandler
Browse files- 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([
|
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:
|