File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 11"""Function to dump the schema of all data models into OpenAPI-compatible format."""
22
33import models .compaction as models_compaction
4+ import models .api .requests as r
45from utils .openapi_schema_dumper import dump_openapi_schema
56
67
@@ -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 )
You can’t perform that action at this time.
0 commit comments