@erdoai/server 0.1.4 → 0.1.5

package/dist/index.js

@@ -1,241 +1,2 @@
-// src/config.ts
-var DEFAULT_ENDPOINT = "https://api.erdo.ai";
-function getConfigFromEnv() {
-  const env = typeof process !== "undefined" ? process.env : {};
-  return {
-    endpoint: env.ERDO_ENDPOINT || env.NEXT_PUBLIC_ERDO_ENDPOINT || DEFAULT_ENDPOINT,
-    authToken: env.ERDO_AUTH_TOKEN || env.NEXT_PUBLIC_ERDO_API_KEY
-  };
-}
-function resolveConfig(userConfig) {
-  const envConfig = getConfigFromEnv();
-  const endpoint = userConfig?.endpoint || envConfig.endpoint || DEFAULT_ENDPOINT;
-  const authToken = userConfig?.authToken || envConfig.authToken;
-  if (!authToken) {
-    throw new Error(
-      "No auth token configured. Set ERDO_AUTH_TOKEN environment variable or pass authToken to ErdoClient."
-    );
-  }
-  return {
-    endpoint,
-    authToken
-  };
-}
-
-// src/streaming/sse-client.ts
-async function* parseSSEStream(response) {
-  if (!response.body) {
-    throw new Error("Response body is null");
-  }
-  const reader = response.body.getReader();
-  const decoder = new TextDecoder();
-  let buffer = "";
-  try {
-    while (true) {
-      const { done, value } = await reader.read();
-      if (done) {
-        if (buffer.trim()) {
-          const event = parseSSELine(buffer);
-          if (event) {
-            yield event;
-          }
-        }
-        break;
-      }
-      buffer += decoder.decode(value, { stream: true });
-      const lines = buffer.split("\n");
-      buffer = lines.pop() || "";
-      for (const line of lines) {
-        const event = parseSSELine(line);
-        if (event) {
-          yield event;
-        }
-      }
-    }
-  } finally {
-    reader.releaseLock();
-  }
-}
-function parseSSELine(line) {
-  const trimmedLine = line.trim();
-  if (!trimmedLine) {
-    return null;
-  }
-  if (trimmedLine.startsWith("data: ")) {
-    const data = trimmedLine.slice(6);
-    if (!data.trim()) {
-      return null;
-    }
-    try {
-      return JSON.parse(data);
-    } catch {
-      return { raw: data };
-    }
-  }
-  return null;
-}
-async function collectSSEEvents(response) {
-  const events = [];
-  for await (const event of parseSSEStream(response)) {
-    events.push(event);
-  }
-  return events;
-}
-
-// src/client.ts
-var ErdoClient = class {
-  endpoint;
-  authToken;
-  constructor(config) {
-    const resolved = resolveConfig(config);
-    this.endpoint = resolved.endpoint;
-    this.authToken = resolved.authToken;
-  }
-  /**
-   * Invoke an agent and wait for the complete result
-   */
-  async invoke(botKey, params = {}) {
-    const events = await this.collectEvents(botKey, params);
-    return this.extractResult(botKey, events);
-  }
-  /**
-   * Invoke an agent and stream SSE events
-   */
-  async *invokeStream(botKey, params = {}) {
-    const response = await this.makeRequest(botKey, params);
-    yield* parseSSEStream(response);
-  }
-  /**
-   * Collect all events from an invocation
-   */
-  async collectEvents(botKey, params) {
-    const response = await this.makeRequest(botKey, params);
-    return collectSSEEvents(response);
-  }
-  /**
-   * Make the HTTP request to invoke an agent
-   */
-  async makeRequest(botKey, params) {
-    const url = `${this.endpoint}/bots/${botKey}/invoke`;
-    const invokeParams = {};
-    if (params.messages) {
-      invokeParams.messages = params.messages;
-    }
-    if (params.parameters) {
-      invokeParams.parameters = params.parameters;
-    }
-    if (params.datasets) {
-      invokeParams.dataset_slugs = params.datasets;
-    }
-    if (params.mode) {
-      invokeParams.mode = params.mode;
-    }
-    if (params.manualMocks) {
-      invokeParams.manual_mocks = params.manualMocks;
-    }
-    const response = await fetch(url, {
-      method: "POST",
-      headers: {
-        Authorization: `Bearer ${this.authToken}`,
-        "Content-Type": "application/json",
-        Accept: "text/event-stream"
-      },
-      body: JSON.stringify(invokeParams)
-    });
-    if (!response.ok) {
-      const errorText = await response.text();
-      throw new Error(`API request failed with status ${response.status}: ${errorText}`);
-    }
-    return response;
-  }
-  /**
-   * Extract structured result from SSE events
-   *
-   * Ported from: erdo-python-sdk/erdo/invoke/invoke.py - _extract_result_data()
-   */
-  extractResult(botKey, events) {
-    const messages = [];
-    const steps = [];
-    const stepsSeenKeys = /* @__PURE__ */ new Set();
-    let finalResult;
-    let invocationId;
-    for (const event of events) {
-      if (!event) continue;
-      const payload = event.payload;
-      const metadata = event.metadata;
-      if (payload && typeof payload === "object") {
-        if ("invocation_id" in payload && !invocationId) {
-          invocationId = payload.invocation_id;
-        }
-      }
-      if (payload && typeof payload === "object" && "action_type" in payload && "key" in payload) {
-        const stepKey = payload.key;
-        if (!stepsSeenKeys.has(stepKey)) {
-          stepsSeenKeys.add(stepKey);
-          steps.push({
-            key: stepKey,
-            action: payload.action_type,
-            status: "completed"
-          });
-        }
-      }
-      if (payload && typeof payload === "object" && "output" in payload && metadata?.user_visibility === "visible") {
-        const output = payload.output;
-        if (output && typeof output === "object" && "content" in output) {
-          const contentArray = output.content;
-          if (Array.isArray(contentArray)) {
-            for (const item of contentArray) {
-              if (item.content_type === "text") {
-                const role = metadata?.role || "assistant";
-                messages.push({
-                  role,
-                  content: item.content || ""
-                });
-              }
-            }
-          }
-        }
-      }
-      if (payload && typeof payload === "object" && "status" in payload && "output" in payload) {
-        finalResult = {
-          status: payload.status,
-          parameters: payload.parameters,
-          output: payload.output,
-          message: payload.message,
-          error: payload.error
-        };
-      }
-    }
-    return {
-      success: true,
-      botId: botKey,
-      invocationId,
-      result: finalResult,
-      messages,
-      events,
-      steps
-    };
-  }
-};
-var defaultClient = null;
-function getDefaultClient() {
-  if (!defaultClient) {
-    defaultClient = new ErdoClient();
-  }
-  return defaultClient;
-}
-async function invoke(botKey, params = {}) {
-  return getDefaultClient().invoke(botKey, params);
-}
-async function* invokeStream(botKey, params = {}) {
-  yield* getDefaultClient().invokeStream(botKey, params);
-}
-export {
-  ErdoClient,
-  collectSSEEvents,
-  getConfigFromEnv,
-  invoke,
-  invokeStream,
-  parseSSEStream,
-  resolveConfig
-};
+var y="https://api.erdo.ai";function S(){let o=typeof process<"u"?process.env:{};return {endpoint:o.ERDO_ENDPOINT||o.NEXT_PUBLIC_ERDO_ENDPOINT||y,authToken:o.ERDO_AUTH_TOKEN||o.NEXT_PUBLIC_ERDO_API_KEY}}function p(o){let t=S(),e=o?.endpoint||t.endpoint||y,s=o?.authToken||t.authToken;if(!s)throw new Error("No auth token configured. Set ERDO_AUTH_TOKEN environment variable or pass authToken to ErdoClient.");return {endpoint:e,authToken:s}}async function*d(o){if(!o.body)throw new Error("Response body is null");let t=o.body.getReader(),e=new TextDecoder,s="";try{for(;;){let{done:r,value:i}=await t.read();if(r){if(s.trim()){let a=h(s);a&&(yield a);}break}s+=e.decode(i,{stream:!0});let u=s.split(`
+`);s=u.pop()||"";for(let a of u){let l=h(a);l&&(yield l);}}}finally{t.releaseLock();}}function h(o){let t=o.trim();if(!t)return null;if(t.startsWith("data: ")){let e=t.slice(6);if(!e.trim())return null;try{return JSON.parse(e)}catch{return {raw:e}}}return null}async function v(o){let t=[];for await(let e of d(o))t.push(e);return t}var f=class{endpoint;authToken;constructor(t){let e=p(t);this.endpoint=e.endpoint,this.authToken=e.authToken;}async invoke(t,e={}){let s=await this.collectEvents(t,e);return this.extractResult(t,s)}async*invokeStream(t,e={}){let s=await this.makeRequest(t,e);yield*d(s);}async collectEvents(t,e){let s=await this.makeRequest(t,e);return v(s)}async makeRequest(t,e){let s=`${this.endpoint}/bots/${t}/invoke`,r={};e.messages&&(r.messages=e.messages),e.parameters&&(r.parameters=e.parameters),e.datasets&&(r.dataset_slugs=e.datasets),e.mode&&(r.mode=e.mode),e.manualMocks&&(r.manual_mocks=e.manualMocks);let i=await fetch(s,{method:"POST",headers:{Authorization:`Bearer ${this.authToken}`,"Content-Type":"application/json",Accept:"text/event-stream"},body:JSON.stringify(r)});if(!i.ok){let u=await i.text();throw new Error(`API request failed with status ${i.status}: ${u}`)}return i}extractResult(t,e){let s=[],r=[],i=new Set,u,a;for(let l of e){if(!l)continue;let n=l.payload,m=l.metadata;if(n&&typeof n=="object"&&"invocation_id"in n&&!a&&(a=n.invocation_id),n&&typeof n=="object"&&"action_type"in n&&"key"in n){let c=n.key;i.has(c)||(i.add(c),r.push({key:c,action:n.action_type,status:"completed"}));}if(n&&typeof n=="object"&&"output"in n&&m?.user_visibility==="visible"){let c=n.output;if(c&&typeof c=="object"&&"content"in c){let g=c.content;if(Array.isArray(g)){for(let k of g)if(k.content_type==="text"){let w=m?.role||"assistant";s.push({role:w,content:k.content||""});}}}}n&&typeof n=="object"&&"status"in n&&"output"in n&&(u={status:n.status,parameters:n.parameters,output:n.output,message:n.message,error:n.error});}return {success:true,botId:t,invocationId:a,result:u,messages:s,events:e,steps:r}}},E=null;function R(){return E||(E=new f),E}async function C(o,t={}){return R().invoke(o,t)}async function*T(o,t={}){yield*R().invokeStream(o,t);}export{f as ErdoClient,v as collectSSEEvents,S as getConfigFromEnv,C as invoke,T as invokeStream,d as parseSSEStream,p as resolveConfig};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@erdoai/server",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "description": "Erdo server SDK for invoking agents",
   "type": "module",
   "main": "./dist/index.js",