openlayer 0.16.1 → 0.16.3

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 (134) hide show
  1. package/CHANGELOG.md +29 -0
  2. package/_shims/bun-runtime.js +1 -2
  3. package/_shims/bun-runtime.js.map +1 -1
  4. package/_shims/node-runtime.js +18 -9
  5. package/_shims/node-runtime.js.map +1 -1
  6. package/_shims/node-runtime.mjs.map +1 -1
  7. package/_shims/registry.js +2 -2
  8. package/_shims/registry.js.map +1 -1
  9. package/_shims/registry.mjs.map +1 -1
  10. package/_shims/web-runtime.js +1 -2
  11. package/_shims/web-runtime.js.map +1 -1
  12. package/_shims/web-runtime.mjs.map +1 -1
  13. package/core.d.ts +2 -4
  14. package/core.d.ts.map +1 -1
  15. package/core.js +6 -6
  16. package/core.js.map +1 -1
  17. package/core.mjs +1 -1
  18. package/core.mjs.map +1 -1
  19. package/error.js.map +1 -1
  20. package/error.mjs.map +1 -1
  21. package/index.js +17 -7
  22. package/index.js.map +1 -1
  23. package/index.mjs.map +1 -1
  24. package/internal/qs/stringify.js +1 -2
  25. package/internal/qs/stringify.js.map +1 -1
  26. package/internal/qs/stringify.mjs.map +1 -1
  27. package/internal/qs/utils.js +9 -9
  28. package/internal/qs/utils.js.map +1 -1
  29. package/internal/qs/utils.mjs.map +1 -1
  30. package/lib/core/cli.d.ts.map +1 -1
  31. package/lib/core/cli.js +17 -8
  32. package/lib/core/cli.js.map +1 -1
  33. package/lib/core/cli.mjs +0 -1
  34. package/lib/core/cli.mjs.map +1 -1
  35. package/lib/core/openai-monitor.d.ts.map +1 -1
  36. package/lib/core/openai-monitor.js +1 -1
  37. package/lib/core/openai-monitor.js.map +1 -1
  38. package/lib/core/openai-monitor.mjs +1 -1
  39. package/lib/core/openai-monitor.mjs.map +1 -1
  40. package/lib/integrations/bedrockAgentTracer.js +49 -50
  41. package/lib/integrations/bedrockAgentTracer.js.map +1 -1
  42. package/lib/integrations/bedrockAgentTracer.mjs +48 -48
  43. package/lib/integrations/bedrockAgentTracer.mjs.map +1 -1
  44. package/lib/integrations/index.d.ts +5 -0
  45. package/lib/integrations/index.d.ts.map +1 -0
  46. package/lib/integrations/index.js +21 -0
  47. package/lib/integrations/index.js.map +1 -0
  48. package/lib/integrations/index.mjs +5 -0
  49. package/lib/integrations/index.mjs.map +1 -0
  50. package/lib/integrations/langchainCallback.d.ts +83 -27
  51. package/lib/integrations/langchainCallback.d.ts.map +1 -1
  52. package/lib/integrations/langchainCallback.js +597 -88
  53. package/lib/integrations/langchainCallback.js.map +1 -1
  54. package/lib/integrations/langchainCallback.mjs +599 -87
  55. package/lib/integrations/langchainCallback.mjs.map +1 -1
  56. package/lib/integrations/openAiTracer.js +1 -2
  57. package/lib/integrations/openAiTracer.js.map +1 -1
  58. package/lib/integrations/openAiTracer.mjs.map +1 -1
  59. package/lib/integrations/tracedTool.d.ts +66 -0
  60. package/lib/integrations/tracedTool.d.ts.map +1 -0
  61. package/lib/integrations/tracedTool.js +186 -0
  62. package/lib/integrations/tracedTool.js.map +1 -0
  63. package/lib/integrations/tracedTool.mjs +149 -0
  64. package/lib/integrations/tracedTool.mjs.map +1 -0
  65. package/lib/tracing/steps.d.ts +120 -1
  66. package/lib/tracing/steps.d.ts.map +1 -1
  67. package/lib/tracing/steps.js +128 -4
  68. package/lib/tracing/steps.js.map +1 -1
  69. package/lib/tracing/steps.mjs +119 -1
  70. package/lib/tracing/steps.mjs.map +1 -1
  71. package/lib/tracing/tracer.d.ts +69 -1
  72. package/lib/tracing/tracer.d.ts.map +1 -1
  73. package/lib/tracing/tracer.js +92 -18
  74. package/lib/tracing/tracer.js.map +1 -1
  75. package/lib/tracing/tracer.mjs +78 -10
  76. package/lib/tracing/tracer.mjs.map +1 -1
  77. package/package.json +3 -1
  78. package/resources/commits/commits.js +17 -7
  79. package/resources/commits/commits.js.map +1 -1
  80. package/resources/commits/test-results.js.map +1 -1
  81. package/resources/commits/test-results.mjs.map +1 -1
  82. package/resources/inference-pipelines/inference-pipelines.js +17 -7
  83. package/resources/inference-pipelines/inference-pipelines.js.map +1 -1
  84. package/resources/inference-pipelines/inference-pipelines.mjs.map +1 -1
  85. package/resources/inference-pipelines/test-results.js.map +1 -1
  86. package/resources/inference-pipelines/test-results.mjs.map +1 -1
  87. package/resources/projects/commits.js.map +1 -1
  88. package/resources/projects/commits.mjs.map +1 -1
  89. package/resources/projects/inference-pipelines.js.map +1 -1
  90. package/resources/projects/inference-pipelines.mjs.map +1 -1
  91. package/resources/projects/projects.js +17 -7
  92. package/resources/projects/projects.js.map +1 -1
  93. package/resources/projects/projects.mjs.map +1 -1
  94. package/resources/projects/tests.js.map +1 -1
  95. package/resources/projects/tests.mjs.map +1 -1
  96. package/resources/storage/storage.js +17 -7
  97. package/resources/storage/storage.js.map +1 -1
  98. package/shims/node.d.ts +0 -4
  99. package/shims/node.d.ts.map +1 -1
  100. package/shims/node.js +0 -25
  101. package/shims/node.js.map +1 -1
  102. package/shims/node.mjs +0 -1
  103. package/shims/node.mjs.map +1 -1
  104. package/shims/web.js +0 -25
  105. package/shims/web.js.map +1 -1
  106. package/shims/web.mjs +0 -1
  107. package/shims/web.mjs.map +1 -1
  108. package/src/core.ts +1 -1
  109. package/src/lib/core/cli.ts +0 -1
  110. package/src/lib/core/openai-monitor.ts +1 -1
  111. package/src/lib/integrations/index.ts +4 -0
  112. package/src/lib/integrations/langchainCallback.ts +802 -88
  113. package/src/lib/integrations/tracedTool.ts +175 -0
  114. package/src/lib/tracing/index.d.ts +10 -0
  115. package/src/lib/tracing/steps.ts +218 -5
  116. package/src/lib/tracing/tracer.ts +153 -9
  117. package/src/uploads.ts +3 -2
  118. package/src/version.ts +1 -1
  119. package/uploads.d.ts +2 -2
  120. package/uploads.d.ts.map +1 -1
  121. package/uploads.js +3 -4
  122. package/uploads.js.map +1 -1
  123. package/uploads.mjs +1 -2
  124. package/uploads.mjs.map +1 -1
  125. package/version.d.ts +1 -1
  126. package/version.js +1 -1
  127. package/version.mjs +1 -1
  128. package/lib/tracing/enums.d.ts +0 -5
  129. package/lib/tracing/enums.d.ts.map +0 -1
  130. package/lib/tracing/enums.js +0 -9
  131. package/lib/tracing/enums.js.map +0 -1
  132. package/lib/tracing/enums.mjs +0 -6
  133. package/lib/tracing/enums.mjs.map +0 -1
  134. package/src/lib/tracing/enums.ts +0 -4
@@ -1,4 +1,18 @@
1
- import { StepType } from "./enums.js";
1
+ /**
2
+ * Module with the different Step classes and helper functions that can be used in a trace.
3
+ *
4
+ * This is the consolidated module for all step-related functionality in the Openlayer SDK,
5
+ * similar to the Python SDK's steps.py module.
6
+ */
7
+ export declare enum StepType {
8
+ USER_CALL = "user_call",
9
+ CHAT_COMPLETION = "chat_completion",
10
+ CHAIN = "chain",
11
+ AGENT = "agent",
12
+ TOOL = "tool",
13
+ RETRIEVER = "retriever",
14
+ FUNCTION_CALL = "function_call"
15
+ }
2
16
  export interface StepData {
3
17
  name: string;
4
18
  id: string;
@@ -21,6 +35,13 @@ export interface ChatCompletionStepData extends StepData {
21
35
  model: string | null;
22
36
  modelParameters: Record<string, any> | null;
23
37
  }
38
+ /**
39
+ * Step, defined as a single function call being traced.
40
+ *
41
+ * This is the base class for all the different types of steps that can be
42
+ * used in a trace. Steps can also contain nested steps, which represent
43
+ * function calls made within the parent step.
44
+ */
24
45
  export declare class Step {
25
46
  name: string;
26
47
  id: string;
@@ -34,13 +55,28 @@ export declare class Step {
34
55
  latency: number | null;
35
56
  steps: Step[];
36
57
  constructor(name: string, inputs?: any, output?: any, metadata?: Record<string, any>, startTime?: number | null, endTime?: number | null);
58
+ /**
59
+ * Adds a nested step to the current step.
60
+ */
37
61
  addNestedStep(nestedStep: Step): void;
62
+ /**
63
+ * Logs step data.
64
+ */
38
65
  log(data: Partial<Record<keyof this, any>>): void;
66
+ /**
67
+ * Dictionary representation of the Step.
68
+ */
39
69
  toJSON(): StepData;
40
70
  }
71
+ /**
72
+ * User call step represents a generic user call in the trace.
73
+ */
41
74
  export declare class UserCallStep extends Step {
42
75
  constructor(name: string, inputs?: any, output?: any, metadata?: Record<string, any>, startTime?: number | null, endTime?: number | null);
43
76
  }
77
+ /**
78
+ * Chat completion step represents an LLM chat completion in the trace.
79
+ */
44
80
  export declare class ChatCompletionStep extends Step {
45
81
  provider: string | null;
46
82
  promptTokens: number | null;
@@ -52,4 +88,87 @@ export declare class ChatCompletionStep extends Step {
52
88
  constructor(name: string, inputs?: any, output?: any, metadata?: Record<string, any>, startTime?: number | null, endTime?: number | null);
53
89
  toJSON(): ChatCompletionStepData;
54
90
  }
91
+ /**
92
+ * Chain step represents sequential operations or workflow chains in AI applications.
93
+ */
94
+ export declare class ChainStep extends Step {
95
+ constructor(name: string, inputs?: any, output?: any, metadata?: Record<string, any>);
96
+ }
97
+ /**
98
+ * Agent step tracks AI agent actions and decision-making processes.
99
+ */
100
+ export declare class AgentStep extends Step {
101
+ tool: string | null;
102
+ action: any;
103
+ constructor(name: string, inputs?: any, output?: any, metadata?: Record<string, any>);
104
+ }
105
+ /**
106
+ * Tool step tracks individual tool executions within AI workflows.
107
+ */
108
+ export declare class ToolStep extends Step {
109
+ constructor(name: string, inputs?: any, output?: any, metadata?: Record<string, any>);
110
+ }
111
+ /**
112
+ * Retriever step tracks document retrieval and search operations.
113
+ */
114
+ export declare class RetrieverStep extends Step {
115
+ documents: any[];
116
+ constructor(name: string, inputs?: any, output?: any, metadata?: Record<string, any>);
117
+ }
118
+ /**
119
+ * Function call step tracks specific function calls with detailed execution metadata.
120
+ */
121
+ export declare class FunctionCallStep extends Step {
122
+ functionName: string | null;
123
+ arguments: Record<string, any> | null;
124
+ result: any;
125
+ executionTime: number | null;
126
+ constructor(name: string, inputs?: any, output?: any, metadata?: Record<string, any>);
127
+ }
128
+ /**
129
+ * Factory function to create a step based on the step type.
130
+ */
131
+ export declare function stepFactory(stepType: StepType, name: string, inputs?: any, output?: any, metadata?: Record<string, any>, startTime?: number | null, endTime?: number | null): Step;
132
+ /**
133
+ * Interface for step creation parameters used by helper functions
134
+ */
135
+ export interface StepCreationParams {
136
+ name: string;
137
+ inputs?: any;
138
+ output?: any;
139
+ metadata?: Record<string, any>;
140
+ }
141
+ export interface ChatCompletionStepParams extends StepCreationParams {
142
+ latency: number;
143
+ tokens?: number | null;
144
+ promptTokens?: number | null;
145
+ completionTokens?: number | null;
146
+ model?: string | null;
147
+ modelParameters?: Record<string, any> | null;
148
+ rawOutput?: string | null;
149
+ provider?: string;
150
+ startTime?: number | null;
151
+ endTime?: number | null;
152
+ }
153
+ export interface AgentStepParams extends StepCreationParams {
154
+ tool?: string;
155
+ action?: any;
156
+ }
157
+ export interface RetrieverStepParams extends StepCreationParams {
158
+ documents?: any[];
159
+ }
160
+ export interface FunctionCallStepParams extends StepCreationParams {
161
+ functionName: string;
162
+ arguments: Record<string, any>;
163
+ }
164
+ export interface EnhancedAgentStepParams extends StepCreationParams {
165
+ agentType: string;
166
+ }
167
+ export interface HandoffStepParams {
168
+ name: string;
169
+ fromComponent: string;
170
+ toComponent: string;
171
+ handoffData: any;
172
+ metadata?: Record<string, any>;
173
+ }
55
174
  //# sourceMappingURL=steps.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"steps.d.ts","sourceRoot":"","sources":["../../src/lib/tracing/steps.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AAEnC,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,QAAQ,CAAC;IACf,MAAM,EAAE,GAAG,CAAC;IACZ,MAAM,EAAE,GAAG,CAAC;IACZ,WAAW,EAAE,GAAG,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC9B,KAAK,EAAE,QAAQ,EAAE,CAAC;IAClB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;CACxB;AAED,MAAM,WAAW,sBAAuB,SAAQ,QAAQ;IACtD,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;IACpB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,CAAC;CAC7C;AAED,qBAAa,IAAI;IACf,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,GAAG,CAAC;IACZ,MAAM,EAAE,GAAG,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC9B,QAAQ,EAAE,QAAQ,GAAG,IAAI,CAAQ;IACjC,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAQ;IAC9B,WAAW,EAAE,GAAG,CAAQ;IACxB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAQ;IAC9B,KAAK,EAAE,IAAI,EAAE,CAAM;gBAEP,IAAI,EAAE,MAAM,EAAE,MAAM,GAAE,GAAU,EAAE,MAAM,GAAE,GAAU,EAAE,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI;IAWxJ,aAAa,CAAC,UAAU,EAAE,IAAI,GAAG,IAAI;IAIrC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,GAAG,CAAC,CAAC,GAAG,IAAI;IASjD,MAAM,IAAI,QAAQ;CAenB;AAED,qBAAa,YAAa,SAAQ,IAAI;gBACxB,IAAI,EAAE,MAAM,EAAE,MAAM,GAAE,GAAU,EAAE,MAAM,GAAE,GAAU,EAAE,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI;CAIzJ;AAED,qBAAa,kBAAmB,SAAQ,IAAI;IAC1C,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAQ;IAC/B,YAAY,EAAE,MAAM,GAAG,IAAI,CAAQ;IACnC,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAQ;IACvC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAQ;IAC7B,IAAI,EAAE,MAAM,GAAG,IAAI,CAAQ;IAC3B,KAAK,EAAE,MAAM,GAAG,IAAI,CAAQ;IAC5B,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,CAAQ;gBAErC,IAAI,EAAE,MAAM,EAAE,MAAM,GAAE,GAAU,EAAE,MAAM,GAAE,GAAU,EAAE,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI;IAKjJ,MAAM,IAAI,sBAAsB;CAa1C"}
1
+ {"version":3,"file":"steps.d.ts","sourceRoot":"","sources":["../../src/lib/tracing/steps.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAMH,oBAAY,QAAQ;IAClB,SAAS,cAAc;IACvB,eAAe,oBAAoB;IACnC,KAAK,UAAU;IACf,KAAK,UAAU;IACf,IAAI,SAAS;IACb,SAAS,cAAc;IACvB,aAAa,kBAAkB;CAChC;AAID,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,QAAQ,CAAC;IACf,MAAM,EAAE,GAAG,CAAC;IACZ,MAAM,EAAE,GAAG,CAAC;IACZ,WAAW,EAAE,GAAG,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC9B,KAAK,EAAE,QAAQ,EAAE,CAAC;IAClB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;CACxB;AAED,MAAM,WAAW,sBAAuB,SAAQ,QAAQ;IACtD,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;IACpB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,CAAC;CAC7C;AAID;;;;;;GAMG;AACH,qBAAa,IAAI;IACf,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,GAAG,CAAC;IACZ,MAAM,EAAE,GAAG,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC9B,QAAQ,EAAE,QAAQ,GAAG,IAAI,CAAQ;IACjC,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAQ;IAC9B,WAAW,EAAE,GAAG,CAAQ;IACxB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAQ;IAC9B,KAAK,EAAE,IAAI,EAAE,CAAM;gBAGjB,IAAI,EAAE,MAAM,EACZ,MAAM,GAAE,GAAU,EAClB,MAAM,GAAE,GAAU,EAClB,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM,EAClC,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,EACzB,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI;IAWzB;;OAEG;IACH,aAAa,CAAC,UAAU,EAAE,IAAI,GAAG,IAAI;IAIrC;;OAEG;IACH,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,GAAG,CAAC,CAAC,GAAG,IAAI;IASjD;;OAEG;IACH,MAAM,IAAI,QAAQ;CAenB;AAED;;GAEG;AACH,qBAAa,YAAa,SAAQ,IAAI;gBAElC,IAAI,EAAE,MAAM,EACZ,MAAM,GAAE,GAAU,EAClB,MAAM,GAAE,GAAU,EAClB,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM,EAClC,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,EACzB,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI;CAK1B;AAED;;GAEG;AACH,qBAAa,kBAAmB,SAAQ,IAAI;IAC1C,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAQ;IAC/B,YAAY,EAAE,MAAM,GAAG,IAAI,CAAQ;IACnC,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAQ;IACvC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAQ;IAC7B,IAAI,EAAE,MAAM,GAAG,IAAI,CAAQ;IAC3B,KAAK,EAAE,MAAM,GAAG,IAAI,CAAQ;IAC5B,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,CAAQ;gBAGjD,IAAI,EAAE,MAAM,EACZ,MAAM,GAAE,GAAU,EAClB,MAAM,GAAE,GAAU,EAClB,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM,EAClC,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,EACzB,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI;IAMhB,MAAM,IAAI,sBAAsB;CAa1C;AAED;;GAEG;AACH,qBAAa,SAAU,SAAQ,IAAI;gBACrB,IAAI,EAAE,MAAM,EAAE,MAAM,GAAE,GAAU,EAAE,MAAM,GAAE,GAAU,EAAE,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM;CAIrG;AAED;;GAEG;AACH,qBAAa,SAAU,SAAQ,IAAI;IACjC,IAAI,EAAE,MAAM,GAAG,IAAI,CAAQ;IAC3B,MAAM,EAAE,GAAG,CAAQ;gBAEP,IAAI,EAAE,MAAM,EAAE,MAAM,GAAE,GAAU,EAAE,MAAM,GAAE,GAAU,EAAE,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM;CAIrG;AAED;;GAEG;AACH,qBAAa,QAAS,SAAQ,IAAI;gBACpB,IAAI,EAAE,MAAM,EAAE,MAAM,GAAE,GAAU,EAAE,MAAM,GAAE,GAAU,EAAE,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM;CAIrG;AAED;;GAEG;AACH,qBAAa,aAAc,SAAQ,IAAI;IACrC,SAAS,EAAE,GAAG,EAAE,CAAM;gBAEV,IAAI,EAAE,MAAM,EAAE,MAAM,GAAE,GAAU,EAAE,MAAM,GAAE,GAAU,EAAE,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM;CAIrG;AAED;;GAEG;AACH,qBAAa,gBAAiB,SAAQ,IAAI;IACxC,YAAY,EAAE,MAAM,GAAG,IAAI,CAAQ;IACnC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,CAAQ;IAC7C,MAAM,EAAE,GAAG,CAAQ;IACnB,aAAa,EAAE,MAAM,GAAG,IAAI,CAAQ;gBAExB,IAAI,EAAE,MAAM,EAAE,MAAM,GAAE,GAAU,EAAE,MAAM,GAAE,GAAU,EAAE,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM;CAIrG;AAID;;GAEG;AACH,wBAAgB,WAAW,CACzB,QAAQ,EAAE,QAAQ,EAClB,IAAI,EAAE,MAAM,EACZ,MAAM,GAAE,GAAU,EAClB,MAAM,GAAE,GAAU,EAClB,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM,EAClC,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,EACzB,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,GACtB,IAAI,CAuBN;AAID;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC;AAED,MAAM,WAAW,wBAAyB,SAAQ,kBAAkB;IAClE,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,gBAAgB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACjC,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,CAAC;IAC7C,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACzB;AAED,MAAM,WAAW,eAAgB,SAAQ,kBAAkB;IACzD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,GAAG,CAAC;CACd;AAED,MAAM,WAAW,mBAAoB,SAAQ,kBAAkB;IAC7D,SAAS,CAAC,EAAE,GAAG,EAAE,CAAC;CACnB;AAED,MAAM,WAAW,sBAAuB,SAAQ,kBAAkB;IAChE,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC;AAED,MAAM,WAAW,uBAAwB,SAAQ,kBAAkB;IACjE,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,GAAG,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC"}
@@ -1,8 +1,33 @@
1
1
  "use strict";
2
+ /**
3
+ * Module with the different Step classes and helper functions that can be used in a trace.
4
+ *
5
+ * This is the consolidated module for all step-related functionality in the Openlayer SDK,
6
+ * similar to the Python SDK's steps.py module.
7
+ */
2
8
  Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.ChatCompletionStep = exports.UserCallStep = exports.Step = void 0;
9
+ exports.FunctionCallStep = exports.RetrieverStep = exports.ToolStep = exports.AgentStep = exports.ChainStep = exports.ChatCompletionStep = exports.UserCallStep = exports.Step = exports.StepType = void 0;
10
+ exports.stepFactory = stepFactory;
4
11
  const uuid_1 = require("uuid");
5
- const enums_1 = require("./enums.js");
12
+ // ============================= ENUMS ============================= //
13
+ var StepType;
14
+ (function (StepType) {
15
+ StepType["USER_CALL"] = "user_call";
16
+ StepType["CHAT_COMPLETION"] = "chat_completion";
17
+ StepType["CHAIN"] = "chain";
18
+ StepType["AGENT"] = "agent";
19
+ StepType["TOOL"] = "tool";
20
+ StepType["RETRIEVER"] = "retriever";
21
+ StepType["FUNCTION_CALL"] = "function_call";
22
+ })(StepType || (exports.StepType = StepType = {}));
23
+ // ============================= STEP CLASSES ============================= //
24
+ /**
25
+ * Step, defined as a single function call being traced.
26
+ *
27
+ * This is the base class for all the different types of steps that can be
28
+ * used in a trace. Steps can also contain nested steps, which represent
29
+ * function calls made within the parent step.
30
+ */
6
31
  class Step {
7
32
  constructor(name, inputs = null, output = null, metadata = {}, startTime, endTime) {
8
33
  this.stepType = null;
@@ -18,9 +43,15 @@ class Step {
18
43
  this.startTime = startTime ?? Date.now();
19
44
  this.endTime = endTime ?? null;
20
45
  }
46
+ /**
47
+ * Adds a nested step to the current step.
48
+ */
21
49
  addNestedStep(nestedStep) {
22
50
  this.steps.push(nestedStep);
23
51
  }
52
+ /**
53
+ * Logs step data.
54
+ */
24
55
  log(data) {
25
56
  Object.keys(data).forEach((key) => {
26
57
  if (key in this) {
@@ -29,6 +60,9 @@ class Step {
29
60
  }
30
61
  });
31
62
  }
63
+ /**
64
+ * Dictionary representation of the Step.
65
+ */
32
66
  toJSON() {
33
67
  return {
34
68
  name: this.name,
@@ -46,13 +80,19 @@ class Step {
46
80
  }
47
81
  }
48
82
  exports.Step = Step;
83
+ /**
84
+ * User call step represents a generic user call in the trace.
85
+ */
49
86
  class UserCallStep extends Step {
50
87
  constructor(name, inputs = null, output = null, metadata = {}, startTime, endTime) {
51
88
  super(name, inputs, output, metadata, startTime, endTime);
52
- this.stepType = enums_1.StepType.USER_CALL;
89
+ this.stepType = StepType.USER_CALL;
53
90
  }
54
91
  }
55
92
  exports.UserCallStep = UserCallStep;
93
+ /**
94
+ * Chat completion step represents an LLM chat completion in the trace.
95
+ */
56
96
  class ChatCompletionStep extends Step {
57
97
  constructor(name, inputs = null, output = null, metadata = {}, startTime, endTime) {
58
98
  super(name, inputs, output, metadata, startTime, endTime);
@@ -63,7 +103,7 @@ class ChatCompletionStep extends Step {
63
103
  this.cost = null;
64
104
  this.model = null;
65
105
  this.modelParameters = null;
66
- this.stepType = enums_1.StepType.CHAT_COMPLETION;
106
+ this.stepType = StepType.CHAT_COMPLETION;
67
107
  }
68
108
  toJSON() {
69
109
  const stepData = super.toJSON();
@@ -80,4 +120,88 @@ class ChatCompletionStep extends Step {
80
120
  }
81
121
  }
82
122
  exports.ChatCompletionStep = ChatCompletionStep;
123
+ /**
124
+ * Chain step represents sequential operations or workflow chains in AI applications.
125
+ */
126
+ class ChainStep extends Step {
127
+ constructor(name, inputs = null, output = null, metadata = {}) {
128
+ super(name, inputs, output, metadata);
129
+ this.stepType = StepType.CHAIN;
130
+ }
131
+ }
132
+ exports.ChainStep = ChainStep;
133
+ /**
134
+ * Agent step tracks AI agent actions and decision-making processes.
135
+ */
136
+ class AgentStep extends Step {
137
+ constructor(name, inputs = null, output = null, metadata = {}) {
138
+ super(name, inputs, output, metadata);
139
+ this.tool = null;
140
+ this.action = null;
141
+ this.stepType = StepType.AGENT;
142
+ }
143
+ }
144
+ exports.AgentStep = AgentStep;
145
+ /**
146
+ * Tool step tracks individual tool executions within AI workflows.
147
+ */
148
+ class ToolStep extends Step {
149
+ constructor(name, inputs = null, output = null, metadata = {}) {
150
+ super(name, inputs, output, metadata);
151
+ this.stepType = StepType.TOOL;
152
+ }
153
+ }
154
+ exports.ToolStep = ToolStep;
155
+ /**
156
+ * Retriever step tracks document retrieval and search operations.
157
+ */
158
+ class RetrieverStep extends Step {
159
+ constructor(name, inputs = null, output = null, metadata = {}) {
160
+ super(name, inputs, output, metadata);
161
+ this.documents = [];
162
+ this.stepType = StepType.RETRIEVER;
163
+ }
164
+ }
165
+ exports.RetrieverStep = RetrieverStep;
166
+ /**
167
+ * Function call step tracks specific function calls with detailed execution metadata.
168
+ */
169
+ class FunctionCallStep extends Step {
170
+ constructor(name, inputs = null, output = null, metadata = {}) {
171
+ super(name, inputs, output, metadata);
172
+ this.functionName = null;
173
+ this.arguments = null;
174
+ this.result = null;
175
+ this.executionTime = null;
176
+ this.stepType = StepType.FUNCTION_CALL;
177
+ }
178
+ }
179
+ exports.FunctionCallStep = FunctionCallStep;
180
+ // ============================= FACTORY FUNCTION ============================= //
181
+ /**
182
+ * Factory function to create a step based on the step type.
183
+ */
184
+ function stepFactory(stepType, name, inputs = null, output = null, metadata = {}, startTime, endTime) {
185
+ if (!Object.values(StepType).includes(stepType)) {
186
+ throw new Error(`Step type ${stepType} not recognized.`);
187
+ }
188
+ switch (stepType) {
189
+ case StepType.USER_CALL:
190
+ return new UserCallStep(name, inputs, output, metadata, startTime, endTime);
191
+ case StepType.CHAT_COMPLETION:
192
+ return new ChatCompletionStep(name, inputs, output, metadata, startTime, endTime);
193
+ case StepType.CHAIN:
194
+ return new ChainStep(name, inputs, output, metadata);
195
+ case StepType.AGENT:
196
+ return new AgentStep(name, inputs, output, metadata);
197
+ case StepType.TOOL:
198
+ return new ToolStep(name, inputs, output, metadata);
199
+ case StepType.RETRIEVER:
200
+ return new RetrieverStep(name, inputs, output, metadata);
201
+ case StepType.FUNCTION_CALL:
202
+ return new FunctionCallStep(name, inputs, output, metadata);
203
+ default:
204
+ throw new Error(`Step type ${stepType} not recognized.`);
205
+ }
206
+ }
83
207
  //# sourceMappingURL=steps.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"steps.js","sourceRoot":"","sources":["../../src/lib/tracing/steps.ts"],"names":[],"mappings":";;;AAAA,+BAAoC;AACpC,sCAAmC;AA0BnC,MAAa,IAAI;IAaf,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE,EAAE,SAAyB,EAAE,OAAuB;QAPxJ,aAAQ,GAAoB,IAAI,CAAC;QAEjC,YAAO,GAAkB,IAAI,CAAC;QAC9B,gBAAW,GAAQ,IAAI,CAAC;QACxB,YAAO,GAAkB,IAAI,CAAC;QAC9B,UAAK,GAAW,EAAE,CAAC;QAGjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,EAAE,GAAG,IAAA,SAAM,GAAE,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAEzB,IAAI,CAAC,SAAS,GAAG,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;QACzC,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,IAAI,CAAC;IACjC,CAAC;IAED,aAAa,CAAC,UAAgB;QAC5B,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC9B,CAAC;IAED,GAAG,CAAC,IAAsC;QACxC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAChC,IAAI,GAAG,IAAI,IAAI,EAAE;gBACf,aAAa;gBACb,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;aACvB;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,MAAM;QACJ,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,QAAS;YACpB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YAC1D,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;IACJ,CAAC;CACF;AApDD,oBAoDC;AAED,MAAa,YAAa,SAAQ,IAAI;IACpC,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE,EAAE,SAAyB,EAAE,OAAuB;QACtJ,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QAC1D,IAAI,CAAC,QAAQ,GAAG,gBAAQ,CAAC,SAAS,CAAC;IACrC,CAAC;CACF;AALD,oCAKC;AAED,MAAa,kBAAmB,SAAQ,IAAI;IASxC,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE,EAAE,SAAyB,EAAE,OAAuB;QACxJ,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QAT5D,aAAQ,GAAkB,IAAI,CAAC;QAC/B,iBAAY,GAAkB,IAAI,CAAC;QACnC,qBAAgB,GAAkB,IAAI,CAAC;QACvC,WAAM,GAAkB,IAAI,CAAC;QAC7B,SAAI,GAAkB,IAAI,CAAC;QAC3B,UAAK,GAAkB,IAAI,CAAC;QAC5B,oBAAe,GAA+B,IAAI,CAAC;QAIjD,IAAI,CAAC,QAAQ,GAAG,gBAAQ,CAAC,eAAe,CAAC;IAC3C,CAAC;IAEQ,MAAM;QACb,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QAChC,OAAO;YACL,GAAG,QAAQ;YACX,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,eAAe,EAAE,IAAI,CAAC,eAAe;SACtC,CAAC;IACJ,CAAC;CACF;AA3BD,gDA2BC"}
1
+ {"version":3,"file":"steps.js","sourceRoot":"","sources":["../../src/lib/tracing/steps.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAiPH,kCA+BC;AA9QD,+BAAoC;AAEpC,uEAAuE;AAEvE,IAAY,QAQX;AARD,WAAY,QAAQ;IAClB,mCAAuB,CAAA;IACvB,+CAAmC,CAAA;IACnC,2BAAe,CAAA;IACf,2BAAe,CAAA;IACf,yBAAa,CAAA;IACb,mCAAuB,CAAA;IACvB,2CAA+B,CAAA;AACjC,CAAC,EARW,QAAQ,wBAAR,QAAQ,QAQnB;AA4BD,8EAA8E;AAE9E;;;;;;GAMG;AACH,MAAa,IAAI;IAaf,YACE,IAAY,EACZ,SAAc,IAAI,EAClB,SAAc,IAAI,EAClB,WAAgC,EAAE,EAClC,SAAyB,EACzB,OAAuB;QAbzB,aAAQ,GAAoB,IAAI,CAAC;QAEjC,YAAO,GAAkB,IAAI,CAAC;QAC9B,gBAAW,GAAQ,IAAI,CAAC;QACxB,YAAO,GAAkB,IAAI,CAAC;QAC9B,UAAK,GAAW,EAAE,CAAC;QAUjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,EAAE,GAAG,IAAA,SAAM,GAAE,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,SAAS,GAAG,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;QACzC,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,IAAI,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,UAAgB;QAC5B,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,GAAG,CAAC,IAAsC;QACxC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAChC,IAAI,GAAG,IAAI,IAAI,EAAE,CAAC;gBAChB,aAAa;gBACb,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;YACxB,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,QAAS;YACpB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YAC1D,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;IACJ,CAAC;CACF;AAnED,oBAmEC;AAED;;GAEG;AACH,MAAa,YAAa,SAAQ,IAAI;IACpC,YACE,IAAY,EACZ,SAAc,IAAI,EAClB,SAAc,IAAI,EAClB,WAAgC,EAAE,EAClC,SAAyB,EACzB,OAAuB;QAEvB,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QAC1D,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,SAAS,CAAC;IACrC,CAAC;CACF;AAZD,oCAYC;AAED;;GAEG;AACH,MAAa,kBAAmB,SAAQ,IAAI;IAS1C,YACE,IAAY,EACZ,SAAc,IAAI,EAClB,SAAc,IAAI,EAClB,WAAgC,EAAE,EAClC,SAAyB,EACzB,OAAuB;QAEvB,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QAhB5D,aAAQ,GAAkB,IAAI,CAAC;QAC/B,iBAAY,GAAkB,IAAI,CAAC;QACnC,qBAAgB,GAAkB,IAAI,CAAC;QACvC,WAAM,GAAkB,IAAI,CAAC;QAC7B,SAAI,GAAkB,IAAI,CAAC;QAC3B,UAAK,GAAkB,IAAI,CAAC;QAC5B,oBAAe,GAA+B,IAAI,CAAC;QAWjD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,eAAe,CAAC;IAC3C,CAAC;IAEQ,MAAM;QACb,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QAChC,OAAO;YACL,GAAG,QAAQ;YACX,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,eAAe,EAAE,IAAI,CAAC,eAAe;SACtC,CAAC;IACJ,CAAC;CACF;AAlCD,gDAkCC;AAED;;GAEG;AACH,MAAa,SAAU,SAAQ,IAAI;IACjC,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE;QAClG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC;IACjC,CAAC;CACF;AALD,8BAKC;AAED;;GAEG;AACH,MAAa,SAAU,SAAQ,IAAI;IAIjC,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE;QAClG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QAJxC,SAAI,GAAkB,IAAI,CAAC;QAC3B,WAAM,GAAQ,IAAI,CAAC;QAIjB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC;IACjC,CAAC;CACF;AARD,8BAQC;AAED;;GAEG;AACH,MAAa,QAAS,SAAQ,IAAI;IAChC,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE;QAClG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC;IAChC,CAAC;CACF;AALD,4BAKC;AAED;;GAEG;AACH,MAAa,aAAc,SAAQ,IAAI;IAGrC,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE;QAClG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QAHxC,cAAS,GAAU,EAAE,CAAC;QAIpB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,SAAS,CAAC;IACrC,CAAC;CACF;AAPD,sCAOC;AAED;;GAEG;AACH,MAAa,gBAAiB,SAAQ,IAAI;IAMxC,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE;QAClG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QANxC,iBAAY,GAAkB,IAAI,CAAC;QACnC,cAAS,GAA+B,IAAI,CAAC;QAC7C,WAAM,GAAQ,IAAI,CAAC;QACnB,kBAAa,GAAkB,IAAI,CAAC;QAIlC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,aAAa,CAAC;IACzC,CAAC;CACF;AAVD,4CAUC;AAED,kFAAkF;AAElF;;GAEG;AACH,SAAgB,WAAW,CACzB,QAAkB,EAClB,IAAY,EACZ,SAAc,IAAI,EAClB,SAAc,IAAI,EAClB,WAAgC,EAAE,EAClC,SAAyB,EACzB,OAAuB;IAEvB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAChD,MAAM,IAAI,KAAK,CAAC,aAAa,QAAQ,kBAAkB,CAAC,CAAC;IAC3D,CAAC;IAED,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,QAAQ,CAAC,SAAS;YACrB,OAAO,IAAI,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QAC9E,KAAK,QAAQ,CAAC,eAAe;YAC3B,OAAO,IAAI,kBAAkB,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QACpF,KAAK,QAAQ,CAAC,KAAK;YACjB,OAAO,IAAI,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACvD,KAAK,QAAQ,CAAC,KAAK;YACjB,OAAO,IAAI,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACvD,KAAK,QAAQ,CAAC,IAAI;YAChB,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACtD,KAAK,QAAQ,CAAC,SAAS;YACrB,OAAO,IAAI,aAAa,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC3D,KAAK,QAAQ,CAAC,aAAa;YACzB,OAAO,IAAI,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC9D;YACE,MAAM,IAAI,KAAK,CAAC,aAAa,QAAQ,kBAAkB,CAAC,CAAC;IAC7D,CAAC;AACH,CAAC"}
@@ -1,5 +1,29 @@
1
+ /**
2
+ * Module with the different Step classes and helper functions that can be used in a trace.
3
+ *
4
+ * This is the consolidated module for all step-related functionality in the Openlayer SDK,
5
+ * similar to the Python SDK's steps.py module.
6
+ */
1
7
  import { v4 as uuidv4 } from 'uuid';
2
- import { StepType } from "./enums.mjs";
8
+ // ============================= ENUMS ============================= //
9
+ export var StepType;
10
+ (function (StepType) {
11
+ StepType["USER_CALL"] = "user_call";
12
+ StepType["CHAT_COMPLETION"] = "chat_completion";
13
+ StepType["CHAIN"] = "chain";
14
+ StepType["AGENT"] = "agent";
15
+ StepType["TOOL"] = "tool";
16
+ StepType["RETRIEVER"] = "retriever";
17
+ StepType["FUNCTION_CALL"] = "function_call";
18
+ })(StepType || (StepType = {}));
19
+ // ============================= STEP CLASSES ============================= //
20
+ /**
21
+ * Step, defined as a single function call being traced.
22
+ *
23
+ * This is the base class for all the different types of steps that can be
24
+ * used in a trace. Steps can also contain nested steps, which represent
25
+ * function calls made within the parent step.
26
+ */
3
27
  export class Step {
4
28
  constructor(name, inputs = null, output = null, metadata = {}, startTime, endTime) {
5
29
  this.stepType = null;
@@ -15,9 +39,15 @@ export class Step {
15
39
  this.startTime = startTime ?? Date.now();
16
40
  this.endTime = endTime ?? null;
17
41
  }
42
+ /**
43
+ * Adds a nested step to the current step.
44
+ */
18
45
  addNestedStep(nestedStep) {
19
46
  this.steps.push(nestedStep);
20
47
  }
48
+ /**
49
+ * Logs step data.
50
+ */
21
51
  log(data) {
22
52
  Object.keys(data).forEach((key) => {
23
53
  if (key in this) {
@@ -26,6 +56,9 @@ export class Step {
26
56
  }
27
57
  });
28
58
  }
59
+ /**
60
+ * Dictionary representation of the Step.
61
+ */
29
62
  toJSON() {
30
63
  return {
31
64
  name: this.name,
@@ -42,12 +75,18 @@ export class Step {
42
75
  };
43
76
  }
44
77
  }
78
+ /**
79
+ * User call step represents a generic user call in the trace.
80
+ */
45
81
  export class UserCallStep extends Step {
46
82
  constructor(name, inputs = null, output = null, metadata = {}, startTime, endTime) {
47
83
  super(name, inputs, output, metadata, startTime, endTime);
48
84
  this.stepType = StepType.USER_CALL;
49
85
  }
50
86
  }
87
+ /**
88
+ * Chat completion step represents an LLM chat completion in the trace.
89
+ */
51
90
  export class ChatCompletionStep extends Step {
52
91
  constructor(name, inputs = null, output = null, metadata = {}, startTime, endTime) {
53
92
  super(name, inputs, output, metadata, startTime, endTime);
@@ -74,4 +113,83 @@ export class ChatCompletionStep extends Step {
74
113
  };
75
114
  }
76
115
  }
116
+ /**
117
+ * Chain step represents sequential operations or workflow chains in AI applications.
118
+ */
119
+ export class ChainStep extends Step {
120
+ constructor(name, inputs = null, output = null, metadata = {}) {
121
+ super(name, inputs, output, metadata);
122
+ this.stepType = StepType.CHAIN;
123
+ }
124
+ }
125
+ /**
126
+ * Agent step tracks AI agent actions and decision-making processes.
127
+ */
128
+ export class AgentStep extends Step {
129
+ constructor(name, inputs = null, output = null, metadata = {}) {
130
+ super(name, inputs, output, metadata);
131
+ this.tool = null;
132
+ this.action = null;
133
+ this.stepType = StepType.AGENT;
134
+ }
135
+ }
136
+ /**
137
+ * Tool step tracks individual tool executions within AI workflows.
138
+ */
139
+ export class ToolStep extends Step {
140
+ constructor(name, inputs = null, output = null, metadata = {}) {
141
+ super(name, inputs, output, metadata);
142
+ this.stepType = StepType.TOOL;
143
+ }
144
+ }
145
+ /**
146
+ * Retriever step tracks document retrieval and search operations.
147
+ */
148
+ export class RetrieverStep extends Step {
149
+ constructor(name, inputs = null, output = null, metadata = {}) {
150
+ super(name, inputs, output, metadata);
151
+ this.documents = [];
152
+ this.stepType = StepType.RETRIEVER;
153
+ }
154
+ }
155
+ /**
156
+ * Function call step tracks specific function calls with detailed execution metadata.
157
+ */
158
+ export class FunctionCallStep extends Step {
159
+ constructor(name, inputs = null, output = null, metadata = {}) {
160
+ super(name, inputs, output, metadata);
161
+ this.functionName = null;
162
+ this.arguments = null;
163
+ this.result = null;
164
+ this.executionTime = null;
165
+ this.stepType = StepType.FUNCTION_CALL;
166
+ }
167
+ }
168
+ // ============================= FACTORY FUNCTION ============================= //
169
+ /**
170
+ * Factory function to create a step based on the step type.
171
+ */
172
+ export function stepFactory(stepType, name, inputs = null, output = null, metadata = {}, startTime, endTime) {
173
+ if (!Object.values(StepType).includes(stepType)) {
174
+ throw new Error(`Step type ${stepType} not recognized.`);
175
+ }
176
+ switch (stepType) {
177
+ case StepType.USER_CALL:
178
+ return new UserCallStep(name, inputs, output, metadata, startTime, endTime);
179
+ case StepType.CHAT_COMPLETION:
180
+ return new ChatCompletionStep(name, inputs, output, metadata, startTime, endTime);
181
+ case StepType.CHAIN:
182
+ return new ChainStep(name, inputs, output, metadata);
183
+ case StepType.AGENT:
184
+ return new AgentStep(name, inputs, output, metadata);
185
+ case StepType.TOOL:
186
+ return new ToolStep(name, inputs, output, metadata);
187
+ case StepType.RETRIEVER:
188
+ return new RetrieverStep(name, inputs, output, metadata);
189
+ case StepType.FUNCTION_CALL:
190
+ return new FunctionCallStep(name, inputs, output, metadata);
191
+ default:
192
+ throw new Error(`Step type ${stepType} not recognized.`);
193
+ }
194
+ }
77
195
  //# sourceMappingURL=steps.mjs.map
@@ -1 +1 @@
1
- {"version":3,"file":"steps.mjs","sourceRoot":"","sources":["../../src/lib/tracing/steps.ts"],"names":[],"mappings":"OAAO,EAAE,EAAE,IAAI,MAAM,EAAE,MAAM,MAAM;OAC5B,EAAE,QAAQ,EAAE;AA0BnB,MAAM,OAAO,IAAI;IAaf,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE,EAAE,SAAyB,EAAE,OAAuB;QAPxJ,aAAQ,GAAoB,IAAI,CAAC;QAEjC,YAAO,GAAkB,IAAI,CAAC;QAC9B,gBAAW,GAAQ,IAAI,CAAC;QACxB,YAAO,GAAkB,IAAI,CAAC;QAC9B,UAAK,GAAW,EAAE,CAAC;QAGjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,EAAE,GAAG,MAAM,EAAE,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAEzB,IAAI,CAAC,SAAS,GAAG,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;QACzC,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,IAAI,CAAC;IACjC,CAAC;IAED,aAAa,CAAC,UAAgB;QAC5B,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC9B,CAAC;IAED,GAAG,CAAC,IAAsC;QACxC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAChC,IAAI,GAAG,IAAI,IAAI,EAAE;gBACf,aAAa;gBACb,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;aACvB;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,MAAM;QACJ,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,QAAS;YACpB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YAC1D,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;IACJ,CAAC;CACF;AAED,MAAM,OAAO,YAAa,SAAQ,IAAI;IACpC,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE,EAAE,SAAyB,EAAE,OAAuB;QACtJ,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QAC1D,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,SAAS,CAAC;IACrC,CAAC;CACF;AAED,MAAM,OAAO,kBAAmB,SAAQ,IAAI;IASxC,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE,EAAE,SAAyB,EAAE,OAAuB;QACxJ,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QAT5D,aAAQ,GAAkB,IAAI,CAAC;QAC/B,iBAAY,GAAkB,IAAI,CAAC;QACnC,qBAAgB,GAAkB,IAAI,CAAC;QACvC,WAAM,GAAkB,IAAI,CAAC;QAC7B,SAAI,GAAkB,IAAI,CAAC;QAC3B,UAAK,GAAkB,IAAI,CAAC;QAC5B,oBAAe,GAA+B,IAAI,CAAC;QAIjD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,eAAe,CAAC;IAC3C,CAAC;IAEQ,MAAM;QACb,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QAChC,OAAO;YACL,GAAG,QAAQ;YACX,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,eAAe,EAAE,IAAI,CAAC,eAAe;SACtC,CAAC;IACJ,CAAC;CACF"}
1
+ {"version":3,"file":"steps.mjs","sourceRoot":"","sources":["../../src/lib/tracing/steps.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;OAEI,EAAE,EAAE,IAAI,MAAM,EAAE,MAAM,MAAM;AAEnC,uEAAuE;AAEvE,MAAM,CAAN,IAAY,QAQX;AARD,WAAY,QAAQ;IAClB,mCAAuB,CAAA;IACvB,+CAAmC,CAAA;IACnC,2BAAe,CAAA;IACf,2BAAe,CAAA;IACf,yBAAa,CAAA;IACb,mCAAuB,CAAA;IACvB,2CAA+B,CAAA;AACjC,CAAC,EARW,QAAQ,KAAR,QAAQ,QAQnB;AA4BD,8EAA8E;AAE9E;;;;;;GAMG;AACH,MAAM,OAAO,IAAI;IAaf,YACE,IAAY,EACZ,SAAc,IAAI,EAClB,SAAc,IAAI,EAClB,WAAgC,EAAE,EAClC,SAAyB,EACzB,OAAuB;QAbzB,aAAQ,GAAoB,IAAI,CAAC;QAEjC,YAAO,GAAkB,IAAI,CAAC;QAC9B,gBAAW,GAAQ,IAAI,CAAC;QACxB,YAAO,GAAkB,IAAI,CAAC;QAC9B,UAAK,GAAW,EAAE,CAAC;QAUjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,EAAE,GAAG,MAAM,EAAE,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,SAAS,GAAG,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;QACzC,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,IAAI,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,UAAgB;QAC5B,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,GAAG,CAAC,IAAsC;QACxC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAChC,IAAI,GAAG,IAAI,IAAI,EAAE,CAAC;gBAChB,aAAa;gBACb,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;YACxB,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,QAAS;YACpB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YAC1D,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;IACJ,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,YAAa,SAAQ,IAAI;IACpC,YACE,IAAY,EACZ,SAAc,IAAI,EAClB,SAAc,IAAI,EAClB,WAAgC,EAAE,EAClC,SAAyB,EACzB,OAAuB;QAEvB,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QAC1D,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,SAAS,CAAC;IACrC,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,kBAAmB,SAAQ,IAAI;IAS1C,YACE,IAAY,EACZ,SAAc,IAAI,EAClB,SAAc,IAAI,EAClB,WAAgC,EAAE,EAClC,SAAyB,EACzB,OAAuB;QAEvB,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QAhB5D,aAAQ,GAAkB,IAAI,CAAC;QAC/B,iBAAY,GAAkB,IAAI,CAAC;QACnC,qBAAgB,GAAkB,IAAI,CAAC;QACvC,WAAM,GAAkB,IAAI,CAAC;QAC7B,SAAI,GAAkB,IAAI,CAAC;QAC3B,UAAK,GAAkB,IAAI,CAAC;QAC5B,oBAAe,GAA+B,IAAI,CAAC;QAWjD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,eAAe,CAAC;IAC3C,CAAC;IAEQ,MAAM;QACb,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QAChC,OAAO;YACL,GAAG,QAAQ;YACX,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,eAAe,EAAE,IAAI,CAAC,eAAe;SACtC,CAAC;IACJ,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,SAAU,SAAQ,IAAI;IACjC,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE;QAClG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC;IACjC,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,SAAU,SAAQ,IAAI;IAIjC,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE;QAClG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QAJxC,SAAI,GAAkB,IAAI,CAAC;QAC3B,WAAM,GAAQ,IAAI,CAAC;QAIjB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC;IACjC,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,QAAS,SAAQ,IAAI;IAChC,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE;QAClG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC;IAChC,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,aAAc,SAAQ,IAAI;IAGrC,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE;QAClG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QAHxC,cAAS,GAAU,EAAE,CAAC;QAIpB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,SAAS,CAAC;IACrC,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,gBAAiB,SAAQ,IAAI;IAMxC,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE;QAClG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QANxC,iBAAY,GAAkB,IAAI,CAAC;QACnC,cAAS,GAA+B,IAAI,CAAC;QAC7C,WAAM,GAAQ,IAAI,CAAC;QACnB,kBAAa,GAAkB,IAAI,CAAC;QAIlC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,aAAa,CAAC;IACzC,CAAC;CACF;AAED,kFAAkF;AAElF;;GAEG;AACH,MAAM,UAAU,WAAW,CACzB,QAAkB,EAClB,IAAY,EACZ,SAAc,IAAI,EAClB,SAAc,IAAI,EAClB,WAAgC,EAAE,EAClC,SAAyB,EACzB,OAAuB;IAEvB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAChD,MAAM,IAAI,KAAK,CAAC,aAAa,QAAQ,kBAAkB,CAAC,CAAC;IAC3D,CAAC;IAED,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,QAAQ,CAAC,SAAS;YACrB,OAAO,IAAI,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QAC9E,KAAK,QAAQ,CAAC,eAAe;YAC3B,OAAO,IAAI,kBAAkB,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QACpF,KAAK,QAAQ,CAAC,KAAK;YACjB,OAAO,IAAI,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACvD,KAAK,QAAQ,CAAC,KAAK;YACjB,OAAO,IAAI,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACvD,KAAK,QAAQ,CAAC,IAAI;YAChB,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACtD,KAAK,QAAQ,CAAC,SAAS;YACrB,OAAO,IAAI,aAAa,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC3D,KAAK,QAAQ,CAAC,aAAa;YACzB,OAAO,IAAI,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC9D;YACE,MAAM,IAAI,KAAK,CAAC,aAAa,QAAQ,kBAAkB,CAAC,CAAC;IAC7D,CAAC;AACH,CAAC"}
@@ -1,5 +1,5 @@
1
- import { StepType } from "./enums.js";
2
1
  import { Trace } from "./traces.js";
2
+ import { Step, StepType } from "./steps.js";
3
3
  export declare function getCurrentTrace(): Trace | null;
4
4
  declare function trace(fn: Function, stepType?: StepType, stepName?: string, openlayerInferencePipelineId?: string): Function;
5
5
  export declare function addChatCompletionStepToTrace({ name, inputs, output, latency, tokens, promptTokens, completionTokens, cost, model, modelParameters, metadata, provider, startTime, endTime, }: {
@@ -18,6 +18,74 @@ export declare function addChatCompletionStepToTrace({ name, inputs, output, lat
18
18
  startTime?: number | null;
19
19
  endTime?: number | null;
20
20
  }, openlayerInferencePipelineId?: string): void;
21
+ export declare function addChainStepToTrace(params: {
22
+ name: string;
23
+ inputs: any;
24
+ output?: any;
25
+ metadata?: Record<string, any>;
26
+ }): {
27
+ step: Step;
28
+ endStep: () => void;
29
+ };
30
+ export declare function addAgentStepToTrace(params: {
31
+ name: string;
32
+ inputs: any;
33
+ output?: any;
34
+ metadata?: Record<string, any>;
35
+ tool?: string;
36
+ action?: any;
37
+ }): {
38
+ step: Step;
39
+ endStep: () => void;
40
+ };
41
+ export declare function addToolStepToTrace(params: {
42
+ name: string;
43
+ inputs: any;
44
+ output?: any;
45
+ metadata?: Record<string, any>;
46
+ }): {
47
+ step: Step;
48
+ endStep: () => void;
49
+ };
50
+ export declare function addRetrieverStepToTrace(params: {
51
+ name: string;
52
+ inputs: any;
53
+ output?: any;
54
+ metadata?: Record<string, any>;
55
+ documents?: any[];
56
+ }): {
57
+ step: Step;
58
+ endStep: () => void;
59
+ };
60
+ export declare function addFunctionCallStepToTrace(params: {
61
+ name: string;
62
+ functionName: string;
63
+ arguments: Record<string, any>;
64
+ output?: any;
65
+ metadata?: Record<string, any>;
66
+ }): {
67
+ step: Step;
68
+ endStep: () => void;
69
+ };
70
+ export declare function startAgentStep(params: {
71
+ name: string;
72
+ agentType: string;
73
+ inputs: any;
74
+ metadata?: Record<string, any>;
75
+ }): {
76
+ step: Step;
77
+ endStep: () => void;
78
+ };
79
+ export declare function addHandoffStepToTrace(params: {
80
+ name: string;
81
+ fromComponent: string;
82
+ toComponent: string;
83
+ handoffData: any;
84
+ metadata?: Record<string, any>;
85
+ }): {
86
+ step: Step;
87
+ endStep: () => void;
88
+ };
21
89
  export declare function postProcessTrace(traceObj: Trace): {
22
90
  traceData: any;
23
91
  inputVariableNames: string[];
@@ -1 +1 @@
1
- {"version":3,"file":"tracer.d.ts","sourceRoot":"","sources":["../../src/lib/tracing/tracer.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AAEnC,OAAO,EAAE,KAAK,EAAE,MAAM,UAAU,CAAC;AAWjC,wBAAgB,eAAe,IAAI,KAAK,GAAG,IAAI,CAE9C;AA2HD,iBAAS,KAAK,CACZ,EAAE,EAAE,QAAQ,EACZ,QAAQ,GAAE,QAA6B,EACvC,QAAQ,CAAC,EAAE,MAAM,EACjB,4BAA4B,CAAC,EAAE,MAAM,GACpC,QAAQ,CAkBV;AAED,wBAAgB,4BAA4B,CAC1C,EACE,IAAI,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,MAAa,EACb,YAAmB,EACnB,gBAAuB,EACvB,IAAW,EACX,KAAY,EACZ,eAAsB,EACtB,QAAa,EACb,QAAmB,EACnB,SAAgB,EAChB,OAAc,GACf,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,GAAG,CAAC;IACZ,MAAM,EAAE,GAAG,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,gBAAgB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACjC,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,CAAC;IAC7C,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC/B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACzB,EACD,4BAA4B,CAAC,EAAE,MAAM,QA2BtC;AAED,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,KAAK,GAAG;IAAE,SAAS,EAAE,GAAG,CAAC;IAAC,kBAAkB,EAAE,MAAM,EAAE,CAAA;CAAE,CAyBlG;AAED,eAAe,KAAK,CAAC"}
1
+ {"version":3,"file":"tracer.d.ts","sourceRoot":"","sources":["../../src/lib/tracing/tracer.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,EAAE,MAAM,UAAU,CAAC;AACjC,OAAO,EACL,IAAI,EAKJ,QAAQ,EAET,MAAM,SAAS,CAAC;AAYjB,wBAAgB,eAAe,IAAI,KAAK,GAAG,IAAI,CAE9C;AAkID,iBAAS,KAAK,CACZ,EAAE,EAAE,QAAQ,EACZ,QAAQ,GAAE,QAA6B,EACvC,QAAQ,CAAC,EAAE,MAAM,EACjB,4BAA4B,CAAC,EAAE,MAAM,GACpC,QAAQ,CAkBV;AAED,wBAAgB,4BAA4B,CAC1C,EACE,IAAI,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,MAAa,EACb,YAAmB,EACnB,gBAAuB,EACvB,IAAW,EACX,KAAY,EACZ,eAAsB,EACtB,QAAa,EACb,QAAmB,EACnB,SAAgB,EAChB,OAAc,GACf,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,GAAG,CAAC;IACZ,MAAM,EAAE,GAAG,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,gBAAgB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACjC,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,CAAC;IAC7C,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC/B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACzB,EACD,4BAA4B,CAAC,EAAE,MAAM,QA2BtC;AAED,wBAAgB,mBAAmB,CAAC,MAAM,EAAE;IAC1C,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,GAAG,CAAC;IACZ,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC;;mBA3Me,IAAI;EAgNnB;AAED,wBAAgB,mBAAmB,CAAC,MAAM,EAAE;IAC1C,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,GAAG,CAAC;IACZ,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC/B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,GAAG,CAAC;CACd;;mBAzNe,IAAI;EAkOnB;AAED,wBAAgB,kBAAkB,CAAC,MAAM,EAAE;IACzC,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,GAAG,CAAC;IACZ,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC;;mBAzOe,IAAI;EAkPnB;AAED,wBAAgB,uBAAuB,CAAC,MAAM,EAAE;IAC9C,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,GAAG,CAAC;IACZ,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC/B,SAAS,CAAC,EAAE,GAAG,EAAE,CAAC;CACnB;;mBA1Pe,IAAI;EAsQnB;AAED,wBAAgB,0BAA0B,CAAC,MAAM,EAAE;IACjD,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC/B,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC;;mBA9Qe,IAAI;EA8RnB;AAGD,wBAAgB,cAAc,CAAC,MAAM,EAAE;IACrC,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,GAAG,CAAC;IACZ,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC;;mBAtSe,IAAI;EAmTnB;AAGD,wBAAgB,qBAAqB,CAAC,MAAM,EAAE;IAC5C,IAAI,EAAE,MAAM,CAAC;IACb,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,GAAG,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC;;mBA5Te,IAAI;EAsUnB;AAED,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,KAAK,GAAG;IAAE,SAAS,EAAE,GAAG,CAAC;IAAC,kBAAkB,EAAE,MAAM,EAAE,CAAA;CAAE,CAyBlG;AAED,eAAe,KAAK,CAAC"}