arize-phoenix 6.2.0__py3-none-any.whl → 7.0.0__py3-none-any.whl

{arize_phoenix-6.2.0.dist-info → arize_phoenix-7.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arize-phoenix
-Version: 6.2.0
+Version: 7.0.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -44,7 +44,7 @@ Requires-Dist: scipy
 Requires-Dist: sqlalchemy[asyncio]<3,>=2.0.4
 Requires-Dist: sqlean-py>=3.45.1
 Requires-Dist: starlette
-Requires-Dist: strawberry-graphql==0.243.1
+Requires-Dist: strawberry-graphql==0.247.0
 Requires-Dist: tqdm
 Requires-Dist: typing-extensions>=4.6
 Requires-Dist: uvicorn
@@ -66,7 +66,7 @@ Requires-Dist: opentelemetry-sdk; extra == 'container'
 Requires-Dist: opentelemetry-semantic-conventions; extra == 'container'
 Requires-Dist: prometheus-client; extra == 'container'
 Requires-Dist: py-grpc-prometheus; extra == 'container'
-Requires-Dist: strawberry-graphql[opentelemetry]==0.243.1; extra == 'container'
+Requires-Dist: strawberry-graphql[opentelemetry]==0.247.0; extra == 'container'
 Requires-Dist: umap-learn; extra == 'container'
 Requires-Dist: uvloop; (platform_system != 'Windows') and extra == 'container'
 Provides-Extra: dev
@@ -97,7 +97,7 @@ Requires-Dist: pytest-postgresql; extra == 'dev'
 Requires-Dist: pytest-xdist; extra == 'dev'
 Requires-Dist: pytest==8.3.3; extra == 'dev'
 Requires-Dist: ruff==0.6.9; extra == 'dev'
-Requires-Dist: strawberry-graphql[debug-server,opentelemetry]==0.243.1; extra == 'dev'
+Requires-Dist: strawberry-graphql[debug-server,opentelemetry]==0.247.0; extra == 'dev'
 Requires-Dist: tabulate; extra == 'dev'
 Requires-Dist: tox-uv==1.11.3; extra == 'dev'
 Requires-Dist: tox==4.18.1; extra == 'dev'

{arize_phoenix-6.2.0.dist-info → arize_phoenix-7.0.0.dist-info}/RECORD

@@ -6,7 +6,7 @@ phoenix/exceptions.py,sha256=n2L2KKuecrdflB9MsCdAYCiSEvGJptIsfRkXMoJle7A,169
 phoenix/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 phoenix/services.py,sha256=kpW1WL0kiB8XJsO6XycvZVJ-lBkNoenhQ7atCvBoSe8,5365
 phoenix/settings.py,sha256=ht-0oN-sMV6SPXrk7Tu1EZlngpAYkGNLYPhO8DyrdQI,661
-phoenix/version.py,sha256=5H_IrrHNTfYkLbC3nRLonK3TwkV0TTf0xGbx1a13Scw,22
+phoenix/version.py,sha256=VgMOOqsYbyb60I1RmlZpqwqQ0C0IyT3R0c8_xX4pRGM,22
 phoenix/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/core/embedding_dimension.py,sha256=zKGbcvwOXgLf-yrJBpQyKtd-LEOPRKHnUToyAU8Owis,87
 phoenix/core/model.py,sha256=qBFraOtmwCCnWJltKNP18DDG0mULXigytlFsa6YOz6k,4837
@@ -21,22 +21,25 @@ phoenix/db/enums.py,sha256=tt7iovXLhVTLZ3_LbHNGgcI44SnNjXfkKtLAZG57T54,428
 phoenix/db/facilitator.py,sha256=sAYqzBXYSVBKPTQVYrd7ZmtqMAr1zP9dVJwjfNGW7hc,4207
 phoenix/db/helpers.py,sha256=daKbpY2QhTPo9a_T1xNHKI4WzWHkMmmrGIws7Hw-RZ4,4884
 phoenix/db/migrate.py,sha256=oUrXH8yEbcpL4eh09aSCuUiSrhFli0eT5D_j4ZmYChY,2797
-phoenix/db/models.py,sha256=wx2XQgf6qhUO2ZQ4oUENoAguf440qaR0Ea_q1XB1bBU,26484
+phoenix/db/models.py,sha256=56_I2oV-OOKfQUIqUlNKWc3GIPag3nuHmoKqrtIis68,27459
 phoenix/db/insertion/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/db/insertion/constants.py,sha256=8wifm7X-1XvroZ__R2Gc96NsgLhTDn0zXl4lehlXtcA,70
 phoenix/db/insertion/dataset.py,sha256=I9OC1ouVx7m6BH_c8hvcxW1dWGRAtpvXee29yBTuFkg,7136
 phoenix/db/insertion/document_annotation.py,sha256=vnszF9L0qHjUY-eWgBVRG5OTpz-ECs1BoCfy66ynzjo,5997
 phoenix/db/insertion/evaluation.py,sha256=SoI85N3MYUSeNgjKa5WzFw14OfNjNTjExv-2m3sxaR8,6371
 phoenix/db/insertion/helpers.py,sha256=-DyRcxzJnjSJFhscPoqiNiQn8fBvGqI8IcNJEu-79Vw,3455
-phoenix/db/insertion/span.py,sha256=-CXn2LlEv2u7xbz7m8X5jALQ-BUNNzTxPJEzSmjhTpA,5958
+phoenix/db/insertion/span.py,sha256=QHjWTc6cEdq1raxHDcQ9x9tBAdUfePJYjRE_wGdgXFk,7811
 phoenix/db/insertion/span_annotation.py,sha256=EtzcjS8GR2rDM_cxIQloGc2SZqfdrACyAPwvyFBa2Ac,5273
 phoenix/db/insertion/trace_annotation.py,sha256=6yzWuU0Fh-mC-rJJ96rH0IYTJg1EQFnj-GZhUwolUKI,5338
 phoenix/db/insertion/types.py,sha256=cWeBceyZZ3efLHKPjnPpVahrpX7-P8Bm_sdudWXkOek,8106
 phoenix/db/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/db/migrations/env.py,sha256=tFO3ceuCx9Es5_2w_BXclaQMmNQKNX21b1UEV5mYdeo,3387
 phoenix/db/migrations/script.py.mako,sha256=MEqL-2qATlST9TAOeYgscMn1uy6HUS9NFvDgl93dMj8,635
+phoenix/db/migrations/data_migration_scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+phoenix/db/migrations/data_migration_scripts/populate_project_sessions.py,sha256=NJPoEX2-OhYxQXl0jRcX5eoq1azgLi8RLArZZw2pTY4,6046
 phoenix/db/migrations/versions/10460e46d750_datasets.py,sha256=3wftHgat1sEeKBl0EcQ24r_oOK_T-Tcmd2Loo73RVCs,8679
 phoenix/db/migrations/versions/3be8647b87d8_add_token_columns_to_spans_table.py,sha256=GxuKD501WSx083uIv21GhJS0l4Xk3exovfycaGAxY30,3638
+phoenix/db/migrations/versions/4ded9e43755f_create_project_sessions_table.py,sha256=HDP3X5yxm5PwxMkODd-wrSwspQpFotTEz40jXRgq320,1787
 phoenix/db/migrations/versions/cd164e83824f_users_and_tokens.py,sha256=fkpmh5PgMZJiZpvLbZIaqlI2cucVpVbbNYpQ-Tznil8,5180
 phoenix/db/migrations/versions/cf03bd6bae1d_init.py,sha256=lorqWUcujC0fsYIOriGZlZw2LKjS3nF34KHxi1fp_Z8,8646
 phoenix/experiments/__init__.py,sha256=6JGwgUd7xCbGpuHqYZlsmErmYvVgv7N_j43bn3dUqsk,123
@@ -73,7 +76,7 @@ phoenix/pointcloud/pointcloud.py,sha256=SN_1wXZcwKrtSnHGZLDZGx71orqE1WyVF7E-D58d
 phoenix/pointcloud/projectors.py,sha256=TQgwc9cJDjJkin1WZyZzgl3HsYrLLiyWD7Czy4jNW3U,1088
 phoenix/pointcloud/umap_parameters.py,sha256=db_WEPoamuWtopZx7tQfAXPnoE0MS8FkAV0_ThjEx_Q,1735
 phoenix/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-phoenix/server/app.py,sha256=O-oawPGpm02KT6_09E66qdEpZqJT1pMUt9LODeI5m5U,37934
+phoenix/server/app.py,sha256=6ahHhsZeiYaWIRAVOGUk66c5BDQnB6EZQhgvtnqS6-M,38691
 phoenix/server/bearer_auth.py,sha256=0UudvkAS_dxna5JEJJhGUYwB6Ny-e22ssX5Mm79QwCk,5907
 phoenix/server/dml_event.py,sha256=MjJmVEKytq75chBOSyvYDusUnEbg1pHpIjR3pZkUaJA,2838
 phoenix/server/dml_event_handler.py,sha256=EZLXmCvx4pJrCkz29gxwKwmvmUkTtPCHw6klR-XM8qE,8258
@@ -89,14 +92,14 @@ phoenix/server/types.py,sha256=Ur73LKRB3VlvBDN95gKJoDIdmauuR-r1rpO1rNHm9kA,7209
 phoenix/server/api/README.md,sha256=Pyq1PLPgTzXAswrfIhGXrjI3Skq8it2jTVnanT6Ba4Q,1162
 phoenix/server/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/api/auth.py,sha256=OhWmv_Hl-K-yhJeUWXKRYYiUOIPo-rYSQXdm4fjeTJ8,901
-phoenix/server/api/context.py,sha256=g5tAYInMifMOdnGTlyfHdB7rDNlC9F2E5bU6IA_3Y0c,5337
+phoenix/server/api/context.py,sha256=oTSBervsrFYeKsln37Mc_fgBD5y5TktYdNVPZSzTJlU,5952
 phoenix/server/api/exceptions.py,sha256=TA0JuY2YRnj35qGuMSQ8d0ToHum9gWm9W--3fSKHrX0,1171
 phoenix/server/api/interceptor.py,sha256=ykDnoC_apUd-llVli3m1CW18kNSIgjz2qZ6m5JmPDu8,1294
-phoenix/server/api/queries.py,sha256=4KJz8TUz3VUTup9MDjr_GoKX0SttWSvHBq2ncWZGxf8,27343
+phoenix/server/api/queries.py,sha256=zRngYE72hxaeAg_1vsnSLwmqs_FSHrRNjQwi632_D-Y,27670
 phoenix/server/api/schema.py,sha256=tHyw2jTbue_-gu0fe9Sw7LUYtzJUCwp9SvccDgOkNPw,1696
 phoenix/server/api/subscriptions.py,sha256=NJXCWrbBERyTwmi3H_9jNpml4klIxs_kpynwXRoH9pc,23572
 phoenix/server/api/utils.py,sha256=quCBRcusc6PUq9tJq7M8PgwFZp7nXgVAxtbw8feribY,833
-phoenix/server/api/dataloaders/__init__.py,sha256=jNYvfXjnZzgA2HWTG7AZdqWGla3ZysBUDUei8Zkz6N8,3290
+phoenix/server/api/dataloaders/__init__.py,sha256=nt5Lbs34q05GSvNCT2IGrkRW0kg65L8h3bZelzgpkYg,3909
 phoenix/server/api/dataloaders/annotation_summaries.py,sha256=2sHmIDX7n8tuPeBTs9bMKtlMKWn_Ph9awTZqmwn2Owc,5505
 phoenix/server/api/dataloaders/average_experiment_run_latency.py,sha256=GLFoFAbztOH-0FVzzZ77mATIO63UcjB50j3qXiNi-tE,1811
 phoenix/server/api/dataloaders/dataset_example_revisions.py,sha256=YG2VSWOkM7q_-RO8OwE8bHYX8d3_OH34SeaD4pvEt0w,5115
@@ -113,12 +116,18 @@ phoenix/server/api/dataloaders/latency_ms_quantile.py,sha256=CHWV9nvEEM3xqpXWuby
 phoenix/server/api/dataloaders/min_start_or_max_end_times.py,sha256=1jYglTXv4S8k7OWSGdgeJ2OxVCmZtX8a6sjJp9GPZ04,2783
 phoenix/server/api/dataloaders/project_by_name.py,sha256=O24Rjs0ZQaYtSnPfA6YBQfbHfUgNsHKUcc_gIKyK9vo,1086
 phoenix/server/api/dataloaders/record_counts.py,sha256=NLlMEaJ-HxiJavqbSkMbzOrIRzqzYxO8-t2mfp1vXqM,4160
+phoenix/server/api/dataloaders/session_io.py,sha256=r-Zvn19qOo_JWATHtt5zwuwhHjiATLAGHRjoi4czd6U,3038
+phoenix/server/api/dataloaders/session_num_traces.py,sha256=lu5EMdQWQzEr_4wCiu7FycnIShjGf0yQXpwBG_DDzGs,1050
+phoenix/server/api/dataloaders/session_num_traces_with_error.py,sha256=Y5xmr4Nme2ZMfvJljW9yhBfuCNdGlxTRka0h9VdXdqA,1149
+phoenix/server/api/dataloaders/session_token_usages.py,sha256=9RXZUyzDmzp5WDp_BpmvvDhk5gIIaV97x0xxUM1H9bA,1586
+phoenix/server/api/dataloaders/session_trace_latency_ms_quantile.py,sha256=s_Ce_BSad1Wvabq9hajdq-i-9rLbZ0esU2gwg9h7Vwg,2157
 phoenix/server/api/dataloaders/span_annotations.py,sha256=y5TvxnljS2P3hm_NiQd24qy313AG76lyjcZFNWGls1A,1028
 phoenix/server/api/dataloaders/span_dataset_examples.py,sha256=rpBStVP7ZMIH11Cpq4-zCJ4amNC5ZzX72NYv4vAQTdc,1255
 phoenix/server/api/dataloaders/span_descendants.py,sha256=1NhD59micO9ozKTF85bFKVCqqsrfDzrK0e0nlg6QyvI,2051
 phoenix/server/api/dataloaders/span_projects.py,sha256=JTfuKn2BBn72QdAP53ZGP2OUCgZJ7AzlzQAx8WivBog,1234
 phoenix/server/api/dataloaders/token_counts.py,sha256=9BjumOJ8uCaQ8ZieJVoIP_zEnI38VxyBoB_ks8YrUYU,4629
-phoenix/server/api/dataloaders/trace_row_ids.py,sha256=yFv752LmBmOnMYVaauU4v0Y660AnSGJCtD7O-UgKk9Q,1070
+phoenix/server/api/dataloaders/trace_by_trace_ids.py,sha256=xqYJOjCgXlTzour4vY72kO-2gJFgWMjD3759o78cq44,874
+phoenix/server/api/dataloaders/trace_root_spans.py,sha256=IpSsoBRF0NHtbFhQasA_mux6v-m6b0yFRINygjCBOtM,1106
 phoenix/server/api/dataloaders/user_roles.py,sha256=TvPWve38TlGFvaFkMeFvnnHBn1OMHUJTErRLeYrXR8I,1029
 phoenix/server/api/dataloaders/users.py,sha256=Uh86Kny_xpqDdEvEklcTBUA_MELgu4Z0jgh45ZqTzXs,1075
 phoenix/server/api/dataloaders/cache/__init__.py,sha256=SYoOM9n8FJaMdQarma5d1blu-jIg2GB8Shqg5ezSzZ8,106
@@ -155,6 +164,7 @@ phoenix/server/api/input_types/PatchAnnotationInput.py,sha256=NWhkcbcGNPwfOYsN3w
 phoenix/server/api/input_types/PatchDatasetExamplesInput.py,sha256=_uMqkAInhLDvzUSASl6HgLNulTsekMcYzyd5J6LF90I,884
 phoenix/server/api/input_types/PatchDatasetInput.py,sha256=OURtTVY8Z_oFEDtKwT1LCMaOK5D4QYo5TVQ6mDrex-g,328
 phoenix/server/api/input_types/PerformanceMetricInput.py,sha256=fElsLTSEYYgGFGMYTEGcYid39tXUKFdV_JkdHavMcbA,591
+phoenix/server/api/input_types/ProjectSessionSort.py,sha256=qFgLmKYeyFpx7An9ZNdDNkkt7tghrG6XXruQbFLiFt8,919
 phoenix/server/api/input_types/SpanAnnotationSort.py,sha256=T5pAGzmh4MiJp9JMAzNDByFVTczfw02FH4WFWwFezyI,361
 phoenix/server/api/input_types/SpanSort.py,sha256=Dhvl8BIoV52yHoqntfOax_gUc15uH8ITI_00Ha7PvYc,5959
 phoenix/server/api/input_types/TemplateOptions.py,sha256=BrKozr604QwthBdKhlsHbxJc5oZ7eHnwviVubW2i0jA,220
@@ -168,7 +178,7 @@ phoenix/server/api/mutations/chat_mutations.py,sha256=X8Rq0_gBYX_Ey2U9EIuWuOt2Je
 phoenix/server/api/mutations/dataset_mutations.py,sha256=OkVek-GtDf_1eloI7Sdz6W6l5FFmwU_41LJM8-fw3q4,26088
 phoenix/server/api/mutations/experiment_mutations.py,sha256=p3CoLAa8nFPa3D759Y2A7De_PVJNGOL98mA3HoZBrRQ,3188
 phoenix/server/api/mutations/export_events_mutations.py,sha256=xoDnVWC7eA_8wNQP0-oyiHojyUZ0EhVVSrsAnztetC0,3993
-phoenix/server/api/mutations/project_mutations.py,sha256=KdcA3c6tgLZ8QhpIv37qJESTBPxs56SbfeYdDKzU7-8,2263
+phoenix/server/api/mutations/project_mutations.py,sha256=u7kLvJbBFR418F9lMamiisZerF18E8h5wGKV4sDTF-g,2674
 phoenix/server/api/mutations/span_annotations_mutations.py,sha256=kvIOUjHRA6wNMh-bLe5B6bRszMw77HyFp3bxHTOL3yU,5940
 phoenix/server/api/mutations/trace_annotations_mutations.py,sha256=qwTnYt6A5BwXC7sn_TT0f2A2PO7M26C22aYMnA3MpYk,5915
 phoenix/server/api/mutations/user_mutations.py,sha256=YWgg1t5P0XVdxOMrhWpRqsCWi8NQR2zxerzuITC-UrY,13311
@@ -222,8 +232,8 @@ phoenix/server/api/types/ExampleRevisionInterface.py,sha256=gV3Gt9-3Oi5wjaVtepC6
 phoenix/server/api/types/Experiment.py,sha256=hqqIwgeWoylOeBAMmQyzefn_p_COwxW0VqEGOqdcTpE,5470
 phoenix/server/api/types/ExperimentAnnotationSummary.py,sha256=Uk3JtxIrsMoZT5tqc4nJdUOM3XegVzjUyoV3pkjNotE,256
 phoenix/server/api/types/ExperimentComparison.py,sha256=cFQZy5rj_klzTV-3jnxIlkrCHXkW8gYnp5uSR1UWCzg,416
-phoenix/server/api/types/ExperimentRun.py,sha256=18VDl9ciENRp-yn1_gyCDnTI4TYchG985GuM6hssN7g,4253
-phoenix/server/api/types/ExperimentRunAnnotation.py,sha256=iBxDaD9DgiF-Qymp5QyxWfJRGYXM1_CeWA_qzsZBqkI,1812
+phoenix/server/api/types/ExperimentRun.py,sha256=5PjvIx-VLx-W9Xw8JURbbfOYYIL-FAaZs1kyJB0icp4,4169
+phoenix/server/api/types/ExperimentRunAnnotation.py,sha256=HEbPygBwbWg10fwY3VZyM10OL1TKlnQk_x5MZIjZKPs,1724
 phoenix/server/api/types/ExportedFile.py,sha256=e3GTn7B5LgsTbqiwjhMCQH7VsiqXitrBO4aCMS1lHsg,163
 phoenix/server/api/types/Functionality.py,sha256=tzV9xdhB8zqfsjWxP66NDC7EZsplYkYO7jRbLWJIeeg,382
 phoenix/server/api/types/GenerativeModel.py,sha256=P7eBUMXbeqaLwSSGBKdZy3a5gOLd9I0fuP8o1st6H08,193
@@ -235,18 +245,21 @@ phoenix/server/api/types/MimeType.py,sha256=Zpi6zCalkSFgsvhzvOs-O1gYA04usAi9H__Q
 phoenix/server/api/types/Model.py,sha256=dirXGnOqLFNaOlXUwqyAjiJjcOrWpryO_5N6V0pAxtk,8045
 phoenix/server/api/types/NumericRange.py,sha256=afEjgF97Go_OvmjMggbPBt-zGM8IONewAyEiKEHRds0,192
 phoenix/server/api/types/PerformanceMetric.py,sha256=W92B7OghEOgzFvmY0LCqpgavHaQggTGshdgfD0yqHX4,350
-phoenix/server/api/types/Project.py,sha256=tNTiDgbjUTVm45XxmMhHYSjqBI8WjHim9NZ69SfMA8w,13107
+phoenix/server/api/types/Project.py,sha256=xbV15SOHcPQeE1TMSWFpXk7ZQkfxCD4ZjFS4C83K4tU,18852
+phoenix/server/api/types/ProjectSession.py,sha256=j3RJrTG9umruav7d81AqnQVz-Zpi60nXmnsr2Wplpac,4646
 phoenix/server/api/types/PromptResponse.py,sha256=Q8HKtpp8GpUOcxPCzZpkkokidDd6u0aZOv_SuPZZd5Q,630
 phoenix/server/api/types/Retrieval.py,sha256=OhMK2ncjoyp5h1yjKhjlKpoTbQrMHuxmgSFw-AO1rWw,285
 phoenix/server/api/types/ScalarDriftMetricEnum.py,sha256=IUAcRPpgL41WdoIgK6cNk2Te38SspXGyEs-S1fY23_A,232
 phoenix/server/api/types/Segments.py,sha256=vT2v0efoa5cuBKxLtxTnsUP5YJJCZfTloM71Spu0tMI,2915
 phoenix/server/api/types/SortDir.py,sha256=OUpXhlCzCxPoXSDkJJygEs9Rw9pMymfaZUG5zPTrw4Y,152
-phoenix/server/api/types/Span.py,sha256=wJGvenSL6d1oDmpJt4hsLjIVTxmemJBRCBKHy4rYztc,15355
+phoenix/server/api/types/Span.py,sha256=3vMWnCH0w7-lytq9fVpD0oWIfC-4huS5U6lOaoTSPrA,15113
 phoenix/server/api/types/SpanAnnotation.py,sha256=6b5G-b_OoRvDL2ayWk7MkbqarLK-F-pQMx21CpUuNGY,1168
+phoenix/server/api/types/SpanIOValue.py,sha256=u_g9QrX-E3fwneoucire73KY5gf3fE4V9IBN3qx2cvU,445
 phoenix/server/api/types/SystemApiKey.py,sha256=2ym8EgsTBIvxx1l9xZ-2YMovz58ZwYb_MaHBTJ9NH2E,166
 phoenix/server/api/types/TemplateLanguage.py,sha256=6j_0uwO_GZIeCpR7sTOnxySXudT7qBSC6LFsjzbvW1o,160
 phoenix/server/api/types/TimeSeries.py,sha256=IIeGVRFdSMozYXxPg736DW_mKvj4-3WjYSYEnn4UEJc,5241
-phoenix/server/api/types/Trace.py,sha256=1RrdEedlPpNmWkQwosOZ81fabc9-B6PDHTYpr-hZj-Y,3240
+phoenix/server/api/types/TokenUsage.py,sha256=g-PjAGVigpchQgkXAuC5sc53fn2YwAgfeXkGmFPi_TE,201
+phoenix/server/api/types/Trace.py,sha256=PG07x8h1VjYNkE40EK2SJHMW1TRWBoguJS6vPjmJOy8,5217
 phoenix/server/api/types/TraceAnnotation.py,sha256=OW6A2zr1gomOuG0XQe55dk15XXX2DSM0DzatRbHWH5A,1256
 phoenix/server/api/types/UMAPPoints.py,sha256=49sWnxjcAJKRzqUY71Fa0tOPti5XjIIFT5cSg6oNu_U,1650
 phoenix/server/api/types/User.py,sha256=iTVUrI8U6-asOhBt1bOZNtoRAaRNC4WHgR1Uv2rHnWQ,1975
@@ -273,16 +286,16 @@ phoenix/server/static/apple-touch-icon-76x76.png,sha256=CT_xT12I0u2i0WU8JzBZBuOQ
 phoenix/server/static/apple-touch-icon.png,sha256=fOfpjqGpWYbJ0eAurKsyoZP1EAs6ZVooBJ_SGk2ZkDs,3801
 phoenix/server/static/favicon.ico,sha256=bY0vvCKRftemZfPShwZtE93DiiQdaYaozkPGwNFr6H8,34494
 phoenix/server/static/modernizr.js,sha256=mvK-XtkNqjOral-QvzoqsyOMECXIMu5BQwSVN_wcU9c,2564
-phoenix/server/static/.vite/manifest.json,sha256=OHnaSfFCgMjSBfa90MdnYJDMisFjQ0KsBhgAJqRxCXo,2163
-phoenix/server/static/assets/components-DrwPLMB6.js,sha256=8Z11dkkE4K_rxpp63EcjuRXrq7HhCiRWPnqP9k10kMo,316774
-phoenix/server/static/assets/index-CTN-OfBU.js,sha256=F5Q2Ru7h1CGEA_nJCVjXpzNhbH5DxudUB6NnCwHFZt0,7954
-phoenix/server/static/assets/pages-Cmqh2i4E.js,sha256=GI_-PAJ9vx1c3_tZxgmmtXke31-baxkeI3pbk6Suv3s,632754
-phoenix/server/static/assets/vendor-Cdrqqth8.js,sha256=XSrmzfFJlLftuuErU4XHtyYlKYyjh0QWF57SHxkF40c,2061915
+phoenix/server/static/.vite/manifest.json,sha256=H-INZnWsmbWImSRNUMxk97n_Gp3Q7kUXeNZme12yLPM,2163
+phoenix/server/static/assets/components-DKH6AzJw.js,sha256=RWkQPJgVGgZqEzBaCuKcxc19e24pfbjaEjwWdxbrkSY,316783
+phoenix/server/static/assets/index-DLV87qiO.js,sha256=F-cdG02QreiLbNUM0Vcp1PTBfjyaHH2XXzHamQ6syQY,8033
+phoenix/server/static/assets/pages-CVY3Nv4Z.js,sha256=QzdU8hUVKqys5cST3gmMy0bNjShoKGbL6btLMoT3cnc,669246
+phoenix/server/static/assets/vendor-Cb3zlNNd.js,sha256=4yHqkg-jnECjHY3bA_-aYIK8paH4j0vY5Vx0qkaW_OU,2061939
 phoenix/server/static/assets/vendor-DxkFTwjz.css,sha256=nZrkr0u6NNElFGvpWHk9GTHeGoibCXCli1bE7mXZGZg,1816
-phoenix/server/static/assets/vendor-arizeai-BSCL03yQ.js,sha256=Dc98gjSV20cL0_jum75PB6YzkxVrjjmWBIWUjZOW4CE,308489
-phoenix/server/static/assets/vendor-codemirror-Utqu7Snw.js,sha256=yakF-gdRQqJHvaxwXGwCq57p6ETOTqvbTFQf4kx7pt8,392748
-phoenix/server/static/assets/vendor-recharts-CNNUvc5T.js,sha256=RwJTuhF_tFrGfEKZEIbnWxXP6_IoWKuFtALElwXyzUc,282859
-phoenix/server/static/assets/vendor-shiki-B6bHerDK.js,sha256=X1XflS-vSyEn9DOgrz0CyvVlkm0BAxvfDW8b_1852UI,8980328
+phoenix/server/static/assets/vendor-arizeai-Buo4e1A6.js,sha256=gDkYYjDuuj7Qt18ZzrQriCjh3pLiqQjyzdYPcrRnmLs,308489
+phoenix/server/static/assets/vendor-codemirror-BuAQiUVf.js,sha256=BwKGD-l2CFA_RJ6J908ykmJkbYR4STxh9Cq3VnIoitg,392748
+phoenix/server/static/assets/vendor-recharts-Cl9dK5tC.js,sha256=02BDRWrMBNjAqY68qazFHrrSIxR_0r1rSlcJChH-ZrU,282859
+phoenix/server/static/assets/vendor-shiki-CazYUixL.js,sha256=yo1zcws-X4sfuIlakSqCUAqp-ohj0OfjXdZC9nzu9oE,8980328
 phoenix/server/static/assets/vendor-three-DwGkEfCM.js,sha256=0D12ZgKzfKCTSdSTKJBFR2RZO_xxeMXrqDp0AszZqHY,620972
 phoenix/server/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/templates/index.html,sha256=ram6sfy2obf_F053ay35V30v-mnRWZ86rK-PstXLy1c,4457
@@ -296,10 +309,10 @@ phoenix/trace/attributes.py,sha256=mHpOtdjGIs-VGSev_dwvxTeZTc9GILovi1Rdnj56HO0,1
 phoenix/trace/errors.py,sha256=wB1z8qdPckngdfU-TORToekvg3344oNFAA83_hC2yFY,180
 phoenix/trace/evaluation_conventions.py,sha256=t8jydM3U0-T5YpiQKRJ3tWdWGlHtzKyttYdw-ddvPOk,1048
 phoenix/trace/exporter.py,sha256=bUXh8fjJIbHurrnt4bAm-cCWqUN5FqNsIc8DZzzklkQ,4695
-phoenix/trace/fixtures.py,sha256=bwDD0rgIRdAi_9zwLLipke50oYuKJvZLvhiGsuvnGbs,19169
+phoenix/trace/fixtures.py,sha256=k5gduf21C5NbZx95O_WB7mqGnrRiig0ti2VVYQiB9vo,19537
 phoenix/trace/otel.py,sha256=FZGzo0NPJqFBLkL_VmQeBne2z-LdHNPTz3oIZOuPPzQ,9997
 phoenix/trace/projects.py,sha256=9dKv1aiKL4IYMFsg2xnC6EOIRO0YHtkR5o9ALHbMK9g,2178
-phoenix/trace/schemas.py,sha256=0tghsJDaLKvSuxDDQGpUEQ0HkMsKmFWPO3lwJrJ9t84,5972
+phoenix/trace/schemas.py,sha256=Su6e567Bei9oo6PsWO2srTcPAj9C2bMgbGtx64Sgqeg,6332
 phoenix/trace/span_evaluations.py,sha256=x3nye9r2SQk1mrR3N05YbuWsgUKpMWwTRBtJTDBSj3Y,13156
 phoenix/trace/span_json_decoder.py,sha256=J1_oDViuUoC4vxPg61U4EOZC1uEMcCzoj-kVjOFEE8k,3224
 phoenix/trace/span_json_encoder.py,sha256=Wa7RvQTZSsFG4pdKSSFSO_4q4DjAOQuG8wLNJvYzsfM,2004
@@ -323,9 +336,9 @@ phoenix/utilities/project.py,sha256=auVpARXkDb-JgeX5f2aStyFIkeKvGwN9l7qrFeJMVxI,
 phoenix/utilities/re.py,sha256=x8Xbk-Wa6qDMAtUd_7JtZvKtrYEuMY-bchB0n163_5c,2006
 phoenix/utilities/span_store.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/utilities/template_formatters.py,sha256=gh9PJD6WEGw7TEYXfSst1UR4pWWwmjxMLrDVQ_CkpkQ,2779
-arize_phoenix-6.2.0.dist-info/METADATA,sha256=7PdpBWXytNhusx8KpjVhN3BtROpDGm6qF2SvpM1Cy4g,23173
-arize_phoenix-6.2.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-arize_phoenix-6.2.0.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
-arize_phoenix-6.2.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-6.2.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-6.2.0.dist-info/RECORD,,
+arize_phoenix-7.0.0.dist-info/METADATA,sha256=JewxawzwVoqTyW-sdqRFKimVnbpgX7UtZfUV3c878SY,23173
+arize_phoenix-7.0.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+arize_phoenix-7.0.0.dist-info/entry_points.txt,sha256=Pgpn8Upxx9P8z8joPXZWl2LlnAlGc3gcQoVchb06X1Q,94
+arize_phoenix-7.0.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-7.0.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-7.0.0.dist-info/RECORD,,

phoenix/db/insertion/span.py

@@ -28,42 +28,77 @@ async def insert_span(
     dialect = SupportedSQLDialect(session.bind.dialect.name)
     if (
         project_rowid := await session.scalar(
-            select(models.Project.id).where(models.Project.name == project_name)
+            select(models.Project.id).filter_by(name=project_name)
         )
     ) is None:
         project_rowid = await session.scalar(
-            insert(models.Project).values(dict(name=project_name)).returning(models.Project.id)
+            insert(models.Project).values(name=project_name).returning(models.Project.id)
         )
     assert project_rowid is not None
-    if trace := await session.scalar(
-        select(models.Trace).where(models.Trace.trace_id == span.context.trace_id)
-    ):
-        trace_rowid = trace.id
-        if span.start_time < trace.start_time or trace.end_time < span.end_time:
-            trace_start_time = min(trace.start_time, span.start_time)
-            trace_end_time = max(trace.end_time, span.end_time)
-            await session.execute(
-                update(models.Trace)
-                .where(models.Trace.id == trace_rowid)
-                .values(
-                    start_time=trace_start_time,
-                    end_time=trace_end_time,
-                )
-            )
+
+    trace_id = span.context.trace_id
+    trace: models.Trace = await session.scalar(
+        select(models.Trace).filter_by(trace_id=trace_id)
+    ) or models.Trace(trace_id=trace_id)
+
+    if trace.id is not None:
+        # Trace record may need to be updated.
+        if trace.end_time < span.end_time:
+            trace.end_time = span.end_time
+            trace.project_rowid = project_rowid
+        if span.start_time < trace.start_time:
+            trace.start_time = span.start_time
     else:
-        trace_rowid = cast(
-            int,
-            await session.scalar(
-                insert(models.Trace)
-                .values(
-                    project_rowid=project_rowid,
-                    trace_id=span.context.trace_id,
-                    start_time=span.start_time,
-                    end_time=span.end_time,
-                )
-                .returning(models.Trace.id)
-            ),
+        # Trace record needs to be persisted for the first time.
+        trace.start_time = span.start_time
+        trace.end_time = span.end_time
+        trace.project_rowid = project_rowid
+        session.add(trace)
+
+    session_id = get_attribute_value(span.attributes, SpanAttributes.SESSION_ID)
+    session_id = str(session_id).strip() if session_id is not None else ""
+    assert isinstance(session_id, str)
+
+    project_session: Optional[models.ProjectSession] = None
+    if trace.project_session_rowid is not None:
+        # ProjectSession record already exists in database for this Trace record, so we fetch
+        # it because it may need to be updated. However, the session_id on the span, if exists,
+        # will be ignored at this point. Otherwise, if session_id is different, we will need
+        # to create a new ProjectSession record, as well as to determine whether the old record
+        # needs to be deleted if this is the last Trace associated with it.
+        project_session = await session.scalar(
+            select(models.ProjectSession).filter_by(id=trace.project_session_rowid)
         )
+    elif session_id:
+        project_session = await session.scalar(
+            select(models.ProjectSession).filter_by(session_id=session_id)
+        ) or models.ProjectSession(session_id=session_id)
+
+    if project_session is not None:
+        if project_session.id is None:
+            # ProjectSession record needs to be persisted for the first time.
+            project_session.start_time = trace.start_time
+            project_session.end_time = trace.end_time
+            project_session.project_id = project_rowid
+            session.add(project_session)
+            await session.flush()
+            assert project_session.id is not None
+            trace.project_session_rowid = project_session.id
+        else:
+            # ProjectSession record may need to be updated.
+            if trace.project_session_rowid is None:
+                trace.project_session_rowid = project_session.id
+            if trace.start_time < project_session.start_time:
+                project_session.start_time = trace.start_time
+            if project_session.end_time < trace.end_time:
+                project_session.end_time = trace.end_time
+
+    await session.flush()
+    assert trace.id is not None
+    assert project_session is None or (
+        project_session.id is not None and project_session.id == trace.project_session_rowid
+    )
+
     cumulative_error_count = int(span.status_code is SpanStatusCode.ERROR)
     cumulative_llm_token_count_prompt = cast(
         int, get_attribute_value(span.attributes, SpanAttributes.LLM_TOKEN_COUNT_PROMPT) or 0
@@ -94,7 +129,7 @@ async def insert_span(
         insert_on_conflict(
             dict(
                 span_id=span.context.span_id,
-                trace_rowid=trace_rowid,
+                trace_rowid=trace.id,
                 parent_id=span.parent_id,
                 span_kind=span.span_kind.value,
                 name=span.name,

phoenix/db/migrations/data_migration_scripts/populate_project_sessions.py

@@ -0,0 +1,199 @@
+# /// script
+# dependencies = [
+#   "arize-phoenix[pg]",
+# ]
+# ///
+"""
+Populate the `project_sessions` table with data from the traces and spans tables.
+
+Environment variables.
+
+- `PHOENIX_SQL_DATABASE_URL` must be set to the database connection string.
+- (optional) Postgresql schema can be set via `PHOENIX_SQL_DATABASE_SCHEMA`.
+"""
+
+import os
+from datetime import datetime
+from time import perf_counter
+from typing import Any, Optional, Union
+
+import sqlean
+from openinference.semconv.trace import SpanAttributes
+from sqlalchemy import (
+    JSON,
+    Engine,
+    NullPool,
+    create_engine,
+    event,
+    func,
+    insert,
+    make_url,
+    select,
+    update,
+)
+from sqlalchemy.dialects import postgresql
+from sqlalchemy.ext.compiler import compiles
+from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, sessionmaker
+
+from phoenix.config import ENV_PHOENIX_SQL_DATABASE_SCHEMA, get_env_database_connection_str
+from phoenix.db.engines import set_postgresql_search_path
+
+
+class JSONB(JSON):
+    # See https://docs.sqlalchemy.org/en/20/core/custom_types.html
+    __visit_name__ = "JSONB"
+
+
+@compiles(JSONB, "sqlite")
+def _(*args: Any, **kwargs: Any) -> str:
+    # See https://docs.sqlalchemy.org/en/20/core/custom_types.html
+    return "JSONB"
+
+
+JSON_ = (
+    JSON()
+    .with_variant(
+        postgresql.JSONB(),  # type: ignore
+        "postgresql",
+    )
+    .with_variant(
+        JSONB(),
+        "sqlite",
+    )
+)
+
+
+class Base(DeclarativeBase): ...
+
+
+class ProjectSession(Base):
+    __tablename__ = "project_sessions"
+    id: Mapped[int] = mapped_column(primary_key=True)
+    session_id: Mapped[str]
+    project_id: Mapped[int]
+    start_time: Mapped[datetime]
+    end_time: Mapped[datetime]
+
+
+class Trace(Base):
+    __tablename__ = "traces"
+    id: Mapped[int] = mapped_column(primary_key=True)
+    project_session_rowid: Mapped[Union[int, None]]
+    project_rowid: Mapped[int]
+    start_time: Mapped[datetime]
+    end_time: Mapped[datetime]
+
+
+class Span(Base):
+    __tablename__ = "spans"
+    id: Mapped[int] = mapped_column(primary_key=True)
+    trace_rowid: Mapped[int]
+    parent_id: Mapped[Optional[str]]
+    attributes: Mapped[dict[str, Any]] = mapped_column(JSON_, nullable=False)
+
+
+SESSION_ID = SpanAttributes.SESSION_ID.split(".")
+USER_ID = SpanAttributes.USER_ID.split(".")
+
+
+def populate_project_sessions(
+    engine: Engine,
+) -> None:
+    sessions_from_span = (
+        select(
+            Span.attributes[SESSION_ID].as_string().label("session_id"),
+            Trace.project_rowid.label("project_id"),
+            Trace.start_time.label("start_time"),
+            func.row_number()
+            .over(
+                partition_by=Span.attributes[SESSION_ID],
+                order_by=[Trace.start_time, Trace.id, Span.id],
+            )
+            .label("rank"),
+            func.max(Trace.end_time)
+            .over(partition_by=Span.attributes[SESSION_ID])
+            .label("end_time"),
+        )
+        .join_from(Span, Trace, Span.trace_rowid == Trace.id)
+        .where(Span.parent_id.is_(None))
+        .where(Span.attributes[SESSION_ID].as_string() != "")
+        .subquery()
+    )
+    sessions_for_trace_id = (
+        select(
+            Span.trace_rowid,
+            ProjectSession.id.label("project_session_rowid"),
+        )
+        .join_from(
+            Span,
+            ProjectSession,
+            Span.attributes[SESSION_ID].as_string() == ProjectSession.session_id,
+        )
+        .where(Span.parent_id.is_(None))
+        .where(Span.attributes[SESSION_ID].as_string() != "")
+        .subquery()
+    )
+    start_time = perf_counter()
+    with sessionmaker(engine).begin() as session:
+        session.execute(
+            insert(ProjectSession).from_select(
+                [
+                    "session_id",
+                    "project_id",
+                    "start_time",
+                    "end_time",
+                ],
+                select(
+                    sessions_from_span.c.session_id,
+                    sessions_from_span.c.project_id,
+                    sessions_from_span.c.start_time,
+                    sessions_from_span.c.end_time,
+                ).where(sessions_from_span.c.rank == 1),
+            )
+        )
+        session.execute(
+            (
+                update(Trace)
+                .values(project_session_rowid=sessions_for_trace_id.c.project_session_rowid)
+                .where(Trace.id == sessions_for_trace_id.c.trace_rowid)
+            )
+        )
+    elapsed_time = perf_counter() - start_time
+    print(f"✅ Populated project_sessions in {elapsed_time:.3f} seconds.")
+
+
+if __name__ == "__main__":
+    sql_database_url = make_url(get_env_database_connection_str())
+    print(f"Using database URL: {sql_database_url}")
+    ans = input("Is that correct? [y]/n: ")
+    if ans.lower().startswith("n"):
+        url = input("Please enter the correct database URL: ")
+        sql_database_url = make_url(url)
+    backend = sql_database_url.get_backend_name()
+    if backend == "sqlite":
+        file = sql_database_url.database
+        engine = create_engine(
+            url=sql_database_url.set(drivername="sqlite"),
+            creator=lambda: sqlean.connect(f"file:///{file}", uri=True),
+            poolclass=NullPool,
+            echo=True,
+        )
+    elif backend == "postgresql":
+        schema = os.getenv(ENV_PHOENIX_SQL_DATABASE_SCHEMA)
+        if schema:
+            print(f"Using schema: {schema}")
+        else:
+            print("No PostgreSQL schema set. (This is the default.)")
+        ans = input("Is that correct? [y]/n: ")
+        if ans.lower().startswith("n"):
+            schema = input("Please enter the correct schema: ")
+        engine = create_engine(
+            url=sql_database_url.set(drivername="postgresql+psycopg"),
+            poolclass=NullPool,
+            echo=True,
+        )
+        if schema:
+            event.listen(engine, "connect", set_postgresql_search_path(schema))
+    else:
+        raise ValueError(f"Unknown database backend: {backend}")
+    populate_project_sessions(engine)

phoenix/db/migrations/versions/4ded9e43755f_create_project_sessions_table.py

@@ -0,0 +1,66 @@
+"""create project_session table
+
+Revision ID: 4ded9e43755f
+Revises: cd164e83824f
+Create Date: 2024-10-08 22:53:24.539786
+
+"""
+
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "4ded9e43755f"
+down_revision: Union[str, None] = "cd164e83824f"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    op.create_table(
+        "project_sessions",
+        sa.Column("id", sa.Integer, primary_key=True),
+        sa.Column("session_id", sa.String, unique=True, nullable=False),
+        sa.Column(
+            "project_id",
+            sa.Integer,
+            sa.ForeignKey("projects.id", ondelete="CASCADE"),
+            nullable=False,
+            index=True,
+        ),
+        sa.Column(
+            "start_time",
+            sa.TIMESTAMP(timezone=True),
+            index=True,
+            nullable=False,
+        ),
+        sa.Column(
+            "end_time",
+            sa.TIMESTAMP(timezone=True),
+            index=True,
+            nullable=False,
+        ),
+    )
+    with op.batch_alter_table("traces") as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                "project_session_rowid",
+                sa.Integer,
+                sa.ForeignKey("project_sessions.id", ondelete="CASCADE"),
+                nullable=True,
+            ),
+        )
+    op.create_index(
+        "ix_traces_project_session_rowid",
+        "traces",
+        ["project_session_rowid"],
+    )
+
+
+def downgrade() -> None:
+    op.drop_index("ix_traces_project_session_rowid")
+    with op.batch_alter_table("traces") as batch_op:
+        batch_op.drop_column("project_session_rowid")
+    op.drop_table("project_sessions")