@azure/ai-projects 2.0.0 → 2.0.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +5 -5
- package/dist/browser/api/aiProjectContext.d.ts.map +1 -1
- package/dist/browser/api/aiProjectContext.js +2 -1
- package/dist/browser/api/aiProjectContext.js.map +1 -1
- package/dist/browser/api/beta/memoryStores/operations.d.ts.map +1 -1
- package/dist/browser/api/beta/memoryStores/operations.js +11 -0
- package/dist/browser/api/beta/memoryStores/operations.js.map +1 -1
- package/dist/browser/api/datasets/operations.d.ts.map +1 -1
- package/dist/browser/api/datasets/operations.js +9 -8
- package/dist/browser/api/datasets/operations.js.map +1 -1
- package/dist/browser/constants.d.ts +1 -1
- package/dist/browser/constants.js +1 -1
- package/dist/browser/constants.js.map +1 -1
- package/dist/browser/static-helpers/pollingHelpers.d.ts +11 -1
- package/dist/browser/static-helpers/pollingHelpers.d.ts.map +1 -1
- package/dist/browser/static-helpers/pollingHelpers.js +21 -9
- package/dist/browser/static-helpers/pollingHelpers.js.map +1 -1
- package/dist/commonjs/aiProjectClient.js +1 -0
- package/dist/commonjs/aiProjectClient.js.map +1 -1
- package/dist/commonjs/api/agents/index.js +1 -0
- package/dist/commonjs/api/agents/index.js.map +1 -1
- package/dist/commonjs/api/agents/operations.js +1 -0
- package/dist/commonjs/api/agents/operations.js.map +1 -1
- package/dist/commonjs/api/agents/options.js +1 -0
- package/dist/commonjs/api/agents/options.js.map +1 -1
- package/dist/commonjs/api/aiProjectContext.d.ts.map +1 -1
- package/dist/commonjs/api/aiProjectContext.js +3 -1
- package/dist/commonjs/api/aiProjectContext.js.map +3 -3
- package/dist/commonjs/api/beta/evaluationTaxonomies/index.js +1 -0
- package/dist/commonjs/api/beta/evaluationTaxonomies/index.js.map +1 -1
- package/dist/commonjs/api/beta/evaluationTaxonomies/operations.js +1 -0
- package/dist/commonjs/api/beta/evaluationTaxonomies/operations.js.map +1 -1
- package/dist/commonjs/api/beta/evaluationTaxonomies/options.js +1 -0
- package/dist/commonjs/api/beta/evaluationTaxonomies/options.js.map +1 -1
- package/dist/commonjs/api/beta/evaluators/index.js +1 -0
- package/dist/commonjs/api/beta/evaluators/index.js.map +1 -1
- package/dist/commonjs/api/beta/evaluators/operations.js +1 -0
- package/dist/commonjs/api/beta/evaluators/operations.js.map +1 -1
- package/dist/commonjs/api/beta/evaluators/options.js +1 -0
- package/dist/commonjs/api/beta/evaluators/options.js.map +1 -1
- package/dist/commonjs/api/beta/insights/index.js +1 -0
- package/dist/commonjs/api/beta/insights/index.js.map +1 -1
- package/dist/commonjs/api/beta/insights/operations.js +1 -0
- package/dist/commonjs/api/beta/insights/operations.js.map +1 -1
- package/dist/commonjs/api/beta/insights/options.js +1 -0
- package/dist/commonjs/api/beta/insights/options.js.map +1 -1
- package/dist/commonjs/api/beta/memoryStores/index.js +1 -0
- package/dist/commonjs/api/beta/memoryStores/index.js.map +1 -1
- package/dist/commonjs/api/beta/memoryStores/operations.d.ts.map +1 -1
- package/dist/commonjs/api/beta/memoryStores/operations.js +13 -1
- package/dist/commonjs/api/beta/memoryStores/operations.js.map +3 -3
- package/dist/commonjs/api/beta/memoryStores/options.js +1 -0
- package/dist/commonjs/api/beta/memoryStores/options.js.map +1 -1
- package/dist/commonjs/api/beta/redTeams/index.js +1 -0
- package/dist/commonjs/api/beta/redTeams/index.js.map +1 -1
- package/dist/commonjs/api/beta/redTeams/operations.js +1 -0
- package/dist/commonjs/api/beta/redTeams/operations.js.map +1 -1
- package/dist/commonjs/api/beta/redTeams/options.js +1 -0
- package/dist/commonjs/api/beta/redTeams/options.js.map +1 -1
- package/dist/commonjs/api/beta/schedules/index.js +1 -0
- package/dist/commonjs/api/beta/schedules/index.js.map +1 -1
- package/dist/commonjs/api/beta/schedules/operations.js +1 -0
- package/dist/commonjs/api/beta/schedules/operations.js.map +1 -1
- package/dist/commonjs/api/beta/schedules/options.js +1 -0
- package/dist/commonjs/api/beta/schedules/options.js.map +1 -1
- package/dist/commonjs/api/connections/index.js +1 -0
- package/dist/commonjs/api/connections/index.js.map +1 -1
- package/dist/commonjs/api/connections/operations.js +1 -0
- package/dist/commonjs/api/connections/operations.js.map +1 -1
- package/dist/commonjs/api/connections/options.js +1 -0
- package/dist/commonjs/api/connections/options.js.map +1 -1
- package/dist/commonjs/api/datasets/index.js +1 -0
- package/dist/commonjs/api/datasets/index.js.map +1 -1
- package/dist/commonjs/api/datasets/operations.d.ts.map +1 -1
- package/dist/commonjs/api/datasets/operations.js +10 -8
- package/dist/commonjs/api/datasets/operations.js.map +3 -3
- package/dist/commonjs/api/datasets/options.js +1 -0
- package/dist/commonjs/api/datasets/options.js.map +1 -1
- package/dist/commonjs/api/deployments/index.js +1 -0
- package/dist/commonjs/api/deployments/index.js.map +1 -1
- package/dist/commonjs/api/deployments/operations.js +1 -0
- package/dist/commonjs/api/deployments/operations.js.map +1 -1
- package/dist/commonjs/api/deployments/options.js +1 -0
- package/dist/commonjs/api/deployments/options.js.map +1 -1
- package/dist/commonjs/api/evaluationRules/index.js +1 -0
- package/dist/commonjs/api/evaluationRules/index.js.map +1 -1
- package/dist/commonjs/api/evaluationRules/operations.js +1 -0
- package/dist/commonjs/api/evaluationRules/operations.js.map +1 -1
- package/dist/commonjs/api/evaluationRules/options.js +1 -0
- package/dist/commonjs/api/evaluationRules/options.js.map +1 -1
- package/dist/commonjs/api/index.js +1 -0
- package/dist/commonjs/api/index.js.map +1 -1
- package/dist/commonjs/api/indexes/index.js +1 -0
- package/dist/commonjs/api/indexes/index.js.map +1 -1
- package/dist/commonjs/api/indexes/operations.js +1 -0
- package/dist/commonjs/api/indexes/operations.js.map +1 -1
- package/dist/commonjs/api/indexes/options.js +1 -0
- package/dist/commonjs/api/indexes/options.js.map +1 -1
- package/dist/commonjs/api/telemetry/index.js +1 -0
- package/dist/commonjs/api/telemetry/index.js.map +1 -1
- package/dist/commonjs/api/telemetry/operations.js +1 -0
- package/dist/commonjs/api/telemetry/operations.js.map +1 -1
- package/dist/commonjs/classic/agents/index.js +1 -0
- package/dist/commonjs/classic/agents/index.js.map +1 -1
- package/dist/commonjs/classic/beta/evaluationTaxonomies/index.js +1 -0
- package/dist/commonjs/classic/beta/evaluationTaxonomies/index.js.map +1 -1
- package/dist/commonjs/classic/beta/evaluators/index.js +1 -0
- package/dist/commonjs/classic/beta/evaluators/index.js.map +1 -1
- package/dist/commonjs/classic/beta/index.js +1 -0
- package/dist/commonjs/classic/beta/index.js.map +1 -1
- package/dist/commonjs/classic/beta/insights/index.js +1 -0
- package/dist/commonjs/classic/beta/insights/index.js.map +1 -1
- package/dist/commonjs/classic/beta/memoryStores/index.js +1 -0
- package/dist/commonjs/classic/beta/memoryStores/index.js.map +1 -1
- package/dist/commonjs/classic/beta/redTeams/index.js +1 -0
- package/dist/commonjs/classic/beta/redTeams/index.js.map +1 -1
- package/dist/commonjs/classic/beta/schedules/index.js +1 -0
- package/dist/commonjs/classic/beta/schedules/index.js.map +1 -1
- package/dist/commonjs/classic/connections/index.js +1 -0
- package/dist/commonjs/classic/connections/index.js.map +1 -1
- package/dist/commonjs/classic/datasets/index.js +1 -0
- package/dist/commonjs/classic/datasets/index.js.map +1 -1
- package/dist/commonjs/classic/deployments/index.js +1 -0
- package/dist/commonjs/classic/deployments/index.js.map +1 -1
- package/dist/commonjs/classic/evaluationRules/index.js +1 -0
- package/dist/commonjs/classic/evaluationRules/index.js.map +1 -1
- package/dist/commonjs/classic/index.js +1 -0
- package/dist/commonjs/classic/index.js.map +1 -1
- package/dist/commonjs/classic/indexes/index.js +1 -0
- package/dist/commonjs/classic/indexes/index.js.map +1 -1
- package/dist/commonjs/classic/telemetry/index.js +1 -0
- package/dist/commonjs/classic/telemetry/index.js.map +1 -1
- package/dist/commonjs/constants.d.ts +1 -1
- package/dist/commonjs/constants.js +2 -1
- package/dist/commonjs/constants.js.map +2 -2
- package/dist/commonjs/getCustomFetch.js +1 -0
- package/dist/commonjs/getCustomFetch.js.map +1 -1
- package/dist/commonjs/index.js +1 -0
- package/dist/commonjs/index.js.map +1 -1
- package/dist/commonjs/logger.js +1 -0
- package/dist/commonjs/logger.js.map +1 -1
- package/dist/commonjs/models/index.js +1 -0
- package/dist/commonjs/models/index.js.map +1 -1
- package/dist/commonjs/models/models.js +1 -0
- package/dist/commonjs/models/models.js.map +1 -1
- package/dist/commonjs/overwriteOpenAIClient.js +1 -0
- package/dist/commonjs/overwriteOpenAIClient.js.map +1 -1
- package/dist/commonjs/static-helpers/pagingHelpers.js +1 -0
- package/dist/commonjs/static-helpers/pagingHelpers.js.map +1 -1
- package/dist/commonjs/static-helpers/pollingHelpers.d.ts +11 -1
- package/dist/commonjs/static-helpers/pollingHelpers.d.ts.map +1 -1
- package/dist/commonjs/static-helpers/pollingHelpers.js +16 -8
- package/dist/commonjs/static-helpers/pollingHelpers.js.map +3 -3
- package/dist/commonjs/static-helpers/serialization/serialize-record.js +1 -0
- package/dist/commonjs/static-helpers/serialization/serialize-record.js.map +1 -1
- package/dist/commonjs/static-helpers/urlTemplate.js +1 -0
- package/dist/commonjs/static-helpers/urlTemplate.js.map +1 -1
- package/dist/commonjs/tsdoc-metadata.json +1 -1
- package/dist/commonjs/util.js +1 -0
- package/dist/commonjs/util.js.map +1 -1
- package/dist/esm/aiProjectClient.js +119 -94
- package/dist/esm/aiProjectClient.js.map +1 -7
- package/dist/esm/api/agents/index.js +4 -28
- package/dist/esm/api/agents/index.js.map +1 -7
- package/dist/esm/api/agents/operations.js +375 -473
- package/dist/esm/api/agents/operations.js.map +1 -7
- package/dist/esm/api/agents/options.js +4 -0
- package/dist/esm/api/agents/options.js.map +1 -7
- package/dist/esm/api/aiProjectContext.d.ts.map +1 -1
- package/dist/esm/api/aiProjectContext.js +22 -19
- package/dist/esm/api/aiProjectContext.js.map +1 -7
- package/dist/esm/api/beta/evaluationTaxonomies/index.js +4 -8
- package/dist/esm/api/beta/evaluationTaxonomies/index.js.map +1 -7
- package/dist/esm/api/beta/evaluationTaxonomies/operations.js +150 -183
- package/dist/esm/api/beta/evaluationTaxonomies/operations.js.map +1 -7
- package/dist/esm/api/beta/evaluationTaxonomies/options.js +4 -0
- package/dist/esm/api/beta/evaluationTaxonomies/options.js.map +1 -7
- package/dist/esm/api/beta/evaluators/index.js +4 -16
- package/dist/esm/api/beta/evaluators/index.js.map +1 -7
- package/dist/esm/api/beta/evaluators/operations.js +192 -243
- package/dist/esm/api/beta/evaluators/operations.js.map +1 -7
- package/dist/esm/api/beta/evaluators/options.js +4 -0
- package/dist/esm/api/beta/evaluators/options.js.map +1 -7
- package/dist/esm/api/beta/insights/index.js +4 -6
- package/dist/esm/api/beta/insights/index.js.map +1 -7
- package/dist/esm/api/beta/insights/operations.js +109 -126
- package/dist/esm/api/beta/insights/operations.js.map +1 -7
- package/dist/esm/api/beta/insights/options.js +4 -0
- package/dist/esm/api/beta/insights/options.js.map +1 -7
- package/dist/esm/api/beta/memoryStores/index.js +4 -22
- package/dist/esm/api/beta/memoryStores/index.js.map +1 -7
- package/dist/esm/api/beta/memoryStores/operations.d.ts.map +1 -1
- package/dist/esm/api/beta/memoryStores/operations.js +305 -363
- package/dist/esm/api/beta/memoryStores/operations.js.map +1 -7
- package/dist/esm/api/beta/memoryStores/options.js +4 -0
- package/dist/esm/api/beta/memoryStores/options.js.map +1 -7
- package/dist/esm/api/beta/redTeams/index.js +4 -6
- package/dist/esm/api/beta/redTeams/index.js.map +1 -7
- package/dist/esm/api/beta/redTeams/operations.js +95 -119
- package/dist/esm/api/beta/redTeams/operations.js.map +1 -7
- package/dist/esm/api/beta/redTeams/options.js +4 -0
- package/dist/esm/api/beta/redTeams/options.js.map +1 -7
- package/dist/esm/api/beta/schedules/index.js +4 -9
- package/dist/esm/api/beta/schedules/index.js.map +1 -7
- package/dist/esm/api/beta/schedules/operations.js +204 -248
- package/dist/esm/api/beta/schedules/operations.js.map +1 -7
- package/dist/esm/api/beta/schedules/options.js +4 -0
- package/dist/esm/api/beta/schedules/options.js.map +1 -7
- package/dist/esm/api/connections/index.js +4 -6
- package/dist/esm/api/connections/index.js.map +1 -7
- package/dist/esm/api/connections/operations.js +109 -116
- package/dist/esm/api/connections/operations.js.map +1 -7
- package/dist/esm/api/connections/options.js +4 -0
- package/dist/esm/api/connections/options.js.map +1 -7
- package/dist/esm/api/datasets/index.js +4 -18
- package/dist/esm/api/datasets/index.js.map +1 -7
- package/dist/esm/api/datasets/operations.d.ts.map +1 -1
- package/dist/esm/api/datasets/operations.js +288 -355
- package/dist/esm/api/datasets/operations.js.map +1 -7
- package/dist/esm/api/datasets/options.js +4 -0
- package/dist/esm/api/datasets/options.js.map +1 -7
- package/dist/esm/api/deployments/index.js +4 -5
- package/dist/esm/api/deployments/index.js.map +1 -7
- package/dist/esm/api/deployments/operations.js +57 -70
- package/dist/esm/api/deployments/operations.js.map +1 -7
- package/dist/esm/api/deployments/options.js +4 -0
- package/dist/esm/api/deployments/options.js.map +1 -7
- package/dist/esm/api/evaluationRules/index.js +4 -7
- package/dist/esm/api/evaluationRules/index.js.map +1 -7
- package/dist/esm/api/evaluationRules/operations.js +121 -144
- package/dist/esm/api/evaluationRules/operations.js.map +1 -7
- package/dist/esm/api/evaluationRules/options.js +4 -0
- package/dist/esm/api/evaluationRules/options.js.map +1 -7
- package/dist/esm/api/index.js +4 -4
- package/dist/esm/api/index.js.map +1 -7
- package/dist/esm/api/indexes/index.js +4 -8
- package/dist/esm/api/indexes/index.js.map +1 -7
- package/dist/esm/api/indexes/operations.js +116 -161
- package/dist/esm/api/indexes/operations.js.map +1 -7
- package/dist/esm/api/indexes/options.js +4 -0
- package/dist/esm/api/indexes/options.js.map +1 -7
- package/dist/esm/api/telemetry/index.js +4 -4
- package/dist/esm/api/telemetry/index.js.map +1 -7
- package/dist/esm/api/telemetry/operations.js +53 -46
- package/dist/esm/api/telemetry/operations.js.map +1 -7
- package/dist/esm/classic/agents/index.js +40 -80
- package/dist/esm/classic/agents/index.js.map +1 -7
- package/dist/esm/classic/beta/evaluationTaxonomies/index.js +15 -21
- package/dist/esm/classic/beta/evaluationTaxonomies/index.js.map +1 -7
- package/dist/esm/classic/beta/evaluators/index.js +16 -23
- package/dist/esm/classic/beta/evaluators/index.js.map +1 -7
- package/dist/esm/classic/beta/index.js +18 -18
- package/dist/esm/classic/beta/index.js.map +1 -7
- package/dist/esm/classic/beta/insights/index.js +12 -12
- package/dist/esm/classic/beta/insights/index.js.map +1 -7
- package/dist/esm/classic/beta/memoryStores/index.js +19 -29
- package/dist/esm/classic/beta/memoryStores/index.js.map +1 -7
- package/dist/esm/classic/beta/redTeams/index.js +12 -12
- package/dist/esm/classic/beta/redTeams/index.js.map +1 -7
- package/dist/esm/classic/beta/schedules/index.js +16 -23
- package/dist/esm/classic/beta/schedules/index.js.map +1 -7
- package/dist/esm/classic/connections/index.js +13 -13
- package/dist/esm/classic/connections/index.js.map +1 -7
- package/dist/esm/classic/datasets/index.js +19 -29
- package/dist/esm/classic/datasets/index.js.map +1 -7
- package/dist/esm/classic/deployments/index.js +11 -11
- package/dist/esm/classic/deployments/index.js.map +1 -7
- package/dist/esm/classic/evaluationRules/index.js +13 -13
- package/dist/esm/classic/evaluationRules/index.js.map +1 -7
- package/dist/esm/classic/index.js +4 -0
- package/dist/esm/classic/index.js.map +1 -7
- package/dist/esm/classic/indexes/index.js +14 -14
- package/dist/esm/classic/indexes/index.js.map +1 -7
- package/dist/esm/classic/telemetry/index.js +7 -9
- package/dist/esm/classic/telemetry/index.js.map +1 -7
- package/dist/esm/constants.d.ts +1 -1
- package/dist/esm/constants.js +11 -6
- package/dist/esm/constants.js.map +1 -7
- package/dist/esm/getCustomFetch.js +40 -33
- package/dist/esm/getCustomFetch.js.map +1 -7
- package/dist/esm/index.js +4 -4
- package/dist/esm/index.js.map +1 -7
- package/dist/esm/logger.js +4 -4
- package/dist/esm/logger.js.map +1 -7
- package/dist/esm/models/index.js +4 -0
- package/dist/esm/models/index.js.map +1 -7
- package/dist/esm/models/models.js +3709 -4160
- package/dist/esm/models/models.js.map +1 -7
- package/dist/esm/overwriteOpenAIClient.js +13 -13
- package/dist/esm/overwriteOpenAIClient.js.map +1 -7
- package/dist/esm/static-helpers/pagingHelpers.js +123 -99
- package/dist/esm/static-helpers/pagingHelpers.js.map +1 -7
- package/dist/esm/static-helpers/pollingHelpers.d.ts +11 -1
- package/dist/esm/static-helpers/pollingHelpers.d.ts.map +1 -1
- package/dist/esm/static-helpers/pollingHelpers.js +99 -69
- package/dist/esm/static-helpers/pollingHelpers.js.map +1 -7
- package/dist/esm/static-helpers/serialization/serialize-record.js +18 -16
- package/dist/esm/static-helpers/serialization/serialize-record.js.map +1 -7
- package/dist/esm/static-helpers/urlTemplate.js +169 -142
- package/dist/esm/static-helpers/urlTemplate.js.map +1 -7
- package/dist/esm/util.js +84 -63
- package/dist/esm/util.js.map +1 -7
- package/dist/react-native/api/aiProjectContext.d.ts.map +1 -1
- package/dist/react-native/api/aiProjectContext.js +2 -1
- package/dist/react-native/api/aiProjectContext.js.map +1 -1
- package/dist/react-native/api/beta/memoryStores/operations.d.ts.map +1 -1
- package/dist/react-native/api/beta/memoryStores/operations.js +11 -0
- package/dist/react-native/api/beta/memoryStores/operations.js.map +1 -1
- package/dist/react-native/api/datasets/operations.d.ts.map +1 -1
- package/dist/react-native/api/datasets/operations.js +9 -8
- package/dist/react-native/api/datasets/operations.js.map +1 -1
- package/dist/react-native/constants.d.ts +1 -1
- package/dist/react-native/constants.js +1 -1
- package/dist/react-native/constants.js.map +1 -1
- package/dist/react-native/static-helpers/pollingHelpers.d.ts +11 -1
- package/dist/react-native/static-helpers/pollingHelpers.d.ts.map +1 -1
- package/dist/react-native/static-helpers/pollingHelpers.js +21 -9
- package/dist/react-native/static-helpers/pollingHelpers.js.map +1 -1
- package/package.json +9 -9
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"version": 3,
|
|
3
|
-
"sources": ["
|
|
3
|
+
"sources": ["../../src/overwriteOpenAIClient.ts"],
|
|
4
4
|
"sourcesContent": ["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type OpenAI from \"openai\";\n\nexport function overwriteOpenAIClient(openaiClient: OpenAI): OpenAI {\n const responsesCreate = openaiClient.responses.create.bind(openaiClient.responses);\n openaiClient.responses.create = ((...args: Parameters<typeof responsesCreate>) => {\n const [body, options = {}] = args;\n const nextBody = { ...body, ...(options.body || {}) };\n const { body: _, ...nextOptions } = options;\n const response = responsesCreate(nextBody, nextOptions);\n return response;\n }) as typeof responsesCreate;\n return openaiClient;\n}\n"],
|
|
5
5
|
"mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAKO,SAAS,sBAAsB,cAA8B;AAClE,QAAM,kBAAkB,aAAa,UAAU,OAAO,KAAK,aAAa,SAAS;AACjF,eAAa,UAAU,UAAU,IAAI,SAA6C;AAChF,UAAM,CAAC,MAAM,UAAU,CAAC,CAAC,IAAI;AAC7B,UAAM,WAAW,EAAE,GAAG,MAAM,GAAI,QAAQ,QAAQ,CAAC,EAAG;AACpD,UAAM,EAAE,MAAM,GAAG,GAAG,YAAY,IAAI;AACpC,UAAM,WAAW,gBAAgB,UAAU,WAAW;AACtD,WAAO;AAAA,EACT;AACA,SAAO;AACT;",
|
|
6
6
|
"names": []
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"version": 3,
|
|
3
|
-
"sources": ["
|
|
3
|
+
"sources": ["../../../src/static-helpers/pagingHelpers.ts"],
|
|
4
4
|
"sourcesContent": ["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { PageSettings, PagedAsyncIterableIterator } from \"@azure/core-paging\";\nimport type { Client, PathUncheckedResponse } from \"@azure-rest/core-client\";\nimport { createRestError } from \"@azure-rest/core-client\";\nimport { RestError } from \"@azure/core-rest-pipeline\";\nimport type { KnownApiVersions } from \"../models/models.js\";\n\n/**\n * An interface that describes a page of results.\n */\nexport type ContinuablePage<TElement, TPage = TElement[]> = TPage & {\n /**\n * The token that keeps track of where to continue the iterator\n */\n continuationToken?: string;\n};\n\n/**\n * An interface that describes how to communicate with the service.\n */\nexport interface PagedResult<\n TElement,\n TPage = TElement[],\n TPageSettings extends PageSettings = PageSettings,\n> {\n /**\n * Link to the first page of results.\n */\n firstPageLink?: string;\n /**\n * A method that returns a page of results.\n */\n getPage: (pageLink?: string) => Promise<{ page: TPage; nextPageLink?: string } | undefined>;\n /**\n * a function to implement the `byPage` method on the paged async iterator.\n */\n byPage?: (settings?: TPageSettings) => AsyncIterableIterator<ContinuablePage<TElement, TPage>>;\n\n /**\n * A function to extract elements from a page.\n */\n toElements?: (page: TPage) => TElement[];\n}\n\n/**\n * Options for the paging helper\n */\nexport interface BuildPagedAsyncIteratorOptions {\n itemName?: string;\n nextLinkName?: string;\n nextLinkMethod?: \"GET\" | \"POST\";\n apiVersion?: KnownApiVersions;\n nextPageRequestOptions?: Record<string, unknown>;\n}\n\n/**\n * Helper to paginate results in a generic way and return a PagedAsyncIterableIterator\n */\nexport function buildPagedAsyncIterator<\n TElement,\n TPage = TElement[],\n TPageSettings extends PageSettings = PageSettings,\n TResponse extends PathUncheckedResponse = PathUncheckedResponse,\n>(\n client: Client,\n getInitialResponse: () => PromiseLike<TResponse>,\n processResponseBody: (result: TResponse) => PromiseLike<unknown>,\n expectedStatuses: string[],\n options: BuildPagedAsyncIteratorOptions = {},\n): PagedAsyncIterableIterator<TElement, TPage, TPageSettings> {\n const itemName = options.itemName ?? \"value\";\n const nextLinkName = options.nextLinkName ?? \"nextLink\";\n const nextLinkMethod = options.nextLinkMethod ?? \"GET\";\n const apiVersion = options.apiVersion;\n const nextPageRequestOptions = options.nextPageRequestOptions;\n const pagedResult: PagedResult<TElement, TPage, TPageSettings> = {\n getPage: async (pageLink?: string) => {\n let result;\n if (pageLink === undefined) {\n result = await getInitialResponse();\n } else {\n const resolvedPageLink = apiVersion ? addApiVersionToUrl(pageLink, apiVersion) : pageLink;\n result =\n nextLinkMethod === \"POST\"\n ? await client.pathUnchecked(resolvedPageLink).post(nextPageRequestOptions)\n : await client.pathUnchecked(resolvedPageLink).get(nextPageRequestOptions);\n }\n checkPagingRequest(result, expectedStatuses);\n const results = await processResponseBody(result as TResponse);\n const nextLink = getNextLink(results, nextLinkName);\n const values = getElements<TElement>(results, itemName) as TPage;\n return {\n page: values,\n nextPageLink: nextLink,\n };\n },\n byPage: (settings?: TPageSettings) => {\n const { continuationToken } = settings ?? {};\n return getPageAsyncIterator(pagedResult, {\n pageLink: continuationToken,\n });\n },\n };\n return getPagedAsyncIterator(pagedResult);\n}\n\n/**\n * returns an async iterator that iterates over results. It also has a `byPage`\n * method that returns pages of items at once.\n *\n * @param pagedResult - an object that specifies how to get pages.\n * @returns a paged async iterator that iterates over results.\n */\n\nfunction getPagedAsyncIterator<\n TElement,\n TPage = TElement[],\n TPageSettings extends PageSettings = PageSettings,\n>(\n pagedResult: PagedResult<TElement, TPage, TPageSettings>,\n): PagedAsyncIterableIterator<TElement, TPage, TPageSettings> {\n const iter = getItemAsyncIterator<TElement, TPage, TPageSettings>(pagedResult);\n return {\n next() {\n return iter.next();\n },\n [Symbol.asyncIterator]() {\n return this;\n },\n byPage:\n pagedResult?.byPage ??\n ((settings?: TPageSettings) => {\n const { continuationToken } = settings ?? {};\n return getPageAsyncIterator(pagedResult, {\n pageLink: continuationToken,\n });\n }),\n };\n}\n\nasync function* getItemAsyncIterator<TElement, TPage, TPageSettings extends PageSettings>(\n pagedResult: PagedResult<TElement, TPage, TPageSettings>,\n): AsyncIterableIterator<TElement> {\n const pages = getPageAsyncIterator(pagedResult);\n for await (const page of pages) {\n yield* page as unknown as TElement[];\n }\n}\n\nasync function* getPageAsyncIterator<TElement, TPage, TPageSettings extends PageSettings>(\n pagedResult: PagedResult<TElement, TPage, TPageSettings>,\n options: {\n pageLink?: string;\n } = {},\n): AsyncIterableIterator<ContinuablePage<TElement, TPage>> {\n const { pageLink } = options;\n let response = await pagedResult.getPage(pageLink ?? pagedResult.firstPageLink);\n if (!response) {\n return;\n }\n let result = response.page as ContinuablePage<TElement, TPage>;\n result.continuationToken = response.nextPageLink;\n yield result;\n while (response.nextPageLink) {\n response = await pagedResult.getPage(response.nextPageLink);\n if (!response) {\n return;\n }\n result = response.page as ContinuablePage<TElement, TPage>;\n result.continuationToken = response.nextPageLink;\n yield result;\n }\n}\n\n/**\n * Gets for the value of nextLink in the body\n */\nfunction getNextLink(body: unknown, nextLinkName?: string): string | undefined {\n if (!nextLinkName) {\n return undefined;\n }\n\n const nextLink = (body as Record<string, unknown>)[nextLinkName];\n\n if (typeof nextLink !== \"string\" && typeof nextLink !== \"undefined\" && nextLink !== null) {\n throw new RestError(\n `Body Property ${nextLinkName} should be a string or undefined or null but got ${typeof nextLink}`,\n );\n }\n\n if (nextLink === null) {\n return undefined;\n }\n\n return nextLink;\n}\n\n/**\n * Gets the elements of the current request in the body.\n */\nfunction getElements<T = unknown>(body: unknown, itemName: string): T[] {\n const value = (body as Record<string, unknown>)[itemName] as T[];\n if (!Array.isArray(value)) {\n throw new RestError(\n `Couldn't paginate response\\n Body doesn't contain an array property with name: ${itemName}`,\n );\n }\n\n return value ?? [];\n}\n\n/**\n * Checks if a request failed\n */\nfunction checkPagingRequest(response: PathUncheckedResponse, expectedStatuses: string[]): void {\n if (!expectedStatuses.includes(response.status)) {\n throw createRestError(\n `Pagination failed with unexpected statusCode ${response.status}`,\n response,\n );\n }\n}\n\n/**\n * Adds the api-version query parameter on a URL if it's not present.\n * @param url - the URL to modify\n * @param apiVersion - the API version to set\n * @returns - the URL with the api-version query parameter set\n */\nfunction addApiVersionToUrl(url: string, apiVersion: string): string {\n // The base URL is only used for parsing and won't appear in the returned URL\n const urlObj = new URL(url, \"https://microsoft.com\");\n if (!urlObj.searchParams.get(\"api-version\")) {\n // Append one if there is no apiVersion\n return `${url}${\n Array.from(urlObj.searchParams.keys()).length > 0 ? \"&\" : \"?\"\n }api-version=${apiVersion}`;\n }\n return url;\n}\n"],
|
|
5
5
|
"mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAKA,yBAAgC;AAChC,gCAA0B;AAsDnB,SAAS,wBAMd,QACA,oBACA,qBACA,kBACA,UAA0C,CAAC,GACiB;AAC5D,QAAM,WAAW,QAAQ,YAAY;AACrC,QAAM,eAAe,QAAQ,gBAAgB;AAC7C,QAAM,iBAAiB,QAAQ,kBAAkB;AACjD,QAAM,aAAa,QAAQ;AAC3B,QAAM,yBAAyB,QAAQ;AACvC,QAAM,cAA2D;AAAA,IAC/D,SAAS,OAAO,aAAsB;AACpC,UAAI;AACJ,UAAI,aAAa,QAAW;AAC1B,iBAAS,MAAM,mBAAmB;AAAA,MACpC,OAAO;AACL,cAAM,mBAAmB,aAAa,mBAAmB,UAAU,UAAU,IAAI;AACjF,iBACE,mBAAmB,SACf,MAAM,OAAO,cAAc,gBAAgB,EAAE,KAAK,sBAAsB,IACxE,MAAM,OAAO,cAAc,gBAAgB,EAAE,IAAI,sBAAsB;AAAA,MAC/E;AACA,yBAAmB,QAAQ,gBAAgB;AAC3C,YAAM,UAAU,MAAM,oBAAoB,MAAmB;AAC7D,YAAM,WAAW,YAAY,SAAS,YAAY;AAClD,YAAM,SAAS,YAAsB,SAAS,QAAQ;AACtD,aAAO;AAAA,QACL,MAAM;AAAA,QACN,cAAc;AAAA,MAChB;AAAA,IACF;AAAA,IACA,QAAQ,CAAC,aAA6B;AACpC,YAAM,EAAE,kBAAkB,IAAI,YAAY,CAAC;AAC3C,aAAO,qBAAqB,aAAa;AAAA,QACvC,UAAU;AAAA,MACZ,CAAC;AAAA,IACH;AAAA,EACF;AACA,SAAO,sBAAsB,WAAW;AAC1C;AAUA,SAAS,sBAKP,aAC4D;AAC5D,QAAM,OAAO,qBAAqD,WAAW;AAC7E,SAAO;AAAA,IACL,OAAO;AACL,aAAO,KAAK,KAAK;AAAA,IACnB;AAAA,IACA,CAAC,OAAO,aAAa,IAAI;AACvB,aAAO;AAAA,IACT;AAAA,IACA,QACE,aAAa,WACZ,CAAC,aAA6B;AAC7B,YAAM,EAAE,kBAAkB,IAAI,YAAY,CAAC;AAC3C,aAAO,qBAAqB,aAAa;AAAA,QACvC,UAAU;AAAA,MACZ,CAAC;AAAA,IACH;AAAA,EACJ;AACF;AAEA,gBAAgB,qBACd,aACiC;AACjC,QAAM,QAAQ,qBAAqB,WAAW;AAC9C,mBAAiB,QAAQ,OAAO;AAC9B,WAAO;AAAA,EACT;AACF;AAEA,gBAAgB,qBACd,aACA,UAEI,CAAC,GACoD;AACzD,QAAM,EAAE,SAAS,IAAI;AACrB,MAAI,WAAW,MAAM,YAAY,QAAQ,YAAY,YAAY,aAAa;AAC9E,MAAI,CAAC,UAAU;AACb;AAAA,EACF;AACA,MAAI,SAAS,SAAS;AACtB,SAAO,oBAAoB,SAAS;AACpC,QAAM;AACN,SAAO,SAAS,cAAc;AAC5B,eAAW,MAAM,YAAY,QAAQ,SAAS,YAAY;AAC1D,QAAI,CAAC,UAAU;AACb;AAAA,IACF;AACA,aAAS,SAAS;AAClB,WAAO,oBAAoB,SAAS;AACpC,UAAM;AAAA,EACR;AACF;AAKA,SAAS,YAAY,MAAe,cAA2C;AAC7E,MAAI,CAAC,cAAc;AACjB,WAAO;AAAA,EACT;AAEA,QAAM,WAAY,KAAiC,YAAY;AAE/D,MAAI,OAAO,aAAa,YAAY,OAAO,aAAa,eAAe,aAAa,MAAM;AACxF,UAAM,IAAI;AAAA,MACR,iBAAiB,YAAY,oDAAoD,OAAO,QAAQ;AAAA,IAClG;AAAA,EACF;AAEA,MAAI,aAAa,MAAM;AACrB,WAAO;AAAA,EACT;AAEA,SAAO;AACT;AAKA,SAAS,YAAyB,MAAe,UAAuB;AACtE,QAAM,QAAS,KAAiC,QAAQ;AACxD,MAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB,UAAM,IAAI;AAAA,MACR;AAAA,qDAAkF,QAAQ;AAAA,IAC5F;AAAA,EACF;AAEA,SAAO,SAAS,CAAC;AACnB;AAKA,SAAS,mBAAmB,UAAiC,kBAAkC;AAC7F,MAAI,CAAC,iBAAiB,SAAS,SAAS,MAAM,GAAG;AAC/C,cAAM;AAAA,MACJ,gDAAgD,SAAS,MAAM;AAAA,MAC/D;AAAA,IACF;AAAA,EACF;AACF;AAQA,SAAS,mBAAmB,KAAa,YAA4B;AAEnE,QAAM,SAAS,IAAI,IAAI,KAAK,uBAAuB;AACnD,MAAI,CAAC,OAAO,aAAa,IAAI,aAAa,GAAG;AAE3C,WAAO,GAAG,GAAG,GACX,MAAM,KAAK,OAAO,aAAa,KAAK,CAAC,EAAE,SAAS,IAAI,MAAM,GAC5D,eAAe,UAAU;AAAA,EAC3B;AACA,SAAO;AACT;",
|
|
6
6
|
"names": []
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import type { PollerLike, OperationState, ResourceLocationConfig } from "@azure/core-lro";
|
|
1
|
+
import type { PollerLike, OperationState, OperationStatus, ResourceLocationConfig } from "@azure/core-lro";
|
|
2
2
|
import type { Client, PathUncheckedResponse } from "@azure-rest/core-client";
|
|
3
3
|
import type { AbortSignalLike } from "@azure/abort-controller";
|
|
4
4
|
import type { KnownApiVersions } from "../models/models.js";
|
|
@@ -26,6 +26,16 @@ export interface GetLongRunningPollerOptions<TResponse> {
|
|
|
26
26
|
* The api-version of the LRO
|
|
27
27
|
*/
|
|
28
28
|
apiVersion?: KnownApiVersions;
|
|
29
|
+
/**
|
|
30
|
+
* Additional headers to include on poll requests (e.g. feature-flag headers required by the endpoint).
|
|
31
|
+
*/
|
|
32
|
+
pollHeaders?: Record<string, string>;
|
|
33
|
+
/**
|
|
34
|
+
* Optional mapping from service-specific status values to the core-lro status values
|
|
35
|
+
* ("running", "succeeded", "failed", "canceled"). Use this when the service returns
|
|
36
|
+
* non-standard terminal status strings (e.g. "completed" instead of "succeeded").
|
|
37
|
+
*/
|
|
38
|
+
statusNormalizations?: Record<string, OperationStatus>;
|
|
29
39
|
/**
|
|
30
40
|
* The function to get the initial response
|
|
31
41
|
*/
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"pollingHelpers.d.ts","sourceRoot":"","sources":["../../../src/static-helpers/pollingHelpers.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,UAAU,EACV,cAAc,EACd,sBAAsB,EAGvB,MAAM,iBAAiB,CAAC;AAGzB,OAAO,KAAK,EAAE,MAAM,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAE7E,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAE5D,MAAM,WAAW,2BAA2B,CAAC,SAAS;IACpD,sDAAsD;IACtD,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B;;OAEG;IACH,WAAW,CAAC,EAAE,eAAe,CAAC;IAC9B;;OAEG;IACH,sBAAsB,CAAC,EAAE,sBAAsB,CAAC;IAChD;;;OAGG;IACH,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;OAEG;IACH,UAAU,CAAC,EAAE,gBAAgB,CAAC;IAC9B;;OAEG;IACH,kBAAkB,CAAC,EAAE,MAAM,WAAW,CAAC,SAAS,CAAC,CAAC;CACnD;AACD,wBAAgB,oBAAoB,CAAC,SAAS,SAAS,qBAAqB,EAAE,OAAO,GAAG,IAAI,EAC1F,MAAM,EAAE,MAAM,EACd,mBAAmB,EAAE,CAAC,MAAM,EAAE,SAAS,KAAK,OAAO,CAAC,OAAO,CAAC,EAC5D,gBAAgB,EAAE,MAAM,EAAE,EAC1B,OAAO,EAAE,2BAA2B,CAAC,SAAS,CAAC,GAC9C,UAAU,CAAC,cAAc,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,
|
|
1
|
+
{"version":3,"file":"pollingHelpers.d.ts","sourceRoot":"","sources":["../../../src/static-helpers/pollingHelpers.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,UAAU,EACV,cAAc,EACd,eAAe,EACf,sBAAsB,EAGvB,MAAM,iBAAiB,CAAC;AAGzB,OAAO,KAAK,EAAE,MAAM,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAE7E,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAE5D,MAAM,WAAW,2BAA2B,CAAC,SAAS;IACpD,sDAAsD;IACtD,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B;;OAEG;IACH,WAAW,CAAC,EAAE,eAAe,CAAC;IAC9B;;OAEG;IACH,sBAAsB,CAAC,EAAE,sBAAsB,CAAC;IAChD;;;OAGG;IACH,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;OAEG;IACH,UAAU,CAAC,EAAE,gBAAgB,CAAC;IAC9B;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACrC;;;;OAIG;IACH,oBAAoB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;IACvD;;OAEG;IACH,kBAAkB,CAAC,EAAE,MAAM,WAAW,CAAC,SAAS,CAAC,CAAC;CACnD;AACD,wBAAgB,oBAAoB,CAAC,SAAS,SAAS,qBAAqB,EAAE,OAAO,GAAG,IAAI,EAC1F,MAAM,EAAE,MAAM,EACd,mBAAmB,EAAE,CAAC,MAAM,EAAE,SAAS,KAAK,OAAO,CAAC,OAAO,CAAC,EAC5D,gBAAgB,EAAE,MAAM,EAAE,EAC1B,OAAO,EAAE,2BAA2B,CAAC,SAAS,CAAC,GAC9C,UAAU,CAAC,cAAc,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,CA6D9C"}
|
|
@@ -23,7 +23,7 @@ module.exports = __toCommonJS(pollingHelpers_exports);
|
|
|
23
23
|
var import_core_lro = require("@azure/core-lro");
|
|
24
24
|
var import_core_client = require("@azure-rest/core-client");
|
|
25
25
|
function getLongRunningPoller(client, processResponseBody, expectedStatuses, options) {
|
|
26
|
-
const { restoreFrom, getInitialResponse, apiVersion } = options;
|
|
26
|
+
const { restoreFrom, getInitialResponse, apiVersion, pollHeaders, statusNormalizations } = options;
|
|
27
27
|
if (!restoreFrom && !getInitialResponse) {
|
|
28
28
|
throw new Error("Either restoreFrom or getInitialResponse must be specified");
|
|
29
29
|
}
|
|
@@ -35,7 +35,7 @@ function getLongRunningPoller(client, processResponseBody, expectedStatuses, opt
|
|
|
35
35
|
throw new Error("getInitialResponse is required when initializing a new poller");
|
|
36
36
|
}
|
|
37
37
|
initialResponse = await getInitialResponse();
|
|
38
|
-
return getLroResponse(initialResponse, expectedStatuses);
|
|
38
|
+
return getLroResponse(initialResponse, expectedStatuses, statusNormalizations);
|
|
39
39
|
},
|
|
40
40
|
sendPollRequest: async (path, pollOptions) => {
|
|
41
41
|
function abortListener() {
|
|
@@ -57,12 +57,12 @@ function getLongRunningPoller(client, processResponseBody, expectedStatuses, opt
|
|
|
57
57
|
let response;
|
|
58
58
|
try {
|
|
59
59
|
const pollingPath = apiVersion ? addApiVersionToUrl(path, apiVersion) : path;
|
|
60
|
-
response = await client.pathUnchecked(pollingPath).get({ abortSignal });
|
|
60
|
+
response = await client.pathUnchecked(pollingPath).get({ abortSignal, headers: pollHeaders });
|
|
61
61
|
} finally {
|
|
62
62
|
options.abortSignal?.removeEventListener("abort", abortListener);
|
|
63
63
|
pollOptions?.abortSignal?.removeEventListener("abort", abortListener);
|
|
64
64
|
}
|
|
65
|
-
return getLroResponse(response, expectedStatuses);
|
|
65
|
+
return getLroResponse(response, expectedStatuses, statusNormalizations);
|
|
66
66
|
}
|
|
67
67
|
};
|
|
68
68
|
return (0, import_core_lro.createHttpPoller)(poller, {
|
|
@@ -74,27 +74,35 @@ function getLongRunningPoller(client, processResponseBody, expectedStatuses, opt
|
|
|
74
74
|
}
|
|
75
75
|
});
|
|
76
76
|
}
|
|
77
|
-
function getLroResponse(response, expectedStatuses) {
|
|
77
|
+
function getLroResponse(response, expectedStatuses, statusNormalizations) {
|
|
78
78
|
if (!expectedStatuses.includes(response.status)) {
|
|
79
79
|
throw (0, import_core_client.createRestError)(response);
|
|
80
80
|
}
|
|
81
|
+
const body = response.body;
|
|
82
|
+
const rawStatus = body?.["status"];
|
|
83
|
+
const normalizedBody = statusNormalizations && typeof rawStatus === "string" && rawStatus in statusNormalizations ? { ...body, status: statusNormalizations[rawStatus] } : body;
|
|
81
84
|
return {
|
|
82
85
|
flatResponse: response,
|
|
83
86
|
rawResponse: {
|
|
84
87
|
...response,
|
|
85
88
|
statusCode: Number.parseInt(response.status),
|
|
86
|
-
body:
|
|
89
|
+
body: normalizedBody
|
|
87
90
|
}
|
|
88
91
|
};
|
|
89
92
|
}
|
|
90
93
|
function addApiVersionToUrl(url, apiVersion) {
|
|
91
94
|
const urlObj = new URL(url, "https://microsoft.com");
|
|
92
|
-
|
|
95
|
+
const existingVersion = urlObj.searchParams.get("api-version");
|
|
96
|
+
if (!existingVersion) {
|
|
93
97
|
return `${url}${Array.from(urlObj.searchParams.keys()).length > 0 ? "&" : "?"}api-version=${apiVersion}`;
|
|
94
98
|
}
|
|
95
|
-
|
|
99
|
+
if (existingVersion === apiVersion) {
|
|
100
|
+
return url;
|
|
101
|
+
}
|
|
102
|
+
return url.replace(`api-version=${existingVersion}`, `api-version=${apiVersion}`);
|
|
96
103
|
}
|
|
97
104
|
// Annotate the CommonJS export names for ESM import in node:
|
|
98
105
|
0 && (module.exports = {
|
|
99
106
|
getLongRunningPoller
|
|
100
107
|
});
|
|
108
|
+
//# sourceMappingURL=pollingHelpers.js.map
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
{
|
|
2
2
|
"version": 3,
|
|
3
|
-
"sources": ["
|
|
4
|
-
"sourcesContent": ["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type {\n PollerLike,\n OperationState,\n ResourceLocationConfig,\n RunningOperation,\n OperationResponse,\n} from \"@azure/core-lro\";\nimport { createHttpPoller } from \"@azure/core-lro\";\n\nimport type { Client, PathUncheckedResponse } from \"@azure-rest/core-client\";\nimport { createRestError } from \"@azure-rest/core-client\";\nimport type { AbortSignalLike } from \"@azure/abort-controller\";\nimport type { KnownApiVersions } from \"../models/models.js\";\n\nexport interface GetLongRunningPollerOptions<TResponse> {\n /** Delay to wait until next poll, in milliseconds. */\n updateIntervalInMs?: number;\n /**\n * The signal which can be used to abort requests.\n */\n abortSignal?: AbortSignalLike;\n /**\n * The potential location of the result of the LRO if specified by the LRO extension in the swagger.\n */\n resourceLocationConfig?: ResourceLocationConfig;\n /**\n * The original url of the LRO\n * Should not be null when restoreFrom is set\n */\n initialRequestUrl?: string;\n /**\n * A serialized poller which can be used to resume an existing paused Long-Running-Operation.\n */\n restoreFrom?: string;\n /**\n * The api-version of the LRO\n */\n apiVersion?: KnownApiVersions;\n /**\n * The function to get the initial response\n */\n getInitialResponse?: () => PromiseLike<TResponse>;\n}\nexport function getLongRunningPoller<TResponse extends PathUncheckedResponse, TResult = void>(\n client: Client,\n processResponseBody: (result: TResponse) => Promise<TResult>,\n expectedStatuses: string[],\n options: GetLongRunningPollerOptions<TResponse>,\n): PollerLike<OperationState<TResult>, TResult> {\n const { restoreFrom, getInitialResponse, apiVersion }
|
|
5
|
-
"mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;
|
|
3
|
+
"sources": ["../../../src/static-helpers/pollingHelpers.ts"],
|
|
4
|
+
"sourcesContent": ["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type {\n PollerLike,\n OperationState,\n OperationStatus,\n ResourceLocationConfig,\n RunningOperation,\n OperationResponse,\n} from \"@azure/core-lro\";\nimport { createHttpPoller } from \"@azure/core-lro\";\n\nimport type { Client, PathUncheckedResponse } from \"@azure-rest/core-client\";\nimport { createRestError } from \"@azure-rest/core-client\";\nimport type { AbortSignalLike } from \"@azure/abort-controller\";\nimport type { KnownApiVersions } from \"../models/models.js\";\n\nexport interface GetLongRunningPollerOptions<TResponse> {\n /** Delay to wait until next poll, in milliseconds. */\n updateIntervalInMs?: number;\n /**\n * The signal which can be used to abort requests.\n */\n abortSignal?: AbortSignalLike;\n /**\n * The potential location of the result of the LRO if specified by the LRO extension in the swagger.\n */\n resourceLocationConfig?: ResourceLocationConfig;\n /**\n * The original url of the LRO\n * Should not be null when restoreFrom is set\n */\n initialRequestUrl?: string;\n /**\n * A serialized poller which can be used to resume an existing paused Long-Running-Operation.\n */\n restoreFrom?: string;\n /**\n * The api-version of the LRO\n */\n apiVersion?: KnownApiVersions;\n /**\n * Additional headers to include on poll requests (e.g. feature-flag headers required by the endpoint).\n */\n pollHeaders?: Record<string, string>;\n /**\n * Optional mapping from service-specific status values to the core-lro status values\n * (\"running\", \"succeeded\", \"failed\", \"canceled\"). Use this when the service returns\n * non-standard terminal status strings (e.g. \"completed\" instead of \"succeeded\").\n */\n statusNormalizations?: Record<string, OperationStatus>;\n /**\n * The function to get the initial response\n */\n getInitialResponse?: () => PromiseLike<TResponse>;\n}\nexport function getLongRunningPoller<TResponse extends PathUncheckedResponse, TResult = void>(\n client: Client,\n processResponseBody: (result: TResponse) => Promise<TResult>,\n expectedStatuses: string[],\n options: GetLongRunningPollerOptions<TResponse>,\n): PollerLike<OperationState<TResult>, TResult> {\n const { restoreFrom, getInitialResponse, apiVersion, pollHeaders, statusNormalizations } =\n options;\n if (!restoreFrom && !getInitialResponse) {\n throw new Error(\"Either restoreFrom or getInitialResponse must be specified\");\n }\n let initialResponse: TResponse | undefined = undefined;\n const pollAbortController = new AbortController();\n const poller: RunningOperation<TResponse> = {\n sendInitialRequest: async () => {\n if (!getInitialResponse) {\n throw new Error(\"getInitialResponse is required when initializing a new poller\");\n }\n initialResponse = await getInitialResponse();\n return getLroResponse(initialResponse, expectedStatuses, statusNormalizations);\n },\n sendPollRequest: async (\n path: string,\n pollOptions?: {\n abortSignal?: AbortSignalLike;\n },\n ) => {\n // The poll request would both listen to the user provided abort signal and the poller's own abort signal\n function abortListener(): void {\n pollAbortController.abort();\n }\n const abortSignal = pollAbortController.signal;\n if (options.abortSignal?.aborted) {\n pollAbortController.abort();\n } else if (pollOptions?.abortSignal?.aborted) {\n pollAbortController.abort();\n } else if (!abortSignal.aborted) {\n options.abortSignal?.addEventListener(\"abort\", abortListener, {\n once: true,\n });\n pollOptions?.abortSignal?.addEventListener(\"abort\", abortListener, {\n once: true,\n });\n }\n let response;\n try {\n const pollingPath = apiVersion ? addApiVersionToUrl(path, apiVersion) : path;\n response = await client\n .pathUnchecked(pollingPath)\n .get({ abortSignal, headers: pollHeaders });\n } finally {\n options.abortSignal?.removeEventListener(\"abort\", abortListener);\n pollOptions?.abortSignal?.removeEventListener(\"abort\", abortListener);\n }\n\n return getLroResponse(response as TResponse, expectedStatuses, statusNormalizations);\n },\n };\n return createHttpPoller(poller, {\n intervalInMs: options?.updateIntervalInMs,\n resourceLocationConfig: options?.resourceLocationConfig,\n restoreFrom: options?.restoreFrom,\n processResult: (result: unknown) => {\n return processResponseBody(result as TResponse);\n },\n });\n}\n/**\n * Converts a Rest Client response to a response that the LRO implementation understands\n * @param response - a rest client http response\n * @param deserializeFn - deserialize function to convert Rest response to modular output\n * @returns - An LRO response that the LRO implementation understands\n */\nfunction getLroResponse<TResponse extends PathUncheckedResponse>(\n response: TResponse,\n expectedStatuses: string[],\n statusNormalizations?: Record<string, OperationStatus>,\n): OperationResponse<TResponse> {\n if (!expectedStatuses.includes(response.status)) {\n throw createRestError(response);\n }\n\n const body = response.body as Record<string, unknown> | undefined;\n const rawStatus = body?.[\"status\"];\n const normalizedBody =\n statusNormalizations && typeof rawStatus === \"string\" && rawStatus in statusNormalizations\n ? { ...body, status: statusNormalizations[rawStatus] }\n : body;\n\n return {\n flatResponse: response,\n rawResponse: {\n ...response,\n statusCode: Number.parseInt(response.status),\n body: normalizedBody,\n },\n };\n}\n\n/**\n * Sets the api-version query parameter on a URL, replacing any existing value.\n * @param url - the URL to modify\n * @param apiVersion - the API version to set\n * @returns - the URL with the api-version query parameter set\n */\nfunction addApiVersionToUrl(url: string, apiVersion: string): string {\n // The base URL is only used for parsing and won't appear in the returned URL\n const urlObj = new URL(url, \"https://microsoft.com\");\n const existingVersion = urlObj.searchParams.get(\"api-version\");\n if (!existingVersion) {\n // Append one if there is no apiVersion\n return `${url}${\n Array.from(urlObj.searchParams.keys()).length > 0 ? \"&\" : \"?\"\n }api-version=${apiVersion}`;\n }\n\n if (existingVersion === apiVersion) {\n return url;\n }\n // Replace the service-returned api-version with the client's configured version\n return url.replace(`api-version=${existingVersion}`, `api-version=${apiVersion}`);\n}\n"],
|
|
5
|
+
"mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAWA,sBAAiC;AAGjC,yBAAgC;AA2CzB,SAAS,qBACd,QACA,qBACA,kBACA,SAC8C;AAC9C,QAAM,EAAE,aAAa,oBAAoB,YAAY,aAAa,qBAAqB,IACrF;AACF,MAAI,CAAC,eAAe,CAAC,oBAAoB;AACvC,UAAM,IAAI,MAAM,4DAA4D;AAAA,EAC9E;AACA,MAAI,kBAAyC;AAC7C,QAAM,sBAAsB,IAAI,gBAAgB;AAChD,QAAM,SAAsC;AAAA,IAC1C,oBAAoB,YAAY;AAC9B,UAAI,CAAC,oBAAoB;AACvB,cAAM,IAAI,MAAM,+DAA+D;AAAA,MACjF;AACA,wBAAkB,MAAM,mBAAmB;AAC3C,aAAO,eAAe,iBAAiB,kBAAkB,oBAAoB;AAAA,IAC/E;AAAA,IACA,iBAAiB,OACf,MACA,gBAGG;AAEH,eAAS,gBAAsB;AAC7B,4BAAoB,MAAM;AAAA,MAC5B;AACA,YAAM,cAAc,oBAAoB;AACxC,UAAI,QAAQ,aAAa,SAAS;AAChC,4BAAoB,MAAM;AAAA,MAC5B,WAAW,aAAa,aAAa,SAAS;AAC5C,4BAAoB,MAAM;AAAA,MAC5B,WAAW,CAAC,YAAY,SAAS;AAC/B,gBAAQ,aAAa,iBAAiB,SAAS,eAAe;AAAA,UAC5D,MAAM;AAAA,QACR,CAAC;AACD,qBAAa,aAAa,iBAAiB,SAAS,eAAe;AAAA,UACjE,MAAM;AAAA,QACR,CAAC;AAAA,MACH;AACA,UAAI;AACJ,UAAI;AACF,cAAM,cAAc,aAAa,mBAAmB,MAAM,UAAU,IAAI;AACxE,mBAAW,MAAM,OACd,cAAc,WAAW,EACzB,IAAI,EAAE,aAAa,SAAS,YAAY,CAAC;AAAA,MAC9C,UAAE;AACA,gBAAQ,aAAa,oBAAoB,SAAS,aAAa;AAC/D,qBAAa,aAAa,oBAAoB,SAAS,aAAa;AAAA,MACtE;AAEA,aAAO,eAAe,UAAuB,kBAAkB,oBAAoB;AAAA,IACrF;AAAA,EACF;AACA,aAAO,kCAAiB,QAAQ;AAAA,IAC9B,cAAc,SAAS;AAAA,IACvB,wBAAwB,SAAS;AAAA,IACjC,aAAa,SAAS;AAAA,IACtB,eAAe,CAAC,WAAoB;AAClC,aAAO,oBAAoB,MAAmB;AAAA,IAChD;AAAA,EACF,CAAC;AACH;AAOA,SAAS,eACP,UACA,kBACA,sBAC8B;AAC9B,MAAI,CAAC,iBAAiB,SAAS,SAAS,MAAM,GAAG;AAC/C,cAAM,oCAAgB,QAAQ;AAAA,EAChC;AAEA,QAAM,OAAO,SAAS;AACtB,QAAM,YAAY,OAAO,QAAQ;AACjC,QAAM,iBACJ,wBAAwB,OAAO,cAAc,YAAY,aAAa,uBAClE,EAAE,GAAG,MAAM,QAAQ,qBAAqB,SAAS,EAAE,IACnD;AAEN,SAAO;AAAA,IACL,cAAc;AAAA,IACd,aAAa;AAAA,MACX,GAAG;AAAA,MACH,YAAY,OAAO,SAAS,SAAS,MAAM;AAAA,MAC3C,MAAM;AAAA,IACR;AAAA,EACF;AACF;AAQA,SAAS,mBAAmB,KAAa,YAA4B;AAEnE,QAAM,SAAS,IAAI,IAAI,KAAK,uBAAuB;AACnD,QAAM,kBAAkB,OAAO,aAAa,IAAI,aAAa;AAC7D,MAAI,CAAC,iBAAiB;AAEpB,WAAO,GAAG,GAAG,GACX,MAAM,KAAK,OAAO,aAAa,KAAK,CAAC,EAAE,SAAS,IAAI,MAAM,GAC5D,eAAe,UAAU;AAAA,EAC3B;AAEA,MAAI,oBAAoB,YAAY;AAClC,WAAO;AAAA,EACT;AAEA,SAAO,IAAI,QAAQ,eAAe,eAAe,IAAI,eAAe,UAAU,EAAE;AAClF;",
|
|
6
6
|
"names": []
|
|
7
7
|
}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"version": 3,
|
|
3
|
-
"sources": ["
|
|
3
|
+
"sources": ["../../../../src/static-helpers/serialization/serialize-record.ts"],
|
|
4
4
|
"sourcesContent": ["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\n// eslint-disable-next-line @typescript-eslint/explicit-function-return-type, @typescript-eslint/explicit-module-boundary-types\nexport function serializeRecord(item: any, excludes?: string[], serializer?: (item: any) => any) {\n const nextExcludes = excludes ?? [];\n const res: any = {};\n for (const key of Object.keys(item)) {\n if (nextExcludes.includes(key) || item[key] === undefined) {\n continue;\n }\n if (serializer) {\n res[key] = serializer(item[key]);\n } else {\n res[key] = item[key] as any;\n }\n }\n return res;\n}\n"],
|
|
5
5
|
"mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAIO,SAAS,gBAAgB,MAAW,UAAqB,YAAiC;AAC/F,QAAM,eAAe,YAAY,CAAC;AAClC,QAAM,MAAW,CAAC;AAClB,aAAW,OAAO,OAAO,KAAK,IAAI,GAAG;AACnC,QAAI,aAAa,SAAS,GAAG,KAAK,KAAK,GAAG,MAAM,QAAW;AACzD;AAAA,IACF;AACA,QAAI,YAAY;AACd,UAAI,GAAG,IAAI,WAAW,KAAK,GAAG,CAAC;AAAA,IACjC,OAAO;AACL,UAAI,GAAG,IAAI,KAAK,GAAG;AAAA,IACrB;AAAA,EACF;AACA,SAAO;AACT;",
|
|
6
6
|
"names": []
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"version": 3,
|
|
3
|
-
"sources": ["
|
|
3
|
+
"sources": ["../../../src/static-helpers/urlTemplate.ts"],
|
|
4
4
|
"sourcesContent": ["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\n// ---------------------\n// interfaces\n// ---------------------\ninterface ValueOptions {\n isFirst: boolean; // is first value in the expression\n op?: string; // operator\n varValue?: any; // variable value\n varName?: string; // variable name\n modifier?: string; // modifier e.g *\n reserved?: boolean; // if true we'll keep reserved words with not encoding\n}\n\nexport interface UrlTemplateOptions {\n // if set to true, reserved characters will not be encoded\n allowReserved?: boolean;\n}\n\n// ---------------------\n// helpers\n// ---------------------\nfunction encodeComponent(val: string, reserved?: boolean, op?: string): string {\n return (reserved ?? op === \"+\") || op === \"#\"\n ? encodeReservedComponent(val)\n : encodeRFC3986URIComponent(val);\n}\n\nfunction encodeReservedComponent(str: string): string {\n return str\n .split(/(%[0-9A-Fa-f]{2})/g)\n .map((part) => (!/%[0-9A-Fa-f]/.test(part) ? encodeURI(part) : part))\n .join(\"\");\n}\n\nfunction encodeRFC3986URIComponent(str: string): string {\n return encodeURIComponent(str).replace(\n /[!'()*]/g,\n (c) => `%${c.charCodeAt(0).toString(16).toUpperCase()}`,\n );\n}\n\nfunction isDefined(val: any): boolean {\n return val !== undefined && val !== null;\n}\n\nfunction getNamedAndIfEmpty(op?: string): [boolean, string] {\n return [!!op && [\";\", \"?\", \"&\"].includes(op), !!op && [\"?\", \"&\"].includes(op) ? \"=\" : \"\"];\n}\n\nfunction getFirstOrSep(op?: string, isFirst = false): string {\n if (isFirst) {\n return !op || op === \"+\" ? \"\" : op;\n } else if (!op || op === \"+\" || op === \"#\") {\n return \",\";\n } else if (op === \"?\") {\n return \"&\";\n } else {\n return op;\n }\n}\n\nfunction getExpandedValue(option: ValueOptions): string {\n let isFirst = option.isFirst;\n const { op, varName, varValue: value, reserved } = option;\n const vals: string[] = [];\n const [named, ifEmpty] = getNamedAndIfEmpty(op);\n\n if (Array.isArray(value)) {\n for (const val of value.filter(isDefined)) {\n // prepare the following parts: separator, varName, value\n vals.push(`${getFirstOrSep(op, isFirst)}`);\n if (named && varName) {\n vals.push(`${encodeURIComponent(varName)}`);\n if (val === \"\") {\n vals.push(ifEmpty);\n } else {\n vals.push(\"=\");\n }\n }\n vals.push(encodeComponent(val, reserved, op));\n isFirst = false;\n }\n } else if (typeof value === \"object\") {\n for (const key of Object.keys(value)) {\n const val = value[key];\n if (!isDefined(val)) {\n continue;\n }\n // prepare the following parts: separator, key, value\n vals.push(`${getFirstOrSep(op, isFirst)}`);\n if (key) {\n vals.push(`${encodeURIComponent(key)}`);\n if (named && val === \"\") {\n vals.push(ifEmpty);\n } else {\n vals.push(\"=\");\n }\n }\n vals.push(encodeComponent(val, reserved, op));\n isFirst = false;\n }\n }\n return vals.join(\"\");\n}\n\nfunction getNonExpandedValue(option: ValueOptions): string | undefined {\n const { op, varName, varValue: value, isFirst, reserved } = option;\n const vals: string[] = [];\n const first = getFirstOrSep(op, isFirst);\n const [named, ifEmpty] = getNamedAndIfEmpty(op);\n if (named && varName) {\n vals.push(encodeComponent(varName, reserved, op));\n if (value === \"\") {\n if (!ifEmpty) {\n vals.push(ifEmpty);\n }\n return !vals.join(\"\") ? undefined : `${first}${vals.join(\"\")}`;\n }\n vals.push(\"=\");\n }\n\n const items = [];\n if (Array.isArray(value)) {\n for (const val of value.filter(isDefined)) {\n items.push(encodeComponent(val, reserved, op));\n }\n } else if (typeof value === \"object\") {\n for (const key of Object.keys(value)) {\n if (!isDefined(value[key])) {\n continue;\n }\n items.push(encodeRFC3986URIComponent(key));\n items.push(encodeComponent(value[key], reserved, op));\n }\n }\n vals.push(items.join(\",\"));\n return !vals.join(\",\") ? undefined : `${first}${vals.join(\"\")}`;\n}\n\nfunction getVarValue(option: ValueOptions): string | undefined {\n const { op, varName, modifier, isFirst, reserved, varValue: value } = option;\n\n if (!isDefined(value)) {\n return undefined;\n } else if ([\"string\", \"number\", \"boolean\"].includes(typeof value)) {\n let val = value.toString();\n const [named, ifEmpty] = getNamedAndIfEmpty(op);\n const vals: string[] = [getFirstOrSep(op, isFirst)];\n if (named && varName) {\n // No need to encode varName considering it is already encoded\n vals.push(varName);\n if (val === \"\") {\n vals.push(ifEmpty);\n } else {\n vals.push(\"=\");\n }\n }\n if (modifier && modifier !== \"*\") {\n val = val.substring(0, parseInt(modifier, 10));\n }\n vals.push(encodeComponent(val, reserved, op));\n return vals.join(\"\");\n } else if (modifier === \"*\") {\n return getExpandedValue(option);\n } else {\n return getNonExpandedValue(option);\n }\n}\n\n// ---------------------------------------------------------------------------------------------------\n// This is an implementation of RFC 6570 URI Template: https://datatracker.ietf.org/doc/html/rfc6570.\n// ---------------------------------------------------------------------------------------------------\nexport function expandUrlTemplate(\n template: string,\n context: Record<string, any>,\n option?: UrlTemplateOptions,\n): string {\n const r = template.replace(/\\{([^{}]+)\\}|([^{}]+)/g, (_, expr, text) => {\n if (!expr) {\n return encodeReservedComponent(text);\n }\n let op;\n if ([\"+\", \"#\", \".\", \"/\", \";\", \"?\", \"&\"].includes(expr[0])) {\n op = expr[0];\n expr = expr.slice(1);\n }\n const varList = expr.split(/,/g);\n const result = [];\n for (const varSpec of varList) {\n const varMatch = /([^:*]*)(?::(\\d+)|(\\*))?/.exec(varSpec);\n if (!varMatch || !varMatch[1]) {\n continue;\n }\n const varValue = getVarValue({\n isFirst: result.length === 0,\n op,\n varValue: context[varMatch[1]],\n varName: varMatch[1],\n modifier: varMatch[2] || varMatch[3],\n reserved: option?.allowReserved,\n });\n if (varValue) {\n result.push(varValue);\n }\n }\n return result.join(\"\");\n });\n\n return normalizeUnreserved(r);\n}\n\n/**\n * Normalize an expanded URI by decoding percent-encoded unreserved characters.\n * RFC 3986 unreserved: \"-\" / \".\" / \"~\"\n */\nfunction normalizeUnreserved(uri: string): string {\n return uri.replace(/%([0-9A-Fa-f]{2})/g, (match, hex) => {\n const char = String.fromCharCode(parseInt(hex, 16));\n // Decode only if it's unreserved\n if (/[-.~]/.test(char)) {\n return char;\n }\n return match; // leave other encodings intact\n });\n}\n"],
|
|
5
5
|
"mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAuBA,SAAS,gBAAgB,KAAa,UAAoB,IAAqB;AAC7E,UAAQ,YAAY,OAAO,QAAQ,OAAO,MACtC,wBAAwB,GAAG,IAC3B,0BAA0B,GAAG;AACnC;AAEA,SAAS,wBAAwB,KAAqB;AACpD,SAAO,IACJ,MAAM,oBAAoB,EAC1B,IAAI,CAAC,SAAU,CAAC,eAAe,KAAK,IAAI,IAAI,UAAU,IAAI,IAAI,IAAK,EACnE,KAAK,EAAE;AACZ;AAEA,SAAS,0BAA0B,KAAqB;AACtD,SAAO,mBAAmB,GAAG,EAAE;AAAA,IAC7B;AAAA,IACA,CAAC,MAAM,IAAI,EAAE,WAAW,CAAC,EAAE,SAAS,EAAE,EAAE,YAAY,CAAC;AAAA,EACvD;AACF;AAEA,SAAS,UAAU,KAAmB;AACpC,SAAO,QAAQ,UAAa,QAAQ;AACtC;AAEA,SAAS,mBAAmB,IAAgC;AAC1D,SAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,KAAK,GAAG,EAAE,SAAS,EAAE,GAAG,CAAC,CAAC,MAAM,CAAC,KAAK,GAAG,EAAE,SAAS,EAAE,IAAI,MAAM,EAAE;AAC1F;AAEA,SAAS,cAAc,IAAa,UAAU,OAAe;AAC3D,MAAI,SAAS;AACX,WAAO,CAAC,MAAM,OAAO,MAAM,KAAK;AAAA,EAClC,WAAW,CAAC,MAAM,OAAO,OAAO,OAAO,KAAK;AAC1C,WAAO;AAAA,EACT,WAAW,OAAO,KAAK;AACrB,WAAO;AAAA,EACT,OAAO;AACL,WAAO;AAAA,EACT;AACF;AAEA,SAAS,iBAAiB,QAA8B;AACtD,MAAI,UAAU,OAAO;AACrB,QAAM,EAAE,IAAI,SAAS,UAAU,OAAO,SAAS,IAAI;AACnD,QAAM,OAAiB,CAAC;AACxB,QAAM,CAAC,OAAO,OAAO,IAAI,mBAAmB,EAAE;AAE9C,MAAI,MAAM,QAAQ,KAAK,GAAG;AACxB,eAAW,OAAO,MAAM,OAAO,SAAS,GAAG;AAEzC,WAAK,KAAK,GAAG,cAAc,IAAI,OAAO,CAAC,EAAE;AACzC,UAAI,SAAS,SAAS;AACpB,aAAK,KAAK,GAAG,mBAAmB,OAAO,CAAC,EAAE;AAC1C,YAAI,QAAQ,IAAI;AACd,eAAK,KAAK,OAAO;AAAA,QACnB,OAAO;AACL,eAAK,KAAK,GAAG;AAAA,QACf;AAAA,MACF;AACA,WAAK,KAAK,gBAAgB,KAAK,UAAU,EAAE,CAAC;AAC5C,gBAAU;AAAA,IACZ;AAAA,EACF,WAAW,OAAO,UAAU,UAAU;AACpC,eAAW,OAAO,OAAO,KAAK,KAAK,GAAG;AACpC,YAAM,MAAM,MAAM,GAAG;AACrB,UAAI,CAAC,UAAU,GAAG,GAAG;AACnB;AAAA,MACF;AAEA,WAAK,KAAK,GAAG,cAAc,IAAI,OAAO,CAAC,EAAE;AACzC,UAAI,KAAK;AACP,aAAK,KAAK,GAAG,mBAAmB,GAAG,CAAC,EAAE;AACtC,YAAI,SAAS,QAAQ,IAAI;AACvB,eAAK,KAAK,OAAO;AAAA,QACnB,OAAO;AACL,eAAK,KAAK,GAAG;AAAA,QACf;AAAA,MACF;AACA,WAAK,KAAK,gBAAgB,KAAK,UAAU,EAAE,CAAC;AAC5C,gBAAU;AAAA,IACZ;AAAA,EACF;AACA,SAAO,KAAK,KAAK,EAAE;AACrB;AAEA,SAAS,oBAAoB,QAA0C;AACrE,QAAM,EAAE,IAAI,SAAS,UAAU,OAAO,SAAS,SAAS,IAAI;AAC5D,QAAM,OAAiB,CAAC;AACxB,QAAM,QAAQ,cAAc,IAAI,OAAO;AACvC,QAAM,CAAC,OAAO,OAAO,IAAI,mBAAmB,EAAE;AAC9C,MAAI,SAAS,SAAS;AACpB,SAAK,KAAK,gBAAgB,SAAS,UAAU,EAAE,CAAC;AAChD,QAAI,UAAU,IAAI;AAChB,UAAI,CAAC,SAAS;AACZ,aAAK,KAAK,OAAO;AAAA,MACnB;AACA,aAAO,CAAC,KAAK,KAAK,EAAE,IAAI,SAAY,GAAG,KAAK,GAAG,KAAK,KAAK,EAAE,CAAC;AAAA,IAC9D;AACA,SAAK,KAAK,GAAG;AAAA,EACf;AAEA,QAAM,QAAQ,CAAC;AACf,MAAI,MAAM,QAAQ,KAAK,GAAG;AACxB,eAAW,OAAO,MAAM,OAAO,SAAS,GAAG;AACzC,YAAM,KAAK,gBAAgB,KAAK,UAAU,EAAE,CAAC;AAAA,IAC/C;AAAA,EACF,WAAW,OAAO,UAAU,UAAU;AACpC,eAAW,OAAO,OAAO,KAAK,KAAK,GAAG;AACpC,UAAI,CAAC,UAAU,MAAM,GAAG,CAAC,GAAG;AAC1B;AAAA,MACF;AACA,YAAM,KAAK,0BAA0B,GAAG,CAAC;AACzC,YAAM,KAAK,gBAAgB,MAAM,GAAG,GAAG,UAAU,EAAE,CAAC;AAAA,IACtD;AAAA,EACF;AACA,OAAK,KAAK,MAAM,KAAK,GAAG,CAAC;AACzB,SAAO,CAAC,KAAK,KAAK,GAAG,IAAI,SAAY,GAAG,KAAK,GAAG,KAAK,KAAK,EAAE,CAAC;AAC/D;AAEA,SAAS,YAAY,QAA0C;AAC7D,QAAM,EAAE,IAAI,SAAS,UAAU,SAAS,UAAU,UAAU,MAAM,IAAI;AAEtE,MAAI,CAAC,UAAU,KAAK,GAAG;AACrB,WAAO;AAAA,EACT,WAAW,CAAC,UAAU,UAAU,SAAS,EAAE,SAAS,OAAO,KAAK,GAAG;AACjE,QAAI,MAAM,MAAM,SAAS;AACzB,UAAM,CAAC,OAAO,OAAO,IAAI,mBAAmB,EAAE;AAC9C,UAAM,OAAiB,CAAC,cAAc,IAAI,OAAO,CAAC;AAClD,QAAI,SAAS,SAAS;AAEpB,WAAK,KAAK,OAAO;AACjB,UAAI,QAAQ,IAAI;AACd,aAAK,KAAK,OAAO;AAAA,MACnB,OAAO;AACL,aAAK,KAAK,GAAG;AAAA,MACf;AAAA,IACF;AACA,QAAI,YAAY,aAAa,KAAK;AAChC,YAAM,IAAI,UAAU,GAAG,SAAS,UAAU,EAAE,CAAC;AAAA,IAC/C;AACA,SAAK,KAAK,gBAAgB,KAAK,UAAU,EAAE,CAAC;AAC5C,WAAO,KAAK,KAAK,EAAE;AAAA,EACrB,WAAW,aAAa,KAAK;AAC3B,WAAO,iBAAiB,MAAM;AAAA,EAChC,OAAO;AACL,WAAO,oBAAoB,MAAM;AAAA,EACnC;AACF;AAKO,SAAS,kBACd,UACA,SACA,QACQ;AACR,QAAM,IAAI,SAAS,QAAQ,0BAA0B,CAAC,GAAG,MAAM,SAAS;AACtE,QAAI,CAAC,MAAM;AACT,aAAO,wBAAwB,IAAI;AAAA,IACrC;AACA,QAAI;AACJ,QAAI,CAAC,KAAK,KAAK,KAAK,KAAK,KAAK,KAAK,GAAG,EAAE,SAAS,KAAK,CAAC,CAAC,GAAG;AACzD,WAAK,KAAK,CAAC;AACX,aAAO,KAAK,MAAM,CAAC;AAAA,IACrB;AACA,UAAM,UAAU,KAAK,MAAM,IAAI;AAC/B,UAAM,SAAS,CAAC;AAChB,eAAW,WAAW,SAAS;AAC7B,YAAM,WAAW,2BAA2B,KAAK,OAAO;AACxD,UAAI,CAAC,YAAY,CAAC,SAAS,CAAC,GAAG;AAC7B;AAAA,MACF;AACA,YAAM,WAAW,YAAY;AAAA,QAC3B,SAAS,OAAO,WAAW;AAAA,QAC3B;AAAA,QACA,UAAU,QAAQ,SAAS,CAAC,CAAC;AAAA,QAC7B,SAAS,SAAS,CAAC;AAAA,QACnB,UAAU,SAAS,CAAC,KAAK,SAAS,CAAC;AAAA,QACnC,UAAU,QAAQ;AAAA,MACpB,CAAC;AACD,UAAI,UAAU;AACZ,eAAO,KAAK,QAAQ;AAAA,MACtB;AAAA,IACF;AACA,WAAO,OAAO,KAAK,EAAE;AAAA,EACvB,CAAC;AAED,SAAO,oBAAoB,CAAC;AAC9B;AAMA,SAAS,oBAAoB,KAAqB;AAChD,SAAO,IAAI,QAAQ,sBAAsB,CAAC,OAAO,QAAQ;AACvD,UAAM,OAAO,OAAO,aAAa,SAAS,KAAK,EAAE,CAAC;AAElD,QAAI,QAAQ,KAAK,IAAI,GAAG;AACtB,aAAO;AAAA,IACT;AACA,WAAO;AAAA,EACT,CAAC;AACH;",
|
|
6
6
|
"names": []
|
package/dist/commonjs/util.js
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"version": 3,
|
|
3
|
-
"sources": ["
|
|
3
|
+
"sources": ["../../src/util.ts"],
|
|
4
4
|
"sourcesContent": ["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport type { ClientOptions as OpenAIClientOptions } from \"openai\";\nimport type { HttpHeaders, RawHttpHeadersInput } from \"@azure/core-rest-pipeline\";\nimport { createHttpHeaders } from \"@azure/core-rest-pipeline\";\n\n/*\n * @param headers the openai defaultHeaders in openai client options\n * @returns HttpHeaders, normalized from various OpenAI header formats\n */\nexport const normalizeOpenAIHeaders = (\n headers?: OpenAIClientOptions[\"defaultHeaders\"],\n): HttpHeaders => {\n if (!headers) {\n return createHttpHeaders();\n }\n\n // Handle NullableHeaders (OpenAI internal type with values and nulls properties)\n // Check this first before other object checks\n if (\"values\" in headers && \"nulls\" in headers) {\n const result = normalizeOpenAIHeaders(headers.values as OpenAIClientOptions[\"defaultHeaders\"]);\n\n // Remove headers that were explicitly set to null\n if (result && headers.nulls instanceof Set && headers.nulls.size > 0) {\n for (const nullHeader of headers.nulls) {\n result.delete(nullHeader);\n }\n }\n\n return result;\n }\n\n // Handle array format: readonly HeaderValue[][]\n // Check arrays before forEach to avoid catching arrays with the forEach check\n if (Array.isArray(headers)) {\n const rawHeaders: RawHttpHeadersInput = {};\n for (const [key, value] of headers) {\n if (key && value !== null && value !== undefined) {\n const strValue = String(value);\n // Merge multiple headers with the same name using comma separation\n if (rawHeaders[key]) {\n rawHeaders[key] = `${rawHeaders[key]}, ${strValue}`;\n } else {\n rawHeaders[key] = strValue;\n }\n }\n }\n return createHttpHeaders(rawHeaders);\n }\n\n // Handle Headers object (Web API Headers) - use duck typing to avoid Node.js version warnings\n if (\"forEach\" in headers && typeof headers.forEach === \"function\") {\n const rawHeaders: RawHttpHeadersInput = {};\n // Type assertion since we've verified it has forEach\n (headers as { forEach: (callback: (value: string, key: string) => void) => void }).forEach(\n (value, key) => {\n // Merge multiple headers with the same name using comma separation\n if (rawHeaders[key]) {\n rawHeaders[key] = `${rawHeaders[key]}, ${value}`;\n } else {\n rawHeaders[key] = value;\n }\n },\n );\n return createHttpHeaders(rawHeaders);\n }\n\n // Handle Record<string, HeaderValue | readonly HeaderValue[]>\n if (typeof headers === \"object\" && headers !== null) {\n const rawHeaders: RawHttpHeadersInput = {};\n for (const [key, value] of Object.entries(headers)) {\n if (value === null || value === undefined) {\n continue;\n }\n if (Array.isArray(value)) {\n // Join multiple values with comma (standard HTTP header behavior)\n const filtered = value.filter((v) => v !== null && v !== undefined).map(String);\n if (filtered.length > 0) {\n rawHeaders[key] = filtered.join(\", \");\n }\n } else {\n rawHeaders[key] = String(value);\n }\n }\n return createHttpHeaders(rawHeaders);\n }\n\n return createHttpHeaders();\n};\n\n/**\n * Get OpenAI default headers with user-agent set appropriately\n * @param defaultHeaders - The default headers from OpenAI client options\n * @param userAgentPrefix - Optional user agent prefix to prepend from AIProjectClient options\n * @returns HttpHeaders with normalized headers and user-agent set\n */\nexport const getOpenAIDefaultHeaders = (\n defaultHeaders?: OpenAIClientOptions[\"defaultHeaders\"],\n userAgentPrefix?: string,\n): HttpHeaders => {\n const headers = normalizeOpenAIHeaders(defaultHeaders);\n\n if (!headers.has(\"User-Agent\")) {\n const clientIdentifier = userAgentPrefix\n ? `${userAgentPrefix}-AIProjectClient`\n : \"AIProjectClient\";\n headers.set(\"User-Agent\", `${clientIdentifier} OpenAI/JS`);\n }\n\n return headers;\n};\n"],
|
|
5
5
|
"mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAIA,gCAAkC;AAM3B,MAAM,yBAAyB,CACpC,YACgB;AAChB,MAAI,CAAC,SAAS;AACZ,eAAO,6CAAkB;AAAA,EAC3B;AAIA,MAAI,YAAY,WAAW,WAAW,SAAS;AAC7C,UAAM,SAAS,uBAAuB,QAAQ,MAA+C;AAG7F,QAAI,UAAU,QAAQ,iBAAiB,OAAO,QAAQ,MAAM,OAAO,GAAG;AACpE,iBAAW,cAAc,QAAQ,OAAO;AACtC,eAAO,OAAO,UAAU;AAAA,MAC1B;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAIA,MAAI,MAAM,QAAQ,OAAO,GAAG;AAC1B,UAAM,aAAkC,CAAC;AACzC,eAAW,CAAC,KAAK,KAAK,KAAK,SAAS;AAClC,UAAI,OAAO,UAAU,QAAQ,UAAU,QAAW;AAChD,cAAM,WAAW,OAAO,KAAK;AAE7B,YAAI,WAAW,GAAG,GAAG;AACnB,qBAAW,GAAG,IAAI,GAAG,WAAW,GAAG,CAAC,KAAK,QAAQ;AAAA,QACnD,OAAO;AACL,qBAAW,GAAG,IAAI;AAAA,QACpB;AAAA,MACF;AAAA,IACF;AACA,eAAO,6CAAkB,UAAU;AAAA,EACrC;AAGA,MAAI,aAAa,WAAW,OAAO,QAAQ,YAAY,YAAY;AACjE,UAAM,aAAkC,CAAC;AAEzC,IAAC,QAAkF;AAAA,MACjF,CAAC,OAAO,QAAQ;AAEd,YAAI,WAAW,GAAG,GAAG;AACnB,qBAAW,GAAG,IAAI,GAAG,WAAW,GAAG,CAAC,KAAK,KAAK;AAAA,QAChD,OAAO;AACL,qBAAW,GAAG,IAAI;AAAA,QACpB;AAAA,MACF;AAAA,IACF;AACA,eAAO,6CAAkB,UAAU;AAAA,EACrC;AAGA,MAAI,OAAO,YAAY,YAAY,YAAY,MAAM;AACnD,UAAM,aAAkC,CAAC;AACzC,eAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,OAAO,GAAG;AAClD,UAAI,UAAU,QAAQ,UAAU,QAAW;AACzC;AAAA,MACF;AACA,UAAI,MAAM,QAAQ,KAAK,GAAG;AAExB,cAAM,WAAW,MAAM,OAAO,CAAC,MAAM,MAAM,QAAQ,MAAM,MAAS,EAAE,IAAI,MAAM;AAC9E,YAAI,SAAS,SAAS,GAAG;AACvB,qBAAW,GAAG,IAAI,SAAS,KAAK,IAAI;AAAA,QACtC;AAAA,MACF,OAAO;AACL,mBAAW,GAAG,IAAI,OAAO,KAAK;AAAA,MAChC;AAAA,IACF;AACA,eAAO,6CAAkB,UAAU;AAAA,EACrC;AAEA,aAAO,6CAAkB;AAC3B;AAQO,MAAM,0BAA0B,CACrC,gBACA,oBACgB;AAChB,QAAM,UAAU,uBAAuB,cAAc;AAErD,MAAI,CAAC,QAAQ,IAAI,YAAY,GAAG;AAC9B,UAAM,mBAAmB,kBACrB,GAAG,eAAe,qBAClB;AACJ,YAAQ,IAAI,cAAc,GAAG,gBAAgB,YAAY;AAAA,EAC3D;AAEA,SAAO;AACT;",
|
|
6
6
|
"names": []
|
|
@@ -1,3 +1,6 @@
|
|
|
1
|
+
// Copyright (c) Microsoft Corporation.
|
|
2
|
+
// Licensed under the MIT License.
|
|
3
|
+
/* eslint-disable tsdoc/syntax */
|
|
1
4
|
import OpenAI from "openai";
|
|
2
5
|
import { getBearerTokenProvider } from "@azure/identity";
|
|
3
6
|
import { createAIProject } from "./api/index.js";
|
|
@@ -12,99 +15,121 @@ import { _getTelemetryOperations } from "./classic/telemetry/index.js";
|
|
|
12
15
|
import { overwriteOpenAIClient } from "./overwriteOpenAIClient.js";
|
|
13
16
|
import { getCustomFetch } from "./getCustomFetch.js";
|
|
14
17
|
import { getOpenAIDefaultHeaders } from "./util.js";
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
|
|
18
|
+
/**
|
|
19
|
+
* The main client for the AIProjectClient service. It provides access to the various operations available in the service.
|
|
20
|
+
* @class AIProjectClient
|
|
21
|
+
* @extends {AIProjectContext}
|
|
22
|
+
* @description The main client for the AIProjectClient
|
|
23
|
+
* @constructor
|
|
24
|
+
* @param {string} endpoint - The endpoint to use
|
|
25
|
+
* @param {TokenCredential} credential - The credential to use
|
|
26
|
+
* @param {AIProjectClientOptionalParams} [options] - Optional parameters for the client.
|
|
27
|
+
* @property {DeploymentsOperations} deployments - The operation groups for deployments
|
|
28
|
+
* @property {IndexesOperations} indexes - The operation groups for indexes
|
|
29
|
+
* @property {DatasetsOperations} datasets - The operation groups for datasets
|
|
30
|
+
* @property {ConnectionsOperations} connections - The operation groups for connections
|
|
31
|
+
* @property {AgentsOperations} agents - The operation groups for agents
|
|
32
|
+
* @property {BetaOperations} beta - The operation groups for beta include beta features:
|
|
33
|
+
* - Memory Stores
|
|
34
|
+
* - Evaluators
|
|
35
|
+
* - Evaluation Rules
|
|
36
|
+
* - Evaluation Taxonomies
|
|
37
|
+
* - Insights
|
|
38
|
+
* - Schedules
|
|
39
|
+
* - Red Teams
|
|
40
|
+
* @property {TelemetryOperations} telemetry - The operation groups for telemetry
|
|
41
|
+
* @property {getEndpointUrl} getEndpointUrl - gets the endpoint of the client
|
|
42
|
+
* @property {getOpenAIClient} getOpenAIClient - gets the OpenAI client with optional OpenAI client options
|
|
43
|
+
*/
|
|
44
|
+
export class AIProjectClient {
|
|
45
|
+
_cognitiveScopeClient;
|
|
46
|
+
_azureScopeClient;
|
|
47
|
+
_endpoint;
|
|
48
|
+
_credential;
|
|
49
|
+
_options;
|
|
50
|
+
constructor(endpoint, credential, options = {}) {
|
|
51
|
+
this._endpoint = endpoint;
|
|
52
|
+
this._credential = credential;
|
|
53
|
+
this._options = options;
|
|
54
|
+
const prefixFromOptions = options?.userAgentOptions?.userAgentPrefix;
|
|
55
|
+
const userAgentPrefix = prefixFromOptions ? `${prefixFromOptions}` : "";
|
|
56
|
+
this._cognitiveScopeClient = createAIProject(endpoint, this._credential, {
|
|
57
|
+
...options,
|
|
58
|
+
userAgentOptions: { userAgentPrefix },
|
|
59
|
+
credentials: {
|
|
60
|
+
...options.credentials,
|
|
61
|
+
scopes: ["https://ai.azure.com/.default"],
|
|
62
|
+
},
|
|
63
|
+
});
|
|
64
|
+
this._azureScopeClient = createAIProject(endpoint, credential, {
|
|
65
|
+
...options,
|
|
66
|
+
userAgentOptions: { userAgentPrefix },
|
|
67
|
+
});
|
|
68
|
+
this.indexes = _getIndexesOperations(this._azureScopeClient);
|
|
69
|
+
this.deployments = _getDeploymentsOperations(this._azureScopeClient);
|
|
70
|
+
this.datasets = _getDatasetsOperations(this._azureScopeClient, this._options);
|
|
71
|
+
this.connections = _getConnectionsOperations(this._azureScopeClient);
|
|
72
|
+
this.evaluationRules = _getEvaluationRulesOperations(this._azureScopeClient);
|
|
73
|
+
this.agents = _getAgentsOperations(this._azureScopeClient);
|
|
74
|
+
this.beta = _getBetaOperations(this._cognitiveScopeClient);
|
|
75
|
+
this.telemetry = _getTelemetryOperations(this.connections);
|
|
76
|
+
}
|
|
77
|
+
/** The operation groups for indexes */
|
|
78
|
+
indexes;
|
|
79
|
+
/** The operation groups for deployments */
|
|
80
|
+
deployments;
|
|
81
|
+
/** The operation groups for datasets */
|
|
82
|
+
datasets;
|
|
83
|
+
/** The operation groups for connections */
|
|
84
|
+
connections;
|
|
85
|
+
/** The operation groups for evaluationRules */
|
|
86
|
+
evaluationRules;
|
|
87
|
+
/** The operation groups for agents */
|
|
88
|
+
agents;
|
|
89
|
+
/** The operation groups for beta include beta features:
|
|
90
|
+
* - Memory Stores
|
|
91
|
+
* - Evaluators
|
|
92
|
+
* - Evaluation Rules
|
|
93
|
+
* - Evaluation Taxonomies
|
|
94
|
+
* - Insights
|
|
95
|
+
* - Schedules
|
|
96
|
+
* - Red Teams
|
|
97
|
+
*/
|
|
98
|
+
beta;
|
|
99
|
+
/** The operation groups for telemetry */
|
|
100
|
+
telemetry;
|
|
101
|
+
/**
|
|
102
|
+
* gets the OpenAI client
|
|
103
|
+
* @returns the OpenAI client
|
|
104
|
+
*/
|
|
105
|
+
// eslint-disable-next-line @azure/azure-sdk/ts-naming-options
|
|
106
|
+
getOpenAIClient(opts) {
|
|
107
|
+
const scope = "https://ai.azure.com/.default";
|
|
108
|
+
let customFetch;
|
|
109
|
+
if (this._options.additionalPolicies?.find((policy) => policy.policy.name === "recording policy")) {
|
|
110
|
+
customFetch = getCustomFetch(this._azureScopeClient.pipeline, this._options.httpClient);
|
|
111
|
+
}
|
|
112
|
+
const defaultHeaders = getOpenAIDefaultHeaders(opts?.defaultHeaders, this._options?.userAgentOptions?.userAgentPrefix);
|
|
113
|
+
// Destructure opts to exclude defaultHeaders, then override specific properties
|
|
114
|
+
const { defaultHeaders: _ignoredHeaders, ...restOpts } = opts || {};
|
|
115
|
+
const openAIOptions = {
|
|
116
|
+
...restOpts,
|
|
117
|
+
apiKey: getBearerTokenProvider(this._credential, scope),
|
|
118
|
+
baseURL: `${this._endpoint}/openai`,
|
|
119
|
+
defaultQuery: { "api-version": this._options?.apiVersion || "2025-11-15-preview" },
|
|
120
|
+
dangerouslyAllowBrowser: true,
|
|
121
|
+
defaultHeaders: defaultHeaders.toJSON({ preserveCase: true }),
|
|
122
|
+
fetch: customFetch,
|
|
123
|
+
};
|
|
124
|
+
const openaiClient = new OpenAI(openAIOptions);
|
|
125
|
+
return overwriteOpenAIClient(openaiClient);
|
|
126
|
+
}
|
|
127
|
+
/**
|
|
128
|
+
* gets the endpoint of the client
|
|
129
|
+
* @returns the endpoint of the client
|
|
130
|
+
*/
|
|
131
|
+
get endpoint() {
|
|
132
|
+
return this._endpoint;
|
|
82
133
|
}
|
|
83
|
-
const defaultHeaders = getOpenAIDefaultHeaders(
|
|
84
|
-
opts?.defaultHeaders,
|
|
85
|
-
this._options?.userAgentOptions?.userAgentPrefix
|
|
86
|
-
);
|
|
87
|
-
const { defaultHeaders: _ignoredHeaders, ...restOpts } = opts || {};
|
|
88
|
-
const openAIOptions = {
|
|
89
|
-
...restOpts,
|
|
90
|
-
apiKey: getBearerTokenProvider(this._credential, scope),
|
|
91
|
-
baseURL: `${this._endpoint}/openai`,
|
|
92
|
-
defaultQuery: { "api-version": this._options?.apiVersion || "2025-11-15-preview" },
|
|
93
|
-
dangerouslyAllowBrowser: true,
|
|
94
|
-
defaultHeaders: defaultHeaders.toJSON({ preserveCase: true }),
|
|
95
|
-
fetch: customFetch
|
|
96
|
-
};
|
|
97
|
-
const openaiClient = new OpenAI(openAIOptions);
|
|
98
|
-
return overwriteOpenAIClient(openaiClient);
|
|
99
|
-
}
|
|
100
|
-
/**
|
|
101
|
-
* gets the endpoint of the client
|
|
102
|
-
* @returns the endpoint of the client
|
|
103
|
-
*/
|
|
104
|
-
get endpoint() {
|
|
105
|
-
return this._endpoint;
|
|
106
|
-
}
|
|
107
134
|
}
|
|
108
|
-
|
|
109
|
-
AIProjectClient
|
|
110
|
-
};
|
|
135
|
+
//# sourceMappingURL=aiProjectClient.js.map
|
|
@@ -1,7 +1 @@
|
|
|
1
|
-
{
|
|
2
|
-
"version": 3,
|
|
3
|
-
"sources": ["/mnt/vss/_work/1/s/sdk/ai/ai-projects/src/aiProjectClient.ts"],
|
|
4
|
-
"sourcesContent": ["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n/* eslint-disable tsdoc/syntax */\n\nimport OpenAI from \"openai\";\nimport type { ClientOptions as OpenAIClientOptions } from \"openai\";\nimport { getBearerTokenProvider } from \"@azure/identity\";\nimport type { AIProjectContext, AIProjectClientOptionalParams } from \"./api/index.js\";\nimport { createAIProject } from \"./api/index.js\";\nimport type { AgentsOperations } from \"./classic/agents/index.js\";\nimport { _getAgentsOperations } from \"./classic/agents/index.js\";\nimport type { BetaOperations } from \"./classic/beta/index.js\";\nimport { _getBetaOperations } from \"./classic/beta/index.js\";\nimport type { ConnectionsOperations } from \"./classic/connections/index.js\";\nimport { _getConnectionsOperations } from \"./classic/connections/index.js\";\nimport type { DatasetsOperations } from \"./classic/datasets/index.js\";\nimport { _getDatasetsOperations } from \"./classic/datasets/index.js\";\nimport type { DeploymentsOperations } from \"./classic/deployments/index.js\";\nimport { _getDeploymentsOperations } from \"./classic/deployments/index.js\";\nimport type { EvaluationRulesOperations } from \"./classic/evaluationRules/index.js\";\nimport { _getEvaluationRulesOperations } from \"./classic/evaluationRules/index.js\";\nimport type { IndexesOperations } from \"./classic/indexes/index.js\";\nimport { _getIndexesOperations } from \"./classic/indexes/index.js\";\nimport type { TelemetryOperations } from \"./classic/telemetry/index.js\";\nimport { _getTelemetryOperations } from \"./classic/telemetry/index.js\";\nimport type { TokenCredential } from \"@azure/core-auth\";\nimport { overwriteOpenAIClient } from \"./overwriteOpenAIClient.js\";\nimport { getCustomFetch } from \"./getCustomFetch.js\";\nimport { getOpenAIDefaultHeaders } from \"./util.js\";\n\nexport type { AIProjectClientOptionalParams } from \"./api/aiProjectContext.js\";\n\n/**\n * The main client for the AIProjectClient service. It provides access to the various operations available in the service.\n * @class AIProjectClient\n * @extends {AIProjectContext}\n * @description The main client for the AIProjectClient\n * @constructor\n * @param {string} endpoint - The endpoint to use\n * @param {TokenCredential} credential - The credential to use\n * @param {AIProjectClientOptionalParams} [options] - Optional parameters for the client.\n * @property {DeploymentsOperations} deployments - The operation groups for deployments\n * @property {IndexesOperations} indexes - The operation groups for indexes\n * @property {DatasetsOperations} datasets - The operation groups for datasets\n * @property {ConnectionsOperations} connections - The operation groups for connections\n * @property {AgentsOperations} agents - The operation groups for agents\n * @property {BetaOperations} beta - The operation groups for beta include beta features:\n * - Memory Stores\n * - Evaluators\n * - Evaluation Rules\n * - Evaluation Taxonomies\n * - Insights\n * - Schedules\n * - Red Teams\n * @property {TelemetryOperations} telemetry - The operation groups for telemetry\n * @property {getEndpointUrl} getEndpointUrl - gets the endpoint of the client\n * @property {getOpenAIClient} getOpenAIClient - gets the OpenAI client with optional OpenAI client options\n */\nexport class AIProjectClient {\n private _cognitiveScopeClient: AIProjectContext;\n private _azureScopeClient: AIProjectContext;\n private _endpoint: string;\n private _credential: TokenCredential;\n private _options: AIProjectClientOptionalParams;\n\n constructor(\n endpoint: string,\n credential: TokenCredential,\n options: AIProjectClientOptionalParams = {},\n ) {\n this._endpoint = endpoint;\n this._credential = credential;\n this._options = options;\n const prefixFromOptions = options?.userAgentOptions?.userAgentPrefix;\n const userAgentPrefix = prefixFromOptions ? `${prefixFromOptions}` : \"\";\n this._cognitiveScopeClient = createAIProject(endpoint, this._credential, {\n ...options,\n userAgentOptions: { userAgentPrefix },\n credentials: {\n ...options.credentials,\n scopes: [\"https://ai.azure.com/.default\"],\n },\n });\n this._azureScopeClient = createAIProject(endpoint, credential, {\n ...options,\n userAgentOptions: { userAgentPrefix },\n });\n\n this.indexes = _getIndexesOperations(this._azureScopeClient);\n this.deployments = _getDeploymentsOperations(this._azureScopeClient);\n this.datasets = _getDatasetsOperations(this._azureScopeClient, this._options);\n this.connections = _getConnectionsOperations(this._azureScopeClient);\n this.evaluationRules = _getEvaluationRulesOperations(this._azureScopeClient);\n this.agents = _getAgentsOperations(this._azureScopeClient);\n this.beta = _getBetaOperations(this._cognitiveScopeClient);\n this.telemetry = _getTelemetryOperations(this.connections);\n }\n\n /** The operation groups for indexes */\n public readonly indexes: IndexesOperations;\n /** The operation groups for deployments */\n public readonly deployments: DeploymentsOperations;\n /** The operation groups for datasets */\n public readonly datasets: DatasetsOperations;\n /** The operation groups for connections */\n public readonly connections: ConnectionsOperations;\n /** The operation groups for evaluationRules */\n public readonly evaluationRules: EvaluationRulesOperations;\n /** The operation groups for agents */\n public readonly agents: AgentsOperations;\n /** The operation groups for beta include beta features:\n * - Memory Stores\n * - Evaluators\n * - Evaluation Rules\n * - Evaluation Taxonomies\n * - Insights\n * - Schedules\n * - Red Teams\n */\n public readonly beta: BetaOperations;\n /** The operation groups for telemetry */\n public readonly telemetry: TelemetryOperations;\n /**\n * gets the OpenAI client\n * @returns the OpenAI client\n */\n // eslint-disable-next-line @azure/azure-sdk/ts-naming-options\n public getOpenAIClient(opts?: OpenAIClientOptions): OpenAI {\n const scope = \"https://ai.azure.com/.default\";\n let customFetch: NonNullable<ConstructorParameters<typeof OpenAI>[0]>[\"fetch\"];\n\n if (\n this._options.additionalPolicies?.find((policy) => policy.policy.name === \"recording policy\")\n ) {\n customFetch = getCustomFetch(this._azureScopeClient.pipeline, this._options.httpClient);\n }\n\n const defaultHeaders = getOpenAIDefaultHeaders(\n opts?.defaultHeaders,\n this._options?.userAgentOptions?.userAgentPrefix,\n );\n\n // Destructure opts to exclude defaultHeaders, then override specific properties\n const { defaultHeaders: _ignoredHeaders, ...restOpts } = opts || {};\n const openAIOptions: ConstructorParameters<typeof OpenAI>[0] = {\n ...restOpts,\n apiKey: getBearerTokenProvider(this._credential, scope),\n baseURL: `${this._endpoint}/openai`,\n defaultQuery: { \"api-version\": this._options?.apiVersion || \"2025-11-15-preview\" },\n dangerouslyAllowBrowser: true,\n defaultHeaders: defaultHeaders.toJSON({ preserveCase: true }),\n fetch: customFetch,\n };\n\n const openaiClient = new OpenAI(openAIOptions);\n return overwriteOpenAIClient(openaiClient);\n }\n /**\n * gets the endpoint of the client\n * @returns the endpoint of the client\n */\n public get endpoint(): string {\n return this._endpoint;\n }\n}\n"],
|
|
5
|
-
"mappings": "AAIA,OAAO,YAAY;AAEnB,SAAS,8BAA8B;AAEvC,SAAS,uBAAuB;AAEhC,SAAS,4BAA4B;AAErC,SAAS,0BAA0B;AAEnC,SAAS,iCAAiC;AAE1C,SAAS,8BAA8B;AAEvC,SAAS,iCAAiC;AAE1C,SAAS,qCAAqC;AAE9C,SAAS,6BAA6B;AAEtC,SAAS,+BAA+B;AAExC,SAAS,6BAA6B;AACtC,SAAS,sBAAsB;AAC/B,SAAS,+BAA+B;AA8BjC,MAAM,gBAAgB;AAAA,EACnB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,YACE,UACA,YACA,UAAyC,CAAC,GAC1C;AACA,SAAK,YAAY;AACjB,SAAK,cAAc;AACnB,SAAK,WAAW;AAChB,UAAM,oBAAoB,SAAS,kBAAkB;AACrD,UAAM,kBAAkB,oBAAoB,GAAG,iBAAiB,KAAK;AACrE,SAAK,wBAAwB,gBAAgB,UAAU,KAAK,aAAa;AAAA,MACvE,GAAG;AAAA,MACH,kBAAkB,EAAE,gBAAgB;AAAA,MACpC,aAAa;AAAA,QACX,GAAG,QAAQ;AAAA,QACX,QAAQ,CAAC,+BAA+B;AAAA,MAC1C;AAAA,IACF,CAAC;AACD,SAAK,oBAAoB,gBAAgB,UAAU,YAAY;AAAA,MAC7D,GAAG;AAAA,MACH,kBAAkB,EAAE,gBAAgB;AAAA,IACtC,CAAC;AAED,SAAK,UAAU,sBAAsB,KAAK,iBAAiB;AAC3D,SAAK,cAAc,0BAA0B,KAAK,iBAAiB;AACnE,SAAK,WAAW,uBAAuB,KAAK,mBAAmB,KAAK,QAAQ;AAC5E,SAAK,cAAc,0BAA0B,KAAK,iBAAiB;AACnE,SAAK,kBAAkB,8BAA8B,KAAK,iBAAiB;AAC3E,SAAK,SAAS,qBAAqB,KAAK,iBAAiB;AACzD,SAAK,OAAO,mBAAmB,KAAK,qBAAqB;AACzD,SAAK,YAAY,wBAAwB,KAAK,WAAW;AAAA,EAC3D;AAAA;AAAA,EAGgB;AAAA;AAAA,EAEA;AAAA;AAAA,EAEA;AAAA;AAAA,EAEA;AAAA;AAAA,EAEA;AAAA;AAAA,EAEA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA;AAAA;AAAA,EAEA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAMT,gBAAgB,MAAoC;AACzD,UAAM,QAAQ;AACd,QAAI;AAEJ,QACE,KAAK,SAAS,oBAAoB,KAAK,CAAC,WAAW,OAAO,OAAO,SAAS,kBAAkB,GAC5F;AACA,oBAAc,eAAe,KAAK,kBAAkB,UAAU,KAAK,SAAS,UAAU;AAAA,IACxF;AAEA,UAAM,iBAAiB;AAAA,MACrB,MAAM;AAAA,MACN,KAAK,UAAU,kBAAkB;AAAA,IACnC;AAGA,UAAM,EAAE,gBAAgB,iBAAiB,GAAG,SAAS,IAAI,QAAQ,CAAC;AAClE,UAAM,gBAAyD;AAAA,MAC7D,GAAG;AAAA,MACH,QAAQ,uBAAuB,KAAK,aAAa,KAAK;AAAA,MACtD,SAAS,GAAG,KAAK,SAAS;AAAA,MAC1B,cAAc,EAAE,eAAe,KAAK,UAAU,cAAc,qBAAqB;AAAA,MACjF,yBAAyB;AAAA,MACzB,gBAAgB,eAAe,OAAO,EAAE,cAAc,KAAK,CAAC;AAAA,MAC5D,OAAO;AAAA,IACT;AAEA,UAAM,eAAe,IAAI,OAAO,aAAa;AAC7C,WAAO,sBAAsB,YAAY;AAAA,EAC3C;AAAA;AAAA;AAAA;AAAA;AAAA,EAKA,IAAW,WAAmB;AAC5B,WAAO,KAAK;AAAA,EACd;AACF;",
|
|
6
|
-
"names": []
|
|
7
|
-
}
|
|
1
|
+
{"version":3,"file":"aiProjectClient.js","sourceRoot":"","sources":["../../src/aiProjectClient.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAClC,iCAAiC;AAEjC,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AAEzD,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAEjE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAE3E,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AAErE,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAE3E,OAAO,EAAE,6BAA6B,EAAE,MAAM,oCAAoC,CAAC;AAEnF,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAEnE,OAAO,EAAE,uBAAuB,EAAE,MAAM,8BAA8B,CAAC;AAEvE,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,uBAAuB,EAAE,MAAM,WAAW,CAAC;AAIpD;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,MAAM,OAAO,eAAe;IAClB,qBAAqB,CAAmB;IACxC,iBAAiB,CAAmB;IACpC,SAAS,CAAS;IAClB,WAAW,CAAkB;IAC7B,QAAQ,CAAgC;IAEhD,YACE,QAAgB,EAChB,UAA2B,EAC3B,UAAyC,EAAE;QAE3C,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAC9B,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QACxB,MAAM,iBAAiB,GAAG,OAAO,EAAE,gBAAgB,EAAE,eAAe,CAAC;QACrE,MAAM,eAAe,GAAG,iBAAiB,CAAC,CAAC,CAAC,GAAG,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACxE,IAAI,CAAC,qBAAqB,GAAG,eAAe,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,EAAE;YACvE,GAAG,OAAO;YACV,gBAAgB,EAAE,EAAE,eAAe,EAAE;YACrC,WAAW,EAAE;gBACX,GAAG,OAAO,CAAC,WAAW;gBACtB,MAAM,EAAE,CAAC,+BAA+B,CAAC;aAC1C;SACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,GAAG,eAAe,CAAC,QAAQ,EAAE,UAAU,EAAE;YAC7D,GAAG,OAAO;YACV,gBAAgB,EAAE,EAAE,eAAe,EAAE;SACtC,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,GAAG,qBAAqB,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC7D,IAAI,CAAC,WAAW,GAAG,yBAAyB,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACrE,IAAI,CAAC,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9E,IAAI,CAAC,WAAW,GAAG,yBAAyB,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACrE,IAAI,CAAC,eAAe,GAAG,6BAA6B,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC7E,IAAI,CAAC,MAAM,GAAG,oBAAoB,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC3D,IAAI,CAAC,IAAI,GAAG,kBAAkB,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC3D,IAAI,CAAC,SAAS,GAAG,uBAAuB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC7D,CAAC;IAED,uCAAuC;IACvB,OAAO,CAAoB;IAC3C,2CAA2C;IAC3B,WAAW,CAAwB;IACnD,wCAAwC;IACxB,QAAQ,CAAqB;IAC7C,2CAA2C;IAC3B,WAAW,CAAwB;IACnD,+CAA+C;IAC/B,eAAe,CAA4B;IAC3D,sCAAsC;IACtB,MAAM,CAAmB;IACzC;;;;;;;;OAQG;IACa,IAAI,CAAiB;IACrC,yCAAyC;IACzB,SAAS,CAAsB;IAC/C;;;OAGG;IACH,8DAA8D;IACvD,eAAe,CAAC,IAA0B;QAC/C,MAAM,KAAK,GAAG,+BAA+B,CAAC;QAC9C,IAAI,WAA0E,CAAC;QAE/E,IACE,IAAI,CAAC,QAAQ,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,kBAAkB,CAAC,EAC7F,CAAC;YACD,WAAW,GAAG,cAAc,CAAC,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC1F,CAAC;QAED,MAAM,cAAc,GAAG,uBAAuB,CAC5C,IAAI,EAAE,cAAc,EACpB,IAAI,CAAC,QAAQ,EAAE,gBAAgB,EAAE,eAAe,CACjD,CAAC;QAEF,gFAAgF;QAChF,MAAM,EAAE,cAAc,EAAE,eAAe,EAAE,GAAG,QAAQ,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC;QACpE,MAAM,aAAa,GAA4C;YAC7D,GAAG,QAAQ;YACX,MAAM,EAAE,sBAAsB,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,CAAC;YACvD,OAAO,EAAE,GAAG,IAAI,CAAC,SAAS,SAAS;YACnC,YAAY,EAAE,EAAE,aAAa,EAAE,IAAI,CAAC,QAAQ,EAAE,UAAU,IAAI,oBAAoB,EAAE;YAClF,uBAAuB,EAAE,IAAI;YAC7B,cAAc,EAAE,cAAc,CAAC,MAAM,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;YAC7D,KAAK,EAAE,WAAW;SACnB,CAAC;QAEF,MAAM,YAAY,GAAG,IAAI,MAAM,CAAC,aAAa,CAAC,CAAC;QAC/C,OAAO,qBAAqB,CAAC,YAAY,CAAC,CAAC;IAC7C,CAAC;IACD;;;OAGG;IACH,IAAW,QAAQ;QACjB,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n/* eslint-disable tsdoc/syntax */\n\nimport OpenAI from \"openai\";\nimport type { ClientOptions as OpenAIClientOptions } from \"openai\";\nimport { getBearerTokenProvider } from \"@azure/identity\";\nimport type { AIProjectContext, AIProjectClientOptionalParams } from \"./api/index.js\";\nimport { createAIProject } from \"./api/index.js\";\nimport type { AgentsOperations } from \"./classic/agents/index.js\";\nimport { _getAgentsOperations } from \"./classic/agents/index.js\";\nimport type { BetaOperations } from \"./classic/beta/index.js\";\nimport { _getBetaOperations } from \"./classic/beta/index.js\";\nimport type { ConnectionsOperations } from \"./classic/connections/index.js\";\nimport { _getConnectionsOperations } from \"./classic/connections/index.js\";\nimport type { DatasetsOperations } from \"./classic/datasets/index.js\";\nimport { _getDatasetsOperations } from \"./classic/datasets/index.js\";\nimport type { DeploymentsOperations } from \"./classic/deployments/index.js\";\nimport { _getDeploymentsOperations } from \"./classic/deployments/index.js\";\nimport type { EvaluationRulesOperations } from \"./classic/evaluationRules/index.js\";\nimport { _getEvaluationRulesOperations } from \"./classic/evaluationRules/index.js\";\nimport type { IndexesOperations } from \"./classic/indexes/index.js\";\nimport { _getIndexesOperations } from \"./classic/indexes/index.js\";\nimport type { TelemetryOperations } from \"./classic/telemetry/index.js\";\nimport { _getTelemetryOperations } from \"./classic/telemetry/index.js\";\nimport type { TokenCredential } from \"@azure/core-auth\";\nimport { overwriteOpenAIClient } from \"./overwriteOpenAIClient.js\";\nimport { getCustomFetch } from \"./getCustomFetch.js\";\nimport { getOpenAIDefaultHeaders } from \"./util.js\";\n\nexport type { AIProjectClientOptionalParams } from \"./api/aiProjectContext.js\";\n\n/**\n * The main client for the AIProjectClient service. It provides access to the various operations available in the service.\n * @class AIProjectClient\n * @extends {AIProjectContext}\n * @description The main client for the AIProjectClient\n * @constructor\n * @param {string} endpoint - The endpoint to use\n * @param {TokenCredential} credential - The credential to use\n * @param {AIProjectClientOptionalParams} [options] - Optional parameters for the client.\n * @property {DeploymentsOperations} deployments - The operation groups for deployments\n * @property {IndexesOperations} indexes - The operation groups for indexes\n * @property {DatasetsOperations} datasets - The operation groups for datasets\n * @property {ConnectionsOperations} connections - The operation groups for connections\n * @property {AgentsOperations} agents - The operation groups for agents\n * @property {BetaOperations} beta - The operation groups for beta include beta features:\n * - Memory Stores\n * - Evaluators\n * - Evaluation Rules\n * - Evaluation Taxonomies\n * - Insights\n * - Schedules\n * - Red Teams\n * @property {TelemetryOperations} telemetry - The operation groups for telemetry\n * @property {getEndpointUrl} getEndpointUrl - gets the endpoint of the client\n * @property {getOpenAIClient} getOpenAIClient - gets the OpenAI client with optional OpenAI client options\n */\nexport class AIProjectClient {\n private _cognitiveScopeClient: AIProjectContext;\n private _azureScopeClient: AIProjectContext;\n private _endpoint: string;\n private _credential: TokenCredential;\n private _options: AIProjectClientOptionalParams;\n\n constructor(\n endpoint: string,\n credential: TokenCredential,\n options: AIProjectClientOptionalParams = {},\n ) {\n this._endpoint = endpoint;\n this._credential = credential;\n this._options = options;\n const prefixFromOptions = options?.userAgentOptions?.userAgentPrefix;\n const userAgentPrefix = prefixFromOptions ? `${prefixFromOptions}` : \"\";\n this._cognitiveScopeClient = createAIProject(endpoint, this._credential, {\n ...options,\n userAgentOptions: { userAgentPrefix },\n credentials: {\n ...options.credentials,\n scopes: [\"https://ai.azure.com/.default\"],\n },\n });\n this._azureScopeClient = createAIProject(endpoint, credential, {\n ...options,\n userAgentOptions: { userAgentPrefix },\n });\n\n this.indexes = _getIndexesOperations(this._azureScopeClient);\n this.deployments = _getDeploymentsOperations(this._azureScopeClient);\n this.datasets = _getDatasetsOperations(this._azureScopeClient, this._options);\n this.connections = _getConnectionsOperations(this._azureScopeClient);\n this.evaluationRules = _getEvaluationRulesOperations(this._azureScopeClient);\n this.agents = _getAgentsOperations(this._azureScopeClient);\n this.beta = _getBetaOperations(this._cognitiveScopeClient);\n this.telemetry = _getTelemetryOperations(this.connections);\n }\n\n /** The operation groups for indexes */\n public readonly indexes: IndexesOperations;\n /** The operation groups for deployments */\n public readonly deployments: DeploymentsOperations;\n /** The operation groups for datasets */\n public readonly datasets: DatasetsOperations;\n /** The operation groups for connections */\n public readonly connections: ConnectionsOperations;\n /** The operation groups for evaluationRules */\n public readonly evaluationRules: EvaluationRulesOperations;\n /** The operation groups for agents */\n public readonly agents: AgentsOperations;\n /** The operation groups for beta include beta features:\n * - Memory Stores\n * - Evaluators\n * - Evaluation Rules\n * - Evaluation Taxonomies\n * - Insights\n * - Schedules\n * - Red Teams\n */\n public readonly beta: BetaOperations;\n /** The operation groups for telemetry */\n public readonly telemetry: TelemetryOperations;\n /**\n * gets the OpenAI client\n * @returns the OpenAI client\n */\n // eslint-disable-next-line @azure/azure-sdk/ts-naming-options\n public getOpenAIClient(opts?: OpenAIClientOptions): OpenAI {\n const scope = \"https://ai.azure.com/.default\";\n let customFetch: NonNullable<ConstructorParameters<typeof OpenAI>[0]>[\"fetch\"];\n\n if (\n this._options.additionalPolicies?.find((policy) => policy.policy.name === \"recording policy\")\n ) {\n customFetch = getCustomFetch(this._azureScopeClient.pipeline, this._options.httpClient);\n }\n\n const defaultHeaders = getOpenAIDefaultHeaders(\n opts?.defaultHeaders,\n this._options?.userAgentOptions?.userAgentPrefix,\n );\n\n // Destructure opts to exclude defaultHeaders, then override specific properties\n const { defaultHeaders: _ignoredHeaders, ...restOpts } = opts || {};\n const openAIOptions: ConstructorParameters<typeof OpenAI>[0] = {\n ...restOpts,\n apiKey: getBearerTokenProvider(this._credential, scope),\n baseURL: `${this._endpoint}/openai`,\n defaultQuery: { \"api-version\": this._options?.apiVersion || \"2025-11-15-preview\" },\n dangerouslyAllowBrowser: true,\n defaultHeaders: defaultHeaders.toJSON({ preserveCase: true }),\n fetch: customFetch,\n };\n\n const openaiClient = new OpenAI(openAIOptions);\n return overwriteOpenAIClient(openaiClient);\n }\n /**\n * gets the endpoint of the client\n * @returns the endpoint of the client\n */\n public get endpoint(): string {\n return this._endpoint;\n }\n}\n"]}
|