jRddlZddlZddlmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZd d lmZmZej&ej(d ZGd d ZGddZy)N)AsyncClientWrapperSyncClientWrapper)RequestOptions)Result)WorkspaceConfiguration)WorkspaceMetricResponse)WorkspaceMetricsSummaryResponse)AsyncRawWorkspacesClientRawWorkspacesClient.c $eZdZdefdZedefdZeeddde jde jd e je jed e jed e jedef d Zdd d e jedefdZeeeddde jede jede je j*eefd e jedef dZdd d e jeddfdZeeddde jde jd e je jed e jed e jedef dZeeddde jde jd e je jed e jed e jedef dZeeddde jde jd e je jed e jed e jedef dZy)WorkspacesClientclient_wrapperc&t||_yN)r)r _raw_clientselfrs x/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/rest_api/workspaces/client.py__init__zWorkspacesClient.__init__s.nMreturnc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawWorkspacesClient rrs rwith_raw_responsez"WorkspacesClient.with_raw_responserN project_idsstart_before_endrequest_optionsinterval_start interval_endr r!r"cZ|jj|||||}|jS)a2 Get costs summary Parameters ---------- interval_start : dt.datetime interval_end : dt.datetime project_ids : typing.Optional[typing.Sequence[str]] start_before_end : typing.Optional[bool] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- Result Workspace Metrics Examples -------- from Opik import OpikApi import datetime client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.workspaces.costs_summary(interval_start=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), interval_end=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), ) r#r$r r!r"r costs_summarydatarr#r$r r!r" _responses rr(zWorkspacesClient.costs_summary!s;J$$22)%#-+ 3 ~~rr"cR|jj|}|jS)a Get workspace configuration Parameters ---------- request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- WorkspaceConfiguration Workspace Configuration Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.workspaces.get_workspace_configuration() r,rget_workspace_configurationr)rr"r+s rr/z,WorkspacesClient.get_workspace_configurationOs(,$$@@Q`@a ~~r"timeout_to_mark_thread_as_inactivetruncation_on_tables color_mapr"r2r3r4cX|jj||||}|jS)a Upsert workspace configuration Parameters ---------- timeout_to_mark_thread_as_inactive : typing.Optional[str] Duration in ISO-8601 format (e.g., PT30M for 30 minutes, PT2H for 2 hours, P1D for 1 day). Minimum precision supported is seconds, please use a duration with seconds precision or higher. Also, the max duration allowed is 7 days. truncation_on_tables : typing.Optional[bool] Enable or disable data truncation in table views. When disabled, the frontend will limit pagination to prevent performance issues. Default: true (truncation enabled). color_map : typing.Optional[typing.Dict[str, str]] Workspace-level color map. Maps label names to hex color values (e.g. #FF0000). Max 10000 entries. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- WorkspaceConfiguration Configuration Updated Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.workspaces.upsert_workspace_configuration() r1rupsert_workspace_configurationr)rr2r3r4r"r+s rr7z/WorkspacesClient.upsert_workspace_configurationhs9H$$CC/Q!5+ D ~~rcR|jj|}|jS)a Delete workspace configuration Parameters ---------- request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.workspaces.delete_workspace_configuration() r,rdelete_workspace_configurationr)r0s rr;z/WorkspacesClient.delete_workspace_configurations(&$$CCTcCd ~~rcZ|jj|||||}|jS)aJ Get cost daily data Parameters ---------- interval_start : dt.datetime interval_end : dt.datetime project_ids : typing.Optional[typing.Sequence[str]] start_before_end : typing.Optional[bool] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- WorkspaceMetricResponse Workspace cost data by days Examples -------- from Opik import OpikApi import datetime client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.workspaces.get_cost(interval_start=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), interval_end=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), ) r&rget_costr)r*s rr>zWorkspacesClient.get_costs;J$$--)%#-+ . ~~rcZ|jj|||||}|jS)aP Get metric daily data Parameters ---------- interval_start : dt.datetime interval_end : dt.datetime project_ids : typing.Optional[typing.Sequence[str]] start_before_end : typing.Optional[bool] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- WorkspaceMetricResponse Workspace metric data by days Examples -------- from Opik import OpikApi import datetime client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.workspaces.get_metric(interval_start=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), interval_end=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), ) r&r get_metricr)r*s rrAzWorkspacesClient.get_metrics;J$$//)%#-+ 0 ~~rcZ|jj|||||}|jS)aO Get metrics summary Parameters ---------- interval_start : dt.datetime interval_end : dt.datetime project_ids : typing.Optional[typing.Sequence[str]] start_before_end : typing.Optional[bool] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- WorkspaceMetricsSummaryResponse Workspace Metrics Examples -------- from Opik import OpikApi import datetime client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.workspaces.metrics_summary(interval_start=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), interval_end=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), ) r&rmetrics_summaryr)r*s rrDz WorkspacesClient.metrics_summarys;J$$44)%#-+ 5 ~~r)__name__ __module__ __qualname__rrpropertyr rOMITdtdatetimetypingOptionalSequencestrboolrrr(rr/Dictr7r;r r>rAr rDrrrrsN*;N #6  >B26;?, ,kk , __V__S%9: , !//$/ , 8, ,^EI"(//."A 8DH6:<@;? *-3OOC,@*%ood3 * ??6;;sCx#89 *  8 * *XdhQ_A`lp6>B26;?, ,kk , __V__S%9: , !//$/ , 8, !,f>B26;?, ,kk , __V__S%9: , !//$/ , 8, !,f>B26;?, ,kk , __V__S%9: , !//$/ , 8, ),rrc $eZdZdefdZedefdZeeddde jde jd e je jed e jed e jedef d Zdd d e jedefdZeeeddde jede jede je j*eefd e jedef dZdd d e jeddfdZeeddde jde jd e je jed e jed e jedef dZeeddde jde jd e je jed e jed e jedef dZeeddde jde jd e je jed e jed e jedef dZy)AsyncWorkspacesClientrc&t||_yr)r rrs rrzAsyncWorkspacesClient.__init__6s3>Rrrc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawWorkspacesClient rrs rrz'AsyncWorkspacesClient.with_raw_response9rrNrr#r$r r!r"cvK|jj|||||d{}|jS7w)a Get costs summary Parameters ---------- interval_start : dt.datetime interval_end : dt.datetime project_ids : typing.Optional[typing.Sequence[str]] start_before_end : typing.Optional[bool] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- Result Workspace Metrics Examples -------- from Opik import AsyncOpikApi import datetime import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.workspaces.costs_summary(interval_start=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), interval_end=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), ) asyncio.run(main()) r&Nr'r*s rr(z#AsyncWorkspacesClient.costs_summaryDsIP**88)%#-+ 9  ~~ $979r,cnK|jj|d{}|jS7w)a\ Get workspace configuration Parameters ---------- request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- WorkspaceConfiguration Workspace Configuration Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.workspaces.get_workspace_configuration() asyncio.run(main()) r,Nr.r0s rr/z1AsyncWorkspacesClient.get_workspace_configurationus52**FFWfFgg ~~h 535r1r2r3r4ctK|jj||||d{}|jS7w)a( Upsert workspace configuration Parameters ---------- timeout_to_mark_thread_as_inactive : typing.Optional[str] Duration in ISO-8601 format (e.g., PT30M for 30 minutes, PT2H for 2 hours, P1D for 1 day). Minimum precision supported is seconds, please use a duration with seconds precision or higher. Also, the max duration allowed is 7 days. truncation_on_tables : typing.Optional[bool] Enable or disable data truncation in table views. When disabled, the frontend will limit pagination to prevent performance issues. Default: true (truncation enabled). color_map : typing.Optional[typing.Dict[str, str]] Workspace-level color map. Maps label names to hex color values (e.g. #FF0000). Max 10000 entries. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- WorkspaceConfiguration Configuration Updated Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.workspaces.upsert_workspace_configuration() asyncio.run(main()) r1Nr6r8s rr7z4AsyncWorkspacesClient.upsert_workspace_configurationsGN**II/Q!5+ J  ~~  s #868cnK|jj|d{}|jS7w)a, Delete workspace configuration Parameters ---------- request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.workspaces.delete_workspace_configuration() asyncio.run(main()) r,Nr:r0s rr;z4AsyncWorkspacesClient.delete_workspace_configurations5,**IIZiIjj ~~krZcvK|jj|||||d{}|jS7w)a Get cost daily data Parameters ---------- interval_start : dt.datetime interval_end : dt.datetime project_ids : typing.Optional[typing.Sequence[str]] start_before_end : typing.Optional[bool] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- WorkspaceMetricResponse Workspace cost data by days Examples -------- from Opik import AsyncOpikApi import datetime import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.workspaces.get_cost(interval_start=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), interval_end=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), ) asyncio.run(main()) r&Nr=r*s rr>zAsyncWorkspacesClient.get_costsIP**33)%#-+ 4  ~~ rXcvK|jj|||||d{}|jS7w)a Get metric daily data Parameters ---------- interval_start : dt.datetime interval_end : dt.datetime project_ids : typing.Optional[typing.Sequence[str]] start_before_end : typing.Optional[bool] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- WorkspaceMetricResponse Workspace metric data by days Examples -------- from Opik import AsyncOpikApi import datetime import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.workspaces.get_metric(interval_start=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), interval_end=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), ) asyncio.run(main()) r&Nr@r*s rrAz AsyncWorkspacesClient.get_metric sIP**55)%#-+ 6  ~~ rXcvK|jj|||||d{}|jS7w)a Get metrics summary Parameters ---------- interval_start : dt.datetime interval_end : dt.datetime project_ids : typing.Optional[typing.Sequence[str]] start_before_end : typing.Optional[bool] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- WorkspaceMetricsSummaryResponse Workspace Metrics Examples -------- from Opik import AsyncOpikApi import datetime import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.workspaces.metrics_summary(interval_start=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), interval_end=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), ) asyncio.run(main()) r&NrCr*s rrDz%AsyncWorkspacesClient.metrics_summary;sIP**::)%#-+ ;  ~~ rX)rErFrGrrrHr rrIrJrKrLrMrNrOrPrrr(rr/rQr7r;r r>rAr rDrRrrrTrT5sS*<S #;  >B26;?/ /kk / __V__S%9: / !//$/ / 8/ /dEI"(//."A >DH6:<@;? --3OOC,@-%ood3 - ??6;;sCx#89 -  8 - -^jnvWeGfrv<>B26;?/ /kk / __V__S%9: / !//$/ / 8/ !/l>B26;?/ /kk / __V__S%9: / !//$/ / 8/ !/l>B26;?/ /kk / __V__S%9: / !//$/ / 8/ )/rrT)rKrJrLcore.client_wrapperrrcore.request_optionsr types.resultrtypes.workspace_configurationrtypes.workspace_metric_responser (types.workspace_metrics_summary_responser raw_clientr r castAnyrIrrTrRrrrisS G1!BEVEv{{6::s#``F uur