nbaldwin commited on
Commit
11dd0a0
1 Parent(s): a5ead65

changed naming of messaging methods

Browse files
Files changed (2) hide show
  1. LCToolFlow.py +2 -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._package_output_message(
97
  input_message=input_message,
98
  response = {"observation": observation}
99
  )
100
 
101
- self.reply_to_message(reply=reply, to=input_message)
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._package_input_message(data = data)
80
 
81
  #7. ~~~ Run inference ~~~
82
- future = proxy_flow.send_message_blocking(input_message)
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()