mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-04 20:14:13 +00:00
update generator & yaml spec
This commit is contained in:
parent
ee32de4c3f
commit
838ab91ebf
3 changed files with 214 additions and 45 deletions
|
@ -468,15 +468,11 @@ class Generator:
|
|||
builder = ContentBuilder(self.schema_builder)
|
||||
first = next(iter(op.request_params))
|
||||
request_name, request_type = first
|
||||
if len(op.request_params) == 1 and "Request" in first[1].__name__:
|
||||
# TODO(ashwin): Undo the "Request" hack and this entire block eventually
|
||||
request_name, request_type = first
|
||||
else:
|
||||
from dataclasses import make_dataclass
|
||||
|
||||
op_name = "".join(word.capitalize() for word in op.name.split("_"))
|
||||
request_name = f"{op_name}Request"
|
||||
request_type = make_dataclass(request_name, op.request_params)
|
||||
from dataclasses import make_dataclass
|
||||
op_name = "".join(word.capitalize() for word in op.name.split("_"))
|
||||
request_name = f"{op_name}Request"
|
||||
request_type = make_dataclass(request_name, op.request_params)
|
||||
|
||||
requestBody = RequestBody(
|
||||
content={
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue