Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-07-30 13:48:07 +02:00
Code Issues Projects Releases Wiki Activity

Cleanup embeddings typing

Browse Source
This commit is contained in:
Nicolas Mowen 2025-06-06 08:23:25 -06:00
parent 81d52bcec6
commit 629e04105b
2 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
frigate/comms/embeddings_updater.py
View File

@ -1,10 +1,14 @@
"""Facilitates communication between processes."""
import logging
from enum import Enum
from typing import Any, Callable
import zmq
logger = logging.getLogger(__name__)
SOCKET_REP_REQ = "ipc:///tmp/cache/embeddings"
@ -41,9 +45,16 @@ class EmbeddingsResponder:
break
try:
(topic, value) = self.socket.recv_json(flags=zmq.NOBLOCK)
raw = self.socket.recv_json(flags=zmq.NOBLOCK)
if isinstance(raw, list):
(topic, value) = raw
response = process(topic, value)
else:
logging.warning(
f"Received unexpected data type in ZMQ recv_json: {type(raw)}"
)
response = None
if response is not None:
self.socket.send_json(response)
@ -65,7 +76,7 @@ class EmbeddingsRequestor:
self.socket = self.context.socket(zmq.REQ)
self.socket.connect(SOCKET_REP_REQ)
def send_data(self, topic: str, data: Any) -> str:
def send_data(self, topic: str, data: Any) -> Any:
"""Sends data and then waits for reply."""
try:
self.socket.send_json((topic, data))

2
frigate/comms/inter_process.py
View File

@ -47,7 +47,7 @@ class InterProcessCommunicator(Communicator):
response = self._dispatcher(topic, value)
else:
logging.warning(
f"Received unexpected data type in ZMQ inter-process: {type(raw)}"
f"Received unexpected data type in ZMQ recv_json: {type(raw)}"
)
response = None