|
65 | 65 | from rag_core_lib.impl.settings.ollama_llm_settings import OllamaSettings |
66 | 66 | from rag_core_lib.impl.settings.rag_class_types_settings import RAGClassTypeSettings |
67 | 67 | from rag_core_lib.impl.settings.stackit_vllm_settings import StackitVllmSettings |
68 | | -from rag_core_lib.impl.tracers.langfuse_traced_chain import LangfuseTracedGraph |
| 68 | +from rag_core_lib.impl.tracers.langfuse_traced_runnable import LangfuseTracedRunnable |
69 | 69 | from rag_core_lib.impl.utils.async_threadsafe_semaphore import AsyncThreadsafeSemaphore |
70 | 70 |
|
71 | 71 |
|
@@ -138,19 +138,19 @@ class DependencyContainer(DeclarativeContainer): |
138 | 138 | chunker=summary_text_splitter, |
139 | 139 | semaphore=Singleton(AsyncThreadsafeSemaphore, summarizer_settings.maximum_concurrreny), |
140 | 140 | ) |
| 141 | + traced_summarizer = Singleton( |
| 142 | + LangfuseTracedRunnable, |
| 143 | + inner_chain=summarizer, |
| 144 | + settings=langfuse_settings, |
| 145 | + ) |
141 | 146 |
|
142 | 147 | summary_enhancer = List( |
143 | | - Singleton(PageSummaryEnhancer, summarizer, chunker_settings), |
| 148 | + Singleton(PageSummaryEnhancer, traced_summarizer, chunker_settings), |
144 | 149 | ) |
145 | | - untraced_information_enhancer = Singleton( |
| 150 | + information_enhancer = Singleton( |
146 | 151 | GeneralEnhancer, |
147 | 152 | summary_enhancer, |
148 | 153 | ) |
149 | | - information_enhancer = Singleton( |
150 | | - LangfuseTracedGraph, |
151 | | - inner_chain=untraced_information_enhancer, |
152 | | - settings=langfuse_settings, |
153 | | - ) |
154 | 154 |
|
155 | 155 | document_deleter = Singleton( |
156 | 156 | DefaultDocumentDeleter, rag_api=rag_api, file_service=file_service, key_value_store=key_value_store |
|
0 commit comments