UNPKG

langsmith

Version:

Client library to connect to the LangSmith Observability and Evaluation Platform.

47 lines (46 loc) 2.06 kB
import { getLangSmithEnvironmentVariable } from "../utils/env.js"; // Wrap the default fetch call due to issues with illegal invocations // in some environments: // https://stackoverflow.com/questions/69876859/why-does-bind-fix-failed-to-execute-fetch-on-window-illegal-invocation-err // @ts-expect-error Broad typing to support a range of fetch implementations const DEFAULT_FETCH_IMPLEMENTATION = (...args) => fetch(...args); let globalFetchSupportsWebStreaming = undefined; const LANGSMITH_FETCH_IMPLEMENTATION_KEY = Symbol.for("ls:fetch_implementation"); /** * Overrides the fetch implementation used for LangSmith calls. * You should use this if you need to use an implementation of fetch * other than the default global (e.g. for dealing with proxies). * @param fetch The new fetch function to use. */ export const overrideFetchImplementation = (fetch, supportsWebStreaming) => { globalThis[LANGSMITH_FETCH_IMPLEMENTATION_KEY] = fetch; globalFetchSupportsWebStreaming = supportsWebStreaming; }; export const clearFetchImplementation = () => { delete globalThis[LANGSMITH_FETCH_IMPLEMENTATION_KEY]; globalFetchSupportsWebStreaming = undefined; }; export const _shouldStreamForGlobalFetchImplementation = () => { const overriddenFetchImpl = globalThis[LANGSMITH_FETCH_IMPLEMENTATION_KEY]; if (overriddenFetchImpl === undefined) { return true; } return globalFetchSupportsWebStreaming ?? false; }; /** * @internal */ export const _getFetchImplementation = (debug) => { return async (...args) => { if (debug || getLangSmithEnvironmentVariable("DEBUG") === "true") { const [url, options] = args; console.log(`→ ${options?.method || "GET"} ${url}`); } const res = await (globalThis[LANGSMITH_FETCH_IMPLEMENTATION_KEY] ?? DEFAULT_FETCH_IMPLEMENTATION)(...args); if (debug || getLangSmithEnvironmentVariable("DEBUG") === "true") { console.log(`← ${res.status} ${res.statusText} ${res.url}`); } return res; }; };