Spaces:
Build error
Build error
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:
|