[1/2] delete label type field

app-server/src/db/labels.rs

@@ -5,13 +5,6 @@ use serde_json::Value;
 use sqlx::{FromRow, PgPool, QueryBuilder};
 use uuid::Uuid;
 
-#[derive(sqlx::Type, Deserialize, Serialize, Debug, Clone, PartialEq)]
-#[sqlx(type_name = "label_type")]
-pub enum LabelType {
-    BOOLEAN,
-    CATEGORICAL,
-}
-
 #[derive(sqlx::Type, Serialize, Deserialize, Clone, PartialEq)]
 #[sqlx(type_name = "label_source")]
 pub enum LabelSource {
@@ -27,7 +20,6 @@ pub struct LabelClass {
     pub created_at: DateTime<Utc>,
     pub name: String,
     pub project_id: Uuid,
-    pub label_type: LabelType,
     pub value_map: Value, // HashMap<String, f64>
     pub description: Option<String>,
     pub evaluator_runnable_graph: Option<Value>,
@@ -59,7 +51,6 @@ pub struct SpanLabel {
     pub label_source: LabelSource,
     pub reasoning: Option<String>,
 
-    pub label_type: LabelType,
     pub class_name: String,
     pub value_map: Value, // Vec<Value>
     pub description: Option<String>,
@@ -90,7 +81,6 @@ pub async fn get_label_classes_by_project_id(
             created_at,
             name,
             project_id,
-            label_type,
             value_map,
             description,
             evaluator_runnable_graph
@@ -141,7 +131,6 @@ pub async fn update_label_class(
             created_at,
             name,
             project_id,
-            label_type,
             value_map,
             description,
             evaluator_runnable_graph",
@@ -237,7 +226,6 @@ pub async fn get_span_labels(pool: &PgPool, span_id: Uuid) -> Result<Vec<SpanLab
             labels.label_source,
             labels.reasoning,
             users.email as user_email,
-            label_classes.label_type,
             label_classes.value_map,
             label_classes.name as class_name,
             label_classes.description

app-server/src/main.rs

@@ -530,7 +530,7 @@ fn main() -> anyhow::Result<()> {
                                         .service(routes::traces::get_single_trace)
                                         .service(routes::traces::get_single_span)
                                         .service(routes::traces::get_sessions)
-                                        .service(routes::labels::get_label_types)
+                                        .service(routes::labels::get_label_classes)
                                         .service(routes::labels::get_span_labels)
                                         .service(routes::labels::update_span_label)
                                         .service(routes::labels::delete_span_label)

app-server/src/routes/labels.rs

@@ -10,7 +10,7 @@ use crate::db::{self, labels::LabelSource, user::User, DB};
 use super::ResponseResult;
 
 #[get("label-classes")]
-pub async fn get_label_types(path: web::Path<Uuid>, db: web::Data<DB>) -> ResponseResult {
+pub async fn get_label_classes(path: web::Path<Uuid>, db: web::Data<DB>) -> ResponseResult {
     let project_id = path.into_inner();
     let label_classes =
         db::labels::get_label_classes_by_project_id(&db.pool, project_id, None).await?;

frontend/app/api/projects/[projectId]/label-classes/route.ts

@@ -24,15 +24,13 @@ export async function POST(req: Request, props: { params: Promise<{ projectId: s
   const params = await props.params;
   const projectId = params.projectId;
 
-
-
   const body = await req.json();
 
   const res = await db.insert(labelClasses).values({
     projectId: projectId,
     name: body.name,
+    labelType: "CATEGORICAL",
     description: body.description,
-    labelType: body.labelType,
     evaluatorRunnableGraph: body.evaluatorRunnableGraph,
     valueMap: body.valueMap
   }).returning();

frontend/components/traces/add-label.tsx

@@ -2,7 +2,7 @@ import { ArrowLeft, HelpCircle, Loader2, Trash2 } from 'lucide-react';
 import { useState } from 'react';
 
 import { useProjectContext } from '@/contexts/project-context';
-import { LabelClass, LabelType, Span } from '@/lib/traces/types';
+import { LabelClass, Span } from '@/lib/traces/types';
 
 import { EvaluatorEditorDialog } from '../evaluator/evaluator-editor-dialog';
 import { Button } from '../ui/button';
@@ -24,18 +24,15 @@ interface AddLabelProps {
 
 export function AddLabel({ span, onClose }: AddLabelProps) {
   const { projectId } = useProjectContext();
-  const [selectedType, setSelectedType] = useState<LabelType>(
-    LabelType.BOOLEAN
-  );
   const [isSaving, setIsSaving] = useState(false);
   const [showEvaluator, setShowEvaluator] = useState(false);
 
   const [labelClass, setLabelClass] = useState<LabelClass>({
     id: '',
     name: '',
     projectId: projectId,
+    labelType: 'CATEGORICAL',
     createdAt: '',
-    labelType: LabelType.CATEGORICAL,
     valueMap: {
       "False": 0,
       "True": 1

frontend/lib/db/migrations/schema.ts

@@ -1,5 +1,5 @@
 import { sql } from "drizzle-orm";
-import { bigint, boolean, doublePrecision, foreignKey, index, integer, jsonb, pgEnum,pgPolicy, pgTable, primaryKey, text, timestamp, unique, uuid } from "drizzle-orm/pg-core";
+import { bigint, boolean, doublePrecision, foreignKey, index, integer, jsonb, pgEnum, pgPolicy, pgTable, primaryKey, text, timestamp, unique, uuid } from "drizzle-orm/pg-core";
 
 export const labelSource = pgEnum("label_source", ['MANUAL', 'AUTO', 'CODE']);
 export const labelType = pgEnum("label_type", ['BOOLEAN', 'CATEGORICAL']);
@@ -34,7 +34,7 @@ export const apiKeys = pgTable("api_keys", {
     foreignColumns: [users.id],
     name: "api_keys_user_id_fkey"
   }).onUpdate("cascade").onDelete("cascade"),
-  enableInsertForAuthenticatedUsersOnly: pgPolicy("Enable insert for authenticated users only", { as: "permissive", for: "all", to: ["service_role"], using: sql`true`, withCheck: sql`true`  }),
+  enableInsertForAuthenticatedUsersOnly: pgPolicy("Enable insert for authenticated users only", { as: "permissive", for: "all", to: ["service_role"], using: sql`true`, withCheck: sql`true` }),
 }));
 
 export const labelClasses = pgTable("label_classes", {
@@ -43,7 +43,7 @@ export const labelClasses = pgTable("label_classes", {
   name: text().notNull(),
   projectId: uuid("project_id").notNull(),
   labelType: labelType("label_type").notNull(),
-  valueMap: jsonb("value_map").default([false,true]).notNull(),
+  valueMap: jsonb("value_map").default([false, true]).notNull(),
   description: text(),
   evaluatorRunnableGraph: jsonb("evaluator_runnable_graph"),
   pipelineVersionId: uuid("pipeline_version_id"),
@@ -296,15 +296,15 @@ export const users = pgTable("users", {
   email: text().notNull(),
 }, (table) => ({
   usersEmailKey: unique("users_email_key").on(table.email),
-  enableInsertForAuthenticatedUsersOnly: pgPolicy("Enable insert for authenticated users only", { as: "permissive", for: "insert", to: ["service_role"], withCheck: sql`true`  }),
+  enableInsertForAuthenticatedUsersOnly: pgPolicy("Enable insert for authenticated users only", { as: "permissive", for: "insert", to: ["service_role"], withCheck: sql`true` }),
 }));
 
 export const playgrounds = pgTable("playgrounds", {
   id: uuid().defaultRandom().primaryKey().notNull(),
   createdAt: timestamp("created_at", { withTimezone: true, mode: 'string' }).defaultNow().notNull(),
   name: text().notNull(),
   projectId: uuid("project_id").notNull(),
-  promptMessages: jsonb("prompt_messages").default([{"role":"user","content":""}]).notNull(),
+  promptMessages: jsonb("prompt_messages").default([{ "role": "user", "content": "" }]).notNull(),
   modelId: text("model_id").default('').notNull(),
   outputSchema: text("output_schema"),
 }, (table) => ({
@@ -503,7 +503,7 @@ export const machines = pgTable("machines", {
     foreignColumns: [projects.id],
     name: "machines_project_id_fkey"
   }).onUpdate("cascade").onDelete("cascade"),
-  machinesPkey: primaryKey({ columns: [table.id, table.projectId], name: "machines_pkey"}),
+  machinesPkey: primaryKey({ columns: [table.id, table.projectId], name: "machines_pkey" }),
 }));
 
 export const datapointToSpan = pgTable("datapoint_to_span", {
@@ -522,7 +522,7 @@ export const datapointToSpan = pgTable("datapoint_to_span", {
     foreignColumns: [spans.spanId, spans.projectId],
     name: "datapoint_to_span_span_id_project_id_fkey"
   }).onUpdate("cascade").onDelete("cascade"),
-  datapointToSpanPkey: primaryKey({ columns: [table.datapointId, table.spanId, table.projectId], name: "datapoint_to_span_pkey"}),
+  datapointToSpanPkey: primaryKey({ columns: [table.datapointId, table.spanId, table.projectId], name: "datapoint_to_span_pkey" }),
 }));
 
 export const spans = pgTable("spans", {
@@ -558,6 +558,6 @@ export const spans = pgTable("spans", {
     foreignColumns: [projects.id],
     name: "spans_project_id_fkey"
   }).onUpdate("cascade"),
-  spansPkey: primaryKey({ columns: [table.spanId, table.projectId], name: "spans_pkey"}),
+  spansPkey: primaryKey({ columns: [table.spanId, table.projectId], name: "spans_pkey" }),
   uniqueSpanIdProjectId: unique("unique_span_id_project_id").on(table.spanId, table.projectId),
 }));

frontend/lib/traces/types.ts

@@ -4,11 +4,6 @@ import { GraphMessagePreview } from '../pipeline/types';
 
 export type TraceMessages = { [key: string]: GraphMessagePreview };
 
-export enum LabelType {
-  CATEGORICAL = 'CATEGORICAL',
-  BOOLEAN = 'BOOLEAN'
-}
-
 export enum LabelSource {
   AUTO = 'AUTO',
   MANUAL = 'MANUAL'