@langinsight/ai-sdk 0.0.2 → 0.0.4

package/README.md

@@ -17,6 +17,10 @@ const handler = new LangInsight.CallbackHandler({
   apiKey: LANGINSIGHT_API_KEY,
   endpoint: LANGINSIGHT_ENDPOINT,
   metadata: { userId: "admin", sessionId: "admin" },
+  onSuccess: (input, output) => {
+    // traceId の取得・DB永続化はここで
+    console.log("Trace sent:", output.id);
+  },
 });
 
 const prompt = "Hello!";
@@ -26,8 +30,7 @@ const result = await generateText({
   prompt,
 });
 
-// Report the result to LangInsight (returns output trace id or undefined)
-const traceId = await handler.report(result, { input: prompt });
+await handler.report(result, { input: prompt });
 ```
 
 ### streamText
@@ -43,7 +46,7 @@ const handler = new LangInsight.CallbackHandler({
   apiKey: LANGINSIGHT_API_KEY,
   endpoint: LANGINSIGHT_ENDPOINT,
   metadata: { userId: "admin", sessionId: "admin" },
-  onSuccess: (trace) => console.log("Trace sent:", trace.id),
+  onSuccess: (input, output) => console.log("Trace sent:", output.id),
   onFailure: (err) => console.error("Trace failed:", err),
 });
 
@@ -62,15 +65,25 @@ for await (const chunk of result.textStream) {
 
 ### Feedback (score)
 
-After sending traces, you can submit feedback for the last output trace using `score()` or the trace id from `onSuccess` / `report()`.
+Trace ID の取得や DB 永続化は `onSuccess` コールバックで受け取った `output` を使って行い、その `output.id` を `score()` に渡してフィードバックを送信する。`handler.result` を await することでも取得可能。
 
 ```typescript
-// Using lastTraceId (set after report() or onFinish)
-await handler.score({ traceId: handler.lastTraceId, value: 1 }); // 👍
-await handler.score({ traceId: handler.lastTraceId, value: -1 }); // 👎
+let lastOutputTraceId: string | undefined;
 
-// Or using the trace id from report() / onSuccess
-await handler.score({ traceId, value: 1 });
+const handler = new LangInsight.CallbackHandler({
+  ...options,
+  onSuccess: (input, output) => {
+    lastOutputTraceId = output.id; // DB 永続化などもここで
+  },
+});
+
+// report() または onFinish の後
+await handler.score({ traceId: lastOutputTraceId!, value: 1 }); // 👍
+await handler.score({ traceId: lastOutputTraceId!, value: -1 }); // 👎
+
+// または handler.result を使用
+const { input, output } = await handler.result;
+console.log("Input trace:", input.id, "Output trace:", output.id);
 ```
 
 ## Options
@@ -84,12 +97,12 @@ await handler.score({ traceId, value: 1 });
 | `metadata.sessionId`| `string` | Yes      | Session ID (e.g. conversation id)                                          |
 | `metadata.modelName` | `string` | No       | Model name (defaults to result response when not set)                       |
 | `metadata.*`         | `any`   | No       | Additional key-value pairs                                                  |
-| `onSuccess` | `(trace: Trace) => void`         | No       | Callback invoked with the output trace when trace submission succeeds       |
+| `onSuccess` | `(input: Trace, output: Trace) => void` | No | Callback invoked with both input and output traces when trace submission succeeds. Use `output.id` to get traceId and persist to DB. |
 | `onFailure` | `(error: Error) => void`         | No       | Callback invoked with the error when trace submission fails                 |
 
 ## API
 
-- **`report(result, { input })`** — Sends traces for a `generateText` result. Returns the output trace id, or `undefined` on failure.
-- **`onFinish({ input })`** — Returns a function suitable for `streamText`’s `onFinish`. Sends traces when the stream finishes.
-- **`score({ traceId, value })`** — Submits feedback (1 = thumbs up, -1 = thumbs down) for the given trace.
-- **`lastTraceId`** — The id of the last output trace sent by this handler (set after `report()` or `onFinish()`).
+- **`report(result, { input })`** — Sends traces for a `generateText` result. Both input and output traces are passed to `onSuccess` on success.
+- **`onFinish({ input })`** — Returns a function suitable for `streamText`'s `onFinish`. Sends traces when the stream finishes; both traces are passed to `onSuccess`.
+- **`score({ traceId, value })`** — Submits feedback (1 = thumbs up, -1 = thumbs down) for the given trace. Use the `output.id` received in `onSuccess` as `traceId`.
+- **`result`** — A promise that resolves with `{ input: Trace, output: Trace }` when traces are sent successfully.

package/dist/index.cjs

@@ -321,10 +321,15 @@ var LangInsight;
     constructor(options) {
       this.options = options;
       this.client = hc(options.endpoint, { fetch });
+      this.result = new Promise((resolve, reject) => {
+        this.resolve = resolve;
+        this.reject = reject;
+      });
     }
     client;
-    /** 最後に送信した output trace の ID */
-    lastTraceId;
+    resolve = null;
+    reject = null;
+    result;
     /**
      * streamText の onFinish コールバックとして使用
      *
@@ -349,35 +354,36 @@ var LangInsight;
     }
     /**
      * generateText の結果を送信
-     *
-     * @returns output trace の traceId（送信失敗時は undefined）
+     * traceId の取得・DB永続化は onSuccess コールバックで trace を受け取り行う。
      *
      * @example
      * ```ts
-     * const handler = new LangInsight.CallbackHandler(options);
-     *
-     * const result = await generateText({
-     *   model: openai("gpt-4o"),
-     *   prompt: "Hello!",
+     * const handler = new LangInsight.CallbackHandler({
+     *   ...options,
+     *   onSuccess: (input, output) => {
+     *     saveTraceToDb(output); // traceId = output.id
+     *   },
      * });
-     *
-     * const traceId = await handler.report(result, { input: "Hello!" });
+     * const result = await generateText({ model: openai("gpt-4o"), prompt: "Hello!" });
+     * await handler.report(result, { input: "Hello!" });
      * ```
      */
     // biome-ignore lint/suspicious/noExplicitAny: should accept any output
     async report(result, params) {
-      return await this.sendTraces({
+      await this.sendTraces({
         input: params.input,
         result
       });
     }
     /**
      * フィードバック（👍/👎）を送信
+     * traceId は onSuccess コールバックで受け取った output.id を使用する。
      *
      * @example
      * ```ts
-     * await handler.score({ traceId, value: 1 }); // 👍
-     * await handler.score({ traceId, value: -1 }); // 👎
+     * onSuccess: (input, output) => { lastOutputTraceId = output.id; },
+     * // ...
+     * await handler.score({ traceId: lastOutputTraceId, value: 1 }); // 👍
      * ```
      */
     async score(params) {
@@ -439,18 +445,19 @@ var LangInsight;
           })
         ]);
         if (!res.every((r) => r.ok)) {
-          console.error(await Promise.all(res.map((r) => r.text())));
+          const [a, b] = await Promise.all(res.map((r) => r.text()));
+          console.error(a, b);
           throw new Error("Failed to send traces");
         }
+        const inputTrace = await res[0].json();
         const outputTrace = await res[1].json();
-        const traceId = outputTrace.id;
-        this.lastTraceId = traceId;
-        this.options.onSuccess?.(outputTrace);
-        return traceId;
+        this.resolve?.({ input: inputTrace, output: outputTrace });
+        this.options.onSuccess?.(inputTrace, outputTrace);
       } catch (error) {
-        console.error(error);
-        this.options.onFailure?.(error instanceof Error ? error : new Error(String(error)));
-        return void 0;
+        const err = error instanceof Error ? error : new Error(String(error));
+        this.reject?.(err);
+        console.error(err);
+        this.options.onFailure?.(err);
       }
     }
   }

package/dist/index.d.cts

@@ -123,14 +123,18 @@ declare namespace LangInsight {
             sessionId: string;
             [key: string]: any;
         };
-        onSuccess?: (trace: TraceEntity) => void;
+        onSuccess?: (input: TraceEntity, output: TraceEntity) => void;
         onFailure?: (error: Error) => void;
     }
     class CallbackHandler {
         private readonly options;
         private readonly client;
-        /** 最後に送信した output trace の ID */
-        lastTraceId: string | undefined;
+        private resolve;
+        private reject;
+        readonly result: Promise<{
+            input: TraceEntity;
+            output: TraceEntity;
+        }>;
         constructor(options: Options);
         /**
          * streamText の onFinish コールバックとして使用
@@ -154,31 +158,32 @@ declare namespace LangInsight {
         }) => Promise<void>;
         /**
          * generateText の結果を送信
-         *
-         * @returns output trace の traceId（送信失敗時は undefined）
+         * traceId の取得・DB永続化は onSuccess コールバックで trace を受け取り行う。
          *
          * @example
          * ```ts
-         * const handler = new LangInsight.CallbackHandler(options);
-         *
-         * const result = await generateText({
-         *   model: openai("gpt-4o"),
-         *   prompt: "Hello!",
+         * const handler = new LangInsight.CallbackHandler({
+         *   ...options,
+         *   onSuccess: (input, output) => {
+         *     saveTraceToDb(output); // traceId = output.id
+         *   },
          * });
-         *
-         * const traceId = await handler.report(result, { input: "Hello!" });
+         * const result = await generateText({ model: openai("gpt-4o"), prompt: "Hello!" });
+         * await handler.report(result, { input: "Hello!" });
          * ```
          */
         report(result: GenerateTextResult<ToolSet, any>, params: {
             input: string;
-        }): Promise<string | undefined>;
+        }): Promise<void>;
         /**
          * フィードバック（👍/👎）を送信
+         * traceId は onSuccess コールバックで受け取った output.id を使用する。
          *
          * @example
          * ```ts
-         * await handler.score({ traceId, value: 1 }); // 👍
-         * await handler.score({ traceId, value: -1 }); // 👎
+         * onSuccess: (input, output) => { lastOutputTraceId = output.id; },
+         * // ...
+         * await handler.score({ traceId: lastOutputTraceId, value: 1 }); // 👍
          * ```
          */
         score(params: {

package/package.json

@@ -2,7 +2,7 @@
   "name": "@langinsight/ai-sdk",
   "module": "dist/index.js",
   "types": "dist/index.d.ts",
-  "version": "0.0.2",
+  "version": "0.0.4",
   "publishConfig": {
     "access": "public"
   },
@@ -23,7 +23,7 @@
   },
   "devDependencies": {
     "@types/bun": "1.3.8",
-    "ai": "^6.0.73",
+    "ai": "^6.0.77",
     "tsup": "^8.5.1",
     "typescript": "^5.9.3"
   }