nbaldwin commited on
Commit
6244aa1
·
1 Parent(s): b922a7a

changed naming of messaging methods

Browse files
Files changed (3) hide show
  1. ChromaDBFlow.py +2 -2
  2. VectorStoreFlow.py +2 -2
  3. run.py +5 -5
ChromaDBFlow.py CHANGED
@@ -143,8 +143,8 @@ class ChromaDBFlow(AtomicFlow):
143
  )
144
  response["retrieved"] = ""
145
 
146
- reply = self._package_output_message(
147
  input_message = input_message,
148
  response = response
149
  )
150
- self.reply_to_message(reply = reply, to = input_message)
 
143
  )
144
  response["retrieved"] = ""
145
 
146
+ reply = self.package_output_message(
147
  input_message = input_message,
148
  response = response
149
  )
150
+ self.send_message(reply, is_reply = True)
VectorStoreFlow.py CHANGED
@@ -167,8 +167,8 @@ class VectorStoreFlow(AtomicFlow):
167
  self.vector_db.add_documents(documents)
168
  response["retrieved"] = ""
169
 
170
- reply = self._package_output_message(
171
  input_message = input_message,
172
  response = response
173
  )
174
- self.reply_to_message(reply = reply, to = input_message)
 
167
  self.vector_db.add_documents(documents)
168
  response["retrieved"] = ""
169
 
170
+ reply = self.package_output_message(
171
  input_message = input_message,
172
  response = response
173
  )
174
+ self.send_message(reply, is_reply = True)
run.py CHANGED
@@ -115,14 +115,14 @@ if __name__ == "__main__":
115
  )
116
 
117
  #option2: use the proxy_flow
118
- #input_message = proxy_flow._package_input_message(data = data)
119
 
120
  #7. ~~~ Run inference ~~~
121
  print("##########CHROMA DB DEMO###############")
122
  #write to DB
123
- proxy_flow_cdb.send_message_async(input_message_write)
124
  #read from DB
125
- future = proxy_flow_cdb.send_message_blocking(input_message_read)
126
 
127
  #uncomment this line if you would like to get the full message back
128
  #reply_message = future.get_message()
@@ -135,9 +135,9 @@ if __name__ == "__main__":
135
 
136
  print("##########VECTOR STORE DEMO###############")
137
  #write to DB
138
- proxy_flow_vs.send_message_async(input_message_write)
139
  #read from DB
140
- future = proxy_flow_vs.send_message_blocking(input_message_read)
141
 
142
  #uncomment this line if you would like to get the full message back
143
  #reply_message = future.get_message()
 
115
  )
116
 
117
  #option2: use the proxy_flow
118
+ #input_message = proxy_flow.package_input_message(data = data)
119
 
120
  #7. ~~~ Run inference ~~~
121
  print("##########CHROMA DB DEMO###############")
122
  #write to DB
123
+ proxy_flow_cdb.send_message(input_message_write)
124
  #read from DB
125
+ future = proxy_flow_cdb.get_reply_future(input_message_read)
126
 
127
  #uncomment this line if you would like to get the full message back
128
  #reply_message = future.get_message()
 
135
 
136
  print("##########VECTOR STORE DEMO###############")
137
  #write to DB
138
+ proxy_flow_vs.send_message(input_message_write)
139
  #read from DB
140
+ future = proxy_flow_vs.get_reply_future(input_message_read)
141
 
142
  #uncomment this line if you would like to get the full message back
143
  #reply_message = future.get_message()