fix/ocp_missing_context (#104)

missed OCPQuery class in previous PR
OpenVoiceOS · Jul 12, 2024 · 8f29333 · 8f29333
1 parent 67e3c57
commit 8f29333
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/ovos_bus_client/apis/ocp.py b/ovos_bus_client/apis/ocp.py
@@ -529,20 +529,21 @@ def reset(self):
         else:
             self.has_gui = is_gui_running() or is_gui_connected(self.bus)
 
-    def send(self, skill_id: str = None):
+    @_ensure_message_kwarg()
+    def send(self, skill_id: str = None, source_message: Optional[Message] = None):
         self.query_replies = []
         self.query_timeouts = self.config.get("min_timeout", 5)
         self.search_start = time.time()
         self.searching = True
         self.register_events()
         if skill_id:
-            self.bus.emit(Message(f'ovos.common_play.query.{skill_id}',
-                                  {"phrase": self.query,
-                                   "question_type": self.media_type}))
+            self.bus.emit(source_message.forward(f'ovos.common_play.query.{skill_id}',
+                                                 {"phrase": self.query,
+                                                  "question_type": self.media_type}))
         else:
-            self.bus.emit(Message('ovos.common_play.query',
-                                  {"phrase": self.query,
-                                   "question_type": self.media_type}))
+            self.bus.emit(source_message.forward('ovos.common_play.query',
+                                                 {"phrase": self.query,
+                                                  "question_type": self.media_type}))
 
     def wait(self):
         try: