@n8n/n8n-nodes-langchain 2.43.2 → 2.44.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/dist/credentials/AzureEntraCognitiveServicesOAuth2Api.credentials.js +4 -4
- package/dist/credentials/AzureEntraCognitiveServicesOAuth2Api.credentials.js.map +1 -1
- package/dist/credentials/AzureOpenAiApi.credentials.js +4 -4
- package/dist/credentials/AzureOpenAiApi.credentials.js.map +1 -1
- package/dist/credentials/GoogleVertexAiApi.credentials.d.ts +15 -0
- package/dist/credentials/GoogleVertexAiApi.credentials.js +83 -0
- package/dist/credentials/GoogleVertexAiApi.credentials.js.map +1 -0
- package/dist/credentials/icons/google.svg +1 -0
- package/dist/known/credentials.json +12 -0
- package/dist/known/nodes.json +4 -0
- package/dist/node-definitions/.nodes-hash +1 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/embeddingsAzureOpenAi/index.ts +1 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/embeddingsAzureOpenAi/v1.schema.js +2 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/embeddingsAzureOpenAi/v1.ts +4 -2
- package/dist/node-definitions/nodes/n8n-nodes-langchain/embeddingsGoogleVertex/v1.schema.js +3 -2
- package/dist/node-definitions/nodes/n8n-nodes-langchain/embeddingsGoogleVertex/v1.ts +3 -0
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatAzureOpenAi/index.ts +1 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatAzureOpenAi/v1.schema.js +2 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatAzureOpenAi/v1.ts +11 -2
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatAzureOpenAi/v11.schema.js +3 -2
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatAzureOpenAi/v11.ts +15 -5
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatGoogleGemini/v1.ts +1 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatGoogleGemini/v11.ts +1 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatGoogleGemini/v12.ts +1 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatGoogleVertex/v1.schema.js +3 -2
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatGoogleVertex/v1.ts +3 -0
- package/dist/nodes/agents/Agent/agents/ToolsAgent/V2/execute.d.ts +6 -1
- package/dist/nodes/agents/Agent/agents/ToolsAgent/V2/execute.js +23 -7
- package/dist/nodes/agents/Agent/agents/ToolsAgent/V2/execute.js.map +1 -1
- package/dist/nodes/embeddings/EmbeddingsAzureOpenAi/EmbeddingsAzureOpenAi.node.js +102 -16
- package/dist/nodes/embeddings/EmbeddingsAzureOpenAi/EmbeddingsAzureOpenAi.node.js.map +1 -1
- package/dist/nodes/embeddings/EmbeddingsDatabricks/EmbeddingsDatabricks.node.js +1 -0
- package/dist/nodes/embeddings/EmbeddingsDatabricks/EmbeddingsDatabricks.node.js.map +1 -1
- package/dist/nodes/embeddings/EmbeddingsGoogleVertex/EmbeddingsGoogleVertex.node.d.ts +5 -6
- package/dist/nodes/embeddings/EmbeddingsGoogleVertex/EmbeddingsGoogleVertex.node.js +9 -41
- package/dist/nodes/embeddings/EmbeddingsGoogleVertex/EmbeddingsGoogleVertex.node.js.map +1 -1
- package/dist/nodes/llms/LMChatAnthropic/LmChatAnthropic.node.js +1 -15
- package/dist/nodes/llms/LMChatAnthropic/LmChatAnthropic.node.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/LmChatAzureOpenAi.node.js +59 -8
- package/dist/nodes/llms/LmChatAzureOpenAi/LmChatAzureOpenAi.node.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/api-key.js +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/api-key.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/oauth2.js +1 -2
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/oauth2.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/requireFoundryEndpoint.js +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/requireFoundryEndpoint.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/requireTenantId.js +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/requireTenantId.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/error-handling.d.ts +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/error-handling.js +12 -6
- package/dist/nodes/llms/LmChatAzureOpenAi/error-handling.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/methods/searchModels.js +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/methods/searchModels.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/properties.js +29 -4
- package/dist/nodes/llms/LmChatAzureOpenAi/properties.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/types.d.ts +1 -0
- package/dist/nodes/llms/LmChatAzureOpenAi/types.js +2 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/types.js.map +1 -1
- package/dist/nodes/llms/LmChatGoogleGemini/LmChatGoogleGemini.node.js +22 -13
- package/dist/nodes/llms/LmChatGoogleGemini/LmChatGoogleGemini.node.js.map +1 -1
- package/dist/nodes/llms/LmChatGoogleVertex/LmChatGoogleVertex.node.d.ts +5 -6
- package/dist/nodes/llms/LmChatGoogleVertex/LmChatGoogleVertex.node.js +9 -41
- package/dist/nodes/llms/LmChatGoogleVertex/LmChatGoogleVertex.node.js.map +1 -1
- package/dist/nodes/trigger/ChatTrigger/ChatTrigger.node.js +3 -0
- package/dist/nodes/trigger/ChatTrigger/ChatTrigger.node.js.map +1 -1
- package/dist/nodes/vector_store/VectorStoreDatabricks/DatabricksVectorStore.d.ts +1 -0
- package/dist/nodes/vector_store/VectorStoreDatabricks/DatabricksVectorStore.js +19 -5
- package/dist/nodes/vector_store/VectorStoreDatabricks/DatabricksVectorStore.js.map +1 -1
- package/dist/nodes/vector_store/VectorStoreDatabricks/VectorStoreDatabricks.node.js +4 -1
- package/dist/nodes/vector_store/VectorStoreDatabricks/VectorStoreDatabricks.node.js.map +1 -1
- package/dist/nodes/vector_store/VectorStoreMongoDBAtlas/VectorStoreMongoDBAtlas.node.js +14 -9
- package/dist/nodes/vector_store/VectorStoreMongoDBAtlas/VectorStoreMongoDBAtlas.node.js.map +1 -1
- package/dist/nodes/vector_store/VectorStoreOracleDB/VectorStoreOracleDB.node.js +31 -7
- package/dist/nodes/vector_store/VectorStoreOracleDB/VectorStoreOracleDB.node.js.map +1 -1
- package/dist/nodes/vendors/Databricks/DatabricksAiAgent.node.d.ts +5 -0
- package/dist/nodes/vendors/Databricks/DatabricksAiAgent.node.js +170 -0
- package/dist/nodes/vendors/Databricks/DatabricksAiAgent.node.js.map +1 -0
- package/dist/nodes/vendors/Databricks/databricks.dark.svg +19 -0
- package/dist/nodes/vendors/Databricks/databricks.svg +19 -0
- package/dist/nodes/vendors/Databricks/mlflow/span-collector.d.ts +48 -0
- package/dist/nodes/vendors/Databricks/mlflow/span-collector.js +378 -0
- package/dist/nodes/vendors/Databricks/mlflow/span-collector.js.map +1 -0
- package/dist/nodes/vendors/Databricks/mlflow/trace-writer.d.ts +41 -0
- package/dist/nodes/vendors/Databricks/mlflow/trace-writer.js +129 -0
- package/dist/nodes/vendors/Databricks/mlflow/trace-writer.js.map +1 -0
- package/dist/nodes/vendors/Databricks/mlflow/transport.d.ts +7 -0
- package/dist/nodes/vendors/Databricks/mlflow/transport.js +72 -0
- package/dist/nodes/vendors/Databricks/mlflow/transport.js.map +1 -0
- package/dist/nodes/vendors/Databricks/mlflow/types.d.ts +54 -0
- package/dist/nodes/vendors/Databricks/mlflow/types.js +23 -0
- package/dist/nodes/vendors/Databricks/mlflow/types.js.map +1 -0
- package/dist/nodes/vendors/GoogleGemini/methods/listSearch.js +20 -10
- package/dist/nodes/vendors/GoogleGemini/methods/listSearch.js.map +1 -1
- package/dist/typecheck.tsbuildinfo +1 -1
- package/dist/types/credentials.json +3 -2
- package/dist/types/nodes.json +6 -5
- package/dist/utils/google-vertex.d.ts +14 -0
- package/dist/utils/google-vertex.js +130 -0
- package/dist/utils/google-vertex.js.map +1 -0
- package/dist/utils/sqlIdentifier.d.ts +1 -0
- package/dist/utils/sqlIdentifier.js +22 -0
- package/dist/utils/sqlIdentifier.js.map +1 -1
- package/package.json +14 -12
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"VectorStoreOracleDB.node.js","sourceRoot":"","sources":["../../../../nodes/vector_store/VectorStoreOracleDB/VectorStoreOracleDB.node.ts"],"names":[],"mappings":";;;AAEA,oDAA+E;AAC/E,mEAA6F;AAE7F,8EAAmF;
|
|
1
|
+
{"version":3,"file":"VectorStoreOracleDB.node.js","sourceRoot":"","sources":["../../../../nodes/vector_store/VectorStoreOracleDB/VectorStoreOracleDB.node.ts"],"names":[],"mappings":";;;AAEA,oDAA+E;AAC/E,mEAA6F;AAE7F,8EAAmF;AAEnF,+CAAkD;AAGlD,wDAG8B;AAE9B,MAAM,YAAY,GAAsB;IACvC;QACC,WAAW,EAAE,YAAY;QACzB,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,QAAQ;QACd,OAAO,EAAE,aAAa;QACtB,WAAW,EACV,sFAAsF;KACvF;CACD,CAAC;AAEF,MAAM,2BAA2B,GAAG,sCAAsC,CAAC;AAE3E,MAAM,kBAAkB,GAAG,aAAa,CAAC;AAGzC,SAAS,YAAY,CACpB,OAAiD,EACjD,SAAiB;IAEjB,MAAM,SAAS,GACb,OAAO,CAAC,gBAAgB,CAAC,WAAW,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,CAAY;QACxF,kBAAkB,CAAC;IAEpB,IAAI,CAAC,IAAA,4CAA4B,EAAC,SAAS,CAAC,EAAE,CAAC;QAC9C,MAAM,IAAI,iCAAkB,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,wBAAwB,SAAS,IAAI,EAAE;YACtF,SAAS;YACT,WAAW,EACV,qHAAqH;SACtH,CAAC,CAAC;IACJ,CAAC;IAED,OAAO,SAAS,CAAC;AAClB,CAAC;AAGD,SAAS,qBAAqB,CAAC,IAAW,EAAE,MAAe,EAAE,SAAkB;IAC9E,MAAM,WAAW,GAAG,IAAA,gDAAgC,EAAC,MAAM,CAAC,CAAC;IAC7D,IAAI,WAAW,KAAK,SAAS;QAAE,OAAO;IAEtC,MAAM,IAAI,iCAAkB,CAAC,IAAI,EAAE,iCAAiC,WAAW,IAAI,EAAE;QACpF,SAAS;QACT,WAAW,EACV,iHAAiH;KAClH,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,2BAA2B,GAAG,gBAAgB,CAAC;AAErD,MAAM,kBAAkB,GAAG,CAAC,KAAc,EAAkB,EAAE,CAC7D,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,KAAK,2BAA2B,CAAC;AAIzE,MAAM,aAAa,GAAG,CAAC,KAAc,EAAoC,EAAE,CAC1E,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAYtE,MAAM,YAAY,GAAG,CAAC,IAAmB,EAAE,QAAuB,EAA4B,EAAE;IAC/F,IAAI,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,IAAI;QAAE,OAAO,QAAQ,CAAC;IACzD,IAAI,QAAQ,KAAK,SAAS,IAAI,QAAQ,KAAK,IAAI;QAAE,OAAO,IAAI,CAAC;IAE7D,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACpD,OAAO,CAAC,GAAG,IAAI,EAAE,GAAG,QAAQ,CAA4B,CAAC;IAC1D,CAAC;IAED,IAAI,aAAa,CAAC,IAAI,CAAC,IAAI,aAAa,CAAC,QAAQ,CAAC,EAAE,CAAC;QACpD,MAAM,MAAM,GAA4B,EAAE,GAAG,IAAI,EAAE,CAAC;QAEpD,KAAK,MAAM,CAAC,GAAG,EAAE,aAAa,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC7D,MAAM,CAAC,GAAG,CAAC,GAAG,YAAY,CAAC,IAAI,CAAC,GAAG,CAAiB,EAAE,aAA6B,CAAC,CAAC;QACtF,CAAC;QAED,OAAO,MAAsB,CAAC;IAC/B,CAAC;IAED,OAAO,QAAQ,CAAC;AACjB,CAAC,CAAC;AAGF,MAAM,qBAAqB,GAAoB;IAC9C,WAAW,EAAE,mBAAmB;IAChC,IAAI,EAAE,kBAAkB;IACxB,IAAI,EAAE,SAAS;IACf,OAAO,EAAE,qCAAgB,CAAC,MAAM;IAChC,WAAW,EAAE,0DAA0D;IACvE,OAAO,EAAE;QACR;YACC,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE,qCAAgB,CAAC,MAAM;SAC9B;QACD;YACC,IAAI,EAAE,eAAe;YACrB,KAAK,EAAE,qCAAgB,CAAC,WAAW;SACnC;QACD;YACC,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,qCAAgB,CAAC,SAAS;SACjC;QACD;YACC,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,qCAAgB,CAAC,SAAS;SACjC;QACD;YACC,IAAI,EAAE,mBAAmB;YACzB,KAAK,EAAE,qCAAgB,CAAC,iBAAiB;SACzC;QACD;YACC,IAAI,EAAE,SAAS;YACf,KAAK,EAAE,qCAAgB,CAAC,OAAO;SAC/B;KACD;CACD,CAAC;AAEF,MAAM,cAAc,GAAsB;IACzC;QACC,WAAW,EAAE,SAAS;QACtB,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,YAAY;QAClB,WAAW,EAAE,YAAY;QACzB,OAAO,EAAE,EAAE;QACX,OAAO,EAAE,CAAC,qBAAqB,EAAE,kCAAmB,CAAC;KACrD;CACD,CAAC;AAEF,MAAM,YAAY,GAAsB;IACvC;QACC,WAAW,EAAE,SAAS;QACtB,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,YAAY;QAClB,WAAW,EAAE,YAAY;QACzB,OAAO,EAAE,EAAE;QACX,OAAO,EAAE,EAAE;KACX;CACD,CAAC;AAEF,MAAM,cAAc;IACnB,YAA6B,OAAqC;uBAArC,OAAO;IAAiC,CAAC;IAEtE,KAAK,CAAC,aAAa;QAClB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QAClC,OAAO,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,KAAK;IAEX,CAAC;CACD;AAED,MAAM,oBAAoB,GAAG,CAC5B,OAAiD,EACjD,WAAoC,EACnC,EAAE;IACH,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE,CAAC,MAAM,6BAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IAG/E,OAAO,IAAI,cAAc,CAAC,OAAO,CAA6B,CAAC;AAChE,CAAC,CAAC;AAMF,MAAM,2BAA4B,SAAQ,6BAAQ;IAGjD,MAAM,CAAC,KAAK,CAAC,UAAU,CACtB,UAA+B,EAC/B,IAAyC;QAEzC,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,EAAE,GAAG,IAAI,CAAC;QAClC,MAAM,mBAAmB,GAAG,IAAI,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;QACvD,mBAAmB,CAAC,OAAO,GAAG,OAAO,CAAC;QAEtC,MAAM,mBAAmB,CAAC,UAAU,EAAE,CAAC;QACvC,OAAO,mBAAmB,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,+BAA+B,CACpC,KAAe,EACf,CAAS,EACT,MAA+B;QAE/B,MAAM,YAAY,GAAG,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAGvD,qBAAqB,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QAElD,IAAI,CAAC;YACJ,OAAO,MAAM,KAAK,CAAC,+BAA+B,CAAC,KAAK,EAAE,CAAC,EAAE,YAAY,CAAC,CAAC;QAC5E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,kBAAkB,CAAC,KAAK,CAAC;gBAAE,OAAO,EAAE,CAAC;YACzC,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;CACD;AAED,yBAAiC,SAAQ,IAAA,oCAAqB,EAA8B;IAC3F,IAAI,EAAE;QACL,WAAW,EAAE,gDAAgD;QAC7D,IAAI,EAAE,oCAAoC;QAC1C,WAAW,EAAE,8BAA8B;QAC3C,OAAO,EACN,4GAA4G;QAC7G,IAAI,EAAE,2BAA2B;QACjC,WAAW,EAAE;YACZ;gBACC,IAAI,EAAE,aAAa;gBACnB,QAAQ,EAAE,IAAI;gBACd,QAAQ,EAAE,wBAAwB;aAClC;SACD;QACD,cAAc,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,kBAAkB,CAAC;KAClE;IACD,YAAY;IACZ,YAAY;IACZ,UAAU,EAAE,cAAc;IAC1B,cAAc;IACd,KAAK,CAAC,oBAAoB,CACzB,OAAiD,EACjD,MAAyC,EACzC,UAA+B,EAC/B,SAAiB;QAEjB,MAAM,SAAS,GAAG,YAAY,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QACnD,qBAAqB,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;QAC5D,MAAM,WAAW,GAAG,CAAC,MAAM,OAAO,CAAC,cAAc,CAAC,aAAa,CAAC,CAA4B,CAAC;QAC7F,MAAM,MAAM,GAAG,oBAAoB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QAC1D,MAAM,KAAK,GAAG,2BAA2B,CAAC;QAC1C,MAAM,MAAM,GAAwC;YACnD,MAAM;YACN,SAAS;YACT,KAAK;YACL,MAAM,EAAE,MAA+D;YACvE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE;SAC1B,CAAC;QAEF,MAAM,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CACjD,0BAA0B,EAC1B,SAAS,EACT,qCAAgB,CAAC,MAAM,CACH,CAAC;QAEtB,OAAO,MAAM,2BAA2B,CAAC,UAAU,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IACzE,CAAC;IAED,KAAK,CAAC,mBAAmB,CACxB,OAAiD,EACjD,UAA+B,EAC/B,SAAmD,EACnD,SAAiB;QAEjB,MAAM,SAAS,GAAG,YAAY,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QACnD,MAAM,WAAW,GAAG,CAAC,MAAM,OAAO,CAAC,cAAc,CAAC,aAAa,CAAC,CAA4B,CAAC;QAC7F,MAAM,MAAM,GAAG,oBAAoB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QAC1D,MAAM,KAAK,GAAG,2BAA2B,CAAC;QAC1C,MAAM,MAAM,GAAmB;YAC9B,MAAM;YACN,SAAS;YACT,KAAK;SACL,CAAC;QAEF,MAAM,6BAAQ,CAAC,aAAa,CAAC,SAAS,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IAC7D,CAAC;IAED,wBAAwB,CAAC,WAAW;QACnC,MAAM,IAAI,GAAG,WAAW,CAAC,MAAM,CAAC;QAChC,IAAI,IAAI,IAAI,OAAO,IAAI,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;YAC9C,KAAK,IAAI,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACnC,CAAC;IACF,CAAC;CACD,CAAC;CAAG"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
import { type IExecuteFunctions, type INodeExecutionData, type INodeType, type INodeTypeDescription } from 'n8n-workflow';
|
|
2
|
+
export declare class DatabricksAiAgent implements INodeType {
|
|
3
|
+
description: INodeTypeDescription;
|
|
4
|
+
execute(this: IExecuteFunctions): Promise<INodeExecutionData[][]>;
|
|
5
|
+
}
|
|
@@ -0,0 +1,170 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.DatabricksAiAgent = void 0;
|
|
4
|
+
const sanitize_error_detail_1 = require("@n8n/utils/redaction/sanitize-error-detail");
|
|
5
|
+
const n8n_workflow_1 = require("n8n-workflow");
|
|
6
|
+
const descriptions_1 = require("../../../utils/descriptions");
|
|
7
|
+
const description_1 = require("../../agents/Agent/agents/ToolsAgent/V2/description");
|
|
8
|
+
const execute_1 = require("../../agents/Agent/agents/ToolsAgent/V2/execute");
|
|
9
|
+
const utils_1 = require("../../agents/Agent/utils");
|
|
10
|
+
const span_collector_1 = require("./mlflow/span-collector");
|
|
11
|
+
const trace_writer_1 = require("./mlflow/trace-writer");
|
|
12
|
+
const transport_1 = require("./mlflow/transport");
|
|
13
|
+
const MAX_LOGGED_ERROR_CHARS = 1_000;
|
|
14
|
+
function sanitizedErrorMessage(error) {
|
|
15
|
+
const message = error instanceof Error ? error.message : String(error);
|
|
16
|
+
return (0, sanitize_error_detail_1.sanitizeErrorDetail)(message, MAX_LOGGED_ERROR_CHARS);
|
|
17
|
+
}
|
|
18
|
+
const exportTracesField = {
|
|
19
|
+
displayName: 'Export Traces to MLflow',
|
|
20
|
+
name: 'exportTraces',
|
|
21
|
+
type: 'boolean',
|
|
22
|
+
default: false,
|
|
23
|
+
noDataExpression: true,
|
|
24
|
+
description: 'Whether to send each run to MLflow Tracing in your Databricks workspace. Needs the Databricks credential above. Traces include prompts, model outputs and tool results.',
|
|
25
|
+
};
|
|
26
|
+
const experimentField = {
|
|
27
|
+
displayName: 'Experiment',
|
|
28
|
+
name: 'experiment',
|
|
29
|
+
type: 'resourceLocator',
|
|
30
|
+
default: { mode: 'name', value: '' },
|
|
31
|
+
displayOptions: { show: { exportTraces: [true] } },
|
|
32
|
+
description: 'Where traces are stored. Leave the name empty to create one for this workflow automatically.',
|
|
33
|
+
modes: [
|
|
34
|
+
{
|
|
35
|
+
displayName: 'By Name',
|
|
36
|
+
name: 'name',
|
|
37
|
+
type: 'string',
|
|
38
|
+
placeholder: '/Shared/my-agent-traces',
|
|
39
|
+
hint: 'Created if it does not exist. Empty means /Shared/n8n-workflows-<workflow ID>.',
|
|
40
|
+
},
|
|
41
|
+
{
|
|
42
|
+
displayName: 'By ID',
|
|
43
|
+
name: 'id',
|
|
44
|
+
type: 'string',
|
|
45
|
+
placeholder: '822142731468161',
|
|
46
|
+
},
|
|
47
|
+
],
|
|
48
|
+
};
|
|
49
|
+
async function prepareTraceExport(ctx) {
|
|
50
|
+
try {
|
|
51
|
+
const credential = await ctx.getCredentials('databricksOAuth2Api');
|
|
52
|
+
const { request, upload } = (0, transport_1.createMlflowTransport)(ctx, credential);
|
|
53
|
+
const experiment = ctx.getNodeParameter('experiment', 0, { mode: 'name', value: '' });
|
|
54
|
+
if (experiment.mode === 'id' && experiment.value) {
|
|
55
|
+
return { request, upload, experimentId: experiment.value };
|
|
56
|
+
}
|
|
57
|
+
const name = experiment.value || (0, trace_writer_1.defaultExperimentName)(ctx.getWorkflow().id ?? 'unknown');
|
|
58
|
+
return { request, upload, experimentId: await (0, trace_writer_1.ensureExperiment)(request, name) };
|
|
59
|
+
}
|
|
60
|
+
catch (error) {
|
|
61
|
+
ctx.logger.warn(`Databricks AI Agent: could not prepare MLflow trace export: ${sanitizedErrorMessage(error)}`);
|
|
62
|
+
return undefined;
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
async function flushTrace(ctx, exporter, collector) {
|
|
66
|
+
const trace = collector.finish();
|
|
67
|
+
if (!trace)
|
|
68
|
+
return;
|
|
69
|
+
try {
|
|
70
|
+
await (0, trace_writer_1.writeTrace)(exporter, trace);
|
|
71
|
+
}
|
|
72
|
+
catch (error) {
|
|
73
|
+
ctx.logger.warn(`Databricks AI Agent: MLflow trace export failed: ${sanitizedErrorMessage(error)}`);
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
class DatabricksAiAgent {
|
|
77
|
+
constructor() {
|
|
78
|
+
this.description = {
|
|
79
|
+
displayName: 'Databricks AI Agent',
|
|
80
|
+
name: 'databricksAiAgent',
|
|
81
|
+
icon: { light: 'file:databricks.svg', dark: 'file:databricks.dark.svg' },
|
|
82
|
+
group: ['transform'],
|
|
83
|
+
version: [2.2],
|
|
84
|
+
hidden: true,
|
|
85
|
+
description: 'An AI agent that exports its runs to MLflow Tracing in Databricks',
|
|
86
|
+
defaults: {
|
|
87
|
+
name: 'Databricks AI Agent',
|
|
88
|
+
},
|
|
89
|
+
codex: {
|
|
90
|
+
categories: ['AI'],
|
|
91
|
+
subcategories: {
|
|
92
|
+
AI: ['Agents', 'Root Nodes'],
|
|
93
|
+
},
|
|
94
|
+
},
|
|
95
|
+
inputs: `={{
|
|
96
|
+
((hasOutputParser, needsFallback) => {
|
|
97
|
+
${utils_1.getInputs.toString()};
|
|
98
|
+
return getInputs(true, hasOutputParser, needsFallback);
|
|
99
|
+
})($parameter.hasOutputParser === undefined || $parameter.hasOutputParser === true, $parameter.needsFallback !== undefined && $parameter.needsFallback === true)
|
|
100
|
+
}}`,
|
|
101
|
+
outputs: [n8n_workflow_1.NodeConnectionTypes.Main],
|
|
102
|
+
credentials: [
|
|
103
|
+
{
|
|
104
|
+
name: 'databricksOAuth2Api',
|
|
105
|
+
required: false,
|
|
106
|
+
},
|
|
107
|
+
],
|
|
108
|
+
properties: [
|
|
109
|
+
descriptions_1.promptTypeOptions,
|
|
110
|
+
{
|
|
111
|
+
...descriptions_1.textFromPreviousNode,
|
|
112
|
+
displayOptions: { show: { promptType: ['auto'] } },
|
|
113
|
+
},
|
|
114
|
+
{
|
|
115
|
+
...descriptions_1.textInput,
|
|
116
|
+
displayOptions: { show: { promptType: ['define'] } },
|
|
117
|
+
},
|
|
118
|
+
{
|
|
119
|
+
displayName: 'Require Specific Output Format',
|
|
120
|
+
name: 'hasOutputParser',
|
|
121
|
+
type: 'boolean',
|
|
122
|
+
default: false,
|
|
123
|
+
noDataExpression: true,
|
|
124
|
+
},
|
|
125
|
+
{
|
|
126
|
+
displayName: 'Enable Fallback Model',
|
|
127
|
+
name: 'needsFallback',
|
|
128
|
+
type: 'boolean',
|
|
129
|
+
default: false,
|
|
130
|
+
noDataExpression: true,
|
|
131
|
+
},
|
|
132
|
+
exportTracesField,
|
|
133
|
+
experimentField,
|
|
134
|
+
...(0, description_1.getToolsAgentProperties)({ withStreaming: true }),
|
|
135
|
+
],
|
|
136
|
+
};
|
|
137
|
+
}
|
|
138
|
+
async execute() {
|
|
139
|
+
const exportTraces = this.getNodeParameter('exportTraces', 0, false);
|
|
140
|
+
if (!exportTraces) {
|
|
141
|
+
return await execute_1.toolsAgentExecute.call(this);
|
|
142
|
+
}
|
|
143
|
+
if (!this.getNode().credentials?.databricksOAuth2Api) {
|
|
144
|
+
throw new n8n_workflow_1.NodeOperationError(this.getNode(), 'Select a Databricks credential to export traces to MLflow', {
|
|
145
|
+
description: '"Export Traces to MLflow" is on, but this node has no credential selected.',
|
|
146
|
+
});
|
|
147
|
+
}
|
|
148
|
+
const exporter = await prepareTraceExport(this);
|
|
149
|
+
if (!exporter) {
|
|
150
|
+
return await execute_1.toolsAgentExecute.call(this);
|
|
151
|
+
}
|
|
152
|
+
const collectors = new Map();
|
|
153
|
+
return await execute_1.toolsAgentExecute.call(this, {
|
|
154
|
+
createCallbacks: (itemIndex) => {
|
|
155
|
+
const collector = new span_collector_1.MlflowSpanCollector();
|
|
156
|
+
collectors.set(itemIndex, collector);
|
|
157
|
+
return [collector];
|
|
158
|
+
},
|
|
159
|
+
onItemFinished: async (itemIndex) => {
|
|
160
|
+
const collector = collectors.get(itemIndex);
|
|
161
|
+
if (!collector)
|
|
162
|
+
return;
|
|
163
|
+
collectors.delete(itemIndex);
|
|
164
|
+
await flushTrace(this, exporter, collector);
|
|
165
|
+
},
|
|
166
|
+
});
|
|
167
|
+
}
|
|
168
|
+
}
|
|
169
|
+
exports.DatabricksAiAgent = DatabricksAiAgent;
|
|
170
|
+
//# sourceMappingURL=DatabricksAiAgent.node.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"DatabricksAiAgent.node.js","sourceRoot":"","sources":["../../../../nodes/vendors/Databricks/DatabricksAiAgent.node.ts"],"names":[],"mappings":";;;AAAA,sFAAiF;AACjF,+CAQsB;AAGtB,sDAAyF;AAEzF,qFAA8F;AAC9F,6EAAoF;AACpF,oDAAqD;AACrD,4DAA8D;AAE9D,wDAA4F;AAC5F,kDAA2D;AAE3D,MAAM,sBAAsB,GAAG,KAAK,CAAC;AAGrC,SAAS,qBAAqB,CAAC,KAAc;IAC5C,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACvE,OAAO,IAAA,2CAAmB,EAAC,OAAO,EAAE,sBAAsB,CAAC,CAAC;AAC7D,CAAC;AAOD,MAAM,iBAAiB,GAAoB;IAC1C,WAAW,EAAE,yBAAyB;IACtC,IAAI,EAAE,cAAc;IACpB,IAAI,EAAE,SAAS;IACf,OAAO,EAAE,KAAK;IACd,gBAAgB,EAAE,IAAI;IACtB,WAAW,EACV,yKAAyK;CAC1K,CAAC;AAEF,MAAM,eAAe,GAAoB;IACxC,WAAW,EAAE,YAAY;IACzB,IAAI,EAAE,YAAY;IAClB,IAAI,EAAE,iBAAiB;IACvB,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE;IACpC,cAAc,EAAE,EAAE,IAAI,EAAE,EAAE,YAAY,EAAE,CAAC,IAAI,CAAC,EAAE,EAAE;IAClD,WAAW,EACV,8FAA8F;IAC/F,KAAK,EAAE;QACN;YACC,WAAW,EAAE,SAAS;YACtB,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,yBAAyB;YACtC,IAAI,EAAE,gFAAgF;SACtF;QACD;YACC,WAAW,EAAE,OAAO;YACpB,IAAI,EAAE,IAAI;YACV,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,iBAAiB;SAC9B;KACD;CACD,CAAC;AAYF,KAAK,UAAU,kBAAkB,CAAC,GAAsB;IACvD,IAAI,CAAC;QACJ,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,cAAc,CAA6B,qBAAqB,CAAC,CAAC;QAC/F,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,IAAA,iCAAqB,EAAC,GAAG,EAAE,UAAU,CAAC,CAAC;QAEnE,MAAM,UAAU,GAAG,GAAG,CAAC,gBAAgB,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,CAGnF,CAAC;QACF,IAAI,UAAU,CAAC,IAAI,KAAK,IAAI,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;YAClD,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,UAAU,CAAC,KAAK,EAAE,CAAC;QAC5D,CAAC;QAED,MAAM,IAAI,GAAG,UAAU,CAAC,KAAK,IAAI,IAAA,oCAAqB,EAAC,GAAG,CAAC,WAAW,EAAE,CAAC,EAAE,IAAI,SAAS,CAAC,CAAC;QAC1F,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,IAAA,+BAAgB,EAAC,OAAO,EAAE,IAAI,CAAC,EAAE,CAAC;IACjF,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,GAAG,CAAC,MAAM,CAAC,IAAI,CACd,+DAA+D,qBAAqB,CAAC,KAAK,CAAC,EAAE,CAC7F,CAAC;QACF,OAAO,SAAS,CAAC;IAClB,CAAC;AACF,CAAC;AAGD,KAAK,UAAU,UAAU,CACxB,GAAsB,EACtB,QAAuB,EACvB,SAA8B;IAE9B,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC;IACjC,IAAI,CAAC,KAAK;QAAE,OAAO;IAEnB,IAAI,CAAC;QACJ,MAAM,IAAA,yBAAU,EAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IACnC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,GAAG,CAAC,MAAM,CAAC,IAAI,CACd,oDAAoD,qBAAqB,CAAC,KAAK,CAAC,EAAE,CAClF,CAAC;IACH,CAAC;AACF,CAAC;AAED;IAAA;QACC,gBAAW,GAAyB;YACnC,WAAW,EAAE,qBAAqB;YAClC,IAAI,EAAE,mBAAmB;YACzB,IAAI,EAAE,EAAE,KAAK,EAAE,qBAAqB,EAAE,IAAI,EAAE,0BAA0B,EAAE;YACxE,KAAK,EAAE,CAAC,WAAW,CAAC;YAGpB,OAAO,EAAE,CAAC,GAAG,CAAC;YACd,MAAM,EAAE,IAAI;YACZ,WAAW,EAAE,mEAAmE;YAChF,QAAQ,EAAE;gBACT,IAAI,EAAE,qBAAqB;aAC3B;YACD,KAAK,EAAE;gBACN,UAAU,EAAE,CAAC,IAAI,CAAC;gBAClB,aAAa,EAAE;oBACd,EAAE,EAAE,CAAC,QAAQ,EAAE,YAAY,CAAC;iBAC5B;aACD;YACD,MAAM,EAAE;;MAEJ,iBAAS,CAAC,QAAQ,EAAE;;;KAGrB;YACH,OAAO,EAAE,CAAC,kCAAmB,CAAC,IAAI,CAAC;YAKnC,WAAW,EAAE;gBACZ;oBACC,IAAI,EAAE,qBAAqB;oBAC3B,QAAQ,EAAE,KAAK;iBACf;aACD;YACD,UAAU,EAAE;gBACX,gCAAiB;gBACjB;oBACC,GAAG,mCAAoB;oBACvB,cAAc,EAAE,EAAE,IAAI,EAAE,EAAE,UAAU,EAAE,CAAC,MAAM,CAAC,EAAE,EAAE;iBAClD;gBACD;oBACC,GAAG,wBAAS;oBACZ,cAAc,EAAE,EAAE,IAAI,EAAE,EAAE,UAAU,EAAE,CAAC,QAAQ,CAAC,EAAE,EAAE;iBACpD;gBACD;oBACC,WAAW,EAAE,gCAAgC;oBAC7C,IAAI,EAAE,iBAAiB;oBACvB,IAAI,EAAE,SAAS;oBACf,OAAO,EAAE,KAAK;oBACd,gBAAgB,EAAE,IAAI;iBACtB;gBACD;oBACC,WAAW,EAAE,uBAAuB;oBACpC,IAAI,EAAE,eAAe;oBACrB,IAAI,EAAE,SAAS;oBACf,OAAO,EAAE,KAAK;oBACd,gBAAgB,EAAE,IAAI;iBACtB;gBACD,iBAAiB;gBACjB,eAAe;gBACf,GAAG,IAAA,qCAAuB,EAAC,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;aACnD;SACD,CAAC;IA4CH,CAAC;IA1CA,KAAK,CAAC,OAAO;QACZ,MAAM,YAAY,GAAG,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,CAAC,EAAE,KAAK,CAAY,CAAC;QAChF,IAAI,CAAC,YAAY,EAAE,CAAC;YACnB,OAAO,MAAM,2BAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3C,CAAC;QAID,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,WAAW,EAAE,mBAAmB,EAAE,CAAC;YACtD,MAAM,IAAI,iCAAkB,CAC3B,IAAI,CAAC,OAAO,EAAE,EACd,2DAA2D,EAC3D;gBACC,WAAW,EAAE,4EAA4E;aACzF,CACD,CAAC;QACH,CAAC;QAID,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACf,OAAO,MAAM,2BAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3C,CAAC;QAGD,MAAM,UAAU,GAAG,IAAI,GAAG,EAA+B,CAAC;QAE1D,OAAO,MAAM,2BAAiB,CAAC,IAAI,CAAC,IAAI,EAAE;YACzC,eAAe,EAAE,CAAC,SAAS,EAAE,EAAE;gBAC9B,MAAM,SAAS,GAAG,IAAI,oCAAmB,EAAE,CAAC;gBAC5C,UAAU,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;gBACrC,OAAO,CAAC,SAAS,CAAC,CAAC;YACpB,CAAC;YACD,cAAc,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE;gBACnC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBAC5C,IAAI,CAAC,SAAS;oBAAE,OAAO;gBACvB,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBAC7B,MAAM,UAAU,CAAC,IAAI,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;YAC7C,CAAC;SACD,CAAC,CAAC;IACJ,CAAC;CACD"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
<svg version="1.1" id="Layer_1" xmlns:x="ns_extend;" xmlns:i="ns_ai;" xmlns:graph="ns_graphs;" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" x="0px" y="0px" viewBox="0 0 40.1 42" style="enable-background:new 0 0 40.1 42;" xml:space="preserve">
|
|
2
|
+
<style type="text/css">
|
|
3
|
+
.st0{fill:#FF3621;}
|
|
4
|
+
</style>
|
|
5
|
+
<metadata>
|
|
6
|
+
<sfw xmlns="ns_sfw;">
|
|
7
|
+
<slices>
|
|
8
|
+
</slices>
|
|
9
|
+
<sliceSourceBounds bottomLeftOrigin="true" height="42" width="40.1" x="-69.1" y="-10.5">
|
|
10
|
+
</sliceSourceBounds>
|
|
11
|
+
</sfw>
|
|
12
|
+
</metadata>
|
|
13
|
+
<g>
|
|
14
|
+
<path class="st0" d="M40.1,31.1v-7.4l-0.8-0.5L20.1,33.7l-18.2-10l0-4.3l18.2,9.9l20.1-10.9v-7.3l-0.8-0.5L20.1,21.2L2.6,11.6
|
|
15
|
+
L20.1,2l14.1,7.7l1.1-0.6V8.3L20.1,0L0,10.9V12L20.1,23l18.2-10v4.4l-18.2,10L0.8,16.8L0,17.3v7.4l20.1,10.9l18.2-9.9v4.3l-18.2,10
|
|
16
|
+
L0.8,29.5L0,30v1.1L20.1,42L40.1,31.1z">
|
|
17
|
+
</path>
|
|
18
|
+
</g>
|
|
19
|
+
</svg>
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
<svg version="1.1" id="Layer_1" xmlns:x="ns_extend;" xmlns:i="ns_ai;" xmlns:graph="ns_graphs;" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" x="0px" y="0px" viewBox="0 0 40.1 42" style="enable-background:new 0 0 40.1 42;" xml:space="preserve">
|
|
2
|
+
<style type="text/css">
|
|
3
|
+
.st0{fill:#FF3621;}
|
|
4
|
+
</style>
|
|
5
|
+
<metadata>
|
|
6
|
+
<sfw xmlns="ns_sfw;">
|
|
7
|
+
<slices>
|
|
8
|
+
</slices>
|
|
9
|
+
<sliceSourceBounds bottomLeftOrigin="true" height="42" width="40.1" x="-69.1" y="-10.5">
|
|
10
|
+
</sliceSourceBounds>
|
|
11
|
+
</sfw>
|
|
12
|
+
</metadata>
|
|
13
|
+
<g>
|
|
14
|
+
<path class="st0" d="M40.1,31.1v-7.4l-0.8-0.5L20.1,33.7l-18.2-10l0-4.3l18.2,9.9l20.1-10.9v-7.3l-0.8-0.5L20.1,21.2L2.6,11.6
|
|
15
|
+
L20.1,2l14.1,7.7l1.1-0.6V8.3L20.1,0L0,10.9V12L20.1,23l18.2-10v4.4l-18.2,10L0.8,16.8L0,17.3v7.4l20.1,10.9l18.2-9.9v4.3l-18.2,10
|
|
16
|
+
L0.8,29.5L0,30v1.1L20.1,42L40.1,31.1z">
|
|
17
|
+
</path>
|
|
18
|
+
</g>
|
|
19
|
+
</svg>
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
|
|
2
|
+
import type { DocumentInterface } from '@langchain/core/documents';
|
|
3
|
+
import type { Serialized } from '@langchain/core/load/serializable';
|
|
4
|
+
import { BaseMessage } from '@langchain/core/messages';
|
|
5
|
+
import type { LLMResult } from '@langchain/core/outputs';
|
|
6
|
+
import type { ChainValues } from '@langchain/core/utils/types';
|
|
7
|
+
import type { CollectedTrace } from './types';
|
|
8
|
+
export declare class MlflowSpanCollector extends BaseCallbackHandler {
|
|
9
|
+
name: string;
|
|
10
|
+
awaitHandlers: boolean;
|
|
11
|
+
readonly traceId: string;
|
|
12
|
+
private readonly spanTraceId;
|
|
13
|
+
constructor();
|
|
14
|
+
private readonly spans;
|
|
15
|
+
private readonly parentOf;
|
|
16
|
+
private readonly runOrder;
|
|
17
|
+
private readonly openRuns;
|
|
18
|
+
private rootRunId?;
|
|
19
|
+
private startedAtMs?;
|
|
20
|
+
private lastActivityMs;
|
|
21
|
+
private executionId?;
|
|
22
|
+
private workflowId?;
|
|
23
|
+
private nodeName?;
|
|
24
|
+
private requestPreview?;
|
|
25
|
+
private responsePreview?;
|
|
26
|
+
private tokenUsage?;
|
|
27
|
+
private nearestKeptAncestor;
|
|
28
|
+
private captureRunMetadata;
|
|
29
|
+
private startSpan;
|
|
30
|
+
private closeSpan;
|
|
31
|
+
private endSpan;
|
|
32
|
+
private addTokenUsage;
|
|
33
|
+
private failSpan;
|
|
34
|
+
handleChainStart(_chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, _tags?: string[], metadata?: Record<string, unknown>, _runType?: string, runName?: string): void;
|
|
35
|
+
handleChainEnd(outputs: ChainValues, runId: string): void;
|
|
36
|
+
handleChainError(error: Error, runId: string): void;
|
|
37
|
+
handleChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, _extraParams?: Record<string, unknown>, _tags?: string[], metadata?: Record<string, unknown>, runName?: string): void;
|
|
38
|
+
handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, _extraParams?: Record<string, unknown>, _tags?: string[], metadata?: Record<string, unknown>, runName?: string): void;
|
|
39
|
+
handleLLMEnd(output: LLMResult, runId: string): void;
|
|
40
|
+
handleLLMError(error: Error, runId: string): void;
|
|
41
|
+
handleToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string, _tags?: string[], metadata?: Record<string, unknown>, runName?: string): void;
|
|
42
|
+
handleToolEnd(output: unknown, runId: string): void;
|
|
43
|
+
handleToolError(error: Error, runId: string): void;
|
|
44
|
+
handleRetrieverStart(retriever: Serialized, query: string, runId: string, parentRunId?: string, _tags?: string[], metadata?: Record<string, unknown>, name?: string): void;
|
|
45
|
+
handleRetrieverEnd(documents: DocumentInterface[], runId: string): void;
|
|
46
|
+
handleRetrieverError(error: Error, runId: string): void;
|
|
47
|
+
finish(): CollectedTrace | undefined;
|
|
48
|
+
}
|