@stainless-api/docs 0.1.0-beta.135 → 0.1.0-beta.137

package/stl-docs/chat/stainless-handler/index.ts

@@ -1,126 +0,0 @@
-import type { DocsLanguage } from '@stainless-api/docs-ui/routing';
-import { JSONParser } from '@streamparser/json-whatwg';
-import {
-  type RequestBody,
-  responseChunk,
-  type FeedbackRequestBody,
-  feedbackResponseBody,
-  type MetadataRequestBody,
-  metadataResponseBody,
-} from '../schemas';
-import { streamAsyncIterator } from '../stream-util';
-import { DocsChatHandler } from '../docs-chat-handler';
-
-const API_URL = new URL('https://app.stainless.com/api/');
-const CHAT_ENDPOINT = new URL('ai/get-agentic-help', API_URL);
-
-const FEEDBACK_ENDPOINT = (spanId: string) => new URL(`ai/agentic-help/${spanId}/score`, API_URL);
-const METADATA_ENDPOINT = (spanId: string) => new URL(`ai/agentic-help/${spanId}/metadata`, API_URL);
-
-/**
- * Identifier for tracking unique users in braintrust
- */
-function getClientId() {
-  let clientId = localStorage.getItem('stainless-client-id');
-  if (!clientId) {
-    clientId = crypto.randomUUID();
-    localStorage.setItem('stainless-client-id', clientId);
-  }
-  return clientId;
-}
-
-/** Context on what the user is currently viewing to pass to the agent */
-function getPageContext({ siteTitle }: { siteTitle: string | undefined }) {
-  const { href } = window.location;
-  const markdownUrl = `${href.replace(/\/$/, '')}/index.md`;
-  const pageTitle = document.querySelector('h1')?.textContent;
-  return [
-    `The user is viewing a documentation page${siteTitle ? ` for ${siteTitle}` : ''}.`,
-    `- Content URL: ${markdownUrl}`,
-    pageTitle && `- Page title: "${pageTitle}"`,
-    // TODO: include stainless path here? does the agent know how to use it?
-    // TODO: pass more of the page content into context without the agent having to retrieve it
-  ]
-    .filter(Boolean)
-    .join('\n');
-}
-
-export class StainlessHandler implements DocsChatHandler {
-  constructor(
-    private language: DocsLanguage,
-    private siteTitle: string | undefined,
-    private project: string,
-  ) {}
-  /**
-   * Stream chat response from the server
-   */
-  async *generateResponse(
-    {
-      query,
-      priorMessages,
-    }: {
-      query: string;
-      priorMessages: NonNullable<RequestBody['additionalContext']>['prior_messages'];
-    },
-    abortSignal: AbortSignal,
-  ) {
-    const res = await fetch(CHAT_ENDPOINT, {
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-      },
-      body: JSON.stringify({
-        query,
-        sdk: { project: this.project, language: this.language },
-        stream: true,
-        additionalContext: {
-          prior_messages: priorMessages,
-          intent: getPageContext({ siteTitle: this.siteTitle }),
-        },
-        browser_id: getClientId(),
-      } satisfies RequestBody),
-
-      signal: abortSignal,
-    });
-
-    if (!res.ok || !res.body) throw new Error(`Chat request failed with status ${res.status}`);
-
-    const parser = new JSONParser({ separator: '\n', paths: ['$'] });
-    for await (const chunk of streamAsyncIterator(res.body.pipeThrough(parser))) {
-      const chunkParsed = responseChunk.safeParse(chunk.value);
-      if (chunkParsed.success) yield chunkParsed.data;
-    }
-  }
-
-  /**
-   * Attach a score to a response
-   */
-  async onRate(spanId: string, score: 0 | 1) {
-    const res = await fetch(FEEDBACK_ENDPOINT(spanId), {
-      method: 'PUT',
-      headers: {
-        'Content-Type': 'application/json',
-      },
-      body: JSON.stringify({ score } satisfies FeedbackRequestBody),
-    });
-
-    if (!res.ok) throw new Error(`Feedback request failed with status ${res.status}`);
-    return feedbackResponseBody.parse(await res.json());
-  }
-
-  /**
-   * Attach metadata to a response
-   */
-  async onAssignMetadata(spanId: string, metadata: Record<string, string>) {
-    const res = await fetch(METADATA_ENDPOINT(spanId), {
-      method: 'PUT',
-      headers: {
-        'Content-Type': 'application/json',
-      },
-      body: JSON.stringify({ metadata } satisfies MetadataRequestBody),
-    });
-
-    if (!res.ok) throw new Error(`Metadata request failed with status ${res.status}`);
-    return metadataResponseBody.parse(await res.json());
-  }
-}

package/stl-docs/chat/stream-util.ts

@@ -1,16 +0,0 @@
-/*
- * https://jakearchibald.com/2017/async-iterators-and-generators/#making-streams-iterate
- * safari does not yet support consuming ReadableStream as AsyncIterable
- */
-export async function* streamAsyncIterator<T>(stream: ReadableStream<T>) {
-  const reader = stream.getReader();
-  try {
-    while (true) {
-      const { done, value } = await reader.read();
-      if (done) return;
-      yield value;
-    }
-  } finally {
-    reader.releaseLock();
-  }
-}