From 8140eafba457b443aa4684bd2ecf978ed0dc85df Mon Sep 17 00:00:00 2001
From: JohnGilhuly <jgilhuly@arize.com>
Date: Mon, 16 Sep 2024 21:18:31 -0700
Subject: [PATCH] feat: adds o1 testing demo project (#4636)

* Adds o1 testing demo project
---
 src/phoenix/trace/fixtures.py                 |  8 ++++
 tutorials/tracing/groq_tracing_tutorial.ipynb | 47 +++----------------
 2 files changed, 14 insertions(+), 41 deletions(-)

diff --git a/src/phoenix/trace/fixtures.py b/src/phoenix/trace/fixtures.py
index af41e037dc..fa41a9c612 100644
--- a/src/phoenix/trace/fixtures.py
+++ b/src/phoenix/trace/fixtures.py
@@ -147,6 +147,13 @@ class TracesFixture:
     ),
 )
 
+demo_o1_preview_timeseries_testing_fixture = TracesFixture(
+    name="demo_o1_preview_timeseries_evals",
+    project_name="demo_o1_preview_timeseries",
+    description="Shows the traces for a timeseries evaluation of o1-preview",
+    file_name="o1-preview-testing-traces.parquet",
+)
+
 demo_llama_index_rag_llm_fixture = TracesFixture(
     name="demo_llama_index_rag_llm",
     project_name="demo_llama_index_rag_llm",
@@ -252,6 +259,7 @@ class TracesFixture:
 TRACES_FIXTURES: List[TracesFixture] = [
     demo_llama_index_rag_fixture,
     demo_llama_index_rag_llm_fixture,
+    demo_o1_preview_timeseries_testing_fixture,
     llama_index_rag_fixture,
     llama_index_rag_fixture_with_davinci,
     langchain_rag_stuff_document_chain_fixture,
diff --git a/tutorials/tracing/groq_tracing_tutorial.ipynb b/tutorials/tracing/groq_tracing_tutorial.ipynb
index 244bc7c625..f8971c7869 100644
--- a/tutorials/tracing/groq_tracing_tutorial.ipynb
+++ b/tutorials/tracing/groq_tracing_tutorial.ipynb
@@ -42,13 +42,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "colab": {
-     "base_uri": "https://localhost:8080/"
-    },
-    "id": "Y52gaK9n_oTZ",
-    "outputId": "be9068b4-17e8-419e-a393-04d057c2d7ce"
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "! pip install -q openinference-instrumentation-groq groq arize-phoenix"
@@ -57,9 +51,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "id": "Z_II5NGXA3k-"
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "import os\n",
@@ -85,14 +77,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "colab": {
-     "base_uri": "https://localhost:8080/",
-     "height": 86
-    },
-    "id": "9o1L0rXTD_ED",
-    "outputId": "fa273488-b603-4c2e-d028-d412dc025e46"
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "import phoenix as px\n",
@@ -116,9 +101,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "id": "jdP_j9YF_qAr"
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "from openinference.instrumentation.groq import GroqInstrumentor\n",
@@ -140,9 +123,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "id": "0VE3hHDF_9TB"
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "from groq import Groq\n",
@@ -192,24 +173,8 @@
   }
  ],
  "metadata": {
-  "colab": {
-   "provenance": []
-  },
-  "kernelspec": {
-   "display_name": "Python 3",
-   "name": "python3"
-  },
   "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.10.14"
+   "name": "python"
   }
  },
  "nbformat": 4,