[object Object]

← back to Exo

move udp and tailscale into their own modules

073b3ffce872ab8ddc38160e2f075edfe1748672 · 2024-09-28 02:59:47 +0400 · Alex Cheema

Files touched

Diff

commit 073b3ffce872ab8ddc38160e2f075edfe1748672
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Sat Sep 28 02:59:47 2024 +0400

    move udp and tailscale into their own modules
---
 exo/networking/{ => tailscale}/tailscale_discovery.py      | 4 ++--
 exo/networking/{ => tailscale}/tailscale_helpers.py        | 0
 exo/networking/{ => tailscale}/test_tailscale_discovery.py | 2 +-
 exo/networking/{ => udp}/test_udp_discovery.py             | 2 +-
 exo/networking/{ => udp}/udp_discovery.py                  | 4 ++--
 main.py                                                    | 4 ++--
 6 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/exo/networking/tailscale_discovery.py b/exo/networking/tailscale/tailscale_discovery.py
similarity index 98%
rename from exo/networking/tailscale_discovery.py
rename to exo/networking/tailscale/tailscale_discovery.py
index 5613a53e..f48bbaed 100644
--- a/exo/networking/tailscale_discovery.py
+++ b/exo/networking/tailscale/tailscale_discovery.py
@@ -3,8 +3,8 @@ import time
 import traceback
 from typing import List, Dict, Callable, Tuple
 from tailscale import Tailscale, Device
-from .discovery import Discovery
-from .peer_handle import PeerHandle
+from exo.networking.discovery import Discovery
+from exo.networking.peer_handle import PeerHandle
 from exo.topology.device_capabilities import DeviceCapabilities, device_capabilities, UNKNOWN_DEVICE_CAPABILITIES
 from exo.helpers import DEBUG, DEBUG_DISCOVERY
 from .tailscale_helpers import get_device_id, update_device_attributes, get_device_attributes, update_device_attributes
diff --git a/exo/networking/tailscale_helpers.py b/exo/networking/tailscale/tailscale_helpers.py
similarity index 100%
rename from exo/networking/tailscale_helpers.py
rename to exo/networking/tailscale/tailscale_helpers.py
diff --git a/exo/networking/test_tailscale_discovery.py b/exo/networking/tailscale/test_tailscale_discovery.py
similarity index 94%
rename from exo/networking/test_tailscale_discovery.py
rename to exo/networking/tailscale/test_tailscale_discovery.py
index 053084fc..44bcca48 100644
--- a/exo/networking/test_tailscale_discovery.py
+++ b/exo/networking/tailscale/test_tailscale_discovery.py
@@ -2,7 +2,7 @@ import os
 import asyncio
 import unittest
 from unittest import mock
-from exo.networking.tailscale_discovery import TailscaleDiscovery
+from exo.networking.tailscale.tailscale_discovery import TailscaleDiscovery
 from exo.networking.peer_handle import PeerHandle
 
 class TestTailscaleDiscovery(unittest.IsolatedAsyncioTestCase):
diff --git a/exo/networking/test_udp_discovery.py b/exo/networking/udp/test_udp_discovery.py
similarity index 98%
rename from exo/networking/test_udp_discovery.py
rename to exo/networking/udp/test_udp_discovery.py
index fb5094a0..cada016a 100644
--- a/exo/networking/test_udp_discovery.py
+++ b/exo/networking/udp/test_udp_discovery.py
@@ -1,7 +1,7 @@
 import asyncio
 import unittest
 from unittest import mock
-from exo.networking.udp_discovery import UDPDiscovery
+from exo.networking.udp.udp_discovery import UDPDiscovery
 from exo.networking.grpc.grpc_peer_handle import GRPCPeerHandle
 from exo.networking.grpc.grpc_server import GRPCServer
 from exo.orchestration.node import Node
diff --git a/exo/networking/udp_discovery.py b/exo/networking/udp/udp_discovery.py
similarity index 98%
rename from exo/networking/udp_discovery.py
rename to exo/networking/udp/udp_discovery.py
index 9c454f4b..d982981a 100644
--- a/exo/networking/udp_discovery.py
+++ b/exo/networking/udp/udp_discovery.py
@@ -4,8 +4,8 @@ import socket
 import time
 import traceback
 from typing import List, Dict, Callable, Tuple, Coroutine
-from .discovery import Discovery
-from .peer_handle import PeerHandle
+from exo.networking.discovery import Discovery
+from exo.networking.peer_handle import PeerHandle
 from exo.topology.device_capabilities import DeviceCapabilities, device_capabilities, UNKNOWN_DEVICE_CAPABILITIES
 from exo.helpers import DEBUG, DEBUG_DISCOVERY, get_all_ip_addresses
 
diff --git a/main.py b/main.py
index ab5c12f0..6132bc67 100644
--- a/main.py
+++ b/main.py
@@ -7,8 +7,8 @@ import traceback
 import uuid
 from exo.orchestration.standard_node import StandardNode
 from exo.networking.grpc.grpc_server import GRPCServer
-from exo.networking.udp_discovery import UDPDiscovery
-from exo.networking.tailscale_discovery import TailscaleDiscovery
+from exo.networking.udp.udp_discovery import UDPDiscovery
+from exo.networking.tailscale.tailscale_discovery import TailscaleDiscovery
 from exo.networking.grpc.grpc_peer_handle import GRPCPeerHandle
 from exo.topology.ring_memory_weighted_partitioning_strategy import RingMemoryWeightedPartitioningStrategy
 from exo.api import ChatGPTAPI

← 2ebcf5f4 fix llama 3.2 issue with apply_chat_template assuming messag  ·  back to Exo  ·  upgrade mlx to 0.18.0 3e13e5ed →