Spaces:
Running
Running
changes callback_manager to callbacks
Browse files- pipeline.py +1 -3
pipeline.py
CHANGED
@@ -192,8 +192,6 @@ CACHE_SIZE_LIMIT = 1000
|
|
192 |
# raise RuntimeError("ChatGroq initialization failed.") from e
|
193 |
|
194 |
|
195 |
-
from langchain_core.callbacks import CallbackManager
|
196 |
-
from langchain_core.callbacks.base import BaseCallbackHandler
|
197 |
|
198 |
ChatGroq.model_rebuild()
|
199 |
|
@@ -205,7 +203,7 @@ try:
|
|
205 |
temperature=0.7,
|
206 |
groq_api_key=fallback_groq_api_key,
|
207 |
max_tokens=2048,
|
208 |
-
|
209 |
)
|
210 |
|
211 |
|
|
|
192 |
# raise RuntimeError("ChatGroq initialization failed.") from e
|
193 |
|
194 |
|
|
|
|
|
195 |
|
196 |
ChatGroq.model_rebuild()
|
197 |
|
|
|
203 |
temperature=0.7,
|
204 |
groq_api_key=fallback_groq_api_key,
|
205 |
max_tokens=2048,
|
206 |
+
callbacks=[] # Add the callback manager here
|
207 |
)
|
208 |
|
209 |
|