From 184e1425a41b57a3375221addf104ccaba7d407c Mon Sep 17 00:00:00 2001
From: Owen <owen@txv.io>
Date: Thu, 6 Aug 2026 17:22:01 -0400
Subject: [PATCH] Private connection working with traefik

---
 server/private/lib/traefik/getTraefikConfig.ts | 2 +-
 server/routers/aiGateway/pipeline.ts           | 2 +-
 src/components/PrivateResourcesTable.tsx       | 5 ++++-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/server/private/lib/traefik/getTraefikConfig.ts b/server/private/lib/traefik/getTraefikConfig.ts
index b00dbd7c8..f51dad24e 100644
--- a/server/private/lib/traefik/getTraefikConfig.ts
+++ b/server/private/lib/traefik/getTraefikConfig.ts
@@ -1696,7 +1696,7 @@ export async function getTraefikConfig(
             const srKey = `inference-sr${sr.siteResourceId}`;
             const routerName = `${srKey}-router`;
             const serviceName = `${srKey}-service`;
-            const rule = `Host(\`${fullDomain}\`) && ClientIP(${exitNode.address})`; // restrict to coming from the exit node ip range that the client is connected to
+            const rule = `Host(\`${fullDomain}\`) && ClientIP(\`${exitNode.address}\`)`; // restrict to coming from the exit node ip range that the client is connected to
 
             let tls: any = {};
             if (!privateConfig.getRawPrivateConfig().flags.use_pangolin_dns) {
diff --git a/server/routers/aiGateway/pipeline.ts b/server/routers/aiGateway/pipeline.ts
index ff76a5317..08ea6d1bc 100644
--- a/server/routers/aiGateway/pipeline.ts
+++ b/server/routers/aiGateway/pipeline.ts
@@ -273,7 +273,7 @@ async function resolveTarget(host: string): Promise<ResolvedTarget | null> {
         .from(siteResources)
         .where(
             and(
-                eq(siteResources.alias, host),
+                eq(siteResources.fullDomain, host),
                 eq(siteResources.mode, "inference"),
                 eq(siteResources.enabled, true)
             )
diff --git a/src/components/PrivateResourcesTable.tsx b/src/components/PrivateResourcesTable.tsx
index cb7d2645c..bc77fbc38 100644
--- a/src/components/PrivateResourcesTable.tsx
+++ b/src/components/PrivateResourcesTable.tsx
@@ -421,7 +421,10 @@ export default function PrivateResourcesTable({
                             />
                         );
                     }
-                    if (resourceRow.mode === "http") {
+                    if (
+                        resourceRow.mode === "http" ||
+                        resourceRow.mode === "inference"
+                    ) {
                         const domainId = resourceRow.domainId;
                         const fullDomain = resourceRow.fullDomain;
                         const url = `${resourceRow.ssl ? "https" : "http"}://${fullDomain}`;