Refactor langflow processing and langfuse callback initialization (#1413

)

Update version to 0.6.7a2 in pyproject.toml

This pull request includes the following changes:

- Refactoring of langflow processing and langfuse callback initialization

- Updating the version to 0.6.7a2 in pyproject.toml

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langflow"
-version = "0.6.7a1"
+version = "0.6.7a2"
 description = "A Python package with a built-in web application"
 authors = ["Logspace <[email protected]>"]
 maintainers = [

src/backend/langflow/processing/base.py

@@ -2,10 +2,13 @@
 
 from langchain.agents.agent import AgentExecutor
 from langchain.callbacks.base import BaseCallbackHandler
+from loguru import logger
+
 from langflow.api.v1.callback import AsyncStreamingLLMCallbackHandler, StreamingLLMCallbackHandler
 from langflow.processing.process import fix_memory_inputs, format_actions
 from langflow.services.deps import get_plugins_service
-from loguru import logger
+from langflow.processing.process import fix_memory_inputs, format_actions
+from langflow.services.deps import get_plugins_service
 
 if TYPE_CHECKING:
     from langfuse.callback import CallbackHandler  # type: ignore
@@ -28,13 +31,12 @@ def setup_callbacks(sync, trace_id, **kwargs):
 
 def get_langfuse_callback(trace_id):
     from langflow.services.deps import get_plugins_service
-    from langfuse.callback import CreateTrace
 
     logger.debug("Initializing langfuse callback")
     if langfuse := get_plugins_service().get("langfuse"):
         logger.debug("Langfuse credentials found")
         try:
-            trace = langfuse.trace(CreateTrace(name="langflow-" + trace_id, id=trace_id))
+            trace = langfuse.trace(name="langflow-" + trace_id, id=trace_id)
             return trace.getNewHandler()
         except Exception as exc:
             logger.error(f"Error initializing langfuse callback: {exc}")

src/backend/langflow/services/plugins/langfuse_plugin.py

@@ -64,14 +64,13 @@ def get(self):
     def get_callback(self, _id: Optional[str] = None):
         if _id is None:
             _id = "default"
-        from langfuse.callback import CreateTrace  # type: ignore
 
         logger.debug("Initializing langfuse callback")
 
         try:
             langfuse_instance = self.get()
             if langfuse_instance is not None and hasattr(langfuse_instance, "trace"):
-                trace = langfuse_instance.trace(CreateTrace(name="langflow-" + _id, id=_id))
+                trace = langfuse_instance.trace(name="langflow-" + _id, id=_id)
                 if trace:
                     return trace.getNewHandler()