Upload folder using huggingface_hub
Browse files- chat/component.yaml +2 -2
- chat/schemas.py +4 -0
chat/component.yaml
CHANGED
|
@@ -5,7 +5,7 @@ implementation:
|
|
| 5 |
entrypoint: run.py
|
| 6 |
inputs:
|
| 7 |
location: node
|
| 8 |
-
save:
|
| 9 |
system_message: You are a helpful AI assistant.
|
| 10 |
license: MIT
|
| 11 |
models:
|
|
@@ -19,6 +19,6 @@ name: chat
|
|
| 19 |
outputs:
|
| 20 |
filename: output.txt
|
| 21 |
location: node
|
| 22 |
-
save:
|
| 23 |
type: chat
|
| 24 |
version: 0.1.0
|
|
|
|
| 5 |
entrypoint: run.py
|
| 6 |
inputs:
|
| 7 |
location: node
|
| 8 |
+
save: false
|
| 9 |
system_message: You are a helpful AI assistant.
|
| 10 |
license: MIT
|
| 11 |
models:
|
|
|
|
| 19 |
outputs:
|
| 20 |
filename: output.txt
|
| 21 |
location: node
|
| 22 |
+
save: false
|
| 23 |
type: chat
|
| 24 |
version: 0.1.0
|
chat/schemas.py
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from pydantic import BaseModel
|
| 2 |
+
|
| 3 |
+
class InputSchema(BaseModel):
|
| 4 |
+
prompt: str
|