@langinsight/ai-sdk 0.0.3 → 0.0.4

package/README.md

@@ -17,9 +17,9 @@ const handler = new LangInsight.CallbackHandler({
   apiKey: LANGINSIGHT_API_KEY,
   endpoint: LANGINSIGHT_ENDPOINT,
   metadata: { userId: "admin", sessionId: "admin" },
-  onSuccess: (trace) => {
+  onSuccess: (input, output) => {
     // traceId の取得・DB永続化はここで
-    console.log("Trace sent:", trace.id);
+    console.log("Trace sent:", output.id);
   },
 });
 
@@ -46,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),
 });
 
@@ -65,21 +65,25 @@ for await (const chunk of result.textStream) {
 
 ### Feedback (score)
 
-Trace ID の取得や DB 永続化は `onSuccess` コールバックで受け取った `trace` を使って行い、その `trace.id` を `score()` に渡してフィードバックを送信する。
+Trace ID の取得や DB 永続化は `onSuccess` コールバックで受け取った `output` を使って行い、その `output.id` を `score()` に渡してフィードバックを送信する。`handler.result` を await することでも取得可能。
 
 ```typescript
 let lastOutputTraceId: string | undefined;
 
 const handler = new LangInsight.CallbackHandler({
   ...options,
-  onSuccess: (trace) => {
-    lastOutputTraceId = trace.id; // DB 永続化などもここで
+  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
@@ -93,11 +97,12 @@ await handler.score({ traceId: lastOutputTraceId!, 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. Use this to get `trace.id` and persist to DB. |
+| `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. 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`.
+- **`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,8 +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;
+    resolve = null;
+    reject = null;
+    result;
     /**
      * streamText の onFinish コールバックとして使用
      *
@@ -353,8 +360,8 @@ var LangInsight;
      * ```ts
      * const handler = new LangInsight.CallbackHandler({
      *   ...options,
-     *   onSuccess: (trace) => {
-     *     saveTraceToDb(trace); // traceId = trace.id
+     *   onSuccess: (input, output) => {
+     *     saveTraceToDb(output); // traceId = output.id
      *   },
      * });
      * const result = await generateText({ model: openai("gpt-4o"), prompt: "Hello!" });
@@ -370,11 +377,11 @@ var LangInsight;
     }
     /**
      * フィードバック（👍/👎）を送信
-     * traceId は onSuccess コールバックで受け取った trace.id を使用する。
+     * traceId は onSuccess コールバックで受け取った output.id を使用する。
      *
      * @example
      * ```ts
-     * onSuccess: (trace) => { lastOutputTraceId = trace.id; },
+     * onSuccess: (input, output) => { lastOutputTraceId = output.id; },
      * // ...
      * await handler.score({ traceId: lastOutputTraceId, value: 1 }); // 👍
      * ```
@@ -438,14 +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();
-        this.options.onSuccess?.(outputTrace);
+        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)));
+        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,12 +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;
+        private resolve;
+        private reject;
+        readonly result: Promise<{
+            input: TraceEntity;
+            output: TraceEntity;
+        }>;
         constructor(options: Options);
         /**
          * streamText の onFinish コールバックとして使用
@@ -158,8 +164,8 @@ declare namespace LangInsight {
          * ```ts
          * const handler = new LangInsight.CallbackHandler({
          *   ...options,
-         *   onSuccess: (trace) => {
-         *     saveTraceToDb(trace); // traceId = trace.id
+         *   onSuccess: (input, output) => {
+         *     saveTraceToDb(output); // traceId = output.id
          *   },
          * });
          * const result = await generateText({ model: openai("gpt-4o"), prompt: "Hello!" });
@@ -171,11 +177,11 @@ declare namespace LangInsight {
         }): Promise<void>;
         /**
          * フィードバック（👍/👎）を送信
-         * traceId は onSuccess コールバックで受け取った trace.id を使用する。
+         * traceId は onSuccess コールバックで受け取った output.id を使用する。
          *
          * @example
          * ```ts
-         * onSuccess: (trace) => { lastOutputTraceId = trace.id; },
+         * onSuccess: (input, output) => { lastOutputTraceId = output.id; },
          * // ...
          * await handler.score({ traceId: lastOutputTraceId, value: 1 }); // 👍
          * ```

package/package.json

@@ -2,7 +2,7 @@
   "name": "@langinsight/ai-sdk",
   "module": "dist/index.js",
   "types": "dist/index.d.ts",
-  "version": "0.0.3",
+  "version": "0.0.4",
   "publishConfig": {
     "access": "public"
   },