Move terminate code to base_agent

vocodedev · Aug 1, 2023 · 4cccaf9 · 4cccaf9
1 parent e1773cb
commit 4cccaf9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/vocode/streaming/agent/base_agent.py b/vocode/streaming/agent/base_agent.py
@@ -424,6 +424,15 @@ async def get_tracer_name_start(self) -> str:
         self.tracer_name_start: str = tracer_name_start
         return tracer_name_start
 
+    def terminate(self):
+        if (
+            hasattr(self.agent_config, "vector_db_config")
+            and self.agent_config.vector_db_config
+        ):
+            self.logger.debug("Terminating vector db")
+            self.vector_db.tear_down()
+        return super().terminate()
+
     async def respond(
         self,
         human_input,

diff --git a/vocode/streaming/streaming_conversation.py b/vocode/streaming/streaming_conversation.py
@@ -627,9 +627,6 @@ async def terminate(self):
         await self.synthesizer.tear_down()
         self.logger.debug("Terminating agent")
         self.agent.terminate()
-        if hasattr(self.agent.agent_config, "vector_db_config") and self.agent.agent_config.vector_db_config:
-            self.logger.debug("Terminating vector db")
-            self.agent.vector_db.tear_down()
         self.logger.debug("Terminating output device")
         self.output_device.terminate()
         self.logger.debug("Terminating speech transcriber")