changed naming of messaging methods
Browse files- LCToolFlow.py +2 -2
- run.py +2 -2
LCToolFlow.py
CHANGED
|
@@ -93,10 +93,10 @@ class LCToolFlow(AtomicFlow):
|
|
| 93 |
input_data = input_message.data
|
| 94 |
observation = self.backend.run(tool_input=input_data)
|
| 95 |
|
| 96 |
-
reply = self.
|
| 97 |
input_message=input_message,
|
| 98 |
response = {"observation": observation}
|
| 99 |
)
|
| 100 |
|
| 101 |
-
self.
|
| 102 |
|
|
|
|
| 93 |
input_data = input_message.data
|
| 94 |
observation = self.backend.run(tool_input=input_data)
|
| 95 |
|
| 96 |
+
reply = self.package_output_message(
|
| 97 |
input_message=input_message,
|
| 98 |
response = {"observation": observation}
|
| 99 |
)
|
| 100 |
|
| 101 |
+
self.send_message(reply, is_reply=True)
|
| 102 |
|
run.py
CHANGED
|
@@ -76,10 +76,10 @@ if __name__ == "__main__":
|
|
| 76 |
)
|
| 77 |
|
| 78 |
#option2: use the proxy_flow
|
| 79 |
-
#input_message = proxy_flow.
|
| 80 |
|
| 81 |
#7. ~~~ Run inference ~~~
|
| 82 |
-
future = proxy_flow.
|
| 83 |
|
| 84 |
#uncomment this line if you would like to get the full message back
|
| 85 |
#reply_message = future.get_message()
|
|
|
|
| 76 |
)
|
| 77 |
|
| 78 |
#option2: use the proxy_flow
|
| 79 |
+
#input_message = proxy_flow.package_input_message(data = data)
|
| 80 |
|
| 81 |
#7. ~~~ Run inference ~~~
|
| 82 |
+
future = proxy_flow.get_reply_future(input_message)
|
| 83 |
|
| 84 |
#uncomment this line if you would like to get the full message back
|
| 85 |
#reply_message = future.get_message()
|