@orq-ai/node 3.12.14 → 3.12.17

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 (167) hide show
  1. package/README.md +1 -4
  2. package/bin/mcp-server.js +113 -113
  3. package/bin/mcp-server.js.map +31 -31
  4. package/examples/package-lock.json +1 -1
  5. package/jsr.json +1 -1
  6. package/lib/config.d.ts +3 -3
  7. package/lib/config.js +3 -3
  8. package/lib/url.js +1 -1
  9. package/lib/url.js.map +1 -1
  10. package/mcp-server/mcp-server.js +1 -1
  11. package/mcp-server/server.js +1 -1
  12. package/models/operations/createbudget.js +2 -2
  13. package/models/operations/createcontact.js +2 -2
  14. package/models/operations/createdataset.js +2 -2
  15. package/models/operations/createdatasetitem.js +2 -2
  16. package/models/operations/createdatasource.js +2 -2
  17. package/models/operations/createeval.js +16 -16
  18. package/models/operations/fileget.js +2 -2
  19. package/models/operations/filelist.js +2 -2
  20. package/models/operations/fileupload.js +2 -2
  21. package/models/operations/getbudget.js +2 -2
  22. package/models/operations/getevals.js +28 -28
  23. package/models/operations/listbudgets.js +2 -2
  24. package/models/operations/listcontacts.js +2 -2
  25. package/models/operations/listdatasetdatapoints.js +2 -2
  26. package/models/operations/listdatasets.js +2 -2
  27. package/models/operations/listdatasources.js +2 -2
  28. package/models/operations/retrievecontact.js +2 -2
  29. package/models/operations/retrievedatapoint.js +2 -2
  30. package/models/operations/retrievedataset.js +2 -2
  31. package/models/operations/retrievedatasource.js +2 -2
  32. package/models/operations/updatebudget.js +2 -2
  33. package/models/operations/updatecontact.js +2 -2
  34. package/models/operations/updatedatapoint.js +2 -2
  35. package/models/operations/updatedataset.js +2 -2
  36. package/models/operations/updatedatasource.js +2 -2
  37. package/models/operations/updateeval.js +16 -16
  38. package/package.json +1 -1
  39. package/packages/orq-rc/README.md +108 -94
  40. package/packages/orq-rc/docs/sdks/agents/README.md +8 -2
  41. package/packages/orq-rc/docs/sdks/evals/README.md +2 -0
  42. package/packages/orq-rc/docs/sdks/tools/README.md +488 -0
  43. package/packages/orq-rc/examples/package-lock.json +1 -1
  44. package/packages/orq-rc/jsr.json +1 -1
  45. package/packages/orq-rc/package-lock.json +2 -2
  46. package/packages/orq-rc/package.json +1 -1
  47. package/packages/orq-rc/src/funcs/toolsCreate.ts +163 -0
  48. package/packages/orq-rc/src/funcs/toolsDelete.ts +167 -0
  49. package/packages/orq-rc/src/funcs/toolsDuplicate.ts +175 -0
  50. package/packages/orq-rc/src/funcs/toolsList.ts +176 -0
  51. package/packages/orq-rc/src/funcs/toolsRetrieve.ts +166 -0
  52. package/packages/orq-rc/src/funcs/toolsUpdate.ts +176 -0
  53. package/packages/orq-rc/src/lib/config.ts +3 -3
  54. package/packages/orq-rc/src/lib/url.ts +1 -1
  55. package/packages/orq-rc/src/mcp-server/mcp-server.ts +1 -1
  56. package/packages/orq-rc/src/mcp-server/server.ts +13 -1
  57. package/packages/orq-rc/src/mcp-server/tools/toolsCreate.ts +37 -0
  58. package/packages/orq-rc/src/mcp-server/tools/toolsDelete.ts +35 -0
  59. package/packages/orq-rc/src/mcp-server/tools/toolsDuplicate.ts +37 -0
  60. package/packages/orq-rc/src/mcp-server/tools/toolsList.ts +37 -0
  61. package/packages/orq-rc/src/mcp-server/tools/toolsRetrieve.ts +37 -0
  62. package/packages/orq-rc/src/mcp-server/tools/toolsUpdate.ts +37 -0
  63. package/packages/orq-rc/src/models/components/callsubagenttool.ts +109 -0
  64. package/packages/orq-rc/src/models/components/codeexecutiontool.ts +293 -0
  65. package/packages/orq-rc/src/models/components/currentdatetool.ts +107 -0
  66. package/packages/orq-rc/src/models/components/deployments.ts +52 -45
  67. package/packages/orq-rc/src/models/components/functiontool.ts +195 -0
  68. package/packages/orq-rc/src/models/components/googlesearchtool.ts +108 -0
  69. package/packages/orq-rc/src/models/components/httptool.ts +532 -0
  70. package/packages/orq-rc/src/models/components/index.ts +13 -0
  71. package/packages/orq-rc/src/models/components/queryknowledgebasetool.ts +111 -0
  72. package/packages/orq-rc/src/models/components/querymemorystoretool.ts +111 -0
  73. package/packages/orq-rc/src/models/components/retrieveagentstool.ts +109 -0
  74. package/packages/orq-rc/src/models/components/retrieveknowledgebasestool.ts +111 -0
  75. package/packages/orq-rc/src/models/components/retrievememorystorestool.ts +111 -0
  76. package/packages/orq-rc/src/models/components/webscrapertool.ts +105 -0
  77. package/packages/orq-rc/src/models/components/writememorystoretool.ts +111 -0
  78. package/packages/orq-rc/src/models/errors/duplicatetool.ts +91 -0
  79. package/packages/orq-rc/src/models/errors/index.ts +2 -0
  80. package/packages/orq-rc/src/models/errors/updatetool.ts +91 -0
  81. package/packages/orq-rc/src/models/operations/createbudget.ts +2 -2
  82. package/packages/orq-rc/src/models/operations/createcontact.ts +2 -2
  83. package/packages/orq-rc/src/models/operations/createdataset.ts +2 -2
  84. package/packages/orq-rc/src/models/operations/createdatasetitem.ts +10 -8
  85. package/packages/orq-rc/src/models/operations/createdatasource.ts +2 -2
  86. package/packages/orq-rc/src/models/operations/createeval.ts +199 -56
  87. package/packages/orq-rc/src/models/operations/createknowledge.ts +18 -0
  88. package/packages/orq-rc/src/models/operations/creatememorystore.ts +137 -136
  89. package/packages/orq-rc/src/models/operations/createprompt.ts +633 -314
  90. package/packages/orq-rc/src/models/operations/createtool.ts +4027 -0
  91. package/packages/orq-rc/src/models/operations/deletetool.ts +75 -0
  92. package/packages/orq-rc/src/models/operations/deploymentcreatemetric.ts +8 -6
  93. package/packages/orq-rc/src/models/operations/deploymentgetconfig.ts +363 -194
  94. package/packages/orq-rc/src/models/operations/deployments.ts +330 -160
  95. package/packages/orq-rc/src/models/operations/deploymentstream.ts +8 -6
  96. package/packages/orq-rc/src/models/operations/duplicatetool.ts +2311 -0
  97. package/packages/orq-rc/src/models/operations/fileget.ts +2 -2
  98. package/packages/orq-rc/src/models/operations/filelist.ts +2 -2
  99. package/packages/orq-rc/src/models/operations/fileupload.ts +2 -2
  100. package/packages/orq-rc/src/models/operations/getagent.ts +10 -10
  101. package/packages/orq-rc/src/models/operations/getallprompts.ts +334 -168
  102. package/packages/orq-rc/src/models/operations/getalltools.ts +2278 -0
  103. package/packages/orq-rc/src/models/operations/getbudget.ts +2 -2
  104. package/packages/orq-rc/src/models/operations/getevals.ts +28 -28
  105. package/packages/orq-rc/src/models/operations/getoneknowledge.ts +9 -0
  106. package/packages/orq-rc/src/models/operations/getoneprompt.ts +330 -167
  107. package/packages/orq-rc/src/models/operations/getpromptversion.ts +338 -168
  108. package/packages/orq-rc/src/models/operations/index.ts +6 -0
  109. package/packages/orq-rc/src/models/operations/listagents.ts +10 -10
  110. package/packages/orq-rc/src/models/operations/listbudgets.ts +2 -2
  111. package/packages/orq-rc/src/models/operations/listcontacts.ts +2 -2
  112. package/packages/orq-rc/src/models/operations/listdatasetdatapoints.ts +6 -5
  113. package/packages/orq-rc/src/models/operations/listdatasets.ts +2 -2
  114. package/packages/orq-rc/src/models/operations/listdatasources.ts +2 -2
  115. package/packages/orq-rc/src/models/operations/listknowledgebases.ts +9 -0
  116. package/packages/orq-rc/src/models/operations/listpromptversions.ts +355 -183
  117. package/packages/orq-rc/src/models/operations/parse.ts +20 -335
  118. package/packages/orq-rc/src/models/operations/retrievecontact.ts +2 -2
  119. package/packages/orq-rc/src/models/operations/retrievedatapoint.ts +6 -5
  120. package/packages/orq-rc/src/models/operations/retrievedataset.ts +2 -2
  121. package/packages/orq-rc/src/models/operations/retrievedatasource.ts +2 -2
  122. package/packages/orq-rc/src/models/operations/retrievetool.ts +2303 -0
  123. package/packages/orq-rc/src/models/operations/runagent.ts +428 -1384
  124. package/packages/orq-rc/src/models/operations/searchknowledge.ts +2 -2
  125. package/packages/orq-rc/src/models/operations/streamrunagent.ts +409 -1452
  126. package/packages/orq-rc/src/models/operations/updatebudget.ts +2 -2
  127. package/packages/orq-rc/src/models/operations/updatecontact.ts +2 -2
  128. package/packages/orq-rc/src/models/operations/updatedatapoint.ts +10 -8
  129. package/packages/orq-rc/src/models/operations/updatedataset.ts +2 -2
  130. package/packages/orq-rc/src/models/operations/updatedatasource.ts +2 -2
  131. package/packages/orq-rc/src/models/operations/updateeval.ts +210 -57
  132. package/packages/orq-rc/src/models/operations/updateknowledge.ts +18 -0
  133. package/packages/orq-rc/src/models/operations/updatememorystore.ts +136 -168
  134. package/packages/orq-rc/src/models/operations/updateprompt.ts +670 -335
  135. package/packages/orq-rc/src/models/operations/updatetool.ts +4231 -0
  136. package/packages/orq-rc/src/sdk/sdk.ts +6 -0
  137. package/packages/orq-rc/src/sdk/tools.ts +117 -0
  138. package/src/lib/config.ts +3 -3
  139. package/src/lib/url.ts +1 -1
  140. package/src/mcp-server/mcp-server.ts +1 -1
  141. package/src/mcp-server/server.ts +1 -1
  142. package/src/models/operations/createbudget.ts +2 -2
  143. package/src/models/operations/createcontact.ts +2 -2
  144. package/src/models/operations/createdataset.ts +2 -2
  145. package/src/models/operations/createdatasetitem.ts +2 -2
  146. package/src/models/operations/createdatasource.ts +2 -2
  147. package/src/models/operations/createeval.ts +16 -16
  148. package/src/models/operations/fileget.ts +2 -2
  149. package/src/models/operations/filelist.ts +2 -2
  150. package/src/models/operations/fileupload.ts +2 -2
  151. package/src/models/operations/getbudget.ts +2 -2
  152. package/src/models/operations/getevals.ts +28 -28
  153. package/src/models/operations/listbudgets.ts +2 -2
  154. package/src/models/operations/listcontacts.ts +2 -2
  155. package/src/models/operations/listdatasetdatapoints.ts +2 -2
  156. package/src/models/operations/listdatasets.ts +2 -2
  157. package/src/models/operations/listdatasources.ts +2 -2
  158. package/src/models/operations/retrievecontact.ts +2 -2
  159. package/src/models/operations/retrievedatapoint.ts +2 -2
  160. package/src/models/operations/retrievedataset.ts +2 -2
  161. package/src/models/operations/retrievedatasource.ts +2 -2
  162. package/src/models/operations/updatebudget.ts +2 -2
  163. package/src/models/operations/updatecontact.ts +2 -2
  164. package/src/models/operations/updatedatapoint.ts +2 -2
  165. package/src/models/operations/updatedataset.ts +2 -2
  166. package/src/models/operations/updatedatasource.ts +2 -2
  167. package/src/models/operations/updateeval.ts +16 -16
@@ -16,6 +16,7 @@ export * from "./creatememory.js";
16
16
  export * from "./creatememorydocument.js";
17
17
  export * from "./creatememorystore.js";
18
18
  export * from "./createprompt.js";
19
+ export * from "./createtool.js";
19
20
  export * from "./deletebudget.js";
20
21
  export * from "./deletechunk.js";
21
22
  export * from "./deletecontact.js";
@@ -28,11 +29,13 @@ export * from "./deletememory.js";
28
29
  export * from "./deletememorydocument.js";
29
30
  export * from "./deletememorystore.js";
30
31
  export * from "./deleteprompt.js";
32
+ export * from "./deletetool.js";
31
33
  export * from "./deploymentcreatemetric.js";
32
34
  export * from "./deploymentgetconfig.js";
33
35
  export * from "./deploymentinvoke.js";
34
36
  export * from "./deployments.js";
35
37
  export * from "./deploymentstream.js";
38
+ export * from "./duplicatetool.js";
36
39
  export * from "./evalsageappropriate.js";
37
40
  export * from "./evalsbertscore.js";
38
41
  export * from "./evalsbleuscore.js";
@@ -80,6 +83,7 @@ export * from "./getallmemories.js";
80
83
  export * from "./getallmemorydocuments.js";
81
84
  export * from "./getallmemorystores.js";
82
85
  export * from "./getallprompts.js";
86
+ export * from "./getalltools.js";
83
87
  export * from "./getbudget.js";
84
88
  export * from "./getevals.js";
85
89
  export * from "./getonechunk.js";
@@ -108,6 +112,7 @@ export * from "./retrievedatasource.js";
108
112
  export * from "./retrievememory.js";
109
113
  export * from "./retrievememorydocument.js";
110
114
  export * from "./retrievememorystore.js";
115
+ export * from "./retrievetool.js";
111
116
  export * from "./runagent.js";
112
117
  export * from "./searchknowledge.js";
113
118
  export * from "./streamrunagent.js";
@@ -123,3 +128,4 @@ export * from "./updatememory.js";
123
128
  export * from "./updatememorydocument.js";
124
129
  export * from "./updatememorystore.js";
125
130
  export * from "./updateprompt.js";
131
+ export * from "./updatetool.js";
@@ -199,8 +199,8 @@ export type ListAgentsData = {
199
199
  key: string;
200
200
  workspaceId: string;
201
201
  projectId: string;
202
- createdById?: string | undefined;
203
- updatedById?: string | undefined;
202
+ createdById?: string | null | undefined;
203
+ updatedById?: string | null | undefined;
204
204
  created?: string | undefined;
205
205
  updated?: string | undefined;
206
206
  role: string;
@@ -1022,7 +1022,7 @@ export const ListAgentsKnowledgeBases$inboundSchema: z.ZodType<
1022
1022
  z.ZodTypeDef,
1023
1023
  unknown
1024
1024
  > = z.object({
1025
- id: z.string().default("01K5RWRYK6QDX0QEJXTW2178DP"),
1025
+ id: z.string().default("01K5ZT3CQ2P6FDBCDT64FNA01H"),
1026
1026
  knowledge_id: z.string(),
1027
1027
  configuration: z.union([
1028
1028
  z.lazy(() => ListAgentsConfiguration2$inboundSchema),
@@ -1049,7 +1049,7 @@ export const ListAgentsKnowledgeBases$outboundSchema: z.ZodType<
1049
1049
  z.ZodTypeDef,
1050
1050
  ListAgentsKnowledgeBases
1051
1051
  > = z.object({
1052
- id: z.string().default("01K5RWRYK6QDX0QEJXTW2178DP"),
1052
+ id: z.string().default("01K5ZT3CQ2P6FDBCDT64FNA01H"),
1053
1053
  knowledgeId: z.string(),
1054
1054
  configuration: z.union([
1055
1055
  z.lazy(() => ListAgentsConfiguration2$outboundSchema),
@@ -1123,8 +1123,8 @@ export const ListAgentsData$inboundSchema: z.ZodType<
1123
1123
  key: z.string(),
1124
1124
  workspace_id: z.string(),
1125
1125
  project_id: z.string(),
1126
- created_by_id: z.string().optional(),
1127
- updated_by_id: z.string().optional(),
1126
+ created_by_id: z.nullable(z.string()).optional(),
1127
+ updated_by_id: z.nullable(z.string()).optional(),
1128
1128
  created: z.string().optional(),
1129
1129
  updated: z.string().optional(),
1130
1130
  role: z.string(),
@@ -1165,8 +1165,8 @@ export type ListAgentsData$Outbound = {
1165
1165
  key: string;
1166
1166
  workspace_id: string;
1167
1167
  project_id: string;
1168
- created_by_id?: string | undefined;
1169
- updated_by_id?: string | undefined;
1168
+ created_by_id?: string | null | undefined;
1169
+ updated_by_id?: string | null | undefined;
1170
1170
  created?: string | undefined;
1171
1171
  updated?: string | undefined;
1172
1172
  role: string;
@@ -1196,8 +1196,8 @@ export const ListAgentsData$outboundSchema: z.ZodType<
1196
1196
  key: z.string(),
1197
1197
  workspaceId: z.string(),
1198
1198
  projectId: z.string(),
1199
- createdById: z.string().optional(),
1200
- updatedById: z.string().optional(),
1199
+ createdById: z.nullable(z.string()).optional(),
1200
+ updatedById: z.nullable(z.string()).optional(),
1201
1201
  created: z.string().optional(),
1202
1202
  updated: z.string().optional(),
1203
1203
  role: z.string(),
@@ -464,7 +464,7 @@ export const ListBudgetsData$inboundSchema: z.ZodType<
464
464
  created: z.string().datetime({ offset: true }).transform(v => new Date(v))
465
465
  .optional(),
466
466
  updated: z.string().datetime({ offset: true }).default(
467
- "2025-09-22T14:20:40.492Z",
467
+ "2025-09-25T06:48:39.657Z",
468
468
  ).transform(v => new Date(v)),
469
469
  }).transform((v) => {
470
470
  return remap$(v, {
@@ -499,7 +499,7 @@ export const ListBudgetsData$outboundSchema: z.ZodType<
499
499
  isActive: z.boolean(),
500
500
  consumption: z.lazy(() => ListBudgetsConsumption$outboundSchema).optional(),
501
501
  created: z.date().transform(v => v.toISOString()).optional(),
502
- updated: z.date().default(() => new Date("2025-09-22T14:20:40.492Z"))
502
+ updated: z.date().default(() => new Date("2025-09-25T06:48:39.657Z"))
503
503
  .transform(v => v.toISOString()),
504
504
  }).transform((v) => {
505
505
  return remap$(v, {
@@ -352,7 +352,7 @@ export const Data$inboundSchema: z.ZodType<Data, z.ZodTypeDef, unknown> = z
352
352
  created: z.string().datetime({ offset: true }).transform(v => new Date(v))
353
353
  .optional(),
354
354
  updated: z.string().datetime({ offset: true }).default(
355
- "2025-09-22T14:20:40.492Z",
355
+ "2025-09-25T06:48:39.657Z",
356
356
  ).transform(v => new Date(v)),
357
357
  metrics: z.lazy(() => ListContactsMetrics$inboundSchema),
358
358
  }).transform((v) => {
@@ -389,7 +389,7 @@ export const Data$outboundSchema: z.ZodType<Data$Outbound, z.ZodTypeDef, Data> =
389
389
  tags: z.array(z.string()).optional(),
390
390
  metadata: z.record(z.any()).optional(),
391
391
  created: z.date().transform(v => v.toISOString()).optional(),
392
- updated: z.date().default(() => new Date("2025-09-22T14:20:40.492Z"))
392
+ updated: z.date().default(() => new Date("2025-09-25T06:48:39.657Z"))
393
393
  .transform(v => v.toISOString()),
394
394
  metrics: z.lazy(() => ListContactsMetrics$outboundSchema),
395
395
  }).transform((v) => {
@@ -187,13 +187,14 @@ export type ListDatasetDatapointsMessagesDatasetsContent =
187
187
  >;
188
188
 
189
189
  /**
190
- * The role of the messages author, in this case `assistant`.
190
+ * The role of the messages author, in this case `assistant` or `exception`.
191
191
  */
192
192
  export const ListDatasetDatapointsMessagesDatasetsResponse200Role = {
193
193
  Assistant: "assistant",
194
+ Exception: "exception",
194
195
  } as const;
195
196
  /**
196
- * The role of the messages author, in this case `assistant`.
197
+ * The role of the messages author, in this case `assistant` or `exception`.
197
198
  */
198
199
  export type ListDatasetDatapointsMessagesDatasetsResponse200Role = ClosedEnum<
199
200
  typeof ListDatasetDatapointsMessagesDatasetsResponse200Role
@@ -262,7 +263,7 @@ export type ListDatasetDatapointsMessagesAssistantMessage = {
262
263
  */
263
264
  refusal?: string | null | undefined;
264
265
  /**
265
- * The role of the messages author, in this case `assistant`.
266
+ * The role of the messages author, in this case `assistant` or `exception`.
266
267
  */
267
268
  role: ListDatasetDatapointsMessagesDatasetsResponse200Role;
268
269
  /**
@@ -3100,7 +3101,7 @@ export const ListDatasetDatapointsData$inboundSchema: z.ZodType<
3100
3101
  created: z.string().datetime({ offset: true }).transform(v => new Date(v))
3101
3102
  .optional(),
3102
3103
  updated: z.string().datetime({ offset: true }).default(
3103
- "2025-09-22T14:20:40.492Z",
3104
+ "2025-09-25T06:48:39.657Z",
3104
3105
  ).transform(v => new Date(v)),
3105
3106
  }).transform((v) => {
3106
3107
  return remap$(v, {
@@ -3164,7 +3165,7 @@ export const ListDatasetDatapointsData$outboundSchema: z.ZodType<
3164
3165
  createdById: z.string().optional(),
3165
3166
  updatedById: z.string().optional(),
3166
3167
  created: z.date().transform(v => v.toISOString()).optional(),
3167
- updated: z.date().default(() => new Date("2025-09-22T14:20:40.492Z"))
3168
+ updated: z.date().default(() => new Date("2025-09-25T06:48:39.657Z"))
3168
3169
  .transform(v => v.toISOString()),
3169
3170
  }).transform((v) => {
3170
3171
  return remap$(v, {
@@ -253,7 +253,7 @@ export const ListDatasetsData$inboundSchema: z.ZodType<
253
253
  created: z.string().datetime({ offset: true }).transform(v => new Date(v))
254
254
  .optional(),
255
255
  updated: z.string().datetime({ offset: true }).default(
256
- "2025-09-22T14:20:40.492Z",
256
+ "2025-09-25T06:48:39.657Z",
257
257
  ).transform(v => new Date(v)),
258
258
  }).transform((v) => {
259
259
  return remap$(v, {
@@ -293,7 +293,7 @@ export const ListDatasetsData$outboundSchema: z.ZodType<
293
293
  updatedById: z.string().optional(),
294
294
  metadata: z.lazy(() => ListDatasetsMetadata$outboundSchema),
295
295
  created: z.date().transform(v => v.toISOString()).optional(),
296
- updated: z.date().default(() => new Date("2025-09-22T14:20:40.492Z"))
296
+ updated: z.date().default(() => new Date("2025-09-25T06:48:39.657Z"))
297
297
  .transform(v => v.toISOString()),
298
298
  }).transform((v) => {
299
299
  return remap$(v, {
@@ -285,7 +285,7 @@ export const ListDatasourcesData$inboundSchema: z.ZodType<
285
285
  z.ZodTypeDef,
286
286
  unknown
287
287
  > = z.object({
288
- _id: z.string().default("01K5RWRYR8S2EYS00ABE811GS4"),
288
+ _id: z.string().default("01K5ZT3CV84M7WC2SHW3YJHN1W"),
289
289
  display_name: z.string(),
290
290
  description: z.string().optional(),
291
291
  status: ListDatasourcesStatus$inboundSchema,
@@ -329,7 +329,7 @@ export const ListDatasourcesData$outboundSchema: z.ZodType<
329
329
  z.ZodTypeDef,
330
330
  ListDatasourcesData
331
331
  > = z.object({
332
- id: z.string().default("01K5RWRYR8S2EYS00ABE811GS4"),
332
+ id: z.string().default("01K5ZT3CV84M7WC2SHW3YJHN1W"),
333
333
  displayName: z.string(),
334
334
  description: z.string().optional(),
335
335
  status: ListDatasourcesStatus$outboundSchema,
@@ -50,6 +50,10 @@ export type ListKnowledgeBasesRetrievalType = ClosedEnum<
50
50
  * The rerank configuration for the knowledge base. In case the model is provided it will be used to enhance the search precision.
51
51
  */
52
52
  export type ListKnowledgeBasesRerankConfig = {
53
+ /**
54
+ * The number of results to return by the reranking model
55
+ */
56
+ topK?: number | undefined;
53
57
  /**
54
58
  * The threshold value used to filter the rerank results, only documents with a relevance score greater than the threshold will be returned
55
59
  */
@@ -267,10 +271,12 @@ export const ListKnowledgeBasesRerankConfig$inboundSchema: z.ZodType<
267
271
  z.ZodTypeDef,
268
272
  unknown
269
273
  > = z.object({
274
+ top_k: z.number().int().default(5),
270
275
  rerank_threshold: z.number().default(0.5),
271
276
  rerank_model: z.string(),
272
277
  }).transform((v) => {
273
278
  return remap$(v, {
279
+ "top_k": "topK",
274
280
  "rerank_threshold": "rerankThreshold",
275
281
  "rerank_model": "rerankModel",
276
282
  });
@@ -278,6 +284,7 @@ export const ListKnowledgeBasesRerankConfig$inboundSchema: z.ZodType<
278
284
 
279
285
  /** @internal */
280
286
  export type ListKnowledgeBasesRerankConfig$Outbound = {
287
+ top_k: number;
281
288
  rerank_threshold: number;
282
289
  rerank_model: string;
283
290
  };
@@ -288,10 +295,12 @@ export const ListKnowledgeBasesRerankConfig$outboundSchema: z.ZodType<
288
295
  z.ZodTypeDef,
289
296
  ListKnowledgeBasesRerankConfig
290
297
  > = z.object({
298
+ topK: z.number().int().default(5),
291
299
  rerankThreshold: z.number().default(0.5),
292
300
  rerankModel: z.string(),
293
301
  }).transform((v) => {
294
302
  return remap$(v, {
303
+ topK: "top_k",
295
304
  rerankThreshold: "rerank_threshold",
296
305
  rerankModel: "rerank_model",
297
306
  });