@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.
Files changed (103) hide show
  1. package/dist/credentials/AzureEntraCognitiveServicesOAuth2Api.credentials.js +4 -4
  2. package/dist/credentials/AzureEntraCognitiveServicesOAuth2Api.credentials.js.map +1 -1
  3. package/dist/credentials/AzureOpenAiApi.credentials.js +4 -4
  4. package/dist/credentials/AzureOpenAiApi.credentials.js.map +1 -1
  5. package/dist/credentials/GoogleVertexAiApi.credentials.d.ts +15 -0
  6. package/dist/credentials/GoogleVertexAiApi.credentials.js +83 -0
  7. package/dist/credentials/GoogleVertexAiApi.credentials.js.map +1 -0
  8. package/dist/credentials/icons/google.svg +1 -0
  9. package/dist/known/credentials.json +12 -0
  10. package/dist/known/nodes.json +4 -0
  11. package/dist/node-definitions/.nodes-hash +1 -1
  12. package/dist/node-definitions/nodes/n8n-nodes-langchain/embeddingsAzureOpenAi/index.ts +1 -1
  13. package/dist/node-definitions/nodes/n8n-nodes-langchain/embeddingsAzureOpenAi/v1.schema.js +2 -1
  14. package/dist/node-definitions/nodes/n8n-nodes-langchain/embeddingsAzureOpenAi/v1.ts +4 -2
  15. package/dist/node-definitions/nodes/n8n-nodes-langchain/embeddingsGoogleVertex/v1.schema.js +3 -2
  16. package/dist/node-definitions/nodes/n8n-nodes-langchain/embeddingsGoogleVertex/v1.ts +3 -0
  17. package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatAzureOpenAi/index.ts +1 -1
  18. package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatAzureOpenAi/v1.schema.js +2 -1
  19. package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatAzureOpenAi/v1.ts +11 -2
  20. package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatAzureOpenAi/v11.schema.js +3 -2
  21. package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatAzureOpenAi/v11.ts +15 -5
  22. package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatGoogleGemini/v1.ts +1 -1
  23. package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatGoogleGemini/v11.ts +1 -1
  24. package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatGoogleGemini/v12.ts +1 -1
  25. package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatGoogleVertex/v1.schema.js +3 -2
  26. package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatGoogleVertex/v1.ts +3 -0
  27. package/dist/nodes/agents/Agent/agents/ToolsAgent/V2/execute.d.ts +6 -1
  28. package/dist/nodes/agents/Agent/agents/ToolsAgent/V2/execute.js +23 -7
  29. package/dist/nodes/agents/Agent/agents/ToolsAgent/V2/execute.js.map +1 -1
  30. package/dist/nodes/embeddings/EmbeddingsAzureOpenAi/EmbeddingsAzureOpenAi.node.js +102 -16
  31. package/dist/nodes/embeddings/EmbeddingsAzureOpenAi/EmbeddingsAzureOpenAi.node.js.map +1 -1
  32. package/dist/nodes/embeddings/EmbeddingsDatabricks/EmbeddingsDatabricks.node.js +1 -0
  33. package/dist/nodes/embeddings/EmbeddingsDatabricks/EmbeddingsDatabricks.node.js.map +1 -1
  34. package/dist/nodes/embeddings/EmbeddingsGoogleVertex/EmbeddingsGoogleVertex.node.d.ts +5 -6
  35. package/dist/nodes/embeddings/EmbeddingsGoogleVertex/EmbeddingsGoogleVertex.node.js +9 -41
  36. package/dist/nodes/embeddings/EmbeddingsGoogleVertex/EmbeddingsGoogleVertex.node.js.map +1 -1
  37. package/dist/nodes/llms/LMChatAnthropic/LmChatAnthropic.node.js +1 -15
  38. package/dist/nodes/llms/LMChatAnthropic/LmChatAnthropic.node.js.map +1 -1
  39. package/dist/nodes/llms/LmChatAzureOpenAi/LmChatAzureOpenAi.node.js +59 -8
  40. package/dist/nodes/llms/LmChatAzureOpenAi/LmChatAzureOpenAi.node.js.map +1 -1
  41. package/dist/nodes/llms/LmChatAzureOpenAi/credentials/api-key.js +1 -1
  42. package/dist/nodes/llms/LmChatAzureOpenAi/credentials/api-key.js.map +1 -1
  43. package/dist/nodes/llms/LmChatAzureOpenAi/credentials/oauth2.js +1 -2
  44. package/dist/nodes/llms/LmChatAzureOpenAi/credentials/oauth2.js.map +1 -1
  45. package/dist/nodes/llms/LmChatAzureOpenAi/credentials/requireFoundryEndpoint.js +1 -1
  46. package/dist/nodes/llms/LmChatAzureOpenAi/credentials/requireFoundryEndpoint.js.map +1 -1
  47. package/dist/nodes/llms/LmChatAzureOpenAi/credentials/requireTenantId.js +1 -1
  48. package/dist/nodes/llms/LmChatAzureOpenAi/credentials/requireTenantId.js.map +1 -1
  49. package/dist/nodes/llms/LmChatAzureOpenAi/error-handling.d.ts +1 -1
  50. package/dist/nodes/llms/LmChatAzureOpenAi/error-handling.js +12 -6
  51. package/dist/nodes/llms/LmChatAzureOpenAi/error-handling.js.map +1 -1
  52. package/dist/nodes/llms/LmChatAzureOpenAi/methods/searchModels.js +1 -1
  53. package/dist/nodes/llms/LmChatAzureOpenAi/methods/searchModels.js.map +1 -1
  54. package/dist/nodes/llms/LmChatAzureOpenAi/properties.js +29 -4
  55. package/dist/nodes/llms/LmChatAzureOpenAi/properties.js.map +1 -1
  56. package/dist/nodes/llms/LmChatAzureOpenAi/types.d.ts +1 -0
  57. package/dist/nodes/llms/LmChatAzureOpenAi/types.js +2 -1
  58. package/dist/nodes/llms/LmChatAzureOpenAi/types.js.map +1 -1
  59. package/dist/nodes/llms/LmChatGoogleGemini/LmChatGoogleGemini.node.js +22 -13
  60. package/dist/nodes/llms/LmChatGoogleGemini/LmChatGoogleGemini.node.js.map +1 -1
  61. package/dist/nodes/llms/LmChatGoogleVertex/LmChatGoogleVertex.node.d.ts +5 -6
  62. package/dist/nodes/llms/LmChatGoogleVertex/LmChatGoogleVertex.node.js +9 -41
  63. package/dist/nodes/llms/LmChatGoogleVertex/LmChatGoogleVertex.node.js.map +1 -1
  64. package/dist/nodes/trigger/ChatTrigger/ChatTrigger.node.js +3 -0
  65. package/dist/nodes/trigger/ChatTrigger/ChatTrigger.node.js.map +1 -1
  66. package/dist/nodes/vector_store/VectorStoreDatabricks/DatabricksVectorStore.d.ts +1 -0
  67. package/dist/nodes/vector_store/VectorStoreDatabricks/DatabricksVectorStore.js +19 -5
  68. package/dist/nodes/vector_store/VectorStoreDatabricks/DatabricksVectorStore.js.map +1 -1
  69. package/dist/nodes/vector_store/VectorStoreDatabricks/VectorStoreDatabricks.node.js +4 -1
  70. package/dist/nodes/vector_store/VectorStoreDatabricks/VectorStoreDatabricks.node.js.map +1 -1
  71. package/dist/nodes/vector_store/VectorStoreMongoDBAtlas/VectorStoreMongoDBAtlas.node.js +14 -9
  72. package/dist/nodes/vector_store/VectorStoreMongoDBAtlas/VectorStoreMongoDBAtlas.node.js.map +1 -1
  73. package/dist/nodes/vector_store/VectorStoreOracleDB/VectorStoreOracleDB.node.js +31 -7
  74. package/dist/nodes/vector_store/VectorStoreOracleDB/VectorStoreOracleDB.node.js.map +1 -1
  75. package/dist/nodes/vendors/Databricks/DatabricksAiAgent.node.d.ts +5 -0
  76. package/dist/nodes/vendors/Databricks/DatabricksAiAgent.node.js +170 -0
  77. package/dist/nodes/vendors/Databricks/DatabricksAiAgent.node.js.map +1 -0
  78. package/dist/nodes/vendors/Databricks/databricks.dark.svg +19 -0
  79. package/dist/nodes/vendors/Databricks/databricks.svg +19 -0
  80. package/dist/nodes/vendors/Databricks/mlflow/span-collector.d.ts +48 -0
  81. package/dist/nodes/vendors/Databricks/mlflow/span-collector.js +378 -0
  82. package/dist/nodes/vendors/Databricks/mlflow/span-collector.js.map +1 -0
  83. package/dist/nodes/vendors/Databricks/mlflow/trace-writer.d.ts +41 -0
  84. package/dist/nodes/vendors/Databricks/mlflow/trace-writer.js +129 -0
  85. package/dist/nodes/vendors/Databricks/mlflow/trace-writer.js.map +1 -0
  86. package/dist/nodes/vendors/Databricks/mlflow/transport.d.ts +7 -0
  87. package/dist/nodes/vendors/Databricks/mlflow/transport.js +72 -0
  88. package/dist/nodes/vendors/Databricks/mlflow/transport.js.map +1 -0
  89. package/dist/nodes/vendors/Databricks/mlflow/types.d.ts +54 -0
  90. package/dist/nodes/vendors/Databricks/mlflow/types.js +23 -0
  91. package/dist/nodes/vendors/Databricks/mlflow/types.js.map +1 -0
  92. package/dist/nodes/vendors/GoogleGemini/methods/listSearch.js +20 -10
  93. package/dist/nodes/vendors/GoogleGemini/methods/listSearch.js.map +1 -1
  94. package/dist/typecheck.tsbuildinfo +1 -1
  95. package/dist/types/credentials.json +3 -2
  96. package/dist/types/nodes.json +6 -5
  97. package/dist/utils/google-vertex.d.ts +14 -0
  98. package/dist/utils/google-vertex.js +130 -0
  99. package/dist/utils/google-vertex.js.map +1 -0
  100. package/dist/utils/sqlIdentifier.d.ts +1 -0
  101. package/dist/utils/sqlIdentifier.js +22 -0
  102. package/dist/utils/sqlIdentifier.js.map +1 -1
  103. 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;AAInF,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,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;IACjD,MAAM,CAAC,KAAK,CAAC,UAAU,CACtB,UAA+B,EAC/B,IAAoB;QAEpB,MAAM,mBAAmB,GAAG,IAAI,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;QAEvD,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;QAEvD,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,OAAO,CAAC,gBAAgB,CAAC,WAAW,EAAE,SAAS,EAAE,EAAE,EAAE;YACtE,YAAY,EAAE,IAAI;SAClB,CAAW,CAAC;QACb,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;YACL,MAAM,EAAE,MAA+D;SACvE,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,OAAO,CAAC,gBAAgB,CAAC,WAAW,EAAE,SAAS,EAAE,EAAE,EAAE;YACtE,YAAY,EAAE,IAAI;SAClB,CAAW,CAAC;QACb,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"}
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
+ }