File tree Expand file tree Collapse file tree 3 files changed +9
-5
lines changed
Expand file tree Collapse file tree 3 files changed +9
-5
lines changed Original file line number Diff line number Diff line change @@ -46,9 +46,13 @@ def __call__(
4646
4747
4848try :
49- from langfuse import get_client # pyright: ignore[reportPrivateImportUsage]
49+ from langfuse import Langfuse
5050
5151 LANGFUSE_AVAILABLE = True
52+
53+ def get_client ():
54+ """Compatibility shim for langfuse 2.x (returns Langfuse instance)."""
55+ return Langfuse ()
5256except ImportError :
5357 LANGFUSE_AVAILABLE = False
5458
Original file line number Diff line number Diff line change 1111from tests .chinook .dataset import collect_dataset
1212
1313try :
14- from langfuse import get_client , observe # pyright: ignore[reportPrivateImportUsage]
14+ from langfuse import Langfuse , observe
1515 from pydantic_ai .agent import Agent
1616 from pydantic_ai .models .openai import OpenAIChatModel
1717
1818 LANGFUSE_AVAILABLE = True
19- langfuse_client = get_client ()
19+ langfuse_client = Langfuse ()
2020
2121 Agent .instrument_all ()
2222
Original file line number Diff line number Diff line change 2424
2525# Langfuse client setup
2626try :
27- from langfuse import get_client # pyright: ignore[reportPrivateImportUsage]
27+ from langfuse import Langfuse
2828
2929 LANGFUSE_AVAILABLE = True
30- langfuse = get_client ()
30+ langfuse = Langfuse ()
3131except ImportError :
3232 LANGFUSE_AVAILABLE = False
3333 langfuse = None
You can’t perform that action at this time.
0 commit comments