Skip to content

Commit 649fdab

Browse files
feat(api): api update
1 parent 2341051 commit 649fdab

File tree

3 files changed

+11
-1
lines changed

3 files changed

+11
-1
lines changed

.stats.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
configured_endpoints: 18
2-
openapi_spec_hash: 20f058101a252f7500803d66aff58eb3
2+
openapi_spec_hash: 153617b7252b1b12f21043b2a1246f8b
33
config_hash: 30422a4611d93ca69e4f1aff60b9ddb5

src/openlayer/types/inference_pipelines/data_stream_params.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,13 +82,19 @@ class ConfigLlmData(TypedDict, total=False):
8282
Applies to RAG use cases. Providing the question enables RAG-specific metrics.
8383
"""
8484

85+
session_id_column_name: Annotated[Optional[str], PropertyInfo(alias="sessionIdColumnName")]
86+
"""Name of the column with the session id."""
87+
8588
timestamp_column_name: Annotated[str, PropertyInfo(alias="timestampColumnName")]
8689
"""Name of the column with the timestamps.
8790
8891
Timestamps must be in UNIX sec format. If not provided, the upload timestamp is
8992
used.
9093
"""
9194

95+
user_id_column_name: Annotated[Optional[str], PropertyInfo(alias="userIdColumnName")]
96+
"""Name of the column with the user id."""
97+
9298

9399
class ConfigTabularClassificationData(TypedDict, total=False):
94100
class_names: Required[Annotated[SequenceNotStr[str], PropertyInfo(alias="classNames")]]

tests/api_resources/inference_pipelines/test_data.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,9 @@ def test_method_stream_with_all_params(self, client: Openlayer) -> None:
5555
}
5656
],
5757
"question_column_name": "question",
58+
"session_id_column_name": "session_id",
5859
"timestamp_column_name": "timestamp",
60+
"user_id_column_name": "user_id",
5961
},
6062
rows=[
6163
{
@@ -174,7 +176,9 @@ async def test_method_stream_with_all_params(self, async_client: AsyncOpenlayer)
174176
}
175177
],
176178
"question_column_name": "question",
179+
"session_id_column_name": "session_id",
177180
"timestamp_column_name": "timestamp",
181+
"user_id_column_name": "user_id",
178182
},
179183
rows=[
180184
{

0 commit comments

Comments
 (0)