@langinsight/ai-sdk 0.0.2 → 0.0.3

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: (trace) => {
+    // traceId の取得・DB永続化はここで
+    console.log("Trace sent:", trace.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
@@ -62,15 +65,21 @@ 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` コールバックで受け取った `trace` を使って行い、その `trace.id` を `score()` に渡してフィードバックを送信する。
 
 ```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: (trace) => {
+    lastOutputTraceId = trace.id; // DB 永続化などもここで
+  },
+});
+
+// report() または onFinish の後
+await handler.score({ traceId: lastOutputTraceId!, value: 1 }); // 👍
+await handler.score({ traceId: lastOutputTraceId!, value: -1 }); // 👎
 ```
 
 ## Options
@@ -84,12 +93,11 @@ 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` | `(trace: Trace) => void`         | No       | Callback invoked with the output trace when trace submission succeeds. Use this to get `trace.id` 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. Trace is passed to `onSuccess` on success.
+- **`onFinish({ input })`** — Returns a function suitable for `streamText`’s `onFinish`. Sends traces when the stream finishes; trace is passed to `onSuccess`.
+- **`score({ traceId, value })`** — Submits feedback (1 = thumbs up, -1 = thumbs down) for the given trace. Use the `trace.id` received in `onSuccess` as `traceId`.

package/dist/index.cjs

@@ -323,8 +323,6 @@ var LangInsight;
       this.client = hc(options.endpoint, { fetch });
     }
     client;
-    /** 最後に送信した output trace の ID */
-    lastTraceId;
     /**
      * streamText の onFinish コールバックとして使用
      *
@@ -349,35 +347,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: (trace) => {
+     *     saveTraceToDb(trace); // traceId = trace.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 コールバックで受け取った trace.id を使用する。
      *
      * @example
      * ```ts
-     * await handler.score({ traceId, value: 1 }); // 👍
-     * await handler.score({ traceId, value: -1 }); // 👎
+     * onSuccess: (trace) => { lastOutputTraceId = trace.id; },
+     * // ...
+     * await handler.score({ traceId: lastOutputTraceId, value: 1 }); // 👍
      * ```
      */
     async score(params) {
@@ -443,14 +442,10 @@ var LangInsight;
           throw new Error("Failed to send traces");
         }
         const outputTrace = await res[1].json();
-        const traceId = outputTrace.id;
-        this.lastTraceId = traceId;
         this.options.onSuccess?.(outputTrace);
-        return traceId;
       } catch (error) {
         console.error(error);
         this.options.onFailure?.(error instanceof Error ? error : new Error(String(error)));
-        return void 0;
       }
     }
   }

package/dist/index.d.cts

@@ -129,8 +129,6 @@ declare namespace LangInsight {
     class CallbackHandler {
         private readonly options;
         private readonly client;
-        /** 最後に送信した output trace の ID */
-        lastTraceId: string | undefined;
         constructor(options: Options);
         /**
          * streamText の onFinish コールバックとして使用
@@ -154,31 +152,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: (trace) => {
+         *     saveTraceToDb(trace); // traceId = trace.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 コールバックで受け取った trace.id を使用する。
          *
          * @example
          * ```ts
-         * await handler.score({ traceId, value: 1 }); // 👍
-         * await handler.score({ traceId, value: -1 }); // 👎
+         * onSuccess: (trace) => { lastOutputTraceId = trace.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.3",
   "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"
   }