Skip to content

Commit a9e0a01

Browse files
authored
Merge pull request #2122 from tisnik/lcore-2755-ability-to-dump-requests-models
LCORE-2755: ability to dump requests models
2 parents 62ad471 + 81fa335 commit a9e0a01

2 files changed

Lines changed: 2928 additions & 51 deletions

File tree

src/utils/models_dumper.py

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
"""Function to dump the schema of all data models into OpenAPI-compatible format."""
22

3+
import models.api.requests as r
34
import models.compaction as models_compaction
45
from utils.openapi_schema_dumper import dump_openapi_schema
56

@@ -20,4 +21,28 @@ def dump_models(filename: str) -> None:
2021
IOError: If the file cannot be written.
2122
"""
2223
models = [models_compaction.ConversationSummary]
24+
25+
# add all requests data models
26+
for model in [
27+
r.ConversationUpdateRequest,
28+
r.FeedbackRequest,
29+
r.FeedbackStatusUpdateRequest,
30+
r.MCPServerRegistrationRequest,
31+
r.ModelFilter,
32+
r.PromptCreateRequest,
33+
r.PromptUpdateRequest,
34+
r.QueryRequest,
35+
r.ResponsesRequest,
36+
r.RlsapiV1Attachment,
37+
r.RlsapiV1CLA,
38+
r.RlsapiV1Context,
39+
r.RlsapiV1InferRequest,
40+
r.RlsapiV1SystemInfo,
41+
r.RlsapiV1Terminal,
42+
r.StreamingInterruptRequest,
43+
r.VectorStoreCreateRequest,
44+
r.VectorStoreFileCreateRequest,
45+
r.VectorStoreUpdateRequest,
46+
]:
47+
models.append(model)
2348
dump_openapi_schema(models, filename)

0 commit comments

Comments
 (0)