diff --git a/src/integrations/pi.ts b/src/integrations/pi.ts
index 616cb7b..ea19385 100644
--- a/src/integrations/pi.ts
+++ b/src/integrations/pi.ts
@@ -7,6 +7,7 @@ import { callDaemon, getDaemonBaseUrl } from "../utils/daemon-client";
 
 type PiModelEntry = {
   id: string;
+  contextWindow?: number;
 };
 
 type PiProviderConfig = {
@@ -59,9 +60,13 @@ export async function installPiIntegration(
       return;
     }
 
-    const providerModels: PiModelEntry[] = models.map((model) => ({
-      id: model.id,
-    }));
+    const providerModels: PiModelEntry[] = models.map((model) => {
+      const entry: PiModelEntry = { id: model.id };
+      if (model.context_length !== undefined && model.context_length > 0) {
+        entry.contextWindow = model.context_length;
+      }
+      return entry;
+    });
 
     piConfig.providers["routstr"] = {
       baseUrl,
diff --git a/src/integrations/registry.ts b/src/integrations/registry.ts
index c69375f..03efe93 100644
--- a/src/integrations/registry.ts
+++ b/src/integrations/registry.ts
@@ -15,6 +15,7 @@ export interface IntegrationConfig {
 export type RoutstrModel = {
   id: string;
   name?: string;
+  context_length?: number;
 };
 
 export type IntegrationFn = (