jHddlZddlmZddlmZddlmZmZddlm Z ddl m Z m Z dd l mZdd lmZdd lmZejejd ZGd dZGddZdS)N)Dataset)DatasetRunWithItems)AsyncClientWrapperSyncClientWrapper)RequestOptions)AsyncRawDatasetsClientRawDatasetsClient)DeleteDatasetRunResponse)PaginatedDatasetRuns)PaginatedDatasets.cXeZdZdefdZedefdZdddddej e dej e d ej e de fd Z dd d ed ej e defd Zeeeedddedej edej ejdej ejdej ejd ej e defdZdd d eded ej e defdZdd d eded ej e defdZddddd edej e dej e d ej e def dZdS)DatasetsClientclient_wrapperc0t||_dSN)r)r _raw_clientselfrs f/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/langfuse/api/datasets/client.py__init__zDatasetsClient.__init__s,NKKKreturnc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawDatasetsClient rrs rwith_raw_responsez DatasetsClient.with_raw_response rNpagelimitrequest_optionsr!r"r#cJ|j|||}|jS)aU Get all datasets Parameters ---------- page : typing.Optional[int] page number, starts at 1 limit : typing.Optional[int] limit of items per page request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- PaginatedDatasets Examples -------- from langfuse import LangfuseAPI client = LangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) client.datasets.list() r rlistdatarr!r"r# _responses rr&zDatasetsClient.list!s3N$))UO*  ~rr# dataset_namecH|j||}|jS)a Get a dataset Parameters ---------- dataset_name : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- Dataset Examples -------- from langfuse import LangfuseAPI client = LangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) client.datasets.get( dataset_name="datasetName", ) r*rgetr'rr+r#r)s rr.zDatasetsClient.getMs(H$(((WW ~r descriptionmetadata input_schemaexpected_output_schemar#namer1r2r3r4cP|j||||||}|jS)a Create a dataset Parameters ---------- name : str description : typing.Optional[str] metadata : typing.Optional[typing.Any] input_schema : typing.Optional[typing.Any] JSON Schema for validating dataset item inputs. When set, all new and existing dataset items will be validated against this schema. expected_output_schema : typing.Optional[typing.Any] JSON Schema for validating dataset item expected outputs. When set, all new and existing dataset items will be validated against this schema. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- Dataset Examples -------- from langfuse import LangfuseAPI client = LangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) client.datasets.create( name="name", ) r5r1r2r3r4r#rcreater'rr5r1r2r3r4r#r)s rr9zDatasetsClient.createts>d$++#%#9+ ,  ~rrun_namecJ|j|||}|jS)aZ Get a dataset run and its items Parameters ---------- dataset_name : str run_name : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- DatasetRunWithItems Examples -------- from langfuse import LangfuseAPI client = LangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) client.datasets.get_run( dataset_name="datasetName", run_name="runName", ) r*rget_runr'rr+r;r#r)s rr>zDatasetsClient.get_runs3P$,, (O-  ~rcJ|j|||}|jS)a Delete a dataset run and all its run items. This action is irreversible. Parameters ---------- dataset_name : str run_name : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- DeleteDatasetRunResponse Examples -------- from langfuse import LangfuseAPI client = LangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) client.datasets.delete_run( dataset_name="datasetName", run_name="runName", ) r*r delete_runr'r?s rrBzDatasetsClient.delete_runs3P$// (O0  ~rcL|j||||}|jS)a Get dataset runs Parameters ---------- dataset_name : str page : typing.Optional[int] page number, starts at 1 limit : typing.Optional[int] limit of items per page request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- PaginatedDatasetRuns Examples -------- from langfuse import LangfuseAPI client = LangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) client.datasets.get_runs( dataset_name="datasetName", ) r rget_runsr'rr+r!r"r#r)s rrEzDatasetsClient.get_runs s5X$-- t5/.  ~r)__name__ __module__ __qualname__rrpropertyr rtypingOptionalintrrr&strrr.OMITAnyr9rr>r rBr rErrrrsL*;LLLL #4   X &*&*;? ***oc"*s# *  8 *  ****`<@ %%%% 8 %  %%%%V-10448>B;?::::_S) : /&*- : ofj1 :!' ;: 8: ::::B<@ +++++  8 +  ++++d<@ +++++  8 + " ++++b&*&*;? ////oc" / s# /  8 / //////rrcXeZdZdefdZedefdZdddddej e dej e d ej e de fd Z dd d ed ej e defd Zeeeedddedej edej ejdej ejdej ejd ej e defdZdd d eded ej e defdZdd d eded ej e defdZddddd edej e dej e d ej e def dZdS)AsyncDatasetsClientrc0t||_dSr)r rrs rrzAsyncDatasetsClient.__init__=s1PPPrrc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawDatasetsClient rrs rrz%AsyncDatasetsClient.with_raw_response@rrNr r!r"r#cZK|j|||d{V}|jS)a Get all datasets Parameters ---------- page : typing.Optional[int] page number, starts at 1 limit : typing.Optional[int] limit of items per page request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- PaginatedDatasets Examples -------- import asyncio from langfuse import AsyncLangfuseAPI client = AsyncLangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) async def main() -> None: await client.datasets.list() asyncio.run(main()) r Nr%r(s rr&zAsyncDatasetsClient.listKsU^*//UO0        ~rr*r+cXK|j||d{V}|jS)av Get a dataset Parameters ---------- dataset_name : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- Dataset Examples -------- import asyncio from langfuse import AsyncLangfuseAPI client = AsyncLangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) async def main() -> None: await client.datasets.get( dataset_name="datasetName", ) asyncio.run(main()) r*Nr-r/s rr.zAsyncDatasetsClient.getsSX*.. //        ~rr0r5r1r2r3r4c`K|j||||||d{V}|jS)a] Create a dataset Parameters ---------- name : str description : typing.Optional[str] metadata : typing.Optional[typing.Any] input_schema : typing.Optional[typing.Any] JSON Schema for validating dataset item inputs. When set, all new and existing dataset items will be validated against this schema. expected_output_schema : typing.Optional[typing.Any] JSON Schema for validating dataset item expected outputs. When set, all new and existing dataset items will be validated against this schema. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- Dataset Examples -------- import asyncio from langfuse import AsyncLangfuseAPI client = AsyncLangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) async def main() -> None: await client.datasets.create( name="name", ) asyncio.run(main()) r7Nr8r:s rr9zAsyncDatasetsClient.creates`t*11#%#9+ 2        ~rr;cZK|j|||d{V}|jS)a Get a dataset run and its items Parameters ---------- dataset_name : str run_name : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- DatasetRunWithItems Examples -------- import asyncio from langfuse import AsyncLangfuseAPI client = AsyncLangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) async def main() -> None: await client.datasets.get_run( dataset_name="datasetName", run_name="runName", ) asyncio.run(main()) r*Nr=r?s rr>zAsyncDatasetsClient.get_runsU`*22 (O3        ~rcZK|j|||d{V}|jS)a Delete a dataset run and all its run items. This action is irreversible. Parameters ---------- dataset_name : str run_name : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- DeleteDatasetRunResponse Examples -------- import asyncio from langfuse import AsyncLangfuseAPI client = AsyncLangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) async def main() -> None: await client.datasets.delete_run( dataset_name="datasetName", run_name="runName", ) asyncio.run(main()) r*NrAr?s rrBzAsyncDatasetsClient.delete_run)sU`*55 (O6        ~rc\K|j||||d{V}|jS)a Get dataset runs Parameters ---------- dataset_name : str page : typing.Optional[int] page number, starts at 1 limit : typing.Optional[int] limit of items per page request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- PaginatedDatasetRuns Examples -------- import asyncio from langfuse import AsyncLangfuseAPI client = AsyncLangfuseAPI( x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME", x_langfuse_sdk_version="YOUR_X_LANGFUSE_SDK_VERSION", x_langfuse_public_key="YOUR_X_LANGFUSE_PUBLIC_KEY", username="YOUR_USERNAME", password="YOUR_PASSWORD", base_url="https://yourhost.com/path/to/api", ) async def main() -> None: await client.datasets.get_runs( dataset_name="datasetName", ) asyncio.run(main()) r NrDrFs rrEzAsyncDatasetsClient.get_runs^sWh*33 t5/4        ~r)rGrHrIrrrJr rrKrLrMrrr&rNrr.rOrPr9rr>r rBr rErQrrrSrS<sQ*<QQQQ #9   X &*&*;? 222oc"2s# 2  8 2  2222p<@ //// 8 /  ////j-10448>B;?BBBB_S) B /&*- B ofj1 B!' ;B 8B BBBBR<@ 33333  8 3  3333t<@ 33333  8 3 " 3333r&*&*;? 7777oc" 7 s# 7  8 7 777777rrS)rKcommons.types.datasetr$commons.types.dataset_run_with_itemsrcore.client_wrapperrrcore.request_optionsr raw_clientr r !types.delete_dataset_run_responser types.paginated_dataset_runsr types.paginated_datasetsrcastrPrOrrSrQrrres0 ++++++FFFFFFGGGGGGGG111111AAAAAAAAGGGGGG>>>>>>777777v{6:s##ggggggggT YYYYYYYYYYr