feat(api): api update

Author: stainless-app[bot]

.stats.yml

@@ -1,3 +1,3 @@
 configured_endpoints: 18
-openapi_spec_hash: 20f058101a252f7500803d66aff58eb3
+openapi_spec_hash: 153617b7252b1b12f21043b2a1246f8b
 config_hash: 30422a4611d93ca69e4f1aff60b9ddb5

src/resources/inference-pipelines/data.ts

@@ -118,11 +118,21 @@ export namespace DataStreamParams {
      */
     questionColumnName?: string;
 
+    /**
+     * Name of the column with the session id.
+     */
+    sessionIdColumnName?: string | null;
+
     /**
      * Name of the column with the timestamps. Timestamps must be in UNIX sec format.
      * If not provided, the upload timestamp is used.
      */
     timestampColumnName?: string;
+
+    /**
+     * Name of the column with the user id.
+     */
+    userIdColumnName?: string | null;
   }
 
   export namespace LlmData {

tests/api-resources/inference-pipelines/data.test.ts

@@ -37,7 +37,9 @@ describe('resource data', () => {
         numOfTokenColumnName: 'tokens',
         prompt: [{ content: '{{ user_query }}', role: 'user' }],
         questionColumnName: 'question',
+        sessionIdColumnName: 'session_id',
         timestampColumnName: 'timestamp',
+        userIdColumnName: 'user_id',
       },
       rows: [{ user_query: 'bar', output: 'bar', tokens: 'bar', cost: 'bar', timestamp: 'bar' }],
     });