@arcgis/ai-agents 5.1.0-next.101 → 5.1.0-next.103

package/dist/agents/arcgisKnowledge/context/arcgisKnowledgeContext.d.ts

@@ -1,15 +1,16 @@
 import { default as KnowledgeGraph } from '@arcgis/core/rest/knowledgeGraph/KnowledgeGraph.js';
 import { default as LinkChartView } from '@arcgis/core/views/LinkChartView.js';
+import { default as MapView } from '@arcgis/core/views/MapView.js';
 /**
  * Context required by the Knowledge Link Chart agent.
  * Supplied by the application via an agent `getContext()` provider.
  */
 export type ArcgisKnowledgeContext = {
     knowledgeGraph: KnowledgeGraph;
-    linkChartView?: never;
+    view?: never;
 } | {
-    linkChartView: LinkChartView;
+    view: LinkChartView | MapView;
     knowledgeGraph?: never;
 };
-export declare const arcgisKnowledgeContextRequiredKeys: readonly ["linkChartView"];
-export declare const serviceContextKeys: readonly ["knowledgeGraph", "linkChartView"];
+export declare const arcgisKnowledgeContextRequiredKeys: readonly ["view"];
+export declare const serviceContextKeys: readonly ["knowledgeGraph", "view"];

package/dist/agents/arcgisKnowledge/context/getArcgisKnowledgeContext.d.ts

@@ -1,8 +1,9 @@
 import { RunnableConfig } from '@langchain/core/runnables';
 import { default as KnowledgeGraph } from '@arcgis/core/rest/knowledgeGraph/KnowledgeGraph.js';
 import { default as LinkChartView } from '@arcgis/core/views/LinkChartView.js';
-export declare function getLinkChartContext(config?: RunnableConfig): {
-    linkChartView: LinkChartView;
+import { default as MapView } from '@arcgis/core/views/MapView.js';
+export declare function getViewContext(config?: RunnableConfig): {
+    view: LinkChartView | MapView;
 };
 export declare function getKnowledgeGraphContext(config?: RunnableConfig): Promise<{
     knowledgeGraph: KnowledgeGraph;

package/dist/agents/arcgisKnowledge/tools/addRecords/core.d.ts

@@ -1,4 +1,4 @@
 import { RunnableConfig } from '@langchain/core/runnables';
 import { default as KnowledgeGraph } from '@arcgis/core/rest/knowledgeGraph/KnowledgeGraph.js';
-import { default as LinkChartView } from '@arcgis/core/views/LinkChartView.js';
-export declare function executeAddRecords(prompt: string, knowledgeGraph: KnowledgeGraph, graphQueryServiceUrl: string, view: LinkChartView, config?: RunnableConfig): Promise<string>;
+import { ArcgisKnowledgeContext } from '../../context/arcgisKnowledgeContext';
+export declare function executeAddRecords(prompt: string, knowledgeGraph: KnowledgeGraph, graphQueryServiceUrl: string, context: ArcgisKnowledgeContext, config?: RunnableConfig): Promise<string>;

package/dist/agents/arcgisKnowledge/tools/generateCypher/core.d.ts

@@ -1,7 +1,8 @@
 import { default as DataModel } from '@arcgis/core/rest/knowledgeGraph/DataModel.js';
 import { RunnableConfig } from '@langchain/core/runnables';
+import { GraphQueryRetryContext } from './types';
 export declare function executeGenerateCypher(prompt: string, dataModel: DataModel, graphQueryServiceUrl: string, config?: RunnableConfig): Promise<string>;
-export declare function getCypherFromPrompt(prompt: string, dataModel: DataModel, graphQueryServiceUrl: string, config?: RunnableConfig): Promise<{
+export declare function getCypherFromPrompt(prompt: string, dataModel: DataModel, graphQueryServiceUrl: string, config?: RunnableConfig, retryContext?: GraphQueryRetryContext): Promise<{
     graphQuery: string;
     explanation: string;
 }>;

package/dist/agents/arcgisKnowledge/tools/generateCypher/types.d.ts

@@ -9,3 +9,8 @@ export interface GenerateCypherJobResponse {
     };
     message?: string;
 }
+export interface GraphQueryRetryContext {
+    failedGraphQuery: string;
+    executionError: string;
+    attemptCount: number;
+}

package/dist/agents/arcgisKnowledge/utils/dataTransformationUtils.d.ts

@@ -3,6 +3,12 @@ import { GraphAnyValue } from '@arcgis/core/rest/knowledgeGraph/GraphAnyValue.js
 import { default as GraphNamedObject } from '@arcgis/core/rest/knowledgeGraph/GraphNamedObject.js';
 import { default as KnowledgeGraph } from '@arcgis/core/rest/knowledgeGraph/KnowledgeGraph.js';
 import { default as WebLinkChart } from '@arcgis/core/WebLinkChart.js';
+import { default as WebMap } from '@arcgis/core/WebMap.js';
 export declare function extractGraphNamedObjectsFromQueryValueRecursive(resultValue: GraphAnyValue, results: Map<string, IdTypePair>): void;
 export declare function forEachGraphNamedObjectInQueryValueRecursive(resultValue: GraphAnyValue, callback: (graphNamedObject: GraphNamedObject) => void): void;
 export declare function getAllIdTypePairsFromLinkChart(linkChart: WebLinkChart, knowledgeGraph: KnowledgeGraph): IdTypePair[];
+export declare function getAllIdTypePairsAndDynamicTypesFromMap(map: WebMap): {
+    allNamedTypesFullyDynamic: boolean;
+    idTypePairs: IdTypePair[];
+    dynamicTypes: string[];
+};

package/dist/agents/arcgisKnowledge/utils/mapUtils.d.ts

@@ -0,0 +1,4 @@
+import { default as LinkChartView } from '@arcgis/core/views/LinkChartView.js';
+import { default as MapView } from '@arcgis/core/views/MapView.js';
+export declare function isMapValid(mapView: MapView): void;
+export declare function isMapView(view: LinkChartView | MapView | undefined): boolean;

package/dist/agents/arcgisKnowledge/utils/queryUtils.d.ts

@@ -3,31 +3,49 @@ import { RequestOptions } from '@arcgis/core/request/types.js';
 import { default as GraphNamedObject } from '@arcgis/core/rest/knowledgeGraph/GraphNamedObject.js';
 import { default as GraphQueryResultHeader } from '@arcgis/core/rest/knowledgeGraph/GraphQueryResultHeader.js';
 import { GraphAnyValue } from '@arcgis/core/rest/knowledgeGraph/GraphAnyValue.js';
-import { default as GraphQueryStreaming } from '@arcgis/core/rest/knowledgeGraph/GraphQueryStreaming.js';
+import { RunnableConfig } from '@langchain/core/runnables';
+export declare const MAX_GRAPH_QUERY_ATTEMPTS = 3;
 export declare function truncateGraphNamedObjectStringProperties(record: GraphNamedObject, maxLength?: number): GraphNamedObject;
-type RunCypherQueryResult<T> = {
+type GenerateAndExecuteFailureResult = {
     type: "aborted";
+    errorMessage: string;
+    graphQuery?: string;
+    explanation?: string;
 } | {
-    type: "error";
+    type: "execution-error";
+    errorMessage: string;
+    error: unknown;
+    graphQuery: string;
+    explanation: string;
+} | {
+    type: "generation-error";
     errorMessage: string;
     error: unknown;
 } | {
+    type: "timeout";
+    errorMessage: string;
+    graphQuery: string;
+    explanation: string;
+};
+type GenerateAndExecuteResult<T> = GenerateAndExecuteFailureResult | {
     type: "success";
+    graphQuery: string;
+    explanation: string;
     rowLimitReached: boolean;
     resultHeader: GraphQueryResultHeader;
     processedResult: T;
-} | {
-    type: "timeout";
 };
-export declare const runCypherQueryAndReduce: <T>({ kg, queryParams, targetStructure, targetStructureReducer, requestOptions, queryRowLimit, queryTimeoutMs, }: {
+export declare function generateAndExecuteCypherQueryAndReduce<T>({ prompt, kg, graphQueryServiceUrl, targetStructure, targetStructureReducer, requestOptions, queryRowLimit, queryTimeoutMs, config, }: {
+    prompt: string;
     kg: KnowledgeGraph;
-    queryParams: GraphQueryStreaming;
+    graphQueryServiceUrl: string;
     targetStructure: T;
     targetStructureReducer: (prev: T, nextRow: GraphAnyValue[]) => T;
     requestOptions?: RequestOptions;
     queryRowLimit?: number;
     queryTimeoutMs?: number;
-}) => Promise<RunCypherQueryResult<T>>;
+    config?: RunnableConfig;
+}): Promise<GenerateAndExecuteResult<T>>;
 type RunGraphSearchResult = {
     type: "error";
     errorMessage: string;

package/dist/arcgis_knowledge_current_map_context-CfNPKOjr.js

@@ -0,0 +1,7 @@
+const e = `## Additional System Context
+
+In addition to the user's inquiry prompt that precedes this system context, the current map has entities and relationships on it already, in the format of an array of objects containing identifiers and typeNames or type names that contain all entities or relationships of that type, in case this is needed for query generation (for example 'Add all records that are connected to people on this map to the map'). If the user is asking for a query that acts on 'the current map', 'my map', or similar then the cypher query should be filtered by the ids. For example, 'Match (n:Person)-[\\r]->(m) WHERE id(n) IN [\\'identifier1','identifier2'] return r, m. Curly braces (the '{' and '}' characters) are often part of identifier strings and should not be stripped out - the process of filtering by ids should not be altering the id strings. The entities and relationships that are on the current map are as follows:
+`;
+export {
+  e as default
+};