@ag-ui/langgraph 0.0.26 → 0.0.27
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/index.d.mts +4 -3
- package/dist/index.d.mts.map +1 -1
- package/dist/index.d.ts +4 -3
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +1 -1
- package/dist/index.js.map +1 -1
- package/dist/index.mjs +1 -1
- package/dist/index.mjs.map +1 -1
- package/package.json +17 -17
- package/LICENSE +0 -21
package/dist/index.d.mts
CHANGED
|
@@ -134,6 +134,7 @@ declare class LangGraphAgent extends AbstractAgent {
|
|
|
134
134
|
graphId: string;
|
|
135
135
|
assistant?: Assistant;
|
|
136
136
|
messagesInProcess: MessagesInProgressRecord;
|
|
137
|
+
emittedToolCallStartIds: Set<string>;
|
|
137
138
|
reasoningProcess: null | ReasoningInProgress;
|
|
138
139
|
activeRun?: RunMetadata;
|
|
139
140
|
private cancelRequested;
|
|
@@ -148,16 +149,16 @@ declare class LangGraphAgent extends AbstractAgent {
|
|
|
148
149
|
run(input: RunAgentInput): Observable<ProcessedEvents>;
|
|
149
150
|
runAgentStream(input: RunAgentExtendedInput, subscriber: Subscriber<ProcessedEvents>): Promise<void>;
|
|
150
151
|
prepareRegenerateStream(input: RegenerateInput, streamMode: StreamMode | StreamMode[]): Promise<void | {
|
|
151
|
-
streamResponse: _langchain_langgraph_sdk_dist_types_stream0.TypedAsyncGenerator<"values" | "
|
|
152
|
+
streamResponse: _langchain_langgraph_sdk_dist_types_stream0.TypedAsyncGenerator<"messages" | "values" | "custom" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "messages-tuple" | StreamMode[], false, _langchain_langgraph_sdk0.DefaultValues, _langchain_langgraph_sdk0.DefaultValues, unknown>;
|
|
152
153
|
state: ThreadState<State>;
|
|
153
154
|
streamMode: StreamMode | StreamMode[];
|
|
154
155
|
}>;
|
|
155
156
|
prepareStream(input: RunAgentExtendedInput, streamMode: StreamMode | StreamMode[]): Promise<void | {
|
|
156
|
-
streamResponse: _langchain_langgraph_sdk_dist_types_stream0.TypedAsyncGenerator<"values" | "
|
|
157
|
+
streamResponse: _langchain_langgraph_sdk_dist_types_stream0.TypedAsyncGenerator<"messages" | "values" | "custom" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "messages-tuple" | StreamMode[], false, _langchain_langgraph_sdk0.DefaultValues, _langchain_langgraph_sdk0.DefaultValues, unknown>;
|
|
157
158
|
state: ThreadState<State>;
|
|
158
159
|
streamMode: StreamMode | StreamMode[];
|
|
159
160
|
} | {
|
|
160
|
-
streamResponse: _langchain_langgraph_sdk_dist_types_stream0.TypedAsyncGenerator<"values" | "
|
|
161
|
+
streamResponse: _langchain_langgraph_sdk_dist_types_stream0.TypedAsyncGenerator<"messages" | "values" | "custom" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "messages-tuple" | StreamMode[], false, _langchain_langgraph_sdk0.DefaultValues, _langchain_langgraph_sdk0.DefaultValues, unknown>;
|
|
161
162
|
state: ThreadState<State>;
|
|
162
163
|
}>;
|
|
163
164
|
handleStreamEvents(stream: Awaited<ReturnType<typeof this.prepareStream> | ReturnType<typeof this.prepareRegenerateStream>>, threadId: string, subscriber: Subscriber<ProcessedEvents>, input: RunAgentExtendedInput, streamModes: StreamMode | StreamMode[]): Promise<void>;
|
package/dist/index.d.mts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.mts","names":[],"sources":["../src/types.ts","../src/agent.ts","../src/index.ts"],"mappings":";;;;;;;;;;aAIY,mBAAA;EACV,YAAA;EACA,aAAA;EACA,UAAA;EACA,gBAAA;EACA,iBAAA;EACA,cAAA;EACA,WAAA;EACA,SAAA;EACA,aAAA;EACA,WAAA;AAAA;AAAA,KAGU,qBAAA;EACV,IAAA;EACA,IAAA;EACA,QAAA;IACE,IAAA;IACA,WAAA;IACA,UAAA;EAAA;AAAA;AAAA,KAIQ,KAAA,iBAAsB,MAAA,+BACpB,aAAA,GAAgB,aAAA,CAAc,CAAA,aACxC,MAAA;AAAA,UACa,eAAA;EACf,QAAA,EAAU,OAAA;EACV,KAAA,EAAO,qBAAA;EACP,OAAA;IACE,KAAA,EAAO,qBAAA;IACP,OAAA,EAAS,eAAA;EAAA;AAAA;AAAA,KAID,UAAA;EACV,KAAA;EACA,MAAA;EACA,OAAA;EACA,MAAA;AAAA;AAAA,KAGU,iBAAA;EACV,EAAA;EACA,UAAA;EACA,YAAA;AAAA;AAAA,KAGU,mBAAA;EACV,KAAA;EACA,IAAA,GAAO,kBAAA;EACP,SAAA;EACA,SAAA;AAAA;AAAA,UAGe,WAAA;EACf,EAAA;EACA,UAAA,GAAa,UAAA;EACb,QAAA;EACA,YAAA;EACA,WAAA;EACA,oBAAA,GAAuB,KAAA;EACvB,QAAA;EACA,SAAA,GAAY,cAAA;EACZ,oBAAA;EAEA,gBAAA;EAEA,eAAA;EAEA,gBAAA,GAAmB,wBAAA;AAAA;AAAA,KAGT,wBAAA,GAA2B,MAAA,SAAe,iBAAA;AAAA,UAGrC,QAAA;EACf,EAAA;EACA,IAAA;EACA,IAAA,EAAM,MAAA;AAAA;AAAA,KAGH,4BAAA,GAA+B,IAAA,CAClC,OAAA;EASA,OAAA;EACA,IAAA;EACA,iBAAA,GAAoB,MAAA;EACpB,IAAA,EAAM,WAAA;AAAA;AAAA,UAGE,8BAAA,SAAuC,4BAAA;EAC/C,YAAA;EACA,IAAA;AAAA;AAAA,UAGQ,uCAAA,SAAgD,4BAAA;EACxD,UAAA,EAAY,QAAA;AAAA;AAAA,KAGF,wBAAA,GACR,uCAAA,GACA,8BAAA,GACA,4BAAA;AAAA,aAEQ,gBAAA;EACV,mBAAA;EACA,oBAAA;EACA,iBAAA;EACA,IAAA;AAAA;AAAA,UAGe,gBAAA;EACf,IAAA;EACA,SAAA;EACA,aAAA;AAAA;AAAA,UAGe,kBAAA;EACf,IAAA;EACA,IAAA;EACA,KAAA;EACA,SAAA;AAAA;;;KC7DU,eAAA,GACR,qBAAA,GACA,uBAAA,GACA,mBAAA,GACA,mBAAA,GACA,0BAAA,GACA,4BAAA,GACA,wBAAA,GACA,iBAAA,GACA,4BAAA,GACA,kBAAA,GACA,iBAAA,GACA,gBAAA,GACA,mBAAA,GACA,kBAAA,GACA,eAAA,GACA,qBAAA,GACA,QAAA,GACA,WAAA,GACA,eAAA,GACA,gBAAA,GACA,aAAA,GACA,gBAAA,GACA,iBAAA;AAAA,KAEC,qBAAA,qBACiB,UAAA,GAAa,UAAA,KAAe,UAAA,wCAE9C,IAAA,CAAK,aAAA;EACP,cAAA,GAAiB,IAAA,CAAK,iBAAA,CAAkB,WAAA,EAAa,UAAA;IACnD,QAAA;IACA,cAAA,GAAiB,MAAA;EAAA;AAAA;AAAA,UAIX,eAAA,SAAwB,qBAAA;EAChC,iBAAA,EAAmB,OAAA;AAAA;AAAA,UAGJ,oBAAA,SAA6B,WAAA;EAC5C,MAAA,GAAS,MAAA;EACT,aAAA;EACA,eAAA;EACA,eAAA,GAAkB,MAAA;EAClB,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;AAAA;AAAA,cAGW,cAAA,SAAuB,aAAA;EAClC,MAAA,EAAQ,MAAA;EACR,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;EACA,SAAA,GAAY,SAAA;EACZ,iBAAA,EAAmB,wBAAA;EACnB,gBAAA,SAAyB,mBAAA;EACzB,SAAA,GAAY,WAAA;EAAA,QAEJ,eAAA;EAAA,QACA,UAAA;EAAA,QAEA,kBAAA;EAER,UAAA,EAAY,UAAA,CAAW,eAAA;EACvB,kBAAA;EACA,MAAA,EAAQ,oBAAA;cAEI,MAAA,EAAQ,oBAAA;EAiBb,KAAA,CAAA;EAoBP,aAAA,CAAc,KAAA,EAAO,eAAA;EAKrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAa,UAAA,CAAA,eAAA;
|
|
1
|
+
{"version":3,"file":"index.d.mts","names":[],"sources":["../src/types.ts","../src/agent.ts","../src/index.ts"],"mappings":";;;;;;;;;;aAIY,mBAAA;EACV,YAAA;EACA,aAAA;EACA,UAAA;EACA,gBAAA;EACA,iBAAA;EACA,cAAA;EACA,WAAA;EACA,SAAA;EACA,aAAA;EACA,WAAA;AAAA;AAAA,KAGU,qBAAA;EACV,IAAA;EACA,IAAA;EACA,QAAA;IACE,IAAA;IACA,WAAA;IACA,UAAA;EAAA;AAAA;AAAA,KAIQ,KAAA,iBAAsB,MAAA,+BACpB,aAAA,GAAgB,aAAA,CAAc,CAAA,aACxC,MAAA;AAAA,UACa,eAAA;EACf,QAAA,EAAU,OAAA;EACV,KAAA,EAAO,qBAAA;EACP,OAAA;IACE,KAAA,EAAO,qBAAA;IACP,OAAA,EAAS,eAAA;EAAA;AAAA;AAAA,KAID,UAAA;EACV,KAAA;EACA,MAAA;EACA,OAAA;EACA,MAAA;AAAA;AAAA,KAGU,iBAAA;EACV,EAAA;EACA,UAAA;EACA,YAAA;AAAA;AAAA,KAGU,mBAAA;EACV,KAAA;EACA,IAAA,GAAO,kBAAA;EACP,SAAA;EACA,SAAA;AAAA;AAAA,UAGe,WAAA;EACf,EAAA;EACA,UAAA,GAAa,UAAA;EACb,QAAA;EACA,YAAA;EACA,WAAA;EACA,oBAAA,GAAuB,KAAA;EACvB,QAAA;EACA,SAAA,GAAY,cAAA;EACZ,oBAAA;EAEA,gBAAA;EAEA,eAAA;EAEA,gBAAA,GAAmB,wBAAA;AAAA;AAAA,KAGT,wBAAA,GAA2B,MAAA,SAAe,iBAAA;AAAA,UAGrC,QAAA;EACf,EAAA;EACA,IAAA;EACA,IAAA,EAAM,MAAA;AAAA;AAAA,KAGH,4BAAA,GAA+B,IAAA,CAClC,OAAA;EASA,OAAA;EACA,IAAA;EACA,iBAAA,GAAoB,MAAA;EACpB,IAAA,EAAM,WAAA;AAAA;AAAA,UAGE,8BAAA,SAAuC,4BAAA;EAC/C,YAAA;EACA,IAAA;AAAA;AAAA,UAGQ,uCAAA,SAAgD,4BAAA;EACxD,UAAA,EAAY,QAAA;AAAA;AAAA,KAGF,wBAAA,GACR,uCAAA,GACA,8BAAA,GACA,4BAAA;AAAA,aAEQ,gBAAA;EACV,mBAAA;EACA,oBAAA;EACA,iBAAA;EACA,IAAA;AAAA;AAAA,UAGe,gBAAA;EACf,IAAA;EACA,SAAA;EACA,aAAA;AAAA;AAAA,UAGe,kBAAA;EACf,IAAA;EACA,IAAA;EACA,KAAA;EACA,SAAA;AAAA;;;KC7DU,eAAA,GACR,qBAAA,GACA,uBAAA,GACA,mBAAA,GACA,mBAAA,GACA,0BAAA,GACA,4BAAA,GACA,wBAAA,GACA,iBAAA,GACA,4BAAA,GACA,kBAAA,GACA,iBAAA,GACA,gBAAA,GACA,mBAAA,GACA,kBAAA,GACA,eAAA,GACA,qBAAA,GACA,QAAA,GACA,WAAA,GACA,eAAA,GACA,gBAAA,GACA,aAAA,GACA,gBAAA,GACA,iBAAA;AAAA,KAEC,qBAAA,qBACiB,UAAA,GAAa,UAAA,KAAe,UAAA,wCAE9C,IAAA,CAAK,aAAA;EACP,cAAA,GAAiB,IAAA,CAAK,iBAAA,CAAkB,WAAA,EAAa,UAAA;IACnD,QAAA;IACA,cAAA,GAAiB,MAAA;EAAA;AAAA;AAAA,UAIX,eAAA,SAAwB,qBAAA;EAChC,iBAAA,EAAmB,OAAA;AAAA;AAAA,UAGJ,oBAAA,SAA6B,WAAA;EAC5C,MAAA,GAAS,MAAA;EACT,aAAA;EACA,eAAA;EACA,eAAA,GAAkB,MAAA;EAClB,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;AAAA;AAAA,cAGW,cAAA,SAAuB,aAAA;EAClC,MAAA,EAAQ,MAAA;EACR,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;EACA,SAAA,GAAY,SAAA;EACZ,iBAAA,EAAmB,wBAAA;EACnB,uBAAA,EAAyB,GAAA;EACzB,gBAAA,SAAyB,mBAAA;EACzB,SAAA,GAAY,WAAA;EAAA,QAEJ,eAAA;EAAA,QACA,UAAA;EAAA,QAEA,kBAAA;EAER,UAAA,EAAY,UAAA,CAAW,eAAA;EACvB,kBAAA;EACA,MAAA,EAAQ,oBAAA;cAEI,MAAA,EAAQ,oBAAA;EAiBb,KAAA,CAAA;EAoBP,aAAA,CAAc,KAAA,EAAO,eAAA;EAKrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAa,UAAA,CAAA,eAAA;EAYlB,cAAA,CAAe,KAAA,EAAO,qBAAA,EAAuB,UAAA,EAAY,UAAA,CAAW,eAAA,IAAgB,OAAA;EA0BpF,uBAAA,CAAwB,KAAA,EAAO,eAAA,EAAiB,UAAA,EAAY,UAAA,GAAa,UAAA,KAAY,OAAA;;WAkCxD,WAAA,CAAY,KAAA;;;EAKzC,aAAA,CAAc,KAAA,EAAO,qBAAA,EAAuB,UAAA,EAAY,UAAA,GAAa,UAAA,KAAY,OAAA;;WALpD,WAAA,CAAY,KAAA;;;;WAoKrB,WAAA,CAAY,KAAA;EAAA;EAIhC,kBAAA,CACJ,MAAA,EAAQ,OAAA,CACN,UAAA,aAAuB,aAAA,IAAiB,UAAA,aAAuB,uBAAA,IAEjE,QAAA,UACA,UAAA,EAAY,UAAA,CAAW,eAAA,GACvB,KAAA,EAAO,qBAAA,EACP,WAAA,EAAa,UAAA,GAAa,UAAA,KAAY,OAAA;EAqOxC,iBAAA,CAAkB,KAAA;EDpnBR;;;;;EAAA,QCy9BF,wBAAA;EA8FD,QAAA,CAAA;EAiBP,oBAAA,CAAqB,aAAA,EAAe,kBAAA;EAyDpC,gBAAA,CAAiB,WAAA,EAAa,WAAA,CAAY,KAAA,IAAM,KAAA;EAW1C,iBAAA,CAAkB,QAAA,UAAkB,cAAA,GAAiB,MAAA,gBAAsB,OAAA,CAAQ,MAAA;EAkBnF,SAAA,CAAU,QAAA,WAAgB,OAAA,CAAA,MAAA,CAlBwD,yBAAA,CAkBxD,aAAA;EAI1B,YAAA,CAAa,OAAA,GAAU,UAAA,aAAuB,MAAA,CAAO,OAAA,CAAQ,MAAA,OAAU,OAAA,CAAA,MAAA,CAAtC,yBAAA,CAAsC,aAAA;EAIvE,YAAA,CAAA;IACJ,OAAA;IACA,SAAA;IACA;EAAA;IAEA,OAAA,EAAS,MAAA;IACT,SAAA,EAAW,SAAA;IACX,UAAA,EAAY,UAAA;EAAA,IACb,OAAA,CAAA,MAAA;EA0CD,oBAAA,CAAqB,KAAA,WAAa,iBAAA;EAIlC,oBAAA,CAAqB,KAAA,UAAe,IAAA,EAAM,iBAAA;EAUpC,YAAA,CAAA,GAAgB,OAAA,CAAQ,SAAA;EA4BxB,aAAA,CAAA,GAAiB,OAAA,CAAQ,UAAA;EAgC/B,0BAAA,CAA2B,KAAA,EAAO,KAAA,EAAO,QAAA,EAAU,OAAA,IAAoB,KAAA,EAAO,qBAAA,GAAwB,KAAA,CAAM,eAAA;EA+C5G,gBAAA,CAAiB,QAAA;EAiBjB,SAAA,CAAU,QAAA;EAOV,OAAA,CAAA;EAOM,sBAAA,CACJ,SAAA,UACA,QAAA,UACA,UAAA;IACE,aAAA;IACA,aAAA;EAAA,IAED,OAAA,CAAQ,WAAA;AAAA;;;cCn5CA,kBAAA,SAA2B,SAAA"}
|
package/dist/index.d.ts
CHANGED
|
@@ -134,6 +134,7 @@ declare class LangGraphAgent extends AbstractAgent {
|
|
|
134
134
|
graphId: string;
|
|
135
135
|
assistant?: Assistant;
|
|
136
136
|
messagesInProcess: MessagesInProgressRecord;
|
|
137
|
+
emittedToolCallStartIds: Set<string>;
|
|
137
138
|
reasoningProcess: null | ReasoningInProgress;
|
|
138
139
|
activeRun?: RunMetadata;
|
|
139
140
|
private cancelRequested;
|
|
@@ -148,16 +149,16 @@ declare class LangGraphAgent extends AbstractAgent {
|
|
|
148
149
|
run(input: RunAgentInput): Observable<ProcessedEvents>;
|
|
149
150
|
runAgentStream(input: RunAgentExtendedInput, subscriber: Subscriber<ProcessedEvents>): Promise<void>;
|
|
150
151
|
prepareRegenerateStream(input: RegenerateInput, streamMode: StreamMode | StreamMode[]): Promise<void | {
|
|
151
|
-
streamResponse: _langchain_langgraph_sdk_dist_types_stream0.TypedAsyncGenerator<"values" | "
|
|
152
|
+
streamResponse: _langchain_langgraph_sdk_dist_types_stream0.TypedAsyncGenerator<"messages" | "values" | "custom" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "messages-tuple" | StreamMode[], false, _langchain_langgraph_sdk0.DefaultValues, _langchain_langgraph_sdk0.DefaultValues, unknown>;
|
|
152
153
|
state: ThreadState<State>;
|
|
153
154
|
streamMode: StreamMode | StreamMode[];
|
|
154
155
|
}>;
|
|
155
156
|
prepareStream(input: RunAgentExtendedInput, streamMode: StreamMode | StreamMode[]): Promise<void | {
|
|
156
|
-
streamResponse: _langchain_langgraph_sdk_dist_types_stream0.TypedAsyncGenerator<"values" | "
|
|
157
|
+
streamResponse: _langchain_langgraph_sdk_dist_types_stream0.TypedAsyncGenerator<"messages" | "values" | "custom" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "messages-tuple" | StreamMode[], false, _langchain_langgraph_sdk0.DefaultValues, _langchain_langgraph_sdk0.DefaultValues, unknown>;
|
|
157
158
|
state: ThreadState<State>;
|
|
158
159
|
streamMode: StreamMode | StreamMode[];
|
|
159
160
|
} | {
|
|
160
|
-
streamResponse: _langchain_langgraph_sdk_dist_types_stream0.TypedAsyncGenerator<"values" | "
|
|
161
|
+
streamResponse: _langchain_langgraph_sdk_dist_types_stream0.TypedAsyncGenerator<"messages" | "values" | "custom" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "messages-tuple" | StreamMode[], false, _langchain_langgraph_sdk0.DefaultValues, _langchain_langgraph_sdk0.DefaultValues, unknown>;
|
|
161
162
|
state: ThreadState<State>;
|
|
162
163
|
}>;
|
|
163
164
|
handleStreamEvents(stream: Awaited<ReturnType<typeof this.prepareStream> | ReturnType<typeof this.prepareRegenerateStream>>, threadId: string, subscriber: Subscriber<ProcessedEvents>, input: RunAgentExtendedInput, streamModes: StreamMode | StreamMode[]): Promise<void>;
|
package/dist/index.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","names":[],"sources":["../src/types.ts","../src/agent.ts","../src/index.ts"],"mappings":";;;;;;;;;;aAIY,mBAAA;EACV,YAAA;EACA,aAAA;EACA,UAAA;EACA,gBAAA;EACA,iBAAA;EACA,cAAA;EACA,WAAA;EACA,SAAA;EACA,aAAA;EACA,WAAA;AAAA;AAAA,KAGU,qBAAA;EACV,IAAA;EACA,IAAA;EACA,QAAA;IACE,IAAA;IACA,WAAA;IACA,UAAA;EAAA;AAAA;AAAA,KAIQ,KAAA,iBAAsB,MAAA,+BACpB,aAAA,GAAgB,aAAA,CAAc,CAAA,aACxC,MAAA;AAAA,UACa,eAAA;EACf,QAAA,EAAU,OAAA;EACV,KAAA,EAAO,qBAAA;EACP,OAAA;IACE,KAAA,EAAO,qBAAA;IACP,OAAA,EAAS,eAAA;EAAA;AAAA;AAAA,KAID,UAAA;EACV,KAAA;EACA,MAAA;EACA,OAAA;EACA,MAAA;AAAA;AAAA,KAGU,iBAAA;EACV,EAAA;EACA,UAAA;EACA,YAAA;AAAA;AAAA,KAGU,mBAAA;EACV,KAAA;EACA,IAAA,GAAO,kBAAA;EACP,SAAA;EACA,SAAA;AAAA;AAAA,UAGe,WAAA;EACf,EAAA;EACA,UAAA,GAAa,UAAA;EACb,QAAA;EACA,YAAA;EACA,WAAA;EACA,oBAAA,GAAuB,KAAA;EACvB,QAAA;EACA,SAAA,GAAY,cAAA;EACZ,oBAAA;EAEA,gBAAA;EAEA,eAAA;EAEA,gBAAA,GAAmB,wBAAA;AAAA;AAAA,KAGT,wBAAA,GAA2B,MAAA,SAAe,iBAAA;AAAA,UAGrC,QAAA;EACf,EAAA;EACA,IAAA;EACA,IAAA,EAAM,MAAA;AAAA;AAAA,KAGH,4BAAA,GAA+B,IAAA,CAClC,OAAA;EASA,OAAA;EACA,IAAA;EACA,iBAAA,GAAoB,MAAA;EACpB,IAAA,EAAM,WAAA;AAAA;AAAA,UAGE,8BAAA,SAAuC,4BAAA;EAC/C,YAAA;EACA,IAAA;AAAA;AAAA,UAGQ,uCAAA,SAAgD,4BAAA;EACxD,UAAA,EAAY,QAAA;AAAA;AAAA,KAGF,wBAAA,GACR,uCAAA,GACA,8BAAA,GACA,4BAAA;AAAA,aAEQ,gBAAA;EACV,mBAAA;EACA,oBAAA;EACA,iBAAA;EACA,IAAA;AAAA;AAAA,UAGe,gBAAA;EACf,IAAA;EACA,SAAA;EACA,aAAA;AAAA;AAAA,UAGe,kBAAA;EACf,IAAA;EACA,IAAA;EACA,KAAA;EACA,SAAA;AAAA;;;KC7DU,eAAA,GACR,qBAAA,GACA,uBAAA,GACA,mBAAA,GACA,mBAAA,GACA,0BAAA,GACA,4BAAA,GACA,wBAAA,GACA,iBAAA,GACA,4BAAA,GACA,kBAAA,GACA,iBAAA,GACA,gBAAA,GACA,mBAAA,GACA,kBAAA,GACA,eAAA,GACA,qBAAA,GACA,QAAA,GACA,WAAA,GACA,eAAA,GACA,gBAAA,GACA,aAAA,GACA,gBAAA,GACA,iBAAA;AAAA,KAEC,qBAAA,qBACiB,UAAA,GAAa,UAAA,KAAe,UAAA,wCAE9C,IAAA,CAAK,aAAA;EACP,cAAA,GAAiB,IAAA,CAAK,iBAAA,CAAkB,WAAA,EAAa,UAAA;IACnD,QAAA;IACA,cAAA,GAAiB,MAAA;EAAA;AAAA;AAAA,UAIX,eAAA,SAAwB,qBAAA;EAChC,iBAAA,EAAmB,OAAA;AAAA;AAAA,UAGJ,oBAAA,SAA6B,WAAA;EAC5C,MAAA,GAAS,MAAA;EACT,aAAA;EACA,eAAA;EACA,eAAA,GAAkB,MAAA;EAClB,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;AAAA;AAAA,cAGW,cAAA,SAAuB,aAAA;EAClC,MAAA,EAAQ,MAAA;EACR,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;EACA,SAAA,GAAY,SAAA;EACZ,iBAAA,EAAmB,wBAAA;EACnB,gBAAA,SAAyB,mBAAA;EACzB,SAAA,GAAY,WAAA;EAAA,QAEJ,eAAA;EAAA,QACA,UAAA;EAAA,QAEA,kBAAA;EAER,UAAA,EAAY,UAAA,CAAW,eAAA;EACvB,kBAAA;EACA,MAAA,EAAQ,oBAAA;cAEI,MAAA,EAAQ,oBAAA;EAiBb,KAAA,CAAA;EAoBP,aAAA,CAAc,KAAA,EAAO,eAAA;EAKrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAa,UAAA,CAAA,eAAA;
|
|
1
|
+
{"version":3,"file":"index.d.ts","names":[],"sources":["../src/types.ts","../src/agent.ts","../src/index.ts"],"mappings":";;;;;;;;;;aAIY,mBAAA;EACV,YAAA;EACA,aAAA;EACA,UAAA;EACA,gBAAA;EACA,iBAAA;EACA,cAAA;EACA,WAAA;EACA,SAAA;EACA,aAAA;EACA,WAAA;AAAA;AAAA,KAGU,qBAAA;EACV,IAAA;EACA,IAAA;EACA,QAAA;IACE,IAAA;IACA,WAAA;IACA,UAAA;EAAA;AAAA;AAAA,KAIQ,KAAA,iBAAsB,MAAA,+BACpB,aAAA,GAAgB,aAAA,CAAc,CAAA,aACxC,MAAA;AAAA,UACa,eAAA;EACf,QAAA,EAAU,OAAA;EACV,KAAA,EAAO,qBAAA;EACP,OAAA;IACE,KAAA,EAAO,qBAAA;IACP,OAAA,EAAS,eAAA;EAAA;AAAA;AAAA,KAID,UAAA;EACV,KAAA;EACA,MAAA;EACA,OAAA;EACA,MAAA;AAAA;AAAA,KAGU,iBAAA;EACV,EAAA;EACA,UAAA;EACA,YAAA;AAAA;AAAA,KAGU,mBAAA;EACV,KAAA;EACA,IAAA,GAAO,kBAAA;EACP,SAAA;EACA,SAAA;AAAA;AAAA,UAGe,WAAA;EACf,EAAA;EACA,UAAA,GAAa,UAAA;EACb,QAAA;EACA,YAAA;EACA,WAAA;EACA,oBAAA,GAAuB,KAAA;EACvB,QAAA;EACA,SAAA,GAAY,cAAA;EACZ,oBAAA;EAEA,gBAAA;EAEA,eAAA;EAEA,gBAAA,GAAmB,wBAAA;AAAA;AAAA,KAGT,wBAAA,GAA2B,MAAA,SAAe,iBAAA;AAAA,UAGrC,QAAA;EACf,EAAA;EACA,IAAA;EACA,IAAA,EAAM,MAAA;AAAA;AAAA,KAGH,4BAAA,GAA+B,IAAA,CAClC,OAAA;EASA,OAAA;EACA,IAAA;EACA,iBAAA,GAAoB,MAAA;EACpB,IAAA,EAAM,WAAA;AAAA;AAAA,UAGE,8BAAA,SAAuC,4BAAA;EAC/C,YAAA;EACA,IAAA;AAAA;AAAA,UAGQ,uCAAA,SAAgD,4BAAA;EACxD,UAAA,EAAY,QAAA;AAAA;AAAA,KAGF,wBAAA,GACR,uCAAA,GACA,8BAAA,GACA,4BAAA;AAAA,aAEQ,gBAAA;EACV,mBAAA;EACA,oBAAA;EACA,iBAAA;EACA,IAAA;AAAA;AAAA,UAGe,gBAAA;EACf,IAAA;EACA,SAAA;EACA,aAAA;AAAA;AAAA,UAGe,kBAAA;EACf,IAAA;EACA,IAAA;EACA,KAAA;EACA,SAAA;AAAA;;;KC7DU,eAAA,GACR,qBAAA,GACA,uBAAA,GACA,mBAAA,GACA,mBAAA,GACA,0BAAA,GACA,4BAAA,GACA,wBAAA,GACA,iBAAA,GACA,4BAAA,GACA,kBAAA,GACA,iBAAA,GACA,gBAAA,GACA,mBAAA,GACA,kBAAA,GACA,eAAA,GACA,qBAAA,GACA,QAAA,GACA,WAAA,GACA,eAAA,GACA,gBAAA,GACA,aAAA,GACA,gBAAA,GACA,iBAAA;AAAA,KAEC,qBAAA,qBACiB,UAAA,GAAa,UAAA,KAAe,UAAA,wCAE9C,IAAA,CAAK,aAAA;EACP,cAAA,GAAiB,IAAA,CAAK,iBAAA,CAAkB,WAAA,EAAa,UAAA;IACnD,QAAA;IACA,cAAA,GAAiB,MAAA;EAAA;AAAA;AAAA,UAIX,eAAA,SAAwB,qBAAA;EAChC,iBAAA,EAAmB,OAAA;AAAA;AAAA,UAGJ,oBAAA,SAA6B,WAAA;EAC5C,MAAA,GAAS,MAAA;EACT,aAAA;EACA,eAAA;EACA,eAAA,GAAkB,MAAA;EAClB,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;AAAA;AAAA,cAGW,cAAA,SAAuB,aAAA;EAClC,MAAA,EAAQ,MAAA;EACR,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;EACA,SAAA,GAAY,SAAA;EACZ,iBAAA,EAAmB,wBAAA;EACnB,uBAAA,EAAyB,GAAA;EACzB,gBAAA,SAAyB,mBAAA;EACzB,SAAA,GAAY,WAAA;EAAA,QAEJ,eAAA;EAAA,QACA,UAAA;EAAA,QAEA,kBAAA;EAER,UAAA,EAAY,UAAA,CAAW,eAAA;EACvB,kBAAA;EACA,MAAA,EAAQ,oBAAA;cAEI,MAAA,EAAQ,oBAAA;EAiBb,KAAA,CAAA;EAoBP,aAAA,CAAc,KAAA,EAAO,eAAA;EAKrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAa,UAAA,CAAA,eAAA;EAYlB,cAAA,CAAe,KAAA,EAAO,qBAAA,EAAuB,UAAA,EAAY,UAAA,CAAW,eAAA,IAAgB,OAAA;EA0BpF,uBAAA,CAAwB,KAAA,EAAO,eAAA,EAAiB,UAAA,EAAY,UAAA,GAAa,UAAA,KAAY,OAAA;;WAkCxD,WAAA,CAAY,KAAA;;;EAKzC,aAAA,CAAc,KAAA,EAAO,qBAAA,EAAuB,UAAA,EAAY,UAAA,GAAa,UAAA,KAAY,OAAA;;WALpD,WAAA,CAAY,KAAA;;;;WAoKrB,WAAA,CAAY,KAAA;EAAA;EAIhC,kBAAA,CACJ,MAAA,EAAQ,OAAA,CACN,UAAA,aAAuB,aAAA,IAAiB,UAAA,aAAuB,uBAAA,IAEjE,QAAA,UACA,UAAA,EAAY,UAAA,CAAW,eAAA,GACvB,KAAA,EAAO,qBAAA,EACP,WAAA,EAAa,UAAA,GAAa,UAAA,KAAY,OAAA;EAqOxC,iBAAA,CAAkB,KAAA;EDpnBR;;;;;EAAA,QCy9BF,wBAAA;EA8FD,QAAA,CAAA;EAiBP,oBAAA,CAAqB,aAAA,EAAe,kBAAA;EAyDpC,gBAAA,CAAiB,WAAA,EAAa,WAAA,CAAY,KAAA,IAAM,KAAA;EAW1C,iBAAA,CAAkB,QAAA,UAAkB,cAAA,GAAiB,MAAA,gBAAsB,OAAA,CAAQ,MAAA;EAkBnF,SAAA,CAAU,QAAA,WAAgB,OAAA,CAAA,MAAA,CAlBwD,yBAAA,CAkBxD,aAAA;EAI1B,YAAA,CAAa,OAAA,GAAU,UAAA,aAAuB,MAAA,CAAO,OAAA,CAAQ,MAAA,OAAU,OAAA,CAAA,MAAA,CAAtC,yBAAA,CAAsC,aAAA;EAIvE,YAAA,CAAA;IACJ,OAAA;IACA,SAAA;IACA;EAAA;IAEA,OAAA,EAAS,MAAA;IACT,SAAA,EAAW,SAAA;IACX,UAAA,EAAY,UAAA;EAAA,IACb,OAAA,CAAA,MAAA;EA0CD,oBAAA,CAAqB,KAAA,WAAa,iBAAA;EAIlC,oBAAA,CAAqB,KAAA,UAAe,IAAA,EAAM,iBAAA;EAUpC,YAAA,CAAA,GAAgB,OAAA,CAAQ,SAAA;EA4BxB,aAAA,CAAA,GAAiB,OAAA,CAAQ,UAAA;EAgC/B,0BAAA,CAA2B,KAAA,EAAO,KAAA,EAAO,QAAA,EAAU,OAAA,IAAoB,KAAA,EAAO,qBAAA,GAAwB,KAAA,CAAM,eAAA;EA+C5G,gBAAA,CAAiB,QAAA;EAiBjB,SAAA,CAAU,QAAA;EAOV,OAAA,CAAA;EAOM,sBAAA,CACJ,SAAA,UACA,QAAA,UACA,UAAA;IACE,aAAA;IACA,aAAA;EAAA,IAED,OAAA,CAAQ,WAAA;AAAA;;;cCn5CA,kBAAA,SAA2B,SAAA"}
|
package/dist/index.js
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
let e=require(`@ag-ui/client`),t=require(`rxjs`),n=require(`@langchain/langgraph-sdk`),r=function(e){return e.OnChainStart=`on_chain_start`,e.OnChainStream=`on_chain_stream`,e.OnChainEnd=`on_chain_end`,e.OnChatModelStart=`on_chat_model_start`,e.OnChatModelStream=`on_chat_model_stream`,e.OnChatModelEnd=`on_chat_model_end`,e.OnToolStart=`on_tool_start`,e.OnToolEnd=`on_tool_end`,e.OnCustomEvent=`on_custom_event`,e.OnInterrupt=`on_interrupt`,e}({}),i=function(e){return e.ManuallyEmitMessage=`manually_emit_message`,e.ManuallyEmitToolCall=`manually_emit_tool_call`,e.ManuallyEmitState=`manually_emit_state`,e.Exit=`exit`,e}({});const a=[`messages`,`tools`];function o(e,t){return Object.fromEntries(Object.entries(e).filter(([e])=>t.includes(e)))}function s({mode:e,state:t,schemaKeys:n}){let r=e===`start`?t:null;return r&&n?.input&&(r=o(r,[...a,...n.input])),r}function c(e){let t=[];for(let n of e)if(n.type===`text`&&n.text)t.push({type:`text`,text:n.text});else if(n.type===`image_url`){let e=typeof n.image_url==`string`?n.image_url:n.image_url?.url;if(!e)continue;if(e.startsWith(`data:`)){let[n,r]=e.split(`,`,2),i=n.includes(`:`)?n.split(`:`)[1].split(`;`)[0]:`image/png`;t.push({type:`binary`,mimeType:i,data:r||``})}else t.push({type:`binary`,mimeType:`image/png`,url:e})}return t}function l(e){let t=[];for(let n of e)if(n.type===`text`)t.push({type:`text`,text:n.text});else if(n.type===`binary`){let e;if(n.url)e=n.url;else if(n.data)e=`data:${n.mimeType};base64,${n.data}`;else if(n.id)e=n.id;else continue;t.push({type:`image_url`,image_url:{url:e}})}return t}function u(e){return e.map(e=>{switch(e.type){case`human`:let t;return t=Array.isArray(e.content)?c(e.content):f(h(e.content)),{id:e.id,role:`user`,content:t};case`ai`:let n=h(e.content);return{id:e.id,role:`assistant`,content:n?f(n):``,toolCalls:e.tool_calls?.map(e=>({id:e.id,type:`function`,function:{name:e.name,arguments:JSON.stringify(e.args)}}))};case`system`:return{id:e.id,role:`system`,content:f(h(e.content))};case`tool`:return{id:e.id,role:`tool`,content:f(h(e.content)),toolCallId:e.tool_call_id};default:throw Error(`message type returned from LangGraph is not supported.`)}})}function d(e){return e.map((e,t)=>{switch(e.role){case`user`:let t;return t=typeof e.content==`string`?e.content:Array.isArray(e.content)?l(e.content):String(e.content),{id:e.id,role:e.role,content:t,type:`human`};case`assistant`:return{id:e.id,type:`ai`,role:e.role,content:e.content??``,tool_calls:(e.toolCalls??[]).map(e=>({id:e.id,name:e.function.name,args:JSON.parse(e.function.arguments),type:`tool_call`}))};case`system`:return{id:e.id,role:e.role,content:e.content,type:`system`};case`tool`:return{content:e.content,role:e.role,type:e.role,tool_call_id:e.toolCallId,id:e.id};default:throw console.error(`Message role ${e.role} is not implemented`),Error(`message role is not supported.`)}})}function f(e){return typeof e==`string`?e:JSON.stringify(e)}function p(e){let t=e.chunk?.content;if(t&&Array.isArray(t)&&t.length&&t[0]){let e=t[0];if(e.type===`thinking`&&e.thinking){let t={text:e.thinking,type:`text`,index:e.index??0};return e.signature&&(t.signature=e.signature),t}if(e.type===`reasoning`&&e.reasoning)return{text:e.reasoning,type:`text`,index:e.index??0};if(e.type===`reasoning`&&e.summary?.[0]?.text)return{type:`text`,text:e.summary[0].text,index:e.summary[0].index??0}}if(e.chunk?.additional_kwargs?.reasoning?.summary?.[0]){let t=e.chunk.additional_kwargs.reasoning.summary[0];return!t||!t.text?null:{type:`text`,text:t.text,index:t.index??0}}return null}function m(e){let t=e.chunk?.content;return!t||!Array.isArray(t)||!t.length||!t[0]?null:t[0].type===`redacted_thinking`&&t[0].data?t[0].data:null}function h(e){return e?typeof e==`string`?e:Array.isArray(e)&&e.length?e.find(e=>e.type===`text`)?.text??null:null:null}var g=class extends e.AbstractAgent{constructor(e){super(e),this.cancelRequested=!1,this.cancelSent=!1,this.eventsStreamActive=!1,this.constantSchemaKeys=a,this.config=e,this.messagesInProcess={},this.agentName=e.agentName,this.graphId=e.graphId,this.assistantConfig=e.assistantConfig,this.reasoningProcess=null,this.client=e?.client??new n.Client({apiUrl:e.deploymentUrl,apiKey:e.langsmithApiKey,defaultHeaders:{...e.propertyHeaders??{}}})}clone(){return Object.assign(super.clone(),{config:this.config,messagesInProcess:structuredClone(this.messagesInProcess),agentName:this.agentName,graphId:this.graphId,assistantConfig:this.assistantConfig,reasoningProcess:this.reasoningProcess?structuredClone(this.reasoningProcess):null,constantSchemaKeys:[...this.constantSchemaKeys],client:this.client,assistant:this.assistant,activeRun:this.activeRun?structuredClone(this.activeRun):void 0,cancelRequested:this.cancelRequested,cancelSent:this.cancelSent})}dispatchEvent(e){return this.subscriber.next(e),!0}run(e){return new t.Observable(t=>(this.runAgentStream(e,t),()=>{}))}async runAgentStream(t,n){this.activeRun={id:t.runId,threadId:t.threadId,hasFunctionStreaming:!1,hasPredictState:!1},this.cancelRequested=!1,this.cancelSent=!1,this.eventsStreamActive=!1,this.subscriber=n,this.assistant||=await this.getAssistant();let r=t.threadId??(0,e.randomUUID)(),i=t.forwardedProps?.streamMode??[`events`,`values`,`updates`,`messages-tuple`],a=await this.prepareStream({...t,threadId:r},i);if(!a)return n.error(`No stream to regenerate`);await this.handleStreamEvents(a,r,n,t,Array.isArray(i)?i:[i])}async prepareRegenerateStream(e,t){let{threadId:n,messageCheckpoint:r}=e,i=await this.getCheckpointByMessage(r.id,n);if(this.assistant||=await this.getAssistant(),!i)return this.subscriber.error(`No checkpoint found for message`);let a=await this.client.threads.updateState(n,{values:this.langGraphDefaultMergeState(i.values,[],e),checkpointId:i.checkpoint.checkpoint_id,asNode:i.next?.[0]??`__start__`}),o={...e.forwardedProps??{},input:this.langGraphDefaultMergeState(i.values,[r],e),checkpointId:a.checkpoint.checkpoint_id,streamMode:t};return{streamResponse:this.client.runs.stream(n,this.assistant.assistant_id,o),state:i,streamMode:t}}async prepareStream(t,n){let{threadId:i,state:a,messages:o,tools:c,context:l,forwardedProps:u}=t;this.activeRun.manuallyEmittedState=null;let f=u?.nodeName,p=i??(0,e.randomUUID)();this.assistant||=await this.getAssistant();let m=await this.getOrCreateThread(p,u?.threadMetadata);this.activeRun.threadId=m.thread_id;let h=await this.client.threads.getState(m.thread_id)??{values:{}},g=h.values.messages??[],_=d(o),v=this.langGraphDefaultMergeState({...a,messages:g},_,t),y={...h,values:{...v,messages:[...g,...v.messages??[]]}},b=y.values;if(this.activeRun.schemaKeys=await this.getSchemaKeys(),(h.values.messages??[]).length>o.filter(e=>e.role!==`system`).length){let e=new Set((h.values.messages??[]).map(e=>e.id).filter(Boolean)),r=new Set(o.filter(e=>e.role!==`tool`&&e.id).map(e=>e.id));if(!(r.size>0&&[...r].every(t=>e.has(t)))){let r=null;for(let e=o.length-1;e>=0;e--)if(o[e].role===`user`){r=d([o[e]])[0];break}let i=r?.id;if(i&&e.has(i))return this.prepareRegenerateStream({...t,messageCheckpoint:r},n)}}this.activeRun.graphInfo=await this.client.assistants.getGraph(this.assistant.assistant_id);let x=!u?.command?.resume&&p&&this.activeRun.nodeName!=`__end__`&&this.activeRun.nodeName?`continue`:`start`;if(x===`continue`){let e=this.activeRun.graphInfo.edges.find(e=>e.target===this.activeRun.nodeName);await this.client.threads.updateState(p,{values:a,asNode:e?.source})}let S=s({mode:x,state:b,schemaKeys:this.activeRun.schemaKeys}),C,w=[this.assistantConfig,u?.config].filter(Boolean);w.length&&(C=await this.mergeConfigs({configs:w,assistant:this.assistant,schemaKeys:this.activeRun.schemaKeys}));let{command:T,...E}=u;if(T?.resume&&typeof T.resume==`string`)try{T.resume=JSON.parse(T.resume)}catch{}let D={...E,command:T,streamMode:n,input:S,config:C,context:{...l,...C?.configurable??{}}},O=h.tasks?.[0]?.interrupts??[];return O?.length&&!u?.command?.resume?(this.dispatchEvent({type:e.EventType.RUN_STARTED,threadId:p,runId:t.runId}),this.handleNodeChange(f),O.forEach(t=>{this.dispatchEvent({type:e.EventType.CUSTOM,name:r.OnInterrupt,value:typeof t.value==`string`?t.value:JSON.stringify(t.value),rawEvent:t})}),this.dispatchEvent({type:e.EventType.RUN_FINISHED,threadId:p,runId:t.runId}),this.subscriber.complete()):{streamResponse:this.client.runs.stream(p,this.assistant.assistant_id,D),state:y}}async handleStreamEvents(t,n,a,o,s){let{forwardedProps:c}=o,l=c?.nodeName;this.subscriber=a;let d=!1;if(!t)return;let{streamResponse:f,state:p}=t;this.activeRun.prevNodeName=null;let m={},h=p;try{this.dispatchEvent({type:e.EventType.RUN_STARTED,threadId:n,runId:this.activeRun.id}),this.handleNodeChange(l);for await(let t of f){if(this.cancelRequested&&!this.cancelSent&&this.activeRun?.threadId&&this.activeRun?.id){try{await this.client.runs.cancel(this.activeRun.threadId,this.activeRun.id)}catch{}finally{this.cancelSent=!0}try{await f?.return?.()}catch{}break}let n=o.forwardedProps?.streamSubgraphs,a=n&&(t.event.startsWith(`events`)||t.event.startsWith(`values`)),c=t.event===`messages`&&(Array.isArray(s)?s:[s]).includes(`messages-tuple`);if(!s.includes(t.event)&&!a&&!c&&t.event!==`error`)continue;let l=t;if(t.event===`error`){this.dispatchEvent({type:e.EventType.RUN_ERROR,message:t.data.message,rawEvent:t});break}if(t.event===`updates`)continue;if(t.event===`values`){m=l.data;continue}else if(n&&l.event.startsWith(`values|`)){m={...m,...l.data};continue}let u=l.data,g=u.metadata??{},_=g.langgraph_node,v=u.event;if(g.run_id&&(this.activeRun.id=g.run_id,this.activeRun.serverRunIdKnown=!0,this.cancelRequested&&!this.cancelSent&&this.activeRun?.threadId))try{await this.client.runs.cancel(this.activeRun.threadId,this.activeRun.id)}catch{}finally{this.cancelSent=!0}if(_&&_!==this.activeRun.nodeName&&this.handleNodeChange(_),d||=v===r.OnCustomEvent&&u.name===i.Exit,v===r.OnChainEnd&&this.activeRun.nodeName===_&&(this.activeRun.exitingNode=!0),this.activeRun.exitingNode&&(this.activeRun.manuallyEmittedState=null),this.activeRun.graphInfo?.nodes.some(e=>e.id===_)&&this.handleNodeChange(_),h.values=this.activeRun.manuallyEmittedState??m,!this.activeRun.nodeName)continue;let y=JSON.stringify(h)!==JSON.stringify(p),b=this.activeRun.exitingNode&&this.activeRun.hasPredictState;(y||this.activeRun.prevNodeName!=this.activeRun.nodeName||this.activeRun.exitingNode)&&!this.getMessageInProgress(this.activeRun.id)&&!b?(p=h,this.activeRun.prevNodeName=this.activeRun.nodeName,this.dispatchEvent({type:e.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(p),rawEvent:l})):b&&console.debug(`[ag-ui/langgraph] Suppressing STATE_SNAPSHOT on node exit (node=${this.activeRun.nodeName}, hasPredictState=${this.activeRun.hasPredictState})`),this.dispatchEvent({type:e.EventType.RAW,event:u}),this.handleSingleEvent(u)}p=await this.client.threads.getState(n);let t=p.tasks?.[0]?.interrupts??[],c=p.next.length===0,g=p.metadata?.writes??{},_=this.activeRun.nodeName;t?.length||(_=c?`__end__`:p.next[0]??Object.keys(g)[0]),t.forEach(t=>{this.dispatchEvent({type:e.EventType.CUSTOM,name:r.OnInterrupt,value:typeof t.value==`string`?t.value:JSON.stringify(t.value),rawEvent:t})}),this.handleNodeChange(_),this.handleNodeChange(void 0),this.dispatchEvent({type:e.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(p)});let v=p.values.messages??[],y=this.activeRun?.streamedMessages??[],b=new Set(v.map(e=>e.id).filter(Boolean)),x=y.filter(e=>e.id&&!b.has(e.id)),S=[...v,...x];return this.dispatchEvent({type:e.EventType.MESSAGES_SNAPSHOT,messages:u(S)}),this.dispatchEvent({type:e.EventType.RUN_FINISHED,threadId:n,runId:this.activeRun.id}),this.cancelRequested=!1,this.cancelSent=!1,this.activeRun=void 0,a.complete()}catch(e){return a.error(e)}}handleSingleEvent(t){if(Array.isArray(t)){this.eventsStreamActive||this.handleMessagesTupleEvent(t);return}switch(t.event===r.OnChatModelStream&&(this.eventsStreamActive=!0),t.event){case r.OnChatModelStream:let n=t.metadata?.[`emit-messages`]??!0,a=t.metadata?.[`emit-tool-calls`]??!0;if(t.data.chunk.response_metadata.finish_reason)return;let o=this.getMessageInProgress(this.activeRun.id),s=!!o?.id,c=t.data.chunk.tool_call_chunks?.[0],l=t.metadata?.predict_state?.some(e=>e.tool===c?.name),u=!s&&c?.name,d=s&&o?.toolCallId&&c?.args,f=s&&o?.toolCallId&&!c;(f||d||u)&&(this.activeRun.hasFunctionStreaming=!0);let g=p(t.data),_=m(t.data),v=h(t.data.chunk.content),y=!!(!c&&v),b=s&&!o?.toolCallId&&!y;if(g){this.handleReasoningEvent(g);break}if(_&&this.reasoningProcess){this.dispatchEvent({type:e.EventType.REASONING_ENCRYPTED_VALUE,subtype:`message`,entityId:this.reasoningProcess.messageId,encryptedValue:_});break}if(!g&&this.reasoningProcess&&(this.reasoningProcess.signature&&this.dispatchEvent({type:e.EventType.REASONING_ENCRYPTED_VALUE,subtype:`message`,entityId:this.reasoningProcess.messageId,encryptedValue:this.reasoningProcess.signature}),this.dispatchEvent({type:e.EventType.REASONING_MESSAGE_END,messageId:this.reasoningProcess.messageId}),this.dispatchEvent({type:e.EventType.REASONING_END,messageId:this.reasoningProcess.messageId}),this.reasoningProcess=null),l&&(this.activeRun.hasPredictState=!0,this.dispatchEvent({type:e.EventType.CUSTOM,name:`PredictState`,value:t.metadata?.predict_state})),f){this.dispatchEvent({type:e.EventType.TOOL_CALL_END,toolCallId:o?.toolCallId,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(b){this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_END,messageId:o.id,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(u&&a){this.dispatchEvent({type:e.EventType.TOOL_CALL_START,toolCallId:c.id,toolCallName:c.name,parentMessageId:t.data.chunk.id,rawEvent:t})&&this.setMessageInProgress(this.activeRun.id,{id:t.data.chunk.id,toolCallId:c.id,toolCallName:c.name});break}if(d&&a){this.dispatchEvent({type:e.EventType.TOOL_CALL_ARGS,toolCallId:o?.toolCallId,delta:c.args,rawEvent:t});break}if(y&&n){o||=(this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_START,role:`assistant`,messageId:t.data.chunk.id,rawEvent:t}),this.setMessageInProgress(this.activeRun.id,{id:t.data.chunk.id,toolCallId:null,toolCallName:null}),this.getMessageInProgress(this.activeRun.id)),this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_CONTENT,messageId:o.id,delta:v,rawEvent:t});break}break;case r.OnChatModelEnd:{let n=t.data?.output;if(n&&typeof n==`object`&&n.id&&n.type&&(this.activeRun.streamedMessages||(this.activeRun.streamedMessages=[]),this.activeRun.streamedMessages.push(n)),this.getMessageInProgress(this.activeRun.id)?.toolCallId){this.dispatchEvent({type:e.EventType.TOOL_CALL_END,toolCallId:this.getMessageInProgress(this.activeRun.id).toolCallId,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(this.getMessageInProgress(this.activeRun.id)?.id){this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_END,messageId:this.getMessageInProgress(this.activeRun.id).id,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}break}case r.OnCustomEvent:if(t.name===i.ManuallyEmitMessage){this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_START,role:`assistant`,messageId:t.data.message_id,rawEvent:t}),this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_CONTENT,messageId:t.data.message_id,delta:t.data.message,rawEvent:t}),this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_END,messageId:t.data.message_id,rawEvent:t});break}if(t.name===i.ManuallyEmitToolCall){this.dispatchEvent({type:e.EventType.TOOL_CALL_START,toolCallId:t.data.id,toolCallName:t.data.name,parentMessageId:t.data.id,rawEvent:t}),this.dispatchEvent({type:e.EventType.TOOL_CALL_ARGS,toolCallId:t.data.id,delta:t.data.args,rawEvent:t}),this.dispatchEvent({type:e.EventType.TOOL_CALL_END,toolCallId:t.data.id,rawEvent:t});break}t.name===i.ManuallyEmitState&&(this.activeRun.manuallyEmittedState=t.data,this.dispatchEvent({type:e.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot({values:this.activeRun.manuallyEmittedState}),rawEvent:t})),this.dispatchEvent({type:e.EventType.CUSTOM,name:t.name,value:t.data,rawEvent:t});break;case r.OnToolEnd:this.activeRun.hasPredictState=!1;let x=t.data?.output;if(x&&!x.tool_call_id&&x.update?.messages?.find(e=>e.type===`tool`)&&(x=x.update?.messages?.find(e=>e.type===`tool`)),x&&x.update?.messages?.length){x.update?.messages.filter(e=>e.type===`tool`).forEach(n=>{this.activeRun.hasFunctionStreaming||(this.dispatchEvent({type:e.EventType.TOOL_CALL_START,toolCallId:n.tool_call_id,toolCallName:n.name??``,parentMessageId:n.id,rawEvent:t}),this.dispatchEvent({type:e.EventType.TOOL_CALL_ARGS,toolCallId:n.tool_call_id,delta:JSON.stringify(t.data.input),rawEvent:t})),this.dispatchEvent({type:e.EventType.TOOL_CALL_RESULT,toolCallId:n.tool_call_id,content:typeof n?.content==`string`?n?.content:JSON.stringify(n?.content),messageId:(0,e.randomUUID)(),rawEvent:t,role:`tool`})});break}this.activeRun.hasFunctionStreaming||(this.dispatchEvent({type:e.EventType.TOOL_CALL_START,toolCallId:x.tool_call_id,toolCallName:x.name,parentMessageId:x.id,rawEvent:t}),this.dispatchEvent({type:e.EventType.TOOL_CALL_ARGS,toolCallId:x.tool_call_id,delta:JSON.stringify(t.data.input),rawEvent:t}),this.dispatchEvent({type:e.EventType.TOOL_CALL_END,toolCallId:x.tool_call_id,rawEvent:t}));let S=Array.isArray(x.content)?x.content.map(e=>typeof e==`string`?e:e.type===`text`?e.text:JSON.stringify(e)).join(``):x.content;this.dispatchEvent({type:e.EventType.TOOL_CALL_RESULT,toolCallId:x.tool_call_id,content:S,messageId:(0,e.randomUUID)(),role:`tool`,rawEvent:t});break}}handleMessagesTupleEvent(t){let n=t[0];if(n.type&&n.type!==`AIMessageChunk`)return;let r=typeof n.content==`string`?n.content:Array.isArray(n.content)?n.content.find(e=>e.type===`text`)?.text:null,i=n.tool_call_chunks,a=n.response_metadata?.finish_reason===`stop`,o=this.getMessageInProgress(this.activeRun.id);if(i?.length>0){let t=i[0];t.name?(o?.id&&!o?.toolCallId&&(this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_END,messageId:o.id}),this.messagesInProcess[this.activeRun.id]=null),this.dispatchEvent({type:e.EventType.TOOL_CALL_START,toolCallId:t.id||n.id,toolCallName:t.name,parentMessageId:n.id}),this.setMessageInProgress(this.activeRun.id,{id:n.id,toolCallId:t.id||n.id,toolCallName:t.name}),this.activeRun.hasFunctionStreaming=!0):t.args&&o?.toolCallId&&this.dispatchEvent({type:e.EventType.TOOL_CALL_ARGS,toolCallId:o.toolCallId,delta:t.args});return}if(a){o?.toolCallId?this.dispatchEvent({type:e.EventType.TOOL_CALL_END,toolCallId:o.toolCallId}):o?.id&&this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_END,messageId:o.id}),this.messagesInProcess[this.activeRun.id]=null;return}!r&&!i?.length||r&&(o||(this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_START,role:`assistant`,messageId:n.id}),this.setMessageInProgress(this.activeRun.id,{id:n.id,toolCallId:null,toolCallName:null})),this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_CONTENT,messageId:(this.getMessageInProgress(this.activeRun.id)??{id:n.id}).id,delta:r}))}abortRun(){this.cancelRequested=!0;let e=this.activeRun?.threadId,t=this.activeRun?.id;e&&t&&!this.cancelSent&&this.client.runs.cancel(e,t).then(()=>{this.cancelSent=!0}).catch(()=>{}),super.abortRun()}handleReasoningEvent(t){if(!t||!t.type||!t.text)return;let n=t.index;if(this.reasoningProcess?.index&&this.reasoningProcess.index!==n&&(this.reasoningProcess.type&&this.dispatchEvent({type:e.EventType.REASONING_MESSAGE_END,messageId:this.reasoningProcess.messageId}),this.dispatchEvent({type:e.EventType.REASONING_END,messageId:this.reasoningProcess.messageId}),this.reasoningProcess=null),!this.reasoningProcess){let t=(0,e.randomUUID)();this.dispatchEvent({type:e.EventType.REASONING_START,messageId:t}),this.reasoningProcess={index:n,messageId:t}}this.reasoningProcess.type!==t.type&&(this.dispatchEvent({type:e.EventType.REASONING_MESSAGE_START,messageId:this.reasoningProcess.messageId,role:`reasoning`}),this.reasoningProcess.type=t.type),t.signature&&(this.reasoningProcess.signature=t.signature),this.reasoningProcess.type&&this.dispatchEvent({type:e.EventType.REASONING_MESSAGE_CONTENT,messageId:this.reasoningProcess.messageId,delta:t.text})}getStateSnapshot(e){let t=e.values,n=this.activeRun.schemaKeys;return n?.output&&(t=o(t,[...this.constantSchemaKeys,...n.output])),t}async getOrCreateThread(e,t){let n;try{try{n=await this.getThread(e)}catch{n=await this.createThread({threadId:e,metadata:t})}}catch(e){throw Error(`Failed to create thread: ${e.message}`)}return n}async getThread(e){return this.client.threads.get(e)}async createThread(e){return this.client.threads.create(e)}async mergeConfigs({configs:e,assistant:t,schemaKeys:n}){return e.reduce((e,t)=>{let r=e.configurable;t.configurable&&(r=n?.config?o(t?.configurable,[...this.constantSchemaKeys,...n?.config??[]]):t?.configurable);let i={...e,...t,configurable:r},a=e.recursion_limit==null&&t.recursion_limit===25,s=JSON.stringify(i)!==JSON.stringify(e),c=a&&JSON.stringify({...i,recursion_limit:null})===JSON.stringify({...e,recursion_limit:null});return s&&!c?{...e,...i}:e},t.config)}getMessageInProgress(e){return this.messagesInProcess[e]}setMessageInProgress(e,t){this.messagesInProcess={...this.messagesInProcess,[e]:{...this.messagesInProcess[e],...t}}}async getAssistant(){try{let e=await this.client.assistants.search(),t=e.find(e=>e.graph_id===this.graphId);if(!t){let t=`
|
|
1
|
+
let e=require(`@ag-ui/client`),t=require(`rxjs`),n=require(`@langchain/langgraph-sdk`),r=function(e){return e.OnChainStart=`on_chain_start`,e.OnChainStream=`on_chain_stream`,e.OnChainEnd=`on_chain_end`,e.OnChatModelStart=`on_chat_model_start`,e.OnChatModelStream=`on_chat_model_stream`,e.OnChatModelEnd=`on_chat_model_end`,e.OnToolStart=`on_tool_start`,e.OnToolEnd=`on_tool_end`,e.OnCustomEvent=`on_custom_event`,e.OnInterrupt=`on_interrupt`,e}({}),i=function(e){return e.ManuallyEmitMessage=`manually_emit_message`,e.ManuallyEmitToolCall=`manually_emit_tool_call`,e.ManuallyEmitState=`manually_emit_state`,e.Exit=`exit`,e}({});const a=[`messages`,`tools`];function o(e,t){return Object.fromEntries(Object.entries(e).filter(([e])=>t.includes(e)))}function s({mode:e,state:t,schemaKeys:n}){let r=e===`start`?t:null;return r&&n?.input&&(r=o(r,[...a,...n.input])),r}function c(e){let t=[];for(let n of e)if(n.type===`text`&&n.text)t.push({type:`text`,text:n.text});else if(n.type===`image_url`){let e=typeof n.image_url==`string`?n.image_url:n.image_url?.url;if(!e)continue;if(e.startsWith(`data:`)){let[n,r]=e.split(`,`,2),i=n.includes(`:`)?n.split(`:`)[1].split(`;`)[0]:`image/png`;t.push({type:`binary`,mimeType:i,data:r||``})}else t.push({type:`binary`,mimeType:`image/png`,url:e})}return t}function l(e){let t=[];for(let n of e)if(n.type===`text`)t.push({type:`text`,text:n.text});else if(n.type===`binary`){let e;if(n.url)e=n.url;else if(n.data)e=`data:${n.mimeType};base64,${n.data}`;else if(n.id)e=n.id;else continue;t.push({type:`image_url`,image_url:{url:e}})}return t}function u(e){return e.map(e=>{switch(e.type){case`human`:let t;return t=Array.isArray(e.content)?c(e.content):f(h(e.content)),{id:e.id,role:`user`,content:t};case`ai`:let n=h(e.content);return{id:e.id,role:`assistant`,content:n?f(n):``,toolCalls:e.tool_calls?.map(e=>({id:e.id,type:`function`,function:{name:e.name,arguments:JSON.stringify(e.args)}}))};case`system`:return{id:e.id,role:`system`,content:f(h(e.content))};case`tool`:return{id:e.id,role:`tool`,content:f(h(e.content)),toolCallId:e.tool_call_id};default:throw Error(`message type returned from LangGraph is not supported.`)}})}function d(e){return e.map((e,t)=>{switch(e.role){case`user`:let t;return t=typeof e.content==`string`?e.content:Array.isArray(e.content)?l(e.content):String(e.content),{id:e.id,role:e.role,content:t,type:`human`};case`assistant`:return{id:e.id,type:`ai`,role:e.role,content:e.content??``,tool_calls:(e.toolCalls??[]).map(e=>({id:e.id,name:e.function.name,args:JSON.parse(e.function.arguments),type:`tool_call`}))};case`system`:return{id:e.id,role:e.role,content:e.content,type:`system`};case`tool`:return{content:e.content,role:e.role,type:e.role,tool_call_id:e.toolCallId,id:e.id};default:throw console.error(`Message role ${e.role} is not implemented`),Error(`message role is not supported.`)}})}function f(e){return typeof e==`string`?e:JSON.stringify(e)}function p(e){let t=e.chunk?.content;if(t&&Array.isArray(t)&&t.length&&t[0]){let e=t[0];if(e.type===`thinking`&&e.thinking){let t={text:e.thinking,type:`text`,index:e.index??0};return e.signature&&(t.signature=e.signature),t}if(e.type===`reasoning`&&e.reasoning)return{text:e.reasoning,type:`text`,index:e.index??0};if(e.type===`reasoning`&&e.summary?.[0]?.text)return{type:`text`,text:e.summary[0].text,index:e.summary[0].index??0}}if(e.chunk?.additional_kwargs?.reasoning?.summary?.[0]){let t=e.chunk.additional_kwargs.reasoning.summary[0];return!t||!t.text?null:{type:`text`,text:t.text,index:t.index??0}}return null}function m(e){let t=e.chunk?.content;return!t||!Array.isArray(t)||!t.length||!t[0]?null:t[0].type===`redacted_thinking`&&t[0].data?t[0].data:null}function h(e){return e?typeof e==`string`?e:Array.isArray(e)&&e.length?e.find(e=>e.type===`text`)?.text??null:null:null}var g=class extends e.AbstractAgent{constructor(e){super(e),this.emittedToolCallStartIds=new Set,this.cancelRequested=!1,this.cancelSent=!1,this.eventsStreamActive=!1,this.constantSchemaKeys=a,this.config=e,this.messagesInProcess={},this.agentName=e.agentName,this.graphId=e.graphId,this.assistantConfig=e.assistantConfig,this.reasoningProcess=null,this.client=e?.client??new n.Client({apiUrl:e.deploymentUrl,apiKey:e.langsmithApiKey,defaultHeaders:{...e.propertyHeaders??{}}})}clone(){return Object.assign(super.clone(),{config:this.config,messagesInProcess:structuredClone(this.messagesInProcess),agentName:this.agentName,graphId:this.graphId,assistantConfig:this.assistantConfig,reasoningProcess:this.reasoningProcess?structuredClone(this.reasoningProcess):null,constantSchemaKeys:[...this.constantSchemaKeys],client:this.client,assistant:this.assistant,activeRun:this.activeRun?structuredClone(this.activeRun):void 0,cancelRequested:this.cancelRequested,cancelSent:this.cancelSent})}dispatchEvent(e){return this.subscriber.next(e),!0}run(e){return new t.Observable(t=>(this.runAgentStream(e,t).catch(e=>{console.error(`[LangGraph] runAgentStream error:`,e),t.closed||t.error(e)}),()=>{}))}async runAgentStream(t,n){this.activeRun={id:t.runId,threadId:t.threadId,hasFunctionStreaming:!1},this.cancelRequested=!1,this.cancelSent=!1,this.eventsStreamActive=!1,this.subscriber=n,this.assistant||=await this.getAssistant();let r=t.threadId??(0,e.randomUUID)(),i=t.forwardedProps?.streamMode??[`events`,`values`,`updates`,`messages-tuple`],a=await this.prepareStream({...t,threadId:r},i);if(!a)return n.error(`No stream to regenerate`);await this.handleStreamEvents(a,r,n,t,Array.isArray(i)?i:[i])}async prepareRegenerateStream(e,t){let{threadId:n,messageCheckpoint:r}=e,i=await this.getCheckpointByMessage(r.id,n);if(this.assistant||=await this.getAssistant(),!i)return this.subscriber.error(`No checkpoint found for message`);let a=await this.client.threads.updateState(n,{values:this.langGraphDefaultMergeState(i.values,[],e),checkpointId:i.checkpoint.checkpoint_id,asNode:i.next?.[0]??`__start__`}),o={...e.forwardedProps??{},input:this.langGraphDefaultMergeState(i.values,[r],e),checkpointId:a.checkpoint.checkpoint_id,streamMode:t};return{streamResponse:this.client.runs.stream(n,this.assistant.assistant_id,o),state:i,streamMode:t}}async prepareStream(t,n){let{threadId:i,state:a,messages:o,tools:c,context:l,forwardedProps:u}=t;this.activeRun.manuallyEmittedState=null;let f=u?.nodeName,p=i??(0,e.randomUUID)();this.assistant||=await this.getAssistant();let m=await this.getOrCreateThread(p,u?.threadMetadata);this.activeRun.threadId=m.thread_id;let h=await this.client.threads.getState(m.thread_id)??{values:{}},g=h.values.messages??[],_=d(o),v=this.langGraphDefaultMergeState({...a,messages:g},_,t),y={...h,values:{...v,messages:[...g,...v.messages??[]]}},b=y.values;if(this.activeRun.schemaKeys=await this.getSchemaKeys(),g.filter(e=>e.type!==`system`).length>o.filter(e=>e.role!==`system`).length){let e=null;for(let t=o.length-1;t>=0;t--)if(o[t].role===`user`){e=d([o[t]])[0];break}return e?this.prepareRegenerateStream({...t,messageCheckpoint:e},n):this.subscriber.error(`No user message found in messages to regenerate`)}this.activeRun.graphInfo=await this.client.assistants.getGraph(this.assistant.assistant_id);let x=!u?.command?.resume&&p&&this.activeRun.nodeName!=`__end__`&&this.activeRun.nodeName?`continue`:`start`;if(x===`continue`){let e=this.activeRun.graphInfo.edges.find(e=>e.target===this.activeRun.nodeName);await this.client.threads.updateState(p,{values:a,asNode:e?.source})}let S=s({mode:x,state:b,schemaKeys:this.activeRun.schemaKeys}),C,w=[this.assistantConfig,u?.config].filter(Boolean);w.length&&(C=await this.mergeConfigs({configs:w,assistant:this.assistant,schemaKeys:this.activeRun.schemaKeys}));let{command:T,...E}=u;if(T?.resume&&typeof T.resume==`string`)try{T.resume=JSON.parse(T.resume)}catch{}let D={...E,command:T,streamMode:n,input:S,config:C,context:{...l,...C?.configurable??{}}},O=h.tasks?.[0]?.interrupts??[];return O?.length&&!u?.command?.resume?(this.dispatchEvent({type:e.EventType.RUN_STARTED,threadId:p,runId:t.runId}),this.handleNodeChange(f),O.forEach(t=>{this.dispatchEvent({type:e.EventType.CUSTOM,name:r.OnInterrupt,value:typeof t.value==`string`?t.value:JSON.stringify(t.value),rawEvent:t})}),this.dispatchEvent({type:e.EventType.RUN_FINISHED,threadId:p,runId:t.runId}),this.subscriber.complete()):{streamResponse:this.client.runs.stream(p,this.assistant.assistant_id,D),state:y}}async handleStreamEvents(t,n,a,o,s){let{forwardedProps:c}=o,l=c?.nodeName;this.subscriber=a;let d=!1;if(!t)return;this.emittedToolCallStartIds=new Set;let{streamResponse:f,state:p}=t;this.activeRun.prevNodeName=null;let m={},h=p;try{this.dispatchEvent({type:e.EventType.RUN_STARTED,threadId:n,runId:this.activeRun.id}),this.handleNodeChange(l);for await(let t of f){if(this.cancelRequested&&!this.cancelSent&&this.activeRun?.threadId&&this.activeRun?.id){try{await this.client.runs.cancel(this.activeRun.threadId,this.activeRun.id)}catch{}finally{this.cancelSent=!0}try{await f?.return?.()}catch{}break}let n=o.forwardedProps?.streamSubgraphs,a=n&&(t.event.startsWith(`events`)||t.event.startsWith(`values`)),c=t.event===`messages`&&(Array.isArray(s)?s:[s]).includes(`messages-tuple`);if(!s.includes(t.event)&&!a&&!c&&t.event!==`error`)continue;let l=t;if(t.event===`error`){this.dispatchEvent({type:e.EventType.RUN_ERROR,message:t.data.message,rawEvent:t});break}if(t.event===`updates`)continue;if(t.event===`values`){m=l.data;continue}else if(n&&l.event.startsWith(`values|`)){m={...m,...l.data};continue}let u=l.data,g=u.metadata??{},_=g.langgraph_node,v=u.event;if(g.run_id&&(this.activeRun.id=g.run_id,this.activeRun.serverRunIdKnown=!0,this.cancelRequested&&!this.cancelSent&&this.activeRun?.threadId))try{await this.client.runs.cancel(this.activeRun.threadId,this.activeRun.id)}catch{}finally{this.cancelSent=!0}_&&_!==this.activeRun.nodeName&&this.handleNodeChange(_),d||=v===r.OnCustomEvent&&u.name===i.Exit,v===r.OnChainEnd&&this.activeRun.nodeName===_&&(this.activeRun.exitingNode=!0),this.activeRun.exitingNode&&(this.activeRun.manuallyEmittedState=null),this.activeRun.graphInfo?.nodes.some(e=>e.id===_)&&this.handleNodeChange(_),h.values=this.activeRun.manuallyEmittedState??m,this.activeRun.nodeName&&((JSON.stringify(h)!==JSON.stringify(p)||this.activeRun.prevNodeName!=this.activeRun.nodeName||this.activeRun.exitingNode)&&!this.getMessageInProgress(this.activeRun.id)&&(p=h,this.activeRun.prevNodeName=this.activeRun.nodeName,this.dispatchEvent({type:e.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(p),rawEvent:l})),this.dispatchEvent({type:e.EventType.RAW,event:u}),this.handleSingleEvent(u))}p=await this.client.threads.getState(n);let t=p.tasks?.[0]?.interrupts??[],c=p.next.length===0,g=p.metadata?.writes??{},_=this.activeRun.nodeName;t?.length||(_=c?`__end__`:p.next[0]??Object.keys(g)[0]),t.forEach(t=>{this.dispatchEvent({type:e.EventType.CUSTOM,name:r.OnInterrupt,value:typeof t.value==`string`?t.value:JSON.stringify(t.value),rawEvent:t})}),this.handleNodeChange(_),this.handleNodeChange(void 0),this.dispatchEvent({type:e.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(p)});let v=p.values.messages??[],y=this.activeRun?.streamedMessages??[],b=new Set(v.map(e=>e.id).filter(Boolean)),x=y.filter(e=>e.id&&!b.has(e.id)),S=[...v,...x];return this.dispatchEvent({type:e.EventType.MESSAGES_SNAPSHOT,messages:u(S)}),this.dispatchEvent({type:e.EventType.RUN_FINISHED,threadId:n,runId:this.activeRun.id}),this.cancelRequested=!1,this.cancelSent=!1,this.activeRun=void 0,a.complete()}catch(e){return a.error(e)}}handleSingleEvent(t){if(Array.isArray(t)){this.eventsStreamActive||this.handleMessagesTupleEvent(t);return}switch(t.event===r.OnChatModelStream&&(this.eventsStreamActive=!0),t.event){case r.OnChatModelStream:let n=t.metadata[`emit-messages`]??!0,a=t.metadata[`emit-tool-calls`]??!0;if(t.data.chunk.response_metadata.finish_reason)return;let o=this.getMessageInProgress(this.activeRun.id),s=!!o?.id,c=t.data.chunk.tool_call_chunks?.[0],l=t.metadata.predict_state?.some(e=>e.tool===c?.name),u=!s&&c?.name,d=s&&o?.toolCallId&&c?.args,f=s&&o?.toolCallId&&!c;(f||d||u)&&(this.activeRun.hasFunctionStreaming=!0);let g=p(t.data),_=m(t.data),v=h(t.data.chunk.content),y=!!(!c&&v),b=s&&!o?.toolCallId&&!y;if(g){this.handleReasoningEvent(g);break}if(_&&this.reasoningProcess){this.dispatchEvent({type:e.EventType.REASONING_ENCRYPTED_VALUE,subtype:`message`,entityId:this.reasoningProcess.messageId,encryptedValue:_});break}if(!g&&this.reasoningProcess&&(this.reasoningProcess.signature&&this.dispatchEvent({type:e.EventType.REASONING_ENCRYPTED_VALUE,subtype:`message`,entityId:this.reasoningProcess.messageId,encryptedValue:this.reasoningProcess.signature}),this.dispatchEvent({type:e.EventType.REASONING_MESSAGE_END,messageId:this.reasoningProcess.messageId}),this.dispatchEvent({type:e.EventType.REASONING_END,messageId:this.reasoningProcess.messageId}),this.reasoningProcess=null),l&&this.dispatchEvent({type:e.EventType.CUSTOM,name:`PredictState`,value:t.metadata.predict_state}),f){this.dispatchEvent({type:e.EventType.TOOL_CALL_END,toolCallId:o?.toolCallId,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(b){this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_END,messageId:o.id,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(u&&a){this.dispatchEvent({type:e.EventType.TOOL_CALL_START,toolCallId:c.id,toolCallName:c.name,parentMessageId:t.data.chunk.id,rawEvent:t})&&(this.emittedToolCallStartIds.add(c.id),this.setMessageInProgress(this.activeRun.id,{id:t.data.chunk.id,toolCallId:c.id,toolCallName:c.name}));break}if(d&&a){this.dispatchEvent({type:e.EventType.TOOL_CALL_ARGS,toolCallId:o?.toolCallId,delta:c.args,rawEvent:t});break}if(y&&n){o||=(this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_START,role:`assistant`,messageId:t.data.chunk.id,rawEvent:t}),this.setMessageInProgress(this.activeRun.id,{id:t.data.chunk.id,toolCallId:null,toolCallName:null}),this.getMessageInProgress(this.activeRun.id)),this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_CONTENT,messageId:o.id,delta:v,rawEvent:t});break}break;case r.OnChatModelEnd:{let n=t.data?.output;if(n&&typeof n==`object`&&n.id&&n.type&&(this.activeRun.streamedMessages||(this.activeRun.streamedMessages=[]),this.activeRun.streamedMessages.push(n)),this.getMessageInProgress(this.activeRun.id)?.toolCallId){this.dispatchEvent({type:e.EventType.TOOL_CALL_END,toolCallId:this.getMessageInProgress(this.activeRun.id).toolCallId,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(this.getMessageInProgress(this.activeRun.id)?.id){this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_END,messageId:this.getMessageInProgress(this.activeRun.id).id,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}break}case r.OnCustomEvent:if(t.name===i.ManuallyEmitMessage){this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_START,role:`assistant`,messageId:t.data.message_id,rawEvent:t}),this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_CONTENT,messageId:t.data.message_id,delta:t.data.message,rawEvent:t}),this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_END,messageId:t.data.message_id,rawEvent:t});break}if(t.name===i.ManuallyEmitToolCall){this.dispatchEvent({type:e.EventType.TOOL_CALL_START,toolCallId:t.data.id,toolCallName:t.data.name,parentMessageId:t.data.id,rawEvent:t}),this.dispatchEvent({type:e.EventType.TOOL_CALL_ARGS,toolCallId:t.data.id,delta:t.data.args,rawEvent:t}),this.dispatchEvent({type:e.EventType.TOOL_CALL_END,toolCallId:t.data.id,rawEvent:t});break}t.name===i.ManuallyEmitState&&(this.activeRun.manuallyEmittedState=t.data,this.dispatchEvent({type:e.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot({values:this.activeRun.manuallyEmittedState}),rawEvent:t})),this.dispatchEvent({type:e.EventType.CUSTOM,name:t.name,value:t.data,rawEvent:t});break;case r.OnToolEnd:let x=t.data?.output;if(x&&!x.tool_call_id&&x.update?.messages?.find(e=>e.type===`tool`)&&(x=x.update?.messages?.find(e=>e.type===`tool`)),x&&x.update?.messages?.length){x.update?.messages.filter(e=>e.type===`tool`).forEach(n=>{this.activeRun.hasFunctionStreaming||(this.dispatchEvent({type:e.EventType.TOOL_CALL_START,toolCallId:n.tool_call_id,toolCallName:n.name??``,parentMessageId:n.id,rawEvent:t}),this.dispatchEvent({type:e.EventType.TOOL_CALL_ARGS,toolCallId:n.tool_call_id,delta:JSON.stringify(t.data.input),rawEvent:t})),this.dispatchEvent({type:e.EventType.TOOL_CALL_RESULT,toolCallId:n.tool_call_id,content:typeof n?.content==`string`?n?.content:JSON.stringify(n?.content),messageId:(0,e.randomUUID)(),rawEvent:t,role:`tool`})});break}this.emittedToolCallStartIds.has(x.tool_call_id)||(this.emittedToolCallStartIds.add(x.tool_call_id),this.dispatchEvent({type:e.EventType.TOOL_CALL_START,toolCallId:x.tool_call_id,toolCallName:x.name,parentMessageId:x.id,rawEvent:t}),this.dispatchEvent({type:e.EventType.TOOL_CALL_ARGS,toolCallId:x.tool_call_id,delta:JSON.stringify(t.data.input),rawEvent:t}),this.dispatchEvent({type:e.EventType.TOOL_CALL_END,toolCallId:x.tool_call_id,rawEvent:t}));let S=Array.isArray(x.content)?x.content.map(e=>typeof e==`string`?e:e.type===`text`?e.text:JSON.stringify(e)).join(``):x.content;this.dispatchEvent({type:e.EventType.TOOL_CALL_RESULT,toolCallId:x.tool_call_id,content:S,messageId:(0,e.randomUUID)(),role:`tool`,rawEvent:t});break}}handleMessagesTupleEvent(t){let n=t[0];if(n.type&&n.type!==`AIMessageChunk`)return;let r=typeof n.content==`string`?n.content:Array.isArray(n.content)?n.content.find(e=>e.type===`text`)?.text:null,i=n.tool_call_chunks,a=n.response_metadata?.finish_reason===`stop`,o=this.getMessageInProgress(this.activeRun.id);if(i?.length>0){let t=i[0];t.name?(o?.id&&!o?.toolCallId&&(this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_END,messageId:o.id}),this.messagesInProcess[this.activeRun.id]=null),this.dispatchEvent({type:e.EventType.TOOL_CALL_START,toolCallId:t.id||n.id,toolCallName:t.name,parentMessageId:n.id}),this.setMessageInProgress(this.activeRun.id,{id:n.id,toolCallId:t.id||n.id,toolCallName:t.name}),this.activeRun.hasFunctionStreaming=!0):t.args&&o?.toolCallId&&this.dispatchEvent({type:e.EventType.TOOL_CALL_ARGS,toolCallId:o.toolCallId,delta:t.args});return}if(a){o?.toolCallId?this.dispatchEvent({type:e.EventType.TOOL_CALL_END,toolCallId:o.toolCallId}):o?.id&&this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_END,messageId:o.id}),this.messagesInProcess[this.activeRun.id]=null;return}!r&&!i?.length||r&&(o||(this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_START,role:`assistant`,messageId:n.id}),this.setMessageInProgress(this.activeRun.id,{id:n.id,toolCallId:null,toolCallName:null})),this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_CONTENT,messageId:(this.getMessageInProgress(this.activeRun.id)??{id:n.id}).id,delta:r}))}abortRun(){this.cancelRequested=!0;let e=this.activeRun?.threadId,t=this.activeRun?.id;e&&t&&!this.cancelSent&&this.client.runs.cancel(e,t).then(()=>{this.cancelSent=!0}).catch(()=>{}),super.abortRun()}handleReasoningEvent(t){if(!t||!t.type||!t.text)return;let n=t.index;if(this.reasoningProcess?.index&&this.reasoningProcess.index!==n&&(this.reasoningProcess.type&&this.dispatchEvent({type:e.EventType.REASONING_MESSAGE_END,messageId:this.reasoningProcess.messageId}),this.dispatchEvent({type:e.EventType.REASONING_END,messageId:this.reasoningProcess.messageId}),this.reasoningProcess=null),!this.reasoningProcess){let t=(0,e.randomUUID)();this.dispatchEvent({type:e.EventType.REASONING_START,messageId:t}),this.reasoningProcess={index:n,messageId:t}}this.reasoningProcess.type!==t.type&&(this.dispatchEvent({type:e.EventType.REASONING_MESSAGE_START,messageId:this.reasoningProcess.messageId,role:`reasoning`}),this.reasoningProcess.type=t.type),t.signature&&(this.reasoningProcess.signature=t.signature),this.reasoningProcess.type&&this.dispatchEvent({type:e.EventType.REASONING_MESSAGE_CONTENT,messageId:this.reasoningProcess.messageId,delta:t.text})}getStateSnapshot(e){let t=e.values,n=this.activeRun.schemaKeys;return n?.output&&(t=o(t,[...this.constantSchemaKeys,...n.output])),t}async getOrCreateThread(e,t){let n;try{try{n=await this.getThread(e)}catch{n=await this.createThread({threadId:e,metadata:t})}}catch(e){throw Error(`Failed to create thread: ${e.message}`)}return n}async getThread(e){return this.client.threads.get(e)}async createThread(e){return this.client.threads.create(e)}async mergeConfigs({configs:e,assistant:t,schemaKeys:n}){return e.reduce((e,t)=>{let r=e.configurable;t.configurable&&(r=n?.config?o(t?.configurable,[...this.constantSchemaKeys,...n?.config??[]]):t?.configurable);let i={...e,...t,configurable:r},a=e.recursion_limit==null&&t.recursion_limit===25,s=JSON.stringify(i)!==JSON.stringify(e),c=a&&JSON.stringify({...i,recursion_limit:null})===JSON.stringify({...e,recursion_limit:null});return s&&!c?{...e,...i}:e},t.config)}getMessageInProgress(e){return this.messagesInProcess[e]}setMessageInProgress(e,t){this.messagesInProcess={...this.messagesInProcess,[e]:{...this.messagesInProcess[e],...t}}}async getAssistant(){try{let e=await this.client.assistants.search({graphId:this.graphId,limit:1}),t=e.find(e=>e.graph_id===this.graphId);if(!t){let t=`
|
|
2
2
|
No agent found with graph ID ${this.graphId} found..\n
|
|
3
3
|
|
|
4
4
|
These are the available agents: [${e.map(e=>`${e.graph_id} (ID: ${e.assistant_id})`).join(`, `)}]
|