fix

Author: hassiebp

langfuse/_utils/parse_error.py

@@ -1,6 +1,8 @@
 import logging
 from typing import Union
 
+# our own api errors
+from langfuse._utils.request import APIError, APIErrors
 from langfuse.api.core import ApiError
 
 # fern api errors
@@ -13,9 +15,6 @@
 )
 from langfuse.api.resources.health.errors import ServiceUnavailableError
 
-# our own api errors
-from langfuse.request import APIError, APIErrors
-
 SUPPORT_URL = "https://langfuse.com/support"
 API_DOCS_URL = "https://api.reference.langfuse.com"
 RBAC_DOCS_URL = "https://langfuse.com/docs/rbac"

langfuse/_utils/request.py

@@ -7,7 +7,7 @@
 
 import httpx
 
-from langfuse.serializer import EventSerializer
+from langfuse._utils.serializer import EventSerializer
 
 
 class LangfuseClient:

tests/load_test.py

@@ -1,9 +1,9 @@
 # create 5 different trace names
 from asyncio import gather
-from langfuse.client import Langfuse
-from langfuse.utils import _get_timestamp
-from tests.utils import create_uuid
 
+from langfuse import Langfuse
+from langfuse._utils import _get_timestamp
+from tests.utils import create_uuid
 
 trace_names = [create_uuid() for _ in range(5)]
 
@@ -18,16 +18,22 @@ async def execute():
     start = _get_timestamp()
 
     async def update_generation(i, langfuse: Langfuse):
-        trace = langfuse.trace(name=trace_names[i % 4], user_id=user_ids[i % 1999])
+        span = langfuse.start_span(name=trace_names[i % 4], user_id=user_ids[i % 1999])
         # random amount of generations, 1-10
         for _ in range(i % 10):
-            generation = trace.generation(name=generation_names[i % 19])
-            generation.update(metadata={"count": str(i)})
+            generation = span.start_generation(name=generation_names[i % 19])
+            generation.update(metadata={"count": str(i)}).end()
+
+        span.end()
 
     langfuse = Langfuse(debug=False, threads=100)
     print("start")
+
     await gather(*(update_generation(i, langfuse) for i in range(100_000)))
+
     print("flush")
+
     langfuse.flush()
     diff = _get_timestamp() - start
+
     print(diff)