feat: update pydantic models for tags and datasets (#286)

Author: marcklingen

langfuse/api/resources/commons/types/dataset.py

@@ -5,7 +5,6 @@
 
 from ....core.datetime_utils import serialize_datetime
 from .dataset_item import DatasetItem
-from .dataset_status import DatasetStatus
 
 try:
     import pydantic.v1 as pydantic  # type: ignore
@@ -16,7 +15,6 @@
 class Dataset(pydantic.BaseModel):
     id: str
     name: str
-    status: DatasetStatus
     project_id: str = pydantic.Field(alias="projectId")
     created_at: dt.datetime = pydantic.Field(alias="createdAt")
     updated_at: dt.datetime = pydantic.Field(alias="updatedAt")

langfuse/api/resources/trace/client.py

@@ -68,6 +68,7 @@ def list(
         limit: typing.Optional[int] = None,
         user_id: typing.Optional[str] = None,
         name: typing.Optional[str] = None,
+        tags: typing.Optional[typing.Union[str, typing.List[str]]] = None,
     ) -> Traces:
         """
         Get list of traces
@@ -80,14 +81,22 @@ def list(
             - user_id: typing.Optional[str].
 
             - name: typing.Optional[str].
+
+            - tags: typing.Optional[typing.Union[str, typing.List[str]]]. Only traces that include all of these tags will be returned.
         """
         _response = self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(
                 f"{self._client_wrapper.get_base_url()}/", "api/public/traces"
             ),
             params=remove_none_from_dict(
-                {"page": page, "limit": limit, "userId": user_id, "name": name}
+                {
+                    "page": page,
+                    "limit": limit,
+                    "userId": user_id,
+                    "name": name,
+                    "tags": tags,
+                }
             ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -160,6 +169,7 @@ async def list(
         limit: typing.Optional[int] = None,
         user_id: typing.Optional[str] = None,
         name: typing.Optional[str] = None,
+        tags: typing.Optional[typing.Union[str, typing.List[str]]] = None,
     ) -> Traces:
         """
         Get list of traces
@@ -172,14 +182,22 @@ async def list(
             - user_id: typing.Optional[str].
 
             - name: typing.Optional[str].
+
+            - tags: typing.Optional[typing.Union[str, typing.List[str]]]. Only traces that include all of these tags will be returned.
         """
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(
                 f"{self._client_wrapper.get_base_url()}/", "api/public/traces"
             ),
             params=remove_none_from_dict(
-                {"page": page, "limit": limit, "userId": user_id, "name": name}
+                {
+                    "page": page,
+                    "limit": limit,
+                    "userId": user_id,
+                    "name": name,
+                    "tags": tags,
+                }
             ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,

langfuse/client.py

@@ -1364,7 +1364,6 @@ def get_langchain_handler(self, *, run_name: str):
 class DatasetClient:
     id: str
     name: str
-    status: DatasetStatus
     project_id: str
     dataset_name: str
     created_at: dt.datetime
@@ -1375,7 +1374,6 @@ class DatasetClient:
     def __init__(self, dataset: Dataset, items: typing.List[DatasetItemClient]):
         self.id = dataset.id
         self.name = dataset.name
-        self.status = dataset.status
         self.project_id = dataset.project_id
         self.dataset_name = dataset.name
         self.created_at = dataset.created_at