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()
|