tau-coding-agent 0.1.0

package/extensions/websearch/providers/shared.ts

@@ -0,0 +1,76 @@
+export function withTimeout(signal: AbortSignal | undefined, timeoutMs: number): AbortSignal {
+  return signal
+    ? AbortSignal.any([signal, AbortSignal.timeout(timeoutMs)])
+    : AbortSignal.timeout(timeoutMs);
+}
+
+export async function fetchJson<T>(url: string, options: RequestInit = {}): Promise<T> {
+  return JSON.parse(await fetchText(url, options)) as T;
+}
+
+export async function fetchText(url: string, options: RequestInit = {}): Promise<string> {
+  const response = await fetch(url, options);
+  const text = await response.text();
+  if (!response.ok) {
+    throw new Error(`${response.status} ${response.statusText}${text ? `\n${text}` : ""}`);
+  }
+  return text;
+}
+
+export async function readEventStream(
+  response: Response,
+  onEvent: (event: { event?: string; data: string }) => void,
+): Promise<void> {
+  if (!response.ok) {
+    const text = await response.text();
+    throw new Error(`${response.status} ${response.statusText}${text ? `\n${text}` : ""}`);
+  }
+
+  if (!response.body) {
+    throw new Error("Missing response body.");
+  }
+
+  const decoder = new TextDecoder();
+  let buffer = "";
+
+  for await (const chunk of response.body) {
+    buffer += decoder.decode(chunk, { stream: true });
+    buffer = buffer.replace(/\r\n/g, "\n");
+
+    while (true) {
+      const separatorIndex = buffer.indexOf("\n\n");
+      if (separatorIndex === -1) break;
+
+      const block = buffer.slice(0, separatorIndex);
+      buffer = buffer.slice(separatorIndex + 2);
+      emitEventBlock(block, onEvent);
+    }
+  }
+
+  buffer = buffer.replace(/\r\n/g, "\n");
+  if (buffer.trim().length > 0) {
+    emitEventBlock(buffer, onEvent);
+  }
+}
+
+function emitEventBlock(
+  block: string,
+  onEvent: (event: { event?: string; data: string }) => void,
+): void {
+  let eventName: string | undefined;
+  const dataLines: string[] = [];
+
+  for (const line of block.split(/\r?\n/)) {
+    if (line.startsWith("event:")) {
+      eventName = line.slice(6).trim();
+      continue;
+    }
+    if (line.startsWith("data:")) {
+      dataLines.push(line.slice(5).trimStart());
+    }
+  }
+
+  if (dataLines.length > 0) {
+    onEvent({ event: eventName, data: dataLines.join("\n") });
+  }
+}

package/extensions/websearch/types.ts

@@ -0,0 +1,52 @@
+export type WebsearchBackendId = "openai-codex" | "anthropic" | "gemini";
+export type WebsearchAuthSource = "pi" | "firefox" | "chromium";
+export type WebsearchBrowserFamily = "firefox" | "chromium";
+export type WebsearchRouteId =
+  | "pi:openai-codex"
+  | "pi:anthropic"
+  | "pi:gemini"
+  | "firefox:openai-codex"
+  | "firefox:gemini"
+  | "chromium:openai-codex"
+  | "chromium:gemini";
+
+export interface BrowserCookie {
+  name: string;
+  value: string;
+  domain: string;
+  path: string;
+  secure: boolean;
+}
+
+export interface WebsearchConfig {
+  routes: WebsearchRouteId[];
+  profiles: Partial<Record<WebsearchBrowserFamily, string>>;
+}
+
+export interface BrowserProfile {
+  family: WebsearchBrowserFamily;
+  browserName: string;
+  profileName: string;
+  profilePath: string;
+}
+
+export interface BrowserSession {
+  profile: BrowserProfile;
+  cookies: BrowserCookie[];
+}
+
+export interface WebsearchSource {
+  title: string;
+  url: string;
+  snippet?: string;
+}
+
+export interface WebsearchResult {
+  backend: WebsearchBackendId;
+  authSource: WebsearchAuthSource;
+  browserName?: string;
+  profile?: string;
+  accountLabel?: string;
+  answer: string;
+  sources: WebsearchSource[];
+}