@arizeai/phoenix-client 3.1.0 → 3.2.0

package/dist/src/spans/index.js

@@ -16,6 +16,8 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./addSpanAnnotation"), exports);
 __exportStar(require("./logSpanAnnotations"), exports);
+__exportStar(require("./addDocumentAnnotation"), exports);
+__exportStar(require("./logDocumentAnnotations"), exports);
 __exportStar(require("./getSpans"), exports);
 __exportStar(require("./getSpanAnnotations"), exports);
 __exportStar(require("./deleteSpan"), exports);

package/dist/src/spans/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/spans/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,sDAAoC;AACpC,uDAAqC;AACrC,6CAA2B;AAC3B,uDAAqC;AACrC,+CAA6B"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/spans/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,sDAAoC;AACpC,uDAAqC;AACrC,0DAAwC;AACxC,2DAAyC;AACzC,6CAA2B;AAC3B,uDAAqC;AACrC,+CAA6B"}

package/dist/src/spans/logDocumentAnnotations.d.ts

@@ -0,0 +1,58 @@
+import { ClientFn } from "../types/core";
+import { DocumentAnnotation } from "./types";
+/**
+ * Parameters to log multiple document annotations
+ */
+export interface LogDocumentAnnotationsParams extends ClientFn {
+    /**
+     * The document annotations to log
+     */
+    documentAnnotations: DocumentAnnotation[];
+    /**
+     * If true, the request will be fulfilled synchronously and return the annotation IDs.
+     * If false, the request will be processed asynchronously and return null.
+     * @default false
+     */
+    sync?: boolean;
+}
+/**
+ * Log multiple document annotations in a single request.
+ *
+ * Each annotation can be of type "LLM", "CODE", or "HUMAN" and can include a label, score, explanation, and metadata.
+ * At least one of label, score, or explanation must be provided for each annotation.
+ *
+ * @param params - The parameters to log document annotations
+ * @returns The IDs of the created annotations
+ *
+ * @example
+ * ```ts
+ * const results = await logDocumentAnnotations({
+ *   documentAnnotations: [
+ *     {
+ *       spanId: "123abc",
+ *       documentPosition: 0,
+ *       name: "relevance_score",
+ *       label: "relevant",
+ *       score: 0.95,
+ *       annotatorKind: "LLM",
+ *       explanation: "Document is highly relevant to the query",
+ *       metadata: {
+ *         model: "gpt-4"
+ *       }
+ *     },
+ *     {
+ *       spanId: "123abc",
+ *       documentPosition: 1,
+ *       name: "relevance_score",
+ *       label: "somewhat_relevant",
+ *       score: 0.6,
+ *       annotatorKind: "LLM"
+ *     }
+ *   ]
+ * });
+ * ```
+ */
+export declare function logDocumentAnnotations({ client: _client, documentAnnotations, sync, }: LogDocumentAnnotationsParams): Promise<{
+    id: string;
+}[]>;
+//# sourceMappingURL=logDocumentAnnotations.d.ts.map

package/dist/src/spans/logDocumentAnnotations.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logDocumentAnnotations.d.ts","sourceRoot":"","sources":["../../../src/spans/logDocumentAnnotations.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,kBAAkB,EAA4B,MAAM,SAAS,CAAC;AAEvE;;GAEG;AACH,MAAM,WAAW,4BAA6B,SAAQ,QAAQ;IAC5D;;OAEG;IACH,mBAAmB,EAAE,kBAAkB,EAAE,CAAC;IAC1C;;;;OAIG;IACH,IAAI,CAAC,EAAE,OAAO,CAAC;CAChB;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAoCG;AACH,wBAAsB,sBAAsB,CAAC,EAC3C,MAAM,EAAE,OAAO,EACf,mBAAmB,EACnB,IAAY,GACb,EAAE,4BAA4B,GAAG,OAAO,CAAC;IAAE,EAAE,EAAE,MAAM,CAAA;CAAE,EAAE,CAAC,CAiB1D"}

package/dist/src/spans/logDocumentAnnotations.js

@@ -0,0 +1,58 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.logDocumentAnnotations = logDocumentAnnotations;
+const client_1 = require("../client");
+const types_1 = require("./types");
+/**
+ * Log multiple document annotations in a single request.
+ *
+ * Each annotation can be of type "LLM", "CODE", or "HUMAN" and can include a label, score, explanation, and metadata.
+ * At least one of label, score, or explanation must be provided for each annotation.
+ *
+ * @param params - The parameters to log document annotations
+ * @returns The IDs of the created annotations
+ *
+ * @example
+ * ```ts
+ * const results = await logDocumentAnnotations({
+ *   documentAnnotations: [
+ *     {
+ *       spanId: "123abc",
+ *       documentPosition: 0,
+ *       name: "relevance_score",
+ *       label: "relevant",
+ *       score: 0.95,
+ *       annotatorKind: "LLM",
+ *       explanation: "Document is highly relevant to the query",
+ *       metadata: {
+ *         model: "gpt-4"
+ *       }
+ *     },
+ *     {
+ *       spanId: "123abc",
+ *       documentPosition: 1,
+ *       name: "relevance_score",
+ *       label: "somewhat_relevant",
+ *       score: 0.6,
+ *       annotatorKind: "LLM"
+ *     }
+ *   ]
+ * });
+ * ```
+ */
+async function logDocumentAnnotations({ client: _client, documentAnnotations, sync = false, }) {
+    const client = _client !== null && _client !== void 0 ? _client : (0, client_1.createClient)();
+    const { data, error } = await client.POST("/v1/document_annotations", {
+        params: {
+            query: { sync },
+        },
+        body: {
+            data: documentAnnotations.map(types_1.toDocumentAnnotationData),
+        },
+    });
+    if (error) {
+        throw new Error(`Failed to log document annotations: ${error}`);
+    }
+    return (data === null || data === void 0 ? void 0 : data.data) || [];
+}
+//# sourceMappingURL=logDocumentAnnotations.js.map

package/dist/src/spans/logDocumentAnnotations.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"logDocumentAnnotations.js","sourceRoot":"","sources":["../../../src/spans/logDocumentAnnotations.ts"],"names":[],"mappings":";;AAyDA,wDAqBC;AA9ED,sCAAyC;AAEzC,mCAAuE;AAkBvE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAoCG;AACI,KAAK,UAAU,sBAAsB,CAAC,EAC3C,MAAM,EAAE,OAAO,EACf,mBAAmB,EACnB,IAAI,GAAG,KAAK,GACiB;IAC7B,MAAM,MAAM,GAAG,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,IAAA,qBAAY,GAAE,CAAC;IAEzC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,0BAA0B,EAAE;QACpE,MAAM,EAAE;YACN,KAAK,EAAE,EAAE,IAAI,EAAE;SAChB;QACD,IAAI,EAAE;YACJ,IAAI,EAAE,mBAAmB,CAAC,GAAG,CAAC,gCAAwB,CAAC;SACxD;KACF,CAAC,CAAC;IAEH,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,IAAI,KAAK,CAAC,uCAAuC,KAAK,EAAE,CAAC,CAAC;IAClE,CAAC;IAED,OAAO,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,KAAI,EAAE,CAAC;AAC1B,CAAC"}

package/dist/src/spans/logSpanAnnotations.d.ts

@@ -8,6 +8,12 @@ export interface LogSpanAnnotationsParams extends ClientFn {
      * The span annotations to log
      */
     spanAnnotations: SpanAnnotation[];
+    /**
+     * If true, the request will be fulfilled synchronously and return the annotation IDs.
+     * If false, the request will be processed asynchronously and return null.
+     * @default false
+     */
+    sync?: boolean;
 }
 /**
  * Log multiple span annotations in a single request.
@@ -44,7 +50,7 @@ export interface LogSpanAnnotationsParams extends ClientFn {
  * });
  * ```
  */
-export declare function logSpanAnnotations({ client: _client, spanAnnotations, }: LogSpanAnnotationsParams): Promise<{
+export declare function logSpanAnnotations({ client: _client, spanAnnotations, sync, }: LogSpanAnnotationsParams): Promise<{
     id: string;
 }[]>;
 //# sourceMappingURL=logSpanAnnotations.d.ts.map

package/dist/src/spans/logSpanAnnotations.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"logSpanAnnotations.d.ts","sourceRoot":"","sources":["../../../src/spans/logSpanAnnotations.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,cAAc,EAAwB,MAAM,SAAS,CAAC;AAE/D;;GAEG;AACH,MAAM,WAAW,wBAAyB,SAAQ,QAAQ;IACxD;;OAEG;IACH,eAAe,EAAE,cAAc,EAAE,CAAC;CACnC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACH,wBAAsB,kBAAkB,CAAC,EACvC,MAAM,EAAE,OAAO,EACf,eAAe,GAChB,EAAE,wBAAwB,GAAG,OAAO,CAAC;IAAE,EAAE,EAAE,MAAM,CAAA;CAAE,EAAE,CAAC,CAkBtD"}
+{"version":3,"file":"logSpanAnnotations.d.ts","sourceRoot":"","sources":["../../../src/spans/logSpanAnnotations.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,cAAc,EAAwB,MAAM,SAAS,CAAC;AAE/D;;GAEG;AACH,MAAM,WAAW,wBAAyB,SAAQ,QAAQ;IACxD;;OAEG;IACH,eAAe,EAAE,cAAc,EAAE,CAAC;IAClC;;;;OAIG;IACH,IAAI,CAAC,EAAE,OAAO,CAAC;CAChB;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACH,wBAAsB,kBAAkB,CAAC,EACvC,MAAM,EAAE,OAAO,EACf,eAAe,EACf,IAAY,GACb,EAAE,wBAAwB,GAAG,OAAO,CAAC;IAAE,EAAE,EAAE,MAAM,CAAA;CAAE,EAAE,CAAC,CAiBtD"}

package/dist/src/spans/logSpanAnnotations.js

@@ -38,10 +38,12 @@ const types_1 = require("./types");
  * });
  * ```
  */
-async function logSpanAnnotations({ client: _client, spanAnnotations, }) {
-    var _a;
+async function logSpanAnnotations({ client: _client, spanAnnotations, sync = false, }) {
     const client = _client !== null && _client !== void 0 ? _client : (0, client_1.createClient)();
     const { data, error } = await client.POST("/v1/span_annotations", {
+        params: {
+            query: { sync },
+        },
         body: {
             data: spanAnnotations.map(types_1.toSpanAnnotationData),
         },
@@ -49,9 +51,6 @@ async function logSpanAnnotations({ client: _client, spanAnnotations, }) {
     if (error) {
         throw new Error(`Failed to log span annotations: ${error}`);
     }
-    if (!((_a = data === null || data === void 0 ? void 0 : data.data) === null || _a === void 0 ? void 0 : _a.length)) {
-        throw new Error("No annotation IDs returned from server");
-    }
-    return data.data;
+    return (data === null || data === void 0 ? void 0 : data.data) || [];
 }
 //# sourceMappingURL=logSpanAnnotations.js.map

package/dist/src/spans/logSpanAnnotations.js.map

@@ -1 +1 @@
-{"version":3,"file":"logSpanAnnotations.js","sourceRoot":"","sources":["../../../src/spans/logSpanAnnotations.ts"],"names":[],"mappings":";;AAiDA,gDAqBC;AAtED,sCAAyC;AAEzC,mCAA+D;AAY/D;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACI,KAAK,UAAU,kBAAkB,CAAC,EACvC,MAAM,EAAE,OAAO,EACf,eAAe,GACU;;IACzB,MAAM,MAAM,GAAG,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,IAAA,qBAAY,GAAE,CAAC;IAEzC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,sBAAsB,EAAE;QAChE,IAAI,EAAE;YACJ,IAAI,EAAE,eAAe,CAAC,GAAG,CAAC,4BAAoB,CAAC;SAChD;KACF,CAAC,CAAC;IAEH,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,IAAI,KAAK,CAAC,mCAAmC,KAAK,EAAE,CAAC,CAAC;IAC9D,CAAC;IAED,IAAI,CAAC,CAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,MAAM,CAAA,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC5D,CAAC;IAED,OAAO,IAAI,CAAC,IAAI,CAAC;AACnB,CAAC"}
+{"version":3,"file":"logSpanAnnotations.js","sourceRoot":"","sources":["../../../src/spans/logSpanAnnotations.ts"],"names":[],"mappings":";;AAuDA,gDAqBC;AA5ED,sCAAyC;AAEzC,mCAA+D;AAkB/D;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACI,KAAK,UAAU,kBAAkB,CAAC,EACvC,MAAM,EAAE,OAAO,EACf,eAAe,EACf,IAAI,GAAG,KAAK,GACa;IACzB,MAAM,MAAM,GAAG,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,IAAA,qBAAY,GAAE,CAAC;IAEzC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,sBAAsB,EAAE;QAChE,MAAM,EAAE;YACN,KAAK,EAAE,EAAE,IAAI,EAAE;SAChB;QACD,IAAI,EAAE;YACJ,IAAI,EAAE,eAAe,CAAC,GAAG,CAAC,4BAAoB,CAAC;SAChD;KACF,CAAC,CAAC;IAEH,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,IAAI,KAAK,CAAC,mCAAmC,KAAK,EAAE,CAAC,CAAC;IAC9D,CAAC;IAED,OAAO,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,KAAI,EAAE,CAAC;AAC1B,CAAC"}

package/dist/src/spans/types.d.ts

@@ -1,5 +1,6 @@
 import { paths } from "../__generated__/api/v1";
 type SpanAnnotationData = paths["/v1/span_annotations"]["post"]["requestBody"]["content"]["application/json"]["data"][0];
+type SpanDocumentAnnotationData = paths["/v1/document_annotations"]["post"]["requestBody"]["content"]["application/json"]["data"][0];
 /**
  * Parameters for a single span annotation
  */
@@ -20,6 +21,10 @@ export interface SpanAnnotation {
      * The score assigned by the annotation
      */
     score?: number;
+    /**
+     * Explanation of the annotation result
+     */
+    explanation?: string;
     /**
      * The identifier of the annotation. If provided, the annotation will be updated if it already exists.
      */
@@ -35,9 +40,52 @@ export interface SpanAnnotation {
      */
     annotatorKind?: SpanAnnotationData["annotator_kind"];
 }
+/**
+ * Parameters for a single document annotation
+ */
+export interface DocumentAnnotation {
+    /**
+     * The OpenTelemetry Span ID (hex format without 0x prefix)
+     */
+    spanId: string;
+    /**
+     * The 0-based index of the document within the span
+     */
+    documentPosition: number;
+    /**
+     * The name of the annotation
+     */
+    name: string;
+    /**
+     * The label assigned by the annotation
+     */
+    label?: string;
+    /**
+     * The score assigned by the annotation
+     */
+    score?: number;
+    /**
+     * Explanation of the annotation result
+     */
+    explanation?: string;
+    /**
+     * Metadata for the annotation
+     */
+    metadata?: Record<string, unknown>;
+    /**
+     * The kind of annotator used for the annotation
+     * Can be "HUMAN", "LLM", or "CODE"
+     * @default "HUMAN"
+     */
+    annotatorKind?: SpanDocumentAnnotationData["annotator_kind"];
+}
 /**
  * Convert a SpanAnnotation to the API format
  */
 export declare function toSpanAnnotationData(annotation: SpanAnnotation): SpanAnnotationData;
+/**
+ * Convert a DocumentAnnotation to the API format
+ */
+export declare function toDocumentAnnotationData(annotation: DocumentAnnotation): SpanDocumentAnnotationData;
 export {};
 //# sourceMappingURL=types.d.ts.map

package/dist/src/spans/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/spans/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC;AAEhD,KAAK,kBAAkB,GACrB,KAAK,CAAC,sBAAsB,CAAC,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,CAAC,kBAAkB,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;AAEjG;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IACb;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC;;;;OAIG;IACH,aAAa,CAAC,EAAE,kBAAkB,CAAC,gBAAgB,CAAC,CAAC;CACtD;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAClC,UAAU,EAAE,cAAc,GACzB,kBAAkB,CAapB"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/spans/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC;AAEhD,KAAK,kBAAkB,GACrB,KAAK,CAAC,sBAAsB,CAAC,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,CAAC,kBAAkB,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;AAEjG,KAAK,0BAA0B,GAC7B,KAAK,CAAC,0BAA0B,CAAC,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,CAAC,kBAAkB,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;AAErG;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IACb;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC;;;;OAIG;IACH,aAAa,CAAC,EAAE,kBAAkB,CAAC,gBAAgB,CAAC,CAAC;CACtD;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,gBAAgB,EAAE,MAAM,CAAC;IACzB;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IACb;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC;;;;OAIG;IACH,aAAa,CAAC,EAAE,0BAA0B,CAAC,gBAAgB,CAAC,CAAC;CAC9D;AA2CD;;GAEG;AACH,wBAAgB,oBAAoB,CAClC,UAAU,EAAE,cAAc,GACzB,kBAAkB,CAWpB;AAED;;GAEG;AACH,wBAAgB,wBAAwB,CACtC,UAAU,EAAE,kBAAkB,GAC7B,0BAA0B,CAW5B"}

package/dist/src/spans/types.js

@@ -1,22 +1,57 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.toSpanAnnotationData = toSpanAnnotationData;
+exports.toDocumentAnnotationData = toDocumentAnnotationData;
+/**
+ * Build and validate annotation result fields
+ */
+function buildAnnotationResult(annotation, annotationType) {
+    const result = {};
+    // Build result with trimming for string fields
+    if (annotation.label !== undefined) {
+        result.label = annotation.label.trim() || null;
+    }
+    if (annotation.score !== undefined) {
+        result.score = annotation.score;
+    }
+    if (annotation.explanation !== undefined) {
+        result.explanation = annotation.explanation.trim() || null;
+    }
+    // Validate that at least one result field is provided
+    const hasValidResult = result.label || result.score !== undefined || result.explanation;
+    if (!hasValidResult) {
+        throw new Error(`At least one of label, score, or explanation must be provided for ${annotationType} annotation`);
+    }
+    return result;
+}
 /**
  * Convert a SpanAnnotation to the API format
  */
 function toSpanAnnotationData(annotation) {
-    var _a, _b, _c, _d, _e;
+    var _a, _b, _c, _d;
+    const result = buildAnnotationResult(annotation, "span");
+    return {
+        span_id: annotation.spanId.trim(),
+        name: annotation.name.trim(),
+        annotator_kind: (_a = annotation.annotatorKind) !== null && _a !== void 0 ? _a : "HUMAN",
+        result,
+        metadata: (_b = annotation.metadata) !== null && _b !== void 0 ? _b : null,
+        identifier: (_d = (_c = annotation.identifier) === null || _c === void 0 ? void 0 : _c.trim()) !== null && _d !== void 0 ? _d : "",
+    };
+}
+/**
+ * Convert a DocumentAnnotation to the API format
+ */
+function toDocumentAnnotationData(annotation) {
+    var _a, _b;
+    const result = buildAnnotationResult(annotation, "document");
     return {
-        span_id: annotation.spanId,
-        name: annotation.name,
+        span_id: annotation.spanId.trim(),
+        document_position: annotation.documentPosition,
+        name: annotation.name.trim(),
         annotator_kind: (_a = annotation.annotatorKind) !== null && _a !== void 0 ? _a : "HUMAN",
-        result: {
-            label: (_b = annotation.label) !== null && _b !== void 0 ? _b : null,
-            score: (_c = annotation.score) !== null && _c !== void 0 ? _c : null,
-            explanation: null,
-        },
-        metadata: (_d = annotation.metadata) !== null && _d !== void 0 ? _d : null,
-        identifier: (_e = annotation.identifier) !== null && _e !== void 0 ? _e : "",
+        result,
+        metadata: (_b = annotation.metadata) !== null && _b !== void 0 ? _b : null,
     };
 }
 //# sourceMappingURL=types.js.map

package/dist/src/spans/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/spans/types.ts"],"names":[],"mappings":";;AA4CA,oDAeC;AAlBD;;GAEG;AACH,SAAgB,oBAAoB,CAClC,UAA0B;;IAE1B,OAAO;QACL,OAAO,EAAE,UAAU,CAAC,MAAM;QAC1B,IAAI,EAAE,UAAU,CAAC,IAAI;QACrB,cAAc,EAAE,MAAA,UAAU,CAAC,aAAa,mCAAI,OAAO;QACnD,MAAM,EAAE;YACN,KAAK,EAAE,MAAA,UAAU,CAAC,KAAK,mCAAI,IAAI;YAC/B,KAAK,EAAE,MAAA,UAAU,CAAC,KAAK,mCAAI,IAAI;YAC/B,WAAW,EAAE,IAAI;SAClB;QACD,QAAQ,EAAE,MAAA,UAAU,CAAC,QAAQ,mCAAI,IAAI;QACrC,UAAU,EAAE,MAAA,UAAU,CAAC,UAAU,mCAAI,EAAE;KACxC,CAAC;AACJ,CAAC"}
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/spans/types.ts"],"names":[],"mappings":";;AAoIA,oDAaC;AAKD,4DAaC;AArED;;GAEG;AACH,SAAS,qBAAqB,CAC5B,UAGC,EACD,cAAmC;IAEnC,MAAM,MAAM,GAAqB,EAAE,CAAC;IAEpC,+CAA+C;IAC/C,IAAI,UAAU,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,IAAI,CAAC;IACjD,CAAC;IACD,IAAI,UAAU,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC;IAClC,CAAC;IACD,IAAI,UAAU,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;QACzC,MAAM,CAAC,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC,IAAI,EAAE,IAAI,IAAI,CAAC;IAC7D,CAAC;IAED,sDAAsD;IACtD,MAAM,cAAc,GAClB,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,IAAI,MAAM,CAAC,WAAW,CAAC;IACnE,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,MAAM,IAAI,KAAK,CACb,qEAAqE,cAAc,aAAa,CACjG,CAAC;IACJ,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAgB,oBAAoB,CAClC,UAA0B;;IAE1B,MAAM,MAAM,GAAG,qBAAqB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAEzD,OAAO;QACL,OAAO,EAAE,UAAU,CAAC,MAAM,CAAC,IAAI,EAAE;QACjC,IAAI,EAAE,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE;QAC5B,cAAc,EAAE,MAAA,UAAU,CAAC,aAAa,mCAAI,OAAO;QACnD,MAAM;QACN,QAAQ,EAAE,MAAA,UAAU,CAAC,QAAQ,mCAAI,IAAI;QACrC,UAAU,EAAE,MAAA,MAAA,UAAU,CAAC,UAAU,0CAAE,IAAI,EAAE,mCAAI,EAAE;KAChD,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAgB,wBAAwB,CACtC,UAA8B;;IAE9B,MAAM,MAAM,GAAG,qBAAqB,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;IAE7D,OAAO;QACL,OAAO,EAAE,UAAU,CAAC,MAAM,CAAC,IAAI,EAAE;QACjC,iBAAiB,EAAE,UAAU,CAAC,gBAAgB;QAC9C,IAAI,EAAE,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE;QAC5B,cAAc,EAAE,MAAA,UAAU,CAAC,aAAa,mCAAI,OAAO;QACnD,MAAM;QACN,QAAQ,EAAE,MAAA,UAAU,CAAC,QAAQ,mCAAI,IAAI;KACtC,CAAC;AACJ,CAAC"}

package/dist/src/utils/formatPromptMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"formatPromptMessages.d.ts","sourceRoot":"","sources":["../../../src/utils/formatPromptMessages.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAClD,OAAO,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AAO3E;;;;;;;GAOG;AACH,wBAAgB,oBAAoB,CAClC,MAAM,EAAE,oBAAoB,EAC5B,cAAc,EAAE,iBAAiB,EAAE,EACnC,SAAS,GAAE,SAAc;;;;;;;mBA4D+t2E,uDAAsB;;;;;;;;;IAlB/w2E"}
+{"version":3,"file":"formatPromptMessages.d.ts","sourceRoot":"","sources":["../../../src/utils/formatPromptMessages.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAClD,OAAO,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AAO3E;;;;;;;GAOG;AACH,wBAAgB,oBAAoB,CAClC,MAAM,EAAE,oBAAoB,EAC5B,cAAc,EAAE,iBAAiB,EAAE,EACnC,SAAS,GAAE,SAAc;;;;;;;mBA4Dmy7E,uDAAsB;;;;;;;;;IAlBn17E"}

package/dist/src/utils/getPromptBySelector.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"getPromptBySelector.d.ts","sourceRoot":"","sources":["../../../src/utils/getPromptBySelector.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAGzC;;GAEG;AACH,MAAM,MAAM,yBAAyB,GAAG,QAAQ,GAAG;IACjD;;OAEG;IACH,MAAM,EAAE,cAAc,CAAC;CACxB,CAAC;AAEF;;;;;;;;;;GAUG;AACH,wBAAsB,mBAAmB,CAAC,EACxC,MAAM,EAAE,OAAO,EACf,MAAM,GACP,EAAE,yBAAyB;;oBAkD0qgE,uDAAsB;;cAAkI,uDAAsB,yBAAwB,uDAAsB;mBAAoD,uDAAsB;qBAAoD,uDAAsB;2BAAqG,uDAAsB,uCAAsC,uDAAsB,4CAA2C,uDAAsB,0CAAyC,uDAAsB,uCAAsC,uDAAsB,yCAAwC,uDAAsB,oCAAmC,uDAAsB,uCAAsC,uDAAsB;YAAsD,uDAAsB;sBAAwF,uDAAsB;;UADrxiE"}
+{"version":3,"file":"getPromptBySelector.d.ts","sourceRoot":"","sources":["../../../src/utils/getPromptBySelector.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAGzC;;GAEG;AACH,MAAM,MAAM,yBAAyB,GAAG,QAAQ,GAAG;IACjD;;OAEG;IACH,MAAM,EAAE,cAAc,CAAC;CACxB,CAAC;AAEF;;;;;;;;;;GAUG;AACH,wBAAsB,mBAAmB,CAAC,EACxC,MAAM,EAAE,OAAO,EACf,MAAM,GACP,EAAE,yBAAyB;;oBAkD86jE,uDAAsB;;cAAkI,uDAAsB,yBAAwB,uDAAsB;mBAAoD,uDAAsB;qBAAoD,uDAAsB;2BAAqG,uDAAsB,uCAAsC,uDAAsB,4CAA2C,uDAAsB,0CAAyC,uDAAsB,uCAAsC,uDAAsB,yCAAwC,uDAAsB,oCAAmC,uDAAsB,uCAAsC,uDAAsB;YAAsD,uDAAsB;sBAAwF,uDAAsB;;UADzhmE"}