Update prompt_expander.py
#2
by
YiYiXu HF Staff - opened
- prompt_expander.py +2 -0
prompt_expander.py
CHANGED
|
@@ -41,6 +41,7 @@ class GeminiPromptExpander(ModularPipelineBlocks):
|
|
| 41 |
type_hint=str,
|
| 42 |
required=True,
|
| 43 |
description="Prompt to use",
|
|
|
|
| 44 |
)
|
| 45 |
]
|
| 46 |
|
|
@@ -55,6 +56,7 @@ class GeminiPromptExpander(ModularPipelineBlocks):
|
|
| 55 |
"prompt",
|
| 56 |
type_hint=str,
|
| 57 |
description="Expanded prompt by the LLM",
|
|
|
|
| 58 |
),
|
| 59 |
OutputParam(
|
| 60 |
"old_prompt",
|
|
|
|
| 41 |
type_hint=str,
|
| 42 |
required=True,
|
| 43 |
description="Prompt to use",
|
| 44 |
+
metadata={"mellon":"textbox"}
|
| 45 |
)
|
| 46 |
]
|
| 47 |
|
|
|
|
| 56 |
"prompt",
|
| 57 |
type_hint=str,
|
| 58 |
description="Expanded prompt by the LLM",
|
| 59 |
+
metadata={"mellon":"text"}
|
| 60 |
),
|
| 61 |
OutputParam(
|
| 62 |
"old_prompt",
|