@grafana/sigil-sdk-js 0.0.1
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/LICENSE +3 -0
- package/README.md +367 -0
- package/dist/client.d.ts +102 -0
- package/dist/client.d.ts.map +1 -0
- package/dist/client.js +1502 -0
- package/dist/client.js.map +1 -0
- package/dist/config.d.ts +9 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +190 -0
- package/dist/config.js.map +1 -0
- package/dist/content_capture.d.ts +35 -0
- package/dist/content_capture.d.ts.map +1 -0
- package/dist/content_capture.js +123 -0
- package/dist/content_capture.js.map +1 -0
- package/dist/context.d.ts +11 -0
- package/dist/context.d.ts.map +1 -0
- package/dist/context.js +53 -0
- package/dist/context.js.map +1 -0
- package/dist/exporters/default.d.ts +3 -0
- package/dist/exporters/default.d.ts.map +1 -0
- package/dist/exporters/default.js +34 -0
- package/dist/exporters/default.js.map +1 -0
- package/dist/exporters/grpc.d.ts +14 -0
- package/dist/exporters/grpc.d.ts.map +1 -0
- package/dist/exporters/grpc.js +369 -0
- package/dist/exporters/grpc.js.map +1 -0
- package/dist/exporters/http.d.ts +8 -0
- package/dist/exporters/http.d.ts.map +1 -0
- package/dist/exporters/http.js +258 -0
- package/dist/exporters/http.js.map +1 -0
- package/dist/frameworks/google-adk/index.d.ts +136 -0
- package/dist/frameworks/google-adk/index.d.ts.map +1 -0
- package/dist/frameworks/google-adk/index.js +509 -0
- package/dist/frameworks/google-adk/index.js.map +1 -0
- package/dist/frameworks/langchain/index.d.ts +29 -0
- package/dist/frameworks/langchain/index.d.ts.map +1 -0
- package/dist/frameworks/langchain/index.js +70 -0
- package/dist/frameworks/langchain/index.js.map +1 -0
- package/dist/frameworks/langgraph/index.d.ts +29 -0
- package/dist/frameworks/langgraph/index.d.ts.map +1 -0
- package/dist/frameworks/langgraph/index.js +70 -0
- package/dist/frameworks/langgraph/index.js.map +1 -0
- package/dist/frameworks/llamaindex/index.d.ts +43 -0
- package/dist/frameworks/llamaindex/index.d.ts.map +1 -0
- package/dist/frameworks/llamaindex/index.js +493 -0
- package/dist/frameworks/llamaindex/index.js.map +1 -0
- package/dist/frameworks/openai-agents/index.d.ts +33 -0
- package/dist/frameworks/openai-agents/index.d.ts.map +1 -0
- package/dist/frameworks/openai-agents/index.js +531 -0
- package/dist/frameworks/openai-agents/index.js.map +1 -0
- package/dist/frameworks/shared.d.ts +57 -0
- package/dist/frameworks/shared.d.ts.map +1 -0
- package/dist/frameworks/shared.js +1032 -0
- package/dist/frameworks/shared.js.map +1 -0
- package/dist/frameworks/vercel-ai-sdk/hooks.d.ts +18 -0
- package/dist/frameworks/vercel-ai-sdk/hooks.d.ts.map +1 -0
- package/dist/frameworks/vercel-ai-sdk/hooks.js +672 -0
- package/dist/frameworks/vercel-ai-sdk/hooks.js.map +1 -0
- package/dist/frameworks/vercel-ai-sdk/index.d.ts +8 -0
- package/dist/frameworks/vercel-ai-sdk/index.d.ts.map +1 -0
- package/dist/frameworks/vercel-ai-sdk/index.js +7 -0
- package/dist/frameworks/vercel-ai-sdk/index.js.map +1 -0
- package/dist/frameworks/vercel-ai-sdk/mapping.d.ts +49 -0
- package/dist/frameworks/vercel-ai-sdk/mapping.d.ts.map +1 -0
- package/dist/frameworks/vercel-ai-sdk/mapping.js +660 -0
- package/dist/frameworks/vercel-ai-sdk/mapping.js.map +1 -0
- package/dist/frameworks/vercel-ai-sdk/types.d.ts +126 -0
- package/dist/frameworks/vercel-ai-sdk/types.d.ts.map +1 -0
- package/dist/frameworks/vercel-ai-sdk/types.js +2 -0
- package/dist/frameworks/vercel-ai-sdk/types.js.map +1 -0
- package/dist/index.d.ts +12 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +12 -0
- package/dist/index.js.map +1 -0
- package/dist/providers/anthropic.d.ts +35 -0
- package/dist/providers/anthropic.d.ts.map +1 -0
- package/dist/providers/anthropic.js +581 -0
- package/dist/providers/anthropic.js.map +1 -0
- package/dist/providers/gemini.d.ts +42 -0
- package/dist/providers/gemini.d.ts.map +1 -0
- package/dist/providers/gemini.js +650 -0
- package/dist/providers/gemini.js.map +1 -0
- package/dist/providers/openai.d.ts +67 -0
- package/dist/providers/openai.d.ts.map +1 -0
- package/dist/providers/openai.js +1007 -0
- package/dist/providers/openai.js.map +1 -0
- package/dist/types.d.ts +461 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/utils.d.ts +25 -0
- package/dist/utils.d.ts.map +1 -0
- package/dist/utils.js +331 -0
- package/dist/utils.js.map +1 -0
- package/docs/frameworks/google-adk.md +84 -0
- package/docs/frameworks/langchain.md +102 -0
- package/docs/frameworks/langgraph.md +137 -0
- package/docs/frameworks/llamaindex.md +82 -0
- package/docs/frameworks/openai-agents.md +88 -0
- package/docs/frameworks/vercel-ai-sdk.md +124 -0
- package/docs/index.md +22 -0
- package/docs/providers/anthropic.md +69 -0
- package/docs/providers/gemini.md +86 -0
- package/docs/providers/openai.md +151 -0
- package/package.json +80 -0
- package/proto/opentelemetry/proto/collector/trace/v1/trace_service.proto +77 -0
- package/proto/opentelemetry/proto/common/v1/common.proto +115 -0
- package/proto/opentelemetry/proto/resource/v1/resource.proto +44 -0
- package/proto/opentelemetry/proto/trace/v1/trace.proto +357 -0
- package/proto/sigil/v1/generation_ingest.proto +145 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"http.js","sourceRoot":"","sources":["../../src/exporters/http.ts"],"names":[],"mappings":"AAYA,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAEvC,MAAM,OAAO,sBAAsB;IAChB,QAAQ,CAAS;IACjB,OAAO,CAAyB;IAEjD,YAAY,QAAgB,EAAE,OAAgC;QAC5D,IAAI,CAAC,QAAQ,GAAG,+BAA+B,CAAC,QAAQ,CAAC,CAAC;QAC1D,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,OAAiC;QACvD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE;YAC1C,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,IAAI,CAAC,OAAO;aAChB;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,WAAW,EAAE,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,wBAAwB,CAAC;aAC/D,CAAC;SACH,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,YAAY,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;YACpD,MAAM,IAAI,KAAK,CAAC,iCAAiC,QAAQ,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC,CAAC;QACvF,CAAC;QAED,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAY,CAAC;QACnD,OAAO,8BAA8B,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAC1D,CAAC;CACF;AAED,SAAS,8BAA8B,CACrC,OAAgB,EAChB,OAAiC;IAEjC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC1D,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;IAChE,CAAC;IAED,MAAM,OAAO,GAA6B,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;QAC9E,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC9D,CAAC;QAED,MAAM,oBAAoB,GAAG,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC;QAClE,OAAO;YACL,YAAY,EACV,OAAO,MAAM,CAAC,YAAY,KAAK,QAAQ;gBACrC,CAAC,CAAC,MAAM,CAAC,YAAY;gBACrB,CAAC,CAAC,OAAO,MAAM,CAAC,aAAa,KAAK,QAAQ;oBACxC,CAAC,CAAC,MAAM,CAAC,aAAa;oBACtB,CAAC,CAAC,oBAAoB;YAC5B,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC;YAClC,KAAK,EAAE,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;SACnE,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,OAAO,EAAE,OAAO,EAAE,CAAC;AACrB,CAAC;AAED,SAAS,+BAA+B,CAAC,QAAgB;IACvD,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;IAChC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC5D,CAAC;IAED,IAAI,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QACpE,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,OAAO,UAAU,OAAO,EAAE,CAAC;AAC7B,CAAC;AAED,SAAS,wBAAwB,CAAC,UAAsB;IACtD,MAAM,KAAK,GAA4B;QACrC,EAAE,EAAE,UAAU,CAAC,EAAE;QACjB,eAAe,EAAE,UAAU,CAAC,cAAc,IAAI,EAAE;QAChD,cAAc,EAAE,UAAU,CAAC,aAAa;QACxC,IAAI,EAAE,UAAU,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,wBAAwB,CAAC,CAAC,CAAC,sBAAsB;QACtF,QAAQ,EAAE,UAAU,CAAC,OAAO,IAAI,EAAE;QAClC,OAAO,EAAE,UAAU,CAAC,MAAM,IAAI,EAAE;QAChC,KAAK,EAAE;YACL,QAAQ,EAAE,UAAU,CAAC,KAAK,CAAC,QAAQ;YACnC,IAAI,EAAE,UAAU,CAAC,KAAK,CAAC,IAAI;SAC5B;QACD,WAAW,EAAE,UAAU,CAAC,UAAU,IAAI,EAAE;QACxC,cAAc,EAAE,UAAU,CAAC,aAAa,IAAI,EAAE;QAC9C,aAAa,EAAE,UAAU,CAAC,YAAY,IAAI,EAAE;QAC5C,KAAK,EAAE,CAAC,UAAU,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,qBAAqB,CAAC;QAC1D,MAAM,EAAE,CAAC,UAAU,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,qBAAqB,CAAC;QAC5D,KAAK,EAAE,CAAC,UAAU,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,kBAAkB,CAAC;QACvD,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,KAAK,CAAC;QAC5C,WAAW,EAAE,UAAU,CAAC,UAAU,IAAI,EAAE;QACxC,UAAU,EAAE,UAAU,CAAC,SAAS,CAAC,WAAW,EAAE;QAC9C,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC,WAAW,EAAE;QAClD,IAAI,EAAE,UAAU,CAAC,IAAI,IAAI,EAAE;QAC3B,QAAQ,EAAE,iBAAiB,CAAC,UAAU,CAAC,QAAQ,CAAC;QAChD,aAAa,EAAE,CAAC,UAAU,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,sBAAsB,CAAC;QACvE,UAAU,EAAE,UAAU,CAAC,SAAS,IAAI,EAAE;QACtC,UAAU,EAAE,UAAU,CAAC,SAAS,IAAI,EAAE;QACtC,aAAa,EAAE,UAAU,CAAC,YAAY,IAAI,EAAE;KAC7C,CAAC;IAEF,IAAI,UAAU,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;QACvC,KAAK,CAAC,UAAU,GAAG,aAAa,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;IACzD,CAAC;IACD,IAAI,UAAU,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;QACzC,KAAK,CAAC,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;IAC7C,CAAC;IACD,IAAI,UAAU,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;QAClC,KAAK,CAAC,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC;IAChC,CAAC;IACD,IAAI,UAAU,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;QACxC,KAAK,CAAC,WAAW,GAAG,UAAU,CAAC,UAAU,CAAC;IAC5C,CAAC;IACD,IAAI,UAAU,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;QAC7C,KAAK,CAAC,gBAAgB,GAAG,UAAU,CAAC,eAAe,CAAC;IACtD,CAAC;IACD,IAAI,UAAU,CAAC,mBAAmB,EAAE,MAAM,EAAE,CAAC;QAC3C,KAAK,CAAC,qBAAqB,GAAG,UAAU,CAAC,mBAAmB,CAAC;IAC/D,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,qBAAqB,CAAC,OAAgB;IAC7C,MAAM,KAAK,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;IACnE,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;QAC9D,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IACxC,CAAC;IAED,OAAO;QACL,IAAI,EAAE,iBAAiB,CAAC,OAAO,CAAC,IAAI,CAAC;QACrC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE;QACxB,KAAK;KACN,CAAC;AACJ,CAAC;AAED,SAAS,yBAAyB,CAAC,IAAiB;IAClD,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,MAAM;YACT,OAAO,gBAAgB,CACrB;gBACE,IAAI,EAAE,IAAI,CAAC,IAAI;aAChB,EACD,IAAI,CAAC,QAAQ,EAAE,YAAY,CAC5B,CAAC;QACJ,KAAK,UAAU;YACb,OAAO,gBAAgB,CACrB;gBACE,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,EACD,IAAI,CAAC,QAAQ,EAAE,YAAY,CAC5B,CAAC;QACJ,KAAK,WAAW;YACd,OAAO,gBAAgB,CACrB;gBACE,SAAS,EAAE;oBACT,EAAE,EAAE,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE;oBAC1B,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI;oBACxB,UAAU,EAAE,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;iBACrD;aACF,EACD,IAAI,CAAC,QAAQ,EAAE,YAAY,CAC5B,CAAC;QACJ,KAAK,aAAa;YAChB,OAAO,gBAAgB,CACrB;gBACE,WAAW,EAAE;oBACX,YAAY,EAAE,IAAI,CAAC,UAAU,CAAC,UAAU,IAAI,EAAE;oBAC9C,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI,IAAI,EAAE;oBAChC,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,IAAI,EAAE;oBACtC,YAAY,EAAE,eAAe,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC;oBAC1D,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,IAAI,KAAK;iBAC3C;aACF,EACD,IAAI,CAAC,QAAQ,EAAE,YAAY,CAC5B,CAAC;IACN,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAoB;IAC9C,OAAO;QACL,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;QACnC,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,EAAE;QACrB,iBAAiB,EAAE,eAAe,CAAC,IAAI,CAAC,eAAe,CAAC;KACzD,CAAC;AACJ,CAAC;AAED,SAAS,mBAAmB,CAAC,KAA6B;IACxD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACxB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC;IAC3C,MAAM,YAAY,GAAG,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;IAC7C,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,IAAI,WAAW,GAAG,YAAY,CAAC;IAEpE,OAAO;QACL,YAAY,EAAE,aAAa,CAAC,WAAW,CAAC;QACxC,aAAa,EAAE,aAAa,CAAC,YAAY,CAAC;QAC1C,YAAY,EAAE,aAAa,CAAC,WAAW,CAAC;QACxC,uBAAuB,EAAE,aAAa,CAAC,KAAK,CAAC,oBAAoB,CAAC;QAClE,wBAAwB,EAAE,aAAa,CAAC,KAAK,CAAC,qBAAqB,CAAC;QACpE,gBAAgB,EAAE,aAAa,CAAC,KAAK,CAAC,eAAe,CAAC;KACvD,CAAC;AACJ,CAAC;AAED,SAAS,sBAAsB,CAAC,QAAkB;IAChD,OAAO;QACL,IAAI,EAAE,kBAAkB,CAAC,QAAQ,CAAC,IAAI,CAAC;QACvC,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,QAAQ,CAAC,IAAI;QACpC,YAAY,EAAE,QAAQ,CAAC,QAAQ,IAAI,kBAAkB;QACrD,OAAO,EAAE,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC;QAC1C,SAAS,EAAE,QAAQ,CAAC,QAAQ,IAAI,EAAE;QAClC,GAAG,EAAE,QAAQ,CAAC,GAAG,IAAI,EAAE;KACxB,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CAAC,IAA6B,EAAE,YAAgC;IACvF,IAAI,YAAY,KAAK,SAAS,IAAI,YAAY,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACnE,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO;QACL,GAAG,IAAI;QACP,QAAQ,EAAE;YACR,aAAa,EAAE,YAAY;SAC5B;KACF,CAAC;AACJ,CAAC;AAED,SAAS,iBAAiB,CAAC,QAA6C;IACtE,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YACvD,IAAI,KAAK,YAAY,IAAI,EAAE,CAAC;gBAC1B,OAAO,KAAK,CAAC,WAAW,EAAE,CAAC;YAC7B,CAAC;YACD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC9B,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC;YAC1B,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC,CAAC,CAAC;QACH,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;YAC1B,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAY,CAAC;QAC/C,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YACvB,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,KAAyB;IAChD,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9C,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;AACvD,CAAC;AAED,SAAS,aAAa,CAAC,KAAyB;IAC9C,IAAI,KAAK,KAAK,SAAS,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1E,OAAO,GAAG,CAAC;IACb,CAAC;IACD,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC;AACtC,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAY;IACrC,QAAQ,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC;QAC1C,KAAK,WAAW;YACd,OAAO,wBAAwB,CAAC;QAClC,KAAK,MAAM;YACT,OAAO,mBAAmB,CAAC;QAC7B,KAAK,MAAM,CAAC;QACZ;YACE,OAAO,mBAAmB,CAAC;IAC/B,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAY;IACtC,QAAQ,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC;QAC1C,KAAK,SAAS;YACZ,OAAO,uBAAuB,CAAC;QACjC,KAAK,UAAU;YACb,OAAO,wBAAwB,CAAC;QAClC,KAAK,OAAO;YACV,OAAO,qBAAqB,CAAC;QAC/B,KAAK,gBAAgB;YACnB,OAAO,8BAA8B,CAAC;QACxC;YACE,OAAO,2BAA2B,CAAC;IACvC,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,136 @@
|
|
|
1
|
+
import type { SigilClient } from '../../client.js';
|
|
2
|
+
import { type FrameworkHandlerOptions, SigilFrameworkHandler } from '../shared.js';
|
|
3
|
+
export type { FrameworkHandlerOptions };
|
|
4
|
+
type PluginConfig = Record<string, unknown> & {
|
|
5
|
+
plugins?: unknown;
|
|
6
|
+
};
|
|
7
|
+
type GoogleAdkInvocationContext = {
|
|
8
|
+
invocationId?: unknown;
|
|
9
|
+
session?: {
|
|
10
|
+
id?: unknown;
|
|
11
|
+
};
|
|
12
|
+
agent?: {
|
|
13
|
+
name?: unknown;
|
|
14
|
+
};
|
|
15
|
+
appName?: unknown;
|
|
16
|
+
userId?: unknown;
|
|
17
|
+
branch?: unknown;
|
|
18
|
+
};
|
|
19
|
+
type GoogleAdkCallbackContext = {
|
|
20
|
+
invocationId?: unknown;
|
|
21
|
+
agentName?: unknown;
|
|
22
|
+
invocationContext?: GoogleAdkInvocationContext;
|
|
23
|
+
};
|
|
24
|
+
type GoogleAdkLlmRequest = {
|
|
25
|
+
model?: unknown;
|
|
26
|
+
contents?: unknown;
|
|
27
|
+
config?: unknown;
|
|
28
|
+
};
|
|
29
|
+
type GoogleAdkLlmResponse = {
|
|
30
|
+
content?: unknown;
|
|
31
|
+
partial?: unknown;
|
|
32
|
+
turnComplete?: unknown;
|
|
33
|
+
finishReason?: unknown;
|
|
34
|
+
usageMetadata?: {
|
|
35
|
+
promptTokenCount?: unknown;
|
|
36
|
+
candidatesTokenCount?: unknown;
|
|
37
|
+
totalTokenCount?: unknown;
|
|
38
|
+
};
|
|
39
|
+
customMetadata?: Record<string, unknown>;
|
|
40
|
+
};
|
|
41
|
+
type GoogleAdkEvent = {
|
|
42
|
+
id?: unknown;
|
|
43
|
+
invocationId?: unknown;
|
|
44
|
+
content?: unknown;
|
|
45
|
+
partial?: unknown;
|
|
46
|
+
text?: unknown;
|
|
47
|
+
delta?: unknown;
|
|
48
|
+
turnComplete?: unknown;
|
|
49
|
+
};
|
|
50
|
+
type GoogleAdkTool = {
|
|
51
|
+
name?: unknown;
|
|
52
|
+
description?: unknown;
|
|
53
|
+
};
|
|
54
|
+
type GoogleAdkToolContext = {
|
|
55
|
+
functionCallId?: unknown;
|
|
56
|
+
invocationId?: unknown;
|
|
57
|
+
agentName?: unknown;
|
|
58
|
+
invocationContext?: GoogleAdkInvocationContext;
|
|
59
|
+
};
|
|
60
|
+
type GoogleAdkPlugin = {
|
|
61
|
+
name: string;
|
|
62
|
+
onUserMessageCallback(params: {
|
|
63
|
+
invocationContext: GoogleAdkInvocationContext;
|
|
64
|
+
userMessage: unknown;
|
|
65
|
+
}): Promise<unknown>;
|
|
66
|
+
beforeRunCallback(params: {
|
|
67
|
+
invocationContext: GoogleAdkInvocationContext;
|
|
68
|
+
}): Promise<unknown>;
|
|
69
|
+
onEventCallback(params: {
|
|
70
|
+
invocationContext: GoogleAdkInvocationContext;
|
|
71
|
+
event: GoogleAdkEvent;
|
|
72
|
+
}): Promise<unknown>;
|
|
73
|
+
afterRunCallback(params: {
|
|
74
|
+
invocationContext: GoogleAdkInvocationContext;
|
|
75
|
+
}): Promise<void>;
|
|
76
|
+
beforeAgentCallback(params: {
|
|
77
|
+
callbackContext: GoogleAdkCallbackContext;
|
|
78
|
+
}): Promise<unknown>;
|
|
79
|
+
afterAgentCallback(params: {
|
|
80
|
+
callbackContext: GoogleAdkCallbackContext;
|
|
81
|
+
}): Promise<unknown>;
|
|
82
|
+
beforeModelCallback(params: {
|
|
83
|
+
callbackContext: GoogleAdkCallbackContext;
|
|
84
|
+
llmRequest: GoogleAdkLlmRequest;
|
|
85
|
+
}): Promise<unknown>;
|
|
86
|
+
afterModelCallback(params: {
|
|
87
|
+
callbackContext: GoogleAdkCallbackContext;
|
|
88
|
+
llmResponse: GoogleAdkLlmResponse;
|
|
89
|
+
}): Promise<unknown>;
|
|
90
|
+
onModelErrorCallback(params: {
|
|
91
|
+
callbackContext: GoogleAdkCallbackContext;
|
|
92
|
+
llmRequest: GoogleAdkLlmRequest;
|
|
93
|
+
error: Error;
|
|
94
|
+
}): Promise<unknown>;
|
|
95
|
+
beforeToolCallback(params: {
|
|
96
|
+
tool: GoogleAdkTool;
|
|
97
|
+
toolArgs: Record<string, unknown>;
|
|
98
|
+
toolContext: GoogleAdkToolContext;
|
|
99
|
+
}): Promise<unknown>;
|
|
100
|
+
afterToolCallback(params: {
|
|
101
|
+
tool: GoogleAdkTool;
|
|
102
|
+
toolArgs: Record<string, unknown>;
|
|
103
|
+
toolContext: GoogleAdkToolContext;
|
|
104
|
+
result: unknown;
|
|
105
|
+
}): Promise<unknown>;
|
|
106
|
+
onToolErrorCallback(params: {
|
|
107
|
+
tool: GoogleAdkTool;
|
|
108
|
+
toolArgs: Record<string, unknown>;
|
|
109
|
+
toolContext: GoogleAdkToolContext;
|
|
110
|
+
error: Error;
|
|
111
|
+
}): Promise<unknown>;
|
|
112
|
+
};
|
|
113
|
+
export declare class SigilGoogleAdkHandler extends SigilFrameworkHandler {
|
|
114
|
+
name: string;
|
|
115
|
+
constructor(client: SigilClient, options?: FrameworkHandlerOptions);
|
|
116
|
+
handleLLMStart(serialized: unknown, prompts: unknown, runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
|
|
117
|
+
handleChatModelStart(serialized: unknown, messages: unknown, runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
|
|
118
|
+
handleLLMNewToken(token: string, _idx: unknown, runId: string): Promise<void>;
|
|
119
|
+
handleLLMEnd(output: unknown, runId: string): Promise<void>;
|
|
120
|
+
handleLLMError(error: unknown, runId: string): Promise<void>;
|
|
121
|
+
handleToolStart(serialized: unknown, input: unknown, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
|
|
122
|
+
handleToolEnd(output: unknown, runId: string): Promise<void>;
|
|
123
|
+
handleToolError(error: unknown, runId: string): Promise<void>;
|
|
124
|
+
handleChainStart(serialized: unknown, _inputs: unknown, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, runType?: string, runName?: string): Promise<void>;
|
|
125
|
+
handleChainEnd(_outputs: unknown, runId: string): Promise<void>;
|
|
126
|
+
handleChainError(error: unknown, runId: string): Promise<void>;
|
|
127
|
+
handleRetrieverStart(serialized: unknown, _query: string, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
|
|
128
|
+
handleRetrieverEnd(_documents: unknown, runId: string): Promise<void>;
|
|
129
|
+
handleRetrieverError(error: unknown, runId: string): Promise<void>;
|
|
130
|
+
}
|
|
131
|
+
export declare function createSigilGoogleAdkHandler(client: SigilClient, options?: FrameworkHandlerOptions): SigilGoogleAdkHandler;
|
|
132
|
+
export declare function createSigilGoogleAdkPlugin(client: SigilClient, options?: FrameworkHandlerOptions): GoogleAdkPlugin;
|
|
133
|
+
export declare function withSigilGoogleAdkPlugins<T extends PluginConfig>(config: T | undefined, client: SigilClient, options?: FrameworkHandlerOptions): T & {
|
|
134
|
+
plugins: unknown[];
|
|
135
|
+
};
|
|
136
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/frameworks/google-adk/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,EAAE,KAAK,uBAAuB,EAAE,qBAAqB,EAAE,MAAM,cAAc,CAAC;AAEnF,YAAY,EAAE,uBAAuB,EAAE,CAAC;AAExC,KAAK,YAAY,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG;IAAE,OAAO,CAAC,EAAE,OAAO,CAAA;CAAE,CAAC;AAIpE,KAAK,0BAA0B,GAAG;IAChC,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,OAAO,CAAC,EAAE;QAAE,EAAE,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC;IAC3B,KAAK,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC;IAC3B,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB,CAAC;AAEF,KAAK,wBAAwB,GAAG;IAC9B,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,iBAAiB,CAAC,EAAE,0BAA0B,CAAC;CAChD,CAAC;AAEF,KAAK,mBAAmB,GAAG;IACzB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB,CAAC;AAEF,KAAK,oBAAoB,GAAG;IAC1B,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,aAAa,CAAC,EAAE;QACd,gBAAgB,CAAC,EAAE,OAAO,CAAC;QAC3B,oBAAoB,CAAC,EAAE,OAAO,CAAC;QAC/B,eAAe,CAAC,EAAE,OAAO,CAAC;KAC3B,CAAC;IACF,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC1C,CAAC;AAEF,KAAK,cAAc,GAAG;IACpB,EAAE,CAAC,EAAE,OAAO,CAAC;IACb,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,IAAI,CAAC,EAAE,OAAO,CAAC;IACf,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB,CAAC;AAEF,KAAK,aAAa,GAAG;IACnB,IAAI,CAAC,EAAE,OAAO,CAAC;IACf,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB,CAAC;AAEF,KAAK,oBAAoB,GAAG;IAC1B,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,iBAAiB,CAAC,EAAE,0BAA0B,CAAC;CAChD,CAAC;AAEF,KAAK,eAAe,GAAG;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,qBAAqB,CAAC,MAAM,EAAE;QAC5B,iBAAiB,EAAE,0BAA0B,CAAC;QAC9C,WAAW,EAAE,OAAO,CAAC;KACtB,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACrB,iBAAiB,CAAC,MAAM,EAAE;QAAE,iBAAiB,EAAE,0BAA0B,CAAA;KAAE,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC/F,eAAe,CAAC,MAAM,EAAE;QAAE,iBAAiB,EAAE,0BAA0B,CAAC;QAAC,KAAK,EAAE,cAAc,CAAA;KAAE,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACpH,gBAAgB,CAAC,MAAM,EAAE;QAAE,iBAAiB,EAAE,0BAA0B,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC3F,mBAAmB,CAAC,MAAM,EAAE;QAAE,eAAe,EAAE,wBAAwB,CAAA;KAAE,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC7F,kBAAkB,CAAC,MAAM,EAAE;QAAE,eAAe,EAAE,wBAAwB,CAAA;KAAE,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC5F,mBAAmB,CAAC,MAAM,EAAE;QAC1B,eAAe,EAAE,wBAAwB,CAAC;QAC1C,UAAU,EAAE,mBAAmB,CAAC;KACjC,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACrB,kBAAkB,CAAC,MAAM,EAAE;QACzB,eAAe,EAAE,wBAAwB,CAAC;QAC1C,WAAW,EAAE,oBAAoB,CAAC;KACnC,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACrB,oBAAoB,CAAC,MAAM,EAAE;QAC3B,eAAe,EAAE,wBAAwB,CAAC;QAC1C,UAAU,EAAE,mBAAmB,CAAC;QAChC,KAAK,EAAE,KAAK,CAAC;KACd,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACrB,kBAAkB,CAAC,MAAM,EAAE;QACzB,IAAI,EAAE,aAAa,CAAC;QACpB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAClC,WAAW,EAAE,oBAAoB,CAAC;KACnC,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACrB,iBAAiB,CAAC,MAAM,EAAE;QACxB,IAAI,EAAE,aAAa,CAAC;QACpB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAClC,WAAW,EAAE,oBAAoB,CAAC;QAClC,MAAM,EAAE,OAAO,CAAC;KACjB,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACrB,mBAAmB,CAAC,MAAM,EAAE;QAC1B,IAAI,EAAE,aAAa,CAAC;QACpB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAClC,WAAW,EAAE,oBAAoB,CAAC;QAClC,KAAK,EAAE,KAAK,CAAC;KACd,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;CACtB,CAAC;AAKF,qBAAa,qBAAsB,SAAQ,qBAAqB;IAC9D,IAAI,SAA8B;gBAEtB,MAAM,EAAE,WAAW,EAAE,OAAO,GAAE,uBAA4B;IAIhE,cAAc,CAClB,UAAU,EAAE,OAAO,EACnB,OAAO,EAAE,OAAO,EAChB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrC,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClC,OAAO,CAAC,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC;IAIV,oBAAoB,CACxB,UAAU,EAAE,OAAO,EACnB,QAAQ,EAAE,OAAO,EACjB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrC,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClC,OAAO,CAAC,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC;IAIV,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAI7E,YAAY,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAI3D,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAI5D,eAAe,CACnB,UAAU,EAAE,OAAO,EACnB,KAAK,EAAE,OAAO,EACd,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClC,OAAO,CAAC,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC;IAIV,aAAa,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAI5D,eAAe,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAI7D,gBAAgB,CACpB,UAAU,EAAE,OAAO,EACnB,OAAO,EAAE,OAAO,EAChB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClC,OAAO,CAAC,EAAE,MAAM,EAChB,OAAO,CAAC,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC;IAIV,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAI/D,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAI9D,oBAAoB,CACxB,UAAU,EAAE,OAAO,EACnB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClC,OAAO,CAAC,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC;IAIV,kBAAkB,CAAC,UAAU,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIrE,oBAAoB,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAGzE;AA+ZD,wBAAgB,2BAA2B,CACzC,MAAM,EAAE,WAAW,EACnB,OAAO,GAAE,uBAA4B,GACpC,qBAAqB,CAEvB;AAED,wBAAgB,0BAA0B,CACxC,MAAM,EAAE,WAAW,EACnB,OAAO,GAAE,uBAA4B,GACpC,eAAe,CAEjB;AAED,wBAAgB,yBAAyB,CAAC,CAAC,SAAS,YAAY,EAC9D,MAAM,EAAE,CAAC,GAAG,SAAS,EACrB,MAAM,EAAE,WAAW,EACnB,OAAO,GAAE,uBAA4B,GACpC,CAAC,GAAG;IAAE,OAAO,EAAE,OAAO,EAAE,CAAA;CAAE,CAgB5B"}
|
|
@@ -0,0 +1,509 @@
|
|
|
1
|
+
import { SigilFrameworkHandler } from '../shared.js';
|
|
2
|
+
const sigilGoogleAdkPluginName = 'sigil_google_adk_plugin';
|
|
3
|
+
const sigilGoogleAdkPluginMarker = Symbol.for('sigil.google_adk.plugin');
|
|
4
|
+
export class SigilGoogleAdkHandler extends SigilFrameworkHandler {
|
|
5
|
+
name = 'sigil_google_adk_handler';
|
|
6
|
+
constructor(client, options = {}) {
|
|
7
|
+
super(client, 'google-adk', 'javascript', options);
|
|
8
|
+
}
|
|
9
|
+
async handleLLMStart(serialized, prompts, runId, parentRunId, extraParams, tags, metadata, runName) {
|
|
10
|
+
this.onLLMStart(serialized, prompts, runId, parentRunId, extraParams, tags, metadata, runName);
|
|
11
|
+
}
|
|
12
|
+
async handleChatModelStart(serialized, messages, runId, parentRunId, extraParams, tags, metadata, runName) {
|
|
13
|
+
this.onChatModelStart(serialized, messages, runId, parentRunId, extraParams, tags, metadata, runName);
|
|
14
|
+
}
|
|
15
|
+
async handleLLMNewToken(token, _idx, runId) {
|
|
16
|
+
this.onLLMNewToken(token, runId);
|
|
17
|
+
}
|
|
18
|
+
async handleLLMEnd(output, runId) {
|
|
19
|
+
this.onLLMEnd(output, runId);
|
|
20
|
+
}
|
|
21
|
+
async handleLLMError(error, runId) {
|
|
22
|
+
this.onLLMError(error, runId);
|
|
23
|
+
}
|
|
24
|
+
async handleToolStart(serialized, input, runId, parentRunId, tags, metadata, runName) {
|
|
25
|
+
this.onToolStart(serialized, input, runId, parentRunId, tags, metadata, runName);
|
|
26
|
+
}
|
|
27
|
+
async handleToolEnd(output, runId) {
|
|
28
|
+
this.onToolEnd(output, runId);
|
|
29
|
+
}
|
|
30
|
+
async handleToolError(error, runId) {
|
|
31
|
+
this.onToolError(error, runId);
|
|
32
|
+
}
|
|
33
|
+
async handleChainStart(serialized, _inputs, runId, parentRunId, tags, metadata, runType, runName) {
|
|
34
|
+
this.onChainStart(serialized, runId, parentRunId, tags, metadata, runType, runName);
|
|
35
|
+
}
|
|
36
|
+
async handleChainEnd(_outputs, runId) {
|
|
37
|
+
this.onChainEnd(runId);
|
|
38
|
+
}
|
|
39
|
+
async handleChainError(error, runId) {
|
|
40
|
+
this.onChainError(error, runId);
|
|
41
|
+
}
|
|
42
|
+
async handleRetrieverStart(serialized, _query, runId, parentRunId, tags, metadata, runName) {
|
|
43
|
+
this.onRetrieverStart(serialized, runId, parentRunId, tags, metadata, runName);
|
|
44
|
+
}
|
|
45
|
+
async handleRetrieverEnd(_documents, runId) {
|
|
46
|
+
this.onRetrieverEnd(runId);
|
|
47
|
+
}
|
|
48
|
+
async handleRetrieverError(error, runId) {
|
|
49
|
+
this.onRetrieverError(error, runId);
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
class SigilGoogleAdkPlugin {
|
|
53
|
+
name = sigilGoogleAdkPluginName;
|
|
54
|
+
[sigilGoogleAdkPluginMarker] = true;
|
|
55
|
+
handler;
|
|
56
|
+
invocationRunIds = new Map();
|
|
57
|
+
agentRunStacks = new Map();
|
|
58
|
+
llmRunStacks = new Map();
|
|
59
|
+
llmTokenSources = new Map();
|
|
60
|
+
toolRunIds = new Map();
|
|
61
|
+
fallbackToolRunIds = new Map();
|
|
62
|
+
pendingUserMessages = new Map();
|
|
63
|
+
fallbackInvocationIds = new WeakMap();
|
|
64
|
+
sequence = 0;
|
|
65
|
+
constructor(client, options = {}) {
|
|
66
|
+
this.handler = new SigilGoogleAdkHandler(client, options);
|
|
67
|
+
}
|
|
68
|
+
async onUserMessageCallback(params) {
|
|
69
|
+
const invocationId = this.resolveInvocationId(params.invocationContext);
|
|
70
|
+
const messages = mapAdkUserMessage(params.userMessage);
|
|
71
|
+
if (messages.length > 0) {
|
|
72
|
+
this.pendingUserMessages.set(invocationId, messages);
|
|
73
|
+
}
|
|
74
|
+
return undefined;
|
|
75
|
+
}
|
|
76
|
+
async beforeRunCallback(params) {
|
|
77
|
+
const invocationContext = params.invocationContext;
|
|
78
|
+
const invocationId = this.resolveInvocationId(invocationContext);
|
|
79
|
+
const runId = `adk_invocation:${invocationId}`;
|
|
80
|
+
this.invocationRunIds.set(invocationId, runId);
|
|
81
|
+
await this.handler.handleChainStart({ name: this.resolveAgentName(invocationContext) || 'google_adk_runner' }, undefined, runId, undefined, undefined, this.metadataFromInvocation(invocationContext), 'invocation', this.resolveAgentName(invocationContext));
|
|
82
|
+
return undefined;
|
|
83
|
+
}
|
|
84
|
+
async onEventCallback(params) {
|
|
85
|
+
const invocationId = this.resolveInvocationId(params.invocationContext, {
|
|
86
|
+
invocationId: params.event?.invocationId,
|
|
87
|
+
});
|
|
88
|
+
const llmRunId = this.peekModelRunId(invocationId);
|
|
89
|
+
if (llmRunId === undefined) {
|
|
90
|
+
return undefined;
|
|
91
|
+
}
|
|
92
|
+
const event = params.event;
|
|
93
|
+
const isPartial = asBoolean(event?.partial);
|
|
94
|
+
const isFinal = asBoolean(event?.turnComplete);
|
|
95
|
+
if (isFinal || !isPartial) {
|
|
96
|
+
return undefined;
|
|
97
|
+
}
|
|
98
|
+
const token = extractEventText(event);
|
|
99
|
+
if (token.length === 0) {
|
|
100
|
+
return undefined;
|
|
101
|
+
}
|
|
102
|
+
this.llmTokenSources.set(llmRunId, 'event');
|
|
103
|
+
await this.handler.handleLLMNewToken(token, undefined, llmRunId);
|
|
104
|
+
return undefined;
|
|
105
|
+
}
|
|
106
|
+
async afterRunCallback(params) {
|
|
107
|
+
const invocationId = this.resolveInvocationId(params.invocationContext);
|
|
108
|
+
const runId = this.invocationRunIds.get(invocationId);
|
|
109
|
+
if (runId !== undefined) {
|
|
110
|
+
await this.handler.handleChainEnd(undefined, runId);
|
|
111
|
+
this.invocationRunIds.delete(invocationId);
|
|
112
|
+
}
|
|
113
|
+
const llmRuns = this.llmRunStacks.get(invocationId) ?? [];
|
|
114
|
+
for (const llmRunId of llmRuns) {
|
|
115
|
+
this.llmTokenSources.delete(llmRunId);
|
|
116
|
+
}
|
|
117
|
+
this.llmRunStacks.delete(invocationId);
|
|
118
|
+
this.agentRunStacks.delete(invocationId);
|
|
119
|
+
this.fallbackToolRunIds.delete(invocationId);
|
|
120
|
+
this.pendingUserMessages.delete(invocationId);
|
|
121
|
+
}
|
|
122
|
+
async beforeAgentCallback(params) {
|
|
123
|
+
const callbackContext = params.callbackContext;
|
|
124
|
+
const invocationContext = callbackContext.invocationContext;
|
|
125
|
+
const invocationId = this.resolveInvocationId(invocationContext, callbackContext);
|
|
126
|
+
const runId = `adk_agent:${invocationId}:${++this.sequence}`;
|
|
127
|
+
const stack = this.agentRunStacks.get(invocationId) ?? [];
|
|
128
|
+
const parentRunId = stack.length > 0 ? stack[stack.length - 1] : this.invocationRunIds.get(invocationId);
|
|
129
|
+
stack.push(runId);
|
|
130
|
+
this.agentRunStacks.set(invocationId, stack);
|
|
131
|
+
const agentName = this.resolveAgentName(invocationContext, callbackContext) || 'google_adk_agent';
|
|
132
|
+
await this.handler.handleChainStart({ name: agentName }, undefined, runId, parentRunId, undefined, this.metadataFromInvocation(invocationContext, callbackContext), 'agent', agentName);
|
|
133
|
+
return undefined;
|
|
134
|
+
}
|
|
135
|
+
async afterAgentCallback(params) {
|
|
136
|
+
const callbackContext = params.callbackContext;
|
|
137
|
+
const invocationContext = callbackContext.invocationContext;
|
|
138
|
+
const invocationId = this.resolveInvocationId(invocationContext, callbackContext);
|
|
139
|
+
const stack = this.agentRunStacks.get(invocationId) ?? [];
|
|
140
|
+
const runId = stack.pop();
|
|
141
|
+
if (runId === undefined) {
|
|
142
|
+
return undefined;
|
|
143
|
+
}
|
|
144
|
+
if (stack.length === 0) {
|
|
145
|
+
this.agentRunStacks.delete(invocationId);
|
|
146
|
+
}
|
|
147
|
+
else {
|
|
148
|
+
this.agentRunStacks.set(invocationId, stack);
|
|
149
|
+
}
|
|
150
|
+
await this.handler.handleChainEnd(undefined, runId);
|
|
151
|
+
return undefined;
|
|
152
|
+
}
|
|
153
|
+
async beforeModelCallback(params) {
|
|
154
|
+
const callbackContext = params.callbackContext;
|
|
155
|
+
const invocationContext = callbackContext.invocationContext;
|
|
156
|
+
const invocationId = this.resolveInvocationId(invocationContext, callbackContext);
|
|
157
|
+
const llmRunId = `adk_llm:${invocationId}:${++this.sequence}`;
|
|
158
|
+
const modelRunStack = this.llmRunStacks.get(invocationId) ?? [];
|
|
159
|
+
const agentRunStack = this.agentRunStacks.get(invocationId) ?? [];
|
|
160
|
+
const parentRunId = modelRunStack.length > 0
|
|
161
|
+
? modelRunStack[modelRunStack.length - 1]
|
|
162
|
+
: agentRunStack.length > 0
|
|
163
|
+
? agentRunStack[agentRunStack.length - 1]
|
|
164
|
+
: this.invocationRunIds.get(invocationId);
|
|
165
|
+
modelRunStack.push(llmRunId);
|
|
166
|
+
this.llmRunStacks.set(invocationId, modelRunStack);
|
|
167
|
+
const modelName = asString(params.llmRequest?.model);
|
|
168
|
+
const agentName = this.resolveAgentName(invocationContext, callbackContext);
|
|
169
|
+
const pendingUserMessages = this.pendingUserMessages.get(invocationId) ?? [];
|
|
170
|
+
const requestMessages = mapAdkContentsToChatMessages(params.llmRequest?.contents);
|
|
171
|
+
const messages = mergeAdkUserMessages(requestMessages, pendingUserMessages);
|
|
172
|
+
this.pendingUserMessages.delete(invocationId);
|
|
173
|
+
await this.handler.handleChatModelStart({
|
|
174
|
+
name: agentName,
|
|
175
|
+
kwargs: modelName.length > 0 ? { model: modelName } : undefined,
|
|
176
|
+
}, [messages], llmRunId, parentRunId, {
|
|
177
|
+
invocation_params: {
|
|
178
|
+
model: modelName,
|
|
179
|
+
stream: resolveRequestStream(params.llmRequest, invocationContext),
|
|
180
|
+
},
|
|
181
|
+
}, undefined, this.metadataFromInvocation(invocationContext, callbackContext), agentName);
|
|
182
|
+
return undefined;
|
|
183
|
+
}
|
|
184
|
+
async afterModelCallback(params) {
|
|
185
|
+
const callbackContext = params.callbackContext;
|
|
186
|
+
const invocationContext = callbackContext.invocationContext;
|
|
187
|
+
const invocationId = this.resolveInvocationId(invocationContext, callbackContext);
|
|
188
|
+
const llmRunId = this.peekModelRunId(invocationId);
|
|
189
|
+
if (llmRunId === undefined) {
|
|
190
|
+
return undefined;
|
|
191
|
+
}
|
|
192
|
+
const text = extractContentText(params.llmResponse?.content);
|
|
193
|
+
const isPartial = asBoolean(params.llmResponse?.partial) || params.llmResponse?.turnComplete === false;
|
|
194
|
+
if (isPartial) {
|
|
195
|
+
if (text.length > 0 && this.llmTokenSources.get(llmRunId) !== 'event') {
|
|
196
|
+
this.llmTokenSources.set(llmRunId, 'model');
|
|
197
|
+
await this.handler.handleLLMNewToken(text, undefined, llmRunId);
|
|
198
|
+
}
|
|
199
|
+
return params.llmResponse;
|
|
200
|
+
}
|
|
201
|
+
this.popModelRunId(invocationId);
|
|
202
|
+
this.llmTokenSources.delete(llmRunId);
|
|
203
|
+
const usageMetadata = params.llmResponse?.usageMetadata;
|
|
204
|
+
const output = {
|
|
205
|
+
generations: text.length > 0 ? [[{ text }]] : [],
|
|
206
|
+
llm_output: {
|
|
207
|
+
model_name: asString(params.llmResponse?.customMetadata?.model) ||
|
|
208
|
+
asString(params.llmResponse?.customMetadata?.model_name) ||
|
|
209
|
+
undefined,
|
|
210
|
+
finish_reason: asString(params.llmResponse?.finishReason) || undefined,
|
|
211
|
+
token_usage: {
|
|
212
|
+
prompt_tokens: asNumber(usageMetadata?.promptTokenCount),
|
|
213
|
+
completion_tokens: asNumber(usageMetadata?.candidatesTokenCount),
|
|
214
|
+
total_tokens: asNumber(usageMetadata?.totalTokenCount),
|
|
215
|
+
},
|
|
216
|
+
},
|
|
217
|
+
};
|
|
218
|
+
await this.handler.handleLLMEnd(output, llmRunId);
|
|
219
|
+
return params.llmResponse;
|
|
220
|
+
}
|
|
221
|
+
async onModelErrorCallback(params) {
|
|
222
|
+
const callbackContext = params.callbackContext;
|
|
223
|
+
const invocationContext = callbackContext.invocationContext;
|
|
224
|
+
const invocationId = this.resolveInvocationId(invocationContext, callbackContext);
|
|
225
|
+
const llmRunId = this.popModelRunId(invocationId);
|
|
226
|
+
if (llmRunId !== undefined) {
|
|
227
|
+
this.llmTokenSources.delete(llmRunId);
|
|
228
|
+
await this.handler.handleLLMError(params.error, llmRunId);
|
|
229
|
+
}
|
|
230
|
+
return undefined;
|
|
231
|
+
}
|
|
232
|
+
async beforeToolCallback(params) {
|
|
233
|
+
const toolContext = params.toolContext;
|
|
234
|
+
const invocationContext = toolContext.invocationContext;
|
|
235
|
+
const invocationId = this.resolveInvocationId(invocationContext, toolContext);
|
|
236
|
+
const callId = asString(toolContext.functionCallId);
|
|
237
|
+
const runId = callId.length > 0 ? `adk_tool:${invocationId}:${callId}` : `adk_tool:${invocationId}:${++this.sequence}`;
|
|
238
|
+
if (callId.length > 0) {
|
|
239
|
+
this.toolRunIds.set(`${invocationId}:${callId}`, runId);
|
|
240
|
+
}
|
|
241
|
+
else {
|
|
242
|
+
const stack = this.fallbackToolRunIds.get(invocationId) ?? [];
|
|
243
|
+
stack.push(runId);
|
|
244
|
+
this.fallbackToolRunIds.set(invocationId, stack);
|
|
245
|
+
}
|
|
246
|
+
await this.handler.handleToolStart({
|
|
247
|
+
name: asString(params.tool?.name),
|
|
248
|
+
description: asString(params.tool?.description),
|
|
249
|
+
}, params.toolArgs, runId, this.peekModelRunId(invocationId), undefined, this.metadataFromInvocation(invocationContext, toolContext, {
|
|
250
|
+
event_id: callId,
|
|
251
|
+
}), asString(params.tool?.name));
|
|
252
|
+
return undefined;
|
|
253
|
+
}
|
|
254
|
+
async afterToolCallback(params) {
|
|
255
|
+
const toolContext = params.toolContext;
|
|
256
|
+
const invocationContext = toolContext.invocationContext;
|
|
257
|
+
const invocationId = this.resolveInvocationId(invocationContext, toolContext);
|
|
258
|
+
const callId = asString(toolContext.functionCallId);
|
|
259
|
+
let runId;
|
|
260
|
+
if (callId.length > 0) {
|
|
261
|
+
runId = this.toolRunIds.get(`${invocationId}:${callId}`);
|
|
262
|
+
}
|
|
263
|
+
else {
|
|
264
|
+
const stack = this.fallbackToolRunIds.get(invocationId);
|
|
265
|
+
runId = stack?.pop();
|
|
266
|
+
if (stack !== undefined && stack.length === 0) {
|
|
267
|
+
this.fallbackToolRunIds.delete(invocationId);
|
|
268
|
+
}
|
|
269
|
+
}
|
|
270
|
+
if (runId !== undefined) {
|
|
271
|
+
if (callId.length > 0) {
|
|
272
|
+
this.toolRunIds.delete(`${invocationId}:${callId}`);
|
|
273
|
+
}
|
|
274
|
+
await this.handler.handleToolEnd(params.result, runId);
|
|
275
|
+
}
|
|
276
|
+
return params.result;
|
|
277
|
+
}
|
|
278
|
+
async onToolErrorCallback(params) {
|
|
279
|
+
const toolContext = params.toolContext;
|
|
280
|
+
const invocationContext = toolContext.invocationContext;
|
|
281
|
+
const invocationId = this.resolveInvocationId(invocationContext, toolContext);
|
|
282
|
+
const callId = asString(toolContext.functionCallId);
|
|
283
|
+
let runId;
|
|
284
|
+
if (callId.length > 0) {
|
|
285
|
+
runId = this.toolRunIds.get(`${invocationId}:${callId}`);
|
|
286
|
+
}
|
|
287
|
+
else {
|
|
288
|
+
const stack = this.fallbackToolRunIds.get(invocationId);
|
|
289
|
+
runId = stack?.pop();
|
|
290
|
+
if (stack !== undefined && stack.length === 0) {
|
|
291
|
+
this.fallbackToolRunIds.delete(invocationId);
|
|
292
|
+
}
|
|
293
|
+
}
|
|
294
|
+
if (runId !== undefined) {
|
|
295
|
+
if (callId.length > 0) {
|
|
296
|
+
this.toolRunIds.delete(`${invocationId}:${callId}`);
|
|
297
|
+
}
|
|
298
|
+
await this.handler.handleToolError(params.error, runId);
|
|
299
|
+
}
|
|
300
|
+
return undefined;
|
|
301
|
+
}
|
|
302
|
+
resolveInvocationId(invocationContext, fallbackContext) {
|
|
303
|
+
const explicit = asString(invocationContext?.invocationId) || asString(fallbackContext?.invocationId);
|
|
304
|
+
if (explicit.length > 0) {
|
|
305
|
+
return explicit;
|
|
306
|
+
}
|
|
307
|
+
const stableContext = this.resolveStableInvocationContext(invocationContext, fallbackContext);
|
|
308
|
+
if (stableContext !== undefined) {
|
|
309
|
+
const existing = this.fallbackInvocationIds.get(stableContext);
|
|
310
|
+
if (existing !== undefined) {
|
|
311
|
+
return existing;
|
|
312
|
+
}
|
|
313
|
+
const assigned = `adk_invocation:${++this.sequence}`;
|
|
314
|
+
this.fallbackInvocationIds.set(stableContext, assigned);
|
|
315
|
+
return assigned;
|
|
316
|
+
}
|
|
317
|
+
return `adk_invocation:${++this.sequence}`;
|
|
318
|
+
}
|
|
319
|
+
resolveStableInvocationContext(invocationContext, fallbackContext) {
|
|
320
|
+
if (isRecord(invocationContext)) {
|
|
321
|
+
return invocationContext;
|
|
322
|
+
}
|
|
323
|
+
if (isRecord(fallbackContext?.invocationContext)) {
|
|
324
|
+
return fallbackContext.invocationContext;
|
|
325
|
+
}
|
|
326
|
+
if (isRecord(fallbackContext)) {
|
|
327
|
+
return fallbackContext;
|
|
328
|
+
}
|
|
329
|
+
return undefined;
|
|
330
|
+
}
|
|
331
|
+
resolveAgentName(invocationContext, callbackContext) {
|
|
332
|
+
return asString(callbackContext?.agentName) || asString(invocationContext?.agent?.name);
|
|
333
|
+
}
|
|
334
|
+
metadataFromInvocation(invocationContext, callbackContext, extra) {
|
|
335
|
+
return {
|
|
336
|
+
session_id: asString(invocationContext?.session?.id),
|
|
337
|
+
conversation_id: asString(invocationContext?.session?.id),
|
|
338
|
+
invocation_id: asString(invocationContext?.invocationId) || asString(callbackContext?.invocationId),
|
|
339
|
+
thread_id: asString(invocationContext?.branch),
|
|
340
|
+
component_name: asString(callbackContext?.agentName) || asString(invocationContext?.agent?.name),
|
|
341
|
+
app_name: asString(invocationContext?.appName),
|
|
342
|
+
user_id: asString(invocationContext?.userId),
|
|
343
|
+
...extra,
|
|
344
|
+
};
|
|
345
|
+
}
|
|
346
|
+
peekModelRunId(invocationId) {
|
|
347
|
+
const stack = this.llmRunStacks.get(invocationId);
|
|
348
|
+
if (stack === undefined || stack.length === 0) {
|
|
349
|
+
return undefined;
|
|
350
|
+
}
|
|
351
|
+
return stack[stack.length - 1];
|
|
352
|
+
}
|
|
353
|
+
popModelRunId(invocationId) {
|
|
354
|
+
const stack = this.llmRunStacks.get(invocationId);
|
|
355
|
+
if (stack === undefined || stack.length === 0) {
|
|
356
|
+
return undefined;
|
|
357
|
+
}
|
|
358
|
+
const runId = stack.pop();
|
|
359
|
+
if (stack.length === 0) {
|
|
360
|
+
this.llmRunStacks.delete(invocationId);
|
|
361
|
+
}
|
|
362
|
+
return runId;
|
|
363
|
+
}
|
|
364
|
+
}
|
|
365
|
+
export function createSigilGoogleAdkHandler(client, options = {}) {
|
|
366
|
+
return new SigilGoogleAdkHandler(client, options);
|
|
367
|
+
}
|
|
368
|
+
export function createSigilGoogleAdkPlugin(client, options = {}) {
|
|
369
|
+
return new SigilGoogleAdkPlugin(client, options);
|
|
370
|
+
}
|
|
371
|
+
export function withSigilGoogleAdkPlugins(config, client, options = {}) {
|
|
372
|
+
const plugin = createSigilGoogleAdkPlugin(client, options);
|
|
373
|
+
const base = { ...(config ?? {}) };
|
|
374
|
+
const existingValue = base.plugins;
|
|
375
|
+
const plugins = Array.isArray(existingValue)
|
|
376
|
+
? [...existingValue]
|
|
377
|
+
: existingValue === undefined
|
|
378
|
+
? []
|
|
379
|
+
: [existingValue];
|
|
380
|
+
if (!plugins.some(isSigilGoogleAdkPlugin)) {
|
|
381
|
+
plugins.push(plugin);
|
|
382
|
+
}
|
|
383
|
+
return {
|
|
384
|
+
...base,
|
|
385
|
+
plugins,
|
|
386
|
+
};
|
|
387
|
+
}
|
|
388
|
+
function isSigilGoogleAdkPlugin(plugin) {
|
|
389
|
+
if (!isRecord(plugin)) {
|
|
390
|
+
return false;
|
|
391
|
+
}
|
|
392
|
+
return plugin[sigilGoogleAdkPluginMarker] === true;
|
|
393
|
+
}
|
|
394
|
+
function mapAdkContentsToChatMessages(contents) {
|
|
395
|
+
if (!Array.isArray(contents)) {
|
|
396
|
+
return [];
|
|
397
|
+
}
|
|
398
|
+
const messages = [];
|
|
399
|
+
for (const content of contents) {
|
|
400
|
+
const text = extractContentText(content);
|
|
401
|
+
if (text.length === 0) {
|
|
402
|
+
continue;
|
|
403
|
+
}
|
|
404
|
+
messages.push({
|
|
405
|
+
role: asString(read(content, 'role')) || 'user',
|
|
406
|
+
content: text,
|
|
407
|
+
});
|
|
408
|
+
}
|
|
409
|
+
return messages;
|
|
410
|
+
}
|
|
411
|
+
function mapAdkUserMessage(userMessage) {
|
|
412
|
+
if (Array.isArray(userMessage)) {
|
|
413
|
+
return mapAdkContentsToChatMessages(userMessage);
|
|
414
|
+
}
|
|
415
|
+
if (isRecord(userMessage)) {
|
|
416
|
+
const role = asString(read(userMessage, 'role')) || 'user';
|
|
417
|
+
const content = extractContentText(userMessage) || asString(read(userMessage, 'text')) || asString(read(userMessage, 'content'));
|
|
418
|
+
if (content.length > 0) {
|
|
419
|
+
return [{ role, content }];
|
|
420
|
+
}
|
|
421
|
+
}
|
|
422
|
+
if (typeof userMessage === 'string' && userMessage.trim().length > 0) {
|
|
423
|
+
return [{ role: 'user', content: userMessage.trim() }];
|
|
424
|
+
}
|
|
425
|
+
return [];
|
|
426
|
+
}
|
|
427
|
+
function mergeAdkUserMessages(requestMessages, pendingMessages) {
|
|
428
|
+
if (requestMessages.length === 0) {
|
|
429
|
+
return pendingMessages;
|
|
430
|
+
}
|
|
431
|
+
if (pendingMessages.length === 0) {
|
|
432
|
+
return requestMessages;
|
|
433
|
+
}
|
|
434
|
+
const firstRequest = requestMessages[0];
|
|
435
|
+
if (firstRequest === undefined) {
|
|
436
|
+
return pendingMessages;
|
|
437
|
+
}
|
|
438
|
+
const duplicate = pendingMessages.some((message) => message.role === firstRequest.role && message.content === firstRequest.content);
|
|
439
|
+
if (duplicate) {
|
|
440
|
+
return requestMessages;
|
|
441
|
+
}
|
|
442
|
+
return [...pendingMessages, ...requestMessages];
|
|
443
|
+
}
|
|
444
|
+
function resolveRequestStream(llmRequest, invocationContext) {
|
|
445
|
+
const config = isRecord(llmRequest?.config) ? llmRequest?.config : undefined;
|
|
446
|
+
const generationConfig = isRecord(read(config, 'generationConfig')) ? read(config, 'generationConfig') : undefined;
|
|
447
|
+
const stream = read(llmRequest, 'stream') ??
|
|
448
|
+
read(llmRequest, 'streaming') ??
|
|
449
|
+
read(config, 'stream') ??
|
|
450
|
+
read(config, 'streaming') ??
|
|
451
|
+
read(generationConfig, 'stream') ??
|
|
452
|
+
read(generationConfig, 'streaming') ??
|
|
453
|
+
read(invocationContext, 'stream') ??
|
|
454
|
+
read(invocationContext, 'streaming');
|
|
455
|
+
return asBoolean(stream);
|
|
456
|
+
}
|
|
457
|
+
function extractEventText(event) {
|
|
458
|
+
if (!event) {
|
|
459
|
+
return '';
|
|
460
|
+
}
|
|
461
|
+
return (extractContentText(event.content) ||
|
|
462
|
+
extractContentText(event.partial) ||
|
|
463
|
+
asString(event.text) ||
|
|
464
|
+
asString(event.delta) ||
|
|
465
|
+
asString(read(event.content, 'text')) ||
|
|
466
|
+
asString(read(event.partial, 'text')));
|
|
467
|
+
}
|
|
468
|
+
function extractContentText(content) {
|
|
469
|
+
if (!isRecord(content)) {
|
|
470
|
+
return '';
|
|
471
|
+
}
|
|
472
|
+
const parts = read(content, 'parts');
|
|
473
|
+
if (!Array.isArray(parts)) {
|
|
474
|
+
return '';
|
|
475
|
+
}
|
|
476
|
+
const textParts = [];
|
|
477
|
+
for (const part of parts) {
|
|
478
|
+
const text = asString(read(part, 'text'));
|
|
479
|
+
if (text.length > 0) {
|
|
480
|
+
textParts.push(text);
|
|
481
|
+
}
|
|
482
|
+
}
|
|
483
|
+
return textParts.join(' ').trim();
|
|
484
|
+
}
|
|
485
|
+
function read(value, key) {
|
|
486
|
+
if (!isRecord(value)) {
|
|
487
|
+
return undefined;
|
|
488
|
+
}
|
|
489
|
+
return value[key];
|
|
490
|
+
}
|
|
491
|
+
function isRecord(value) {
|
|
492
|
+
return typeof value === 'object' && value !== null;
|
|
493
|
+
}
|
|
494
|
+
function asString(value) {
|
|
495
|
+
if (typeof value !== 'string') {
|
|
496
|
+
return '';
|
|
497
|
+
}
|
|
498
|
+
return value.trim();
|
|
499
|
+
}
|
|
500
|
+
function asNumber(value) {
|
|
501
|
+
if (typeof value !== 'number' || !Number.isFinite(value)) {
|
|
502
|
+
return undefined;
|
|
503
|
+
}
|
|
504
|
+
return value;
|
|
505
|
+
}
|
|
506
|
+
function asBoolean(value) {
|
|
507
|
+
return value === true;
|
|
508
|
+
}
|
|
509
|
+
//# sourceMappingURL=index.js.map
|