fix class

holoviz · Oct 16, 2024 · d0efc40 · d0efc40
1 parent 4d47a0c
commit d0efc40
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lumen/ai/llm.py b/lumen/ai/llm.py
@@ -13,7 +13,7 @@
 from instructor.patch import Mode, patch
 from pydantic import BaseModel
 
-from .interceptor import OpenAIInterceptor
+from .interceptor import Interceptor
 
 
 class Llm(param.Parameterized):
@@ -24,7 +24,7 @@ class Llm(param.Parameterized):
 
     use_logfire = param.Boolean(default=False)
 
-    interceptor = param.ClassSelector(default=None, class_=OpenAIInterceptor)
+    interceptor = param.ClassSelector(default=None, class_=Interceptor)
 
     # Allows defining a dictionary of default models.
     model_kwargs = param.Dict(default={})