@cogniformai/instructor-stream/dist/chunk-UDC2UWG6.js

Streaming-first structured data extraction from LLMs with real-time updates

import{b as y,c as w,d as A,e as x,f as N}from"./chunk-ZKNSR3JA.js";var G=e=>typeof e=="object"&&e!==null,M=e=>Array.isArray(e)?e:[],h=e=>typeof e=="string"&&e.length>0,E=e=>{if(h(e.type)){if(e.type==="text"&&h(e.text))return e.text;if(e.type==="tool_call_chunk"&&h(e.args))return e.args}},L=(e={})=>{let{defaultNode:a=null,onMissingNode:r}=e;return new TransformStream({transform(n,c){if(!G(n))return;let d=Array.isArray(n.data)?n.data:[],m=a,l=!1;for(let f=0;f<d.length;f++){let p=d[f];if(!G(p))continue;let t=d[f+1],i=h(p.langgraph_node)?p.langgraph_node:null,s=G(t)&&h(t.langgraph_node)?t.langgraph_node:null;i?m=i:s&&(m=s);let u=i??m,o=M(p.content);if(o.length){if(!u){l=!0;continue}for(let S of o){let g=E(S);g&&c.enqueue({node:u,text:g})}}}l&&r&&r(n)}})};import{Stream as b,Effect as T}from"effect";var k={string:null,number:null,boolean:null},v=(e,a)=>e instanceof y?e:new y({message:a,cause:e}),P=e=>{let a=x(e.schema);if(!a.zod)throw new w({message:`Schema '${a.name}' must provide a Zod schema for LangGraph streaming`});let r=e.validation??"none",n=new N(a.zod,{snapshotMode:"object",autoJSONMode:"off",typeDefaults:{...k}}),c=t=>({_activePath:n.getActivePath(),_completedPaths:n.getCompletedPaths(),_isValid:r==="none",_type:t}),d={};e.defaultNode!==void 0&&(d.defaultNode=e.defaultNode),e.onMissingNode!==void 0&&(d.onMissingNode=e.onMissingNode);let m=e.upstream.pipeThrough(L(d)),l=new Set,f=(async function*(){let t=m.getReader(),i=!1;try{for(;;){let{done:s,value:u}=await t.read();if(s){i=!0;break}if(!u)continue;let o=u,{completions:S,closed:g}=n.ingest([o.node],o.text);if(S.length===0){g&&(n.releaseContext([o.node]),l.delete(o.node));continue}l.add(o.node),yield{data:[n.current()],_meta:c(o.node)},g&&(n.releaseContext([o.node]),l.delete(o.node))}if(r==="final"){let s=await a.zod?.safeParseAsync(n.current());if(!s?.success)throw new A({reason:`Final validation failed for schema '${a.name}'`,issues:s?.error?.issues})}}finally{if(!i)try{await t.cancel()}catch{}try{t.releaseLock()}catch{}for(let s of l)n.releaseContext([s])}})(),p=b.fromAsyncIterable(f,t=>t instanceof A||t instanceof y||t instanceof w?t:v(t,"LangGraph stream failed"));return e.onSnapshot?b.mapEffect(p,t=>T.tryPromise({try:async()=>{let i=t.data[0];return await e.onSnapshot?.(i??{},t._meta),t},catch:i=>v(i,"LangGraph onSnapshot handler failed")})):p};function j(e){let a=Symbol.asyncIterator in e?e[Symbol.asyncIterator]():(async function*(){for(let r of e)yield r})();return new ReadableStream({async pull(r){let{value:n,done:c}=await a.next();c?r.close():r.enqueue(n)},async cancel(){typeof a.return=="function"&&await a.return()}})}export{L as a,P as b,j as c};
//# sourceMappingURL=chunk-UDC2UWG6.js.map