arize-phoenix 5.3.1__tar.gz → 5.4.0__tar.gz

{arize_phoenix-5.3.1 → arize_phoenix-5.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arize-phoenix
-Version: 5.3.1
+Version: 5.4.0
 Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues

{arize_phoenix-5.3.1 → arize_phoenix-5.4.0}/src/phoenix/server/api/subscriptions.py

@@ -45,6 +45,7 @@ from phoenix.server.api.input_types.ChatCompletionMessageInput import ChatComple
 from phoenix.server.api.input_types.InvocationParameters import InvocationParameters
 from phoenix.server.api.types.ChatCompletionMessageRole import ChatCompletionMessageRole
 from phoenix.server.api.types.GenerativeProvider import GenerativeProviderKey
+from phoenix.server.api.types.Span import Span, to_gql_span
 from phoenix.server.dml_event import SpanInsertEvent
 from phoenix.trace.attributes import unflatten
 from phoenix.utilities.json import jsonify
@@ -94,8 +95,14 @@ class ToolCallChunk:
     function: FunctionCallChunk
 
 
-ChatCompletionChunk: TypeAlias = Annotated[
-    Union[TextChunk, ToolCallChunk], strawberry.union("ChatCompletionChunk")
+@strawberry.type
+class FinishedChatCompletion:
+    span: Span
+
+
+ChatCompletionSubscriptionPayload: TypeAlias = Annotated[
+    Union[TextChunk, ToolCallChunk, FinishedChatCompletion],
+    strawberry.union("ChatCompletionSubscriptionPayload"),
 ]
 
 
@@ -160,7 +167,7 @@ class Subscription:
     @strawberry.subscription
     async def chat_completion(
         self, info: Info[Context, None], input: ChatCompletionInput
-    ) -> AsyncIterator[ChatCompletionChunk]:
+    ) -> AsyncIterator[ChatCompletionSubscriptionPayload]:
         from openai import NOT_GIVEN, AsyncAzureOpenAI, AsyncOpenAI
         from openai.types.chat import ChatCompletionStreamOptionsParam
 
@@ -284,36 +291,35 @@ class Subscription:
                         description="Traces from prompt playground",
                     )
                 )
-            trace_rowid = await session.scalar(
-                insert(models.Trace)
-                .returning(models.Trace.id)
-                .values(
-                    project_rowid=playground_project_id,
-                    trace_id=trace_id,
-                    start_time=start_time,
-                    end_time=end_time,
-                )
+            playground_trace = models.Trace(
+                project_rowid=playground_project_id,
+                trace_id=trace_id,
+                start_time=start_time,
+                end_time=end_time,
             )
-            await session.execute(
-                insert(models.Span).values(
-                    trace_rowid=trace_rowid,
-                    span_id=span_id,
-                    parent_id=None,
-                    name=span_name,
-                    span_kind=LLM,
-                    start_time=start_time,
-                    end_time=end_time,
-                    attributes=unflatten(attributes.items()),
-                    events=finished_span.events,
-                    status_code=status.status_code.name,
-                    status_message=status.description or "",
-                    cumulative_error_count=int(not status.is_ok),
-                    cumulative_llm_token_count_prompt=prompt_tokens,
-                    cumulative_llm_token_count_completion=completion_tokens,
-                    llm_token_count_prompt=prompt_tokens,
-                    llm_token_count_completion=completion_tokens,
-                )
+            playground_span = models.Span(
+                trace_rowid=playground_trace.id,
+                span_id=span_id,
+                parent_id=None,
+                name=span_name,
+                span_kind=LLM,
+                start_time=start_time,
+                end_time=end_time,
+                attributes=unflatten(attributes.items()),
+                events=finished_span.events,
+                status_code=status.status_code.name,
+                status_message=status.description or "",
+                cumulative_error_count=int(not status.is_ok),
+                cumulative_llm_token_count_prompt=prompt_tokens,
+                cumulative_llm_token_count_completion=completion_tokens,
+                llm_token_count_prompt=prompt_tokens,
+                llm_token_count_completion=completion_tokens,
+                trace=playground_trace,
             )
+            session.add(playground_trace)
+            session.add(playground_span)
+            await session.flush()
+            yield FinishedChatCompletion(span=to_gql_span(playground_span))
         info.context.event_queue.put(SpanInsertEvent(ids=(playground_project_id,)))
 
 

arize_phoenix-5.4.0/src/phoenix/server/static/.vite/manifest.json

@@ -0,0 +1,78 @@
+{
+  "_components-8zh9kCOG.js": {
+    "file": "assets/components-8zh9kCOG.js",
+    "name": "components",
+    "imports": [
+      "_vendor-6IcPAw_j.js",
+      "_vendor-arizeai-uC2sozJ6.js",
+      "_pages-CFNU-U_5.js",
+      "_vendor-three-DwGkEfCM.js",
+      "_vendor-codemirror-DVE2_WBr.js"
+    ]
+  },
+  "_pages-CFNU-U_5.js": {
+    "file": "assets/pages-CFNU-U_5.js",
+    "name": "pages",
+    "imports": [
+      "_vendor-6IcPAw_j.js",
+      "_vendor-arizeai-uC2sozJ6.js",
+      "_components-8zh9kCOG.js",
+      "_vendor-recharts-DwrexFA4.js",
+      "_vendor-codemirror-DVE2_WBr.js"
+    ]
+  },
+  "_vendor-!~{003}~.js": {
+    "file": "assets/vendor-DxkFTwjz.css",
+    "src": "_vendor-!~{003}~.js"
+  },
+  "_vendor-6IcPAw_j.js": {
+    "file": "assets/vendor-6IcPAw_j.js",
+    "name": "vendor",
+    "imports": [
+      "_vendor-three-DwGkEfCM.js"
+    ],
+    "css": [
+      "assets/vendor-DxkFTwjz.css"
+    ]
+  },
+  "_vendor-arizeai-uC2sozJ6.js": {
+    "file": "assets/vendor-arizeai-uC2sozJ6.js",
+    "name": "vendor-arizeai",
+    "imports": [
+      "_vendor-6IcPAw_j.js"
+    ]
+  },
+  "_vendor-codemirror-DVE2_WBr.js": {
+    "file": "assets/vendor-codemirror-DVE2_WBr.js",
+    "name": "vendor-codemirror",
+    "imports": [
+      "_vendor-6IcPAw_j.js"
+    ]
+  },
+  "_vendor-recharts-DwrexFA4.js": {
+    "file": "assets/vendor-recharts-DwrexFA4.js",
+    "name": "vendor-recharts",
+    "imports": [
+      "_vendor-6IcPAw_j.js"
+    ]
+  },
+  "_vendor-three-DwGkEfCM.js": {
+    "file": "assets/vendor-three-DwGkEfCM.js",
+    "name": "vendor-three"
+  },
+  "index.tsx": {
+    "file": "assets/index-Ci93KI-L.js",
+    "name": "index",
+    "src": "index.tsx",
+    "isEntry": true,
+    "imports": [
+      "_vendor-6IcPAw_j.js",
+      "_vendor-arizeai-uC2sozJ6.js",
+      "_pages-CFNU-U_5.js",
+      "_components-8zh9kCOG.js",
+      "_vendor-three-DwGkEfCM.js",
+      "_vendor-recharts-DwrexFA4.js",
+      "_vendor-codemirror-DVE2_WBr.js"
+    ]
+  }
+}