Skip to content

Commit 238c498

Browse files
authored
Merge pull request #2169 from tisnik/lcore-2755-agents-models-dump
LCORE-2755: Agents models dump
2 parents 110841f + bc90bce commit 238c498

2 files changed

Lines changed: 313 additions & 0 deletions

File tree

src/utils/models_dumper.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import models.api.responses.error as e
55
import models.api.responses.successful as s
66
import models.common as c
7+
import models.common.agents as a
78
import models.compaction as models_compaction
89
from utils.openapi_schema_dumper import dump_openapi_schema
910

@@ -120,6 +121,20 @@ def dump_models(filename: str) -> None:
120121
c.Transcript,
121122
c.TranscriptMetadata,
122123
c.TurnSummary,
124+
a.EndEventData,
125+
a.EndStreamPayload,
126+
a.ErrorEventData,
127+
a.ErrorStreamPayload,
128+
a.InterruptedEventData,
129+
a.InterruptedStreamPayload,
130+
a.StartEventData,
131+
a.StartStreamPayload,
132+
a.StreamPayloadBase,
133+
a.TokenChunkData,
134+
a.TokenStreamPayload,
135+
a.ToolCallStreamPayload,
136+
a.ToolResultStreamPayload,
137+
a.TurnCompleteStreamPayload,
123138
]:
124139
models.append(model)
125140
dump_openapi_schema(models, filename)

0 commit comments

Comments
 (0)