mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-04 12:07:34 +00:00
update wrapper request
This commit is contained in:
parent
838ab91ebf
commit
84b8a53a34
3 changed files with 556 additions and 113 deletions
|
@ -470,9 +470,16 @@ class Generator:
|
|||
request_name, request_type = first
|
||||
|
||||
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)
|
||||
|
||||
if len(op.request_params) == 1 and "Request" in first[1].__name__:
|
||||
# TODO(ashwin): Undo the "Request" hack and this entire block eventually
|
||||
op_name = "".join(word.capitalize() for word in op.name.split("_"))
|
||||
request_name = f"{op_name}RequestWrapper"
|
||||
request_type = make_dataclass(request_name, op.request_params)
|
||||
else:
|
||||
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