changed naming of messaging methods
Browse files- ChromaDBFlow.py +2 -2
- VectorStoreFlow.py +2 -2
- run.py +5 -5
ChromaDBFlow.py
CHANGED
|
@@ -143,8 +143,8 @@ class ChromaDBFlow(AtomicFlow):
|
|
| 143 |
)
|
| 144 |
response["retrieved"] = ""
|
| 145 |
|
| 146 |
-
reply = self.
|
| 147 |
input_message = input_message,
|
| 148 |
response = response
|
| 149 |
)
|
| 150 |
-
self.
|
|
|
|
| 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.
|
| 171 |
input_message = input_message,
|
| 172 |
response = response
|
| 173 |
)
|
| 174 |
-
self.
|
|
|
|
| 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.
|
| 119 |
|
| 120 |
#7. ~~~ Run inference ~~~
|
| 121 |
print("##########CHROMA DB DEMO###############")
|
| 122 |
#write to DB
|
| 123 |
-
proxy_flow_cdb.
|
| 124 |
#read from DB
|
| 125 |
-
future = proxy_flow_cdb.
|
| 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.
|
| 139 |
#read from DB
|
| 140 |
-
future = proxy_flow_vs.
|
| 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()
|