@giselles-ai/sandbox-agent 0.1.8 → 0.1.9

package/dist/react/index.d.ts

@@ -28,6 +28,7 @@ type AgentToolHandler = {
 type UseAgentOptions = {
     endpoint: string;
     tools?: Record<string, AgentToolHandler>;
+    headers?: Record<string, string>;
 };
 type AgentHookState = {
     status: AgentStatus;
@@ -43,6 +44,6 @@ type AgentHookState = {
         document?: string;
     }) => Promise<void>;
 };
-declare function useAgent({ endpoint, tools }: UseAgentOptions): AgentHookState;
+declare function useAgent({ endpoint, tools, headers: customHeaders, }: UseAgentOptions): AgentHookState;
 
 export { type AgentHookState, type AgentMessage, type AgentStatus, type AgentToolContext, type AgentToolHandler, type ToolEvent, type UseAgentOptions, useAgent };

package/dist/react/index.js

@@ -77,7 +77,11 @@ function normalizeTools(tools) {
   }
   return Object.values(tools);
 }
-function useAgent({ endpoint, tools }) {
+function useAgent({
+  endpoint,
+  tools,
+  headers: customHeaders
+}) {
   const [relayStatus, setRelayStatus] = useState("idle");
   const [running, setRunning] = useState(false);
   const [messages, setMessages] = useState([]);
@@ -128,7 +132,8 @@ function useAgent({ endpoint, tools }) {
       const response = await fetch(relayBase, {
         method: "POST",
         headers: {
-          "content-type": "application/json"
+          "content-type": "application/json",
+          ...customHeaders
         },
         body: JSON.stringify({
           type: "relay.respond",
@@ -144,7 +149,7 @@ function useAgent({ endpoint, tools }) {
         ok: response.ok
       });
     },
-    [normalizedEndpoint]
+    [normalizedEndpoint, customHeaders]
   );
   const addWarnings = useCallback((next) => {
     if (next.length === 0) {
@@ -221,9 +226,16 @@ function useAgent({ endpoint, tools }) {
     cleanupRelay();
     setRelayStatus("connecting");
     const relayBase = currentSession.relayUrl ?? normalizedEndpoint;
-    const source = new EventSource(
-      `${relayBase}?type=relay.events&sessionId=${encodeURIComponent(currentSession.sessionId)}&token=${encodeURIComponent(currentSession.token)}`
-    );
+    const sseUrl = new URL(relayBase, window.location.origin);
+    sseUrl.searchParams.set("type", "relay.events");
+    sseUrl.searchParams.set("sessionId", currentSession.sessionId);
+    sseUrl.searchParams.set("token", currentSession.token);
+    if (customHeaders) {
+      for (const [key, value] of Object.entries(customHeaders)) {
+        sseUrl.searchParams.set(key, value);
+      }
+    }
+    const source = new EventSource(sseUrl.toString());
     console.info(`${LOG_PREFIX} sse.connect`, {
       sessionId: currentSession.sessionId
     });
@@ -272,7 +284,7 @@ function useAgent({ endpoint, tools }) {
         }, 1500);
       }
     };
-  }, [cleanupRelay, handleRelayEvent, normalizedEndpoint]);
+  }, [cleanupRelay, handleRelayEvent, normalizedEndpoint, customHeaders]);
   useEffect(() => {
     mountedRef.current = true;
     return () => {
@@ -442,7 +454,8 @@ function useAgent({ endpoint, tools }) {
         const response = await fetch(normalizedEndpoint, {
           method: "POST",
           headers: {
-            "content-type": "application/json"
+            "content-type": "application/json",
+            ...customHeaders
           },
           body: JSON.stringify({
             type: "agent.run",
@@ -499,7 +512,8 @@ function useAgent({ endpoint, tools }) {
       handleStreamEvent,
       normalizedEndpoint,
       running,
-      sandboxId
+      sandboxId,
+      customHeaders
     ]
   );
   const status = running ? "running" : relayStatus;

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@giselles-ai/sandbox-agent",
-	"version": "0.1.8",
+	"version": "0.1.9",
 	"type": "module",
 	"sideEffects": false,
 	"license": "Apache-2.0",