@manta-eu/client 0.2.0 → 0.3.0

package/src/generated/graphql.ts

@@ -530,6 +530,7 @@ export type GQLCreateDataConnectionInput = {
   connectorType: Scalars["String"]["input"];
   name: Scalars["String"]["input"];
   settings: Array<GQLDataConnectionSettingInput>;
+  sourceName?: InputMaybe<Scalars["String"]["input"]>;
 };
 
 export type GQLCreateInferalIntegrationInput = {
@@ -655,6 +656,7 @@ export type GQLCustomAppPage = {
 };
 
 export enum GQLCustomAppPermission {
+  ReadWarehouse = "READ_WAREHOUSE",
   ShareChats = "SHARE_CHATS",
   WriteData = "WRITE_DATA",
 }
@@ -757,6 +759,7 @@ export type GQLDataConnection = {
   id: Scalars["String"]["output"];
   name: Scalars["String"]["output"];
   settings: Array<GQLDataConnectionSetting>;
+  sourceName?: Maybe<Scalars["String"]["output"]>;
   status: GQLDataConnectionStatus;
 };
 
@@ -843,8 +846,12 @@ export type GQLDatastoreProvisioningStatus = {
 
 export type GQLDatastoreSubscription = {
   __typename?: "DatastoreSubscription";
+  amount?: Maybe<Scalars["Int"]["output"]>;
+  currency?: Maybe<Scalars["String"]["output"]>;
   datastoreId: Scalars["String"]["output"];
   inGoodStanding: Scalars["Boolean"]["output"];
+  interval?: Maybe<Scalars["String"]["output"]>;
+  intervalCount?: Maybe<Scalars["Int"]["output"]>;
   status: Scalars["String"]["output"];
 };
 
@@ -1026,6 +1033,14 @@ export type GQLExecuteActionUpdateInput = {
   subjectIri: Scalars["IRI"]["input"];
 };
 
+export type GQLFactExplanationResponse = {
+  __typename?: "FactExplanationResponse";
+  error?: Maybe<Scalars["String"]["output"]>;
+  facts: Scalars["JSON"]["output"];
+  formatted: Scalars["String"]["output"];
+  summary?: Maybe<Scalars["String"]["output"]>;
+};
+
 export enum GQLFactValueStatus {
   Changed = "CHANGED",
   Removed = "REMOVED",
@@ -1040,6 +1055,12 @@ export type GQLFileFactsInput = {
   width?: InputMaybe<Scalars["PositiveInt"]["input"]>;
 };
 
+export type GQLForgetUserNameResponse = {
+  __typename?: "ForgetUserNameResponse";
+  datastoreIds: Array<Scalars["String"]["output"]>;
+  id: Scalars["String"]["output"];
+};
+
 export type GQLGmailIntegration = {
   __typename?: "GmailIntegration";
   email: Scalars["String"]["output"];
@@ -1210,6 +1231,7 @@ export type GQLInitiateUploadResponse = {
 };
 
 export enum GQLIntegration {
+  Canva = "CANVA",
   Dash = "DASH",
   Figma = "FIGMA",
   Google = "GOOGLE",
@@ -1369,7 +1391,7 @@ export type GQLLinkedinSetupProgress = {
   state: Array<Scalars["String"]["output"]>;
 };
 
-export type GQLManageRdFoxRuleInput = {
+export type GQLManageRuleInput = {
   dryRun?: InputMaybe<Scalars["Boolean"]["input"]>;
   expectNoInference?: InputMaybe<Scalars["Boolean"]["input"]>;
   humanIntent: Scalars["String"]["input"];
@@ -1379,8 +1401,8 @@ export type GQLManageRdFoxRuleInput = {
   testTriple?: InputMaybe<GQLTestTripleInput>;
 };
 
-export type GQLManageRdFoxRuleResponse = {
-  __typename?: "ManageRDFoxRuleResponse";
+export type GQLManageRuleResponse = {
+  __typename?: "ManageRuleResponse";
   error?: Maybe<Scalars["String"]["output"]>;
   inferenceTest?: Maybe<Scalars["String"]["output"]>;
   inferenceVerified: Scalars["Boolean"]["output"];
@@ -1613,6 +1635,7 @@ export type GQLMutation = {
   executeActionDeletion: GQLExecuteActionResponse;
   executeActionPlan: GQLExecuteActionPlanResponse;
   executeActionUpdate: GQLExecuteActionResponse;
+  forgetUserName: GQLForgetUserNameResponse;
   generateStructuredData: GQLStructuredDataResponse;
   googleAnalyticsSetupConnect: GQLGoogleAnalyticsSetupProgress;
   googleSetupConnect: GQLGoogleSetupProgress;
@@ -1625,7 +1648,7 @@ export type GQLMutation = {
   linkOrganizationToProvider: GQLLinkOrganizationResponse;
   linkUnlinkedDatastore: GQLProvisionDatastoreResponse;
   linkedinSetupConnect?: Maybe<GQLLinkedinSetupProgress>;
-  manageRDFoxRule: GQLManageRdFoxRuleResponse;
+  manageRule: GQLManageRuleResponse;
   manualActionChangeAssignees?: Maybe<GQLManualActionChangeAssigneesPayload>;
   manualActionResolve?: Maybe<GQLManualActionResolvePayload>;
   materializeOntologyCoverage: GQLMaterializeOntologyCoverageResponse;
@@ -1662,7 +1685,7 @@ export type GQLMutation = {
   updateMemory: GQLUpdateMemoryResponse;
   updateTeamMember: GQLTeamMemberUpdateResponse;
   updateUserPreferences: GQLUpdateUserPreferencesResponse;
-  validateRDFoxRule: GQLValidateRdFoxRuleResponse;
+  validateRule: GQLValidateRuleResponse;
   validateStagedRules: GQLValidateStagedRulesResponse;
 };
 
@@ -1814,6 +1837,10 @@ export type GQLMutationExecuteActionUpdateArgs = {
   input: GQLExecuteActionUpdateInput;
 };
 
+export type GQLMutationForgetUserNameArgs = {
+  userId: Scalars["String"]["input"];
+};
+
 export type GQLMutationGenerateStructuredDataArgs = {
   input: GQLStructuredDataInput;
 };
@@ -1876,8 +1903,8 @@ export type GQLMutationLinkedinSetupConnectArgs = {
   accountIds: Array<Scalars["String"]["input"]>;
 };
 
-export type GQLMutationManageRdFoxRuleArgs = {
-  input: GQLManageRdFoxRuleInput;
+export type GQLMutationManageRuleArgs = {
+  input: GQLManageRuleInput;
 };
 
 export type GQLMutationManualActionChangeAssigneesArgs = {
@@ -1889,6 +1916,7 @@ export type GQLMutationManualActionChangeAssigneesArgs = {
 
 export type GQLMutationManualActionResolveArgs = {
   namedGraphIri: Scalars["String"]["input"];
+  reason?: InputMaybe<Scalars["String"]["input"]>;
   resolvedBy: Scalars["String"]["input"];
   status: GQLManualActionResolveStatus;
   subjectIri: Scalars["String"]["input"];
@@ -2058,8 +2086,8 @@ export type GQLMutationUpdateUserPreferencesArgs = {
   input: GQLUpdateUserPreferencesInput;
 };
 
-export type GQLMutationValidateRdFoxRuleArgs = {
-  input: GQLValidateRdFoxRuleInput;
+export type GQLMutationValidateRuleArgs = {
+  input: GQLValidateRuleInput;
 };
 
 export type GQLMutationValidateStagedRulesArgs = {
@@ -2191,7 +2219,7 @@ export type GQLQuery = {
   datastoreSubscriptions: Array<GQLDatastoreSubscription>;
   datastoreTree: Array<GQLDatastoreTreeNode>;
   detectMemoryConflicts: GQLDetectMemoryConflictsResult;
-  explainRDFoxFact: GQLRdFoxExplanationResponse;
+  explainFact: GQLFactExplanationResponse;
   gmailIntegrations: Array<GQLGmailIntegration>;
   googleAnalyticsSetupProgress: GQLGoogleAnalyticsSetupProgress;
   googleBrandSuggestions: Array<GQLGoogleBrandSuggestion>;
@@ -2225,6 +2253,7 @@ export type GQLQuery = {
   unlinkedDatastores: Array<GQLUnlinkedDatastore>;
   userPreferences: GQLUserPreferences;
   validateActionPlan: GQLValidateActionPlanResponse;
+  warehouseQuery: GQLWarehouseQueryResponse;
 };
 
 export type GQLQueryActionPlanDiffArgs = {
@@ -2323,7 +2352,7 @@ export type GQLQueryDetectMemoryConflictsArgs = {
   input: GQLDetectMemoryConflictsInput;
 };
 
-export type GQLQueryExplainRdFoxFactArgs = {
+export type GQLQueryExplainFactArgs = {
   namedGraph?: InputMaybe<Scalars["IRI"]["input"]>;
   object: Scalars["String"]["input"];
   objectDatatype?: InputMaybe<Scalars["String"]["input"]>;
@@ -2433,12 +2462,8 @@ export type GQLQueryValidateActionPlanArgs = {
   shaclShapeIri?: InputMaybe<Scalars["IRI"]["input"]>;
 };
 
-export type GQLRdFoxExplanationResponse = {
-  __typename?: "RDFoxExplanationResponse";
-  error?: Maybe<Scalars["String"]["output"]>;
-  facts: Scalars["JSON"]["output"];
-  formatted: Scalars["String"]["output"];
-  summary?: Maybe<Scalars["String"]["output"]>;
+export type GQLQueryWarehouseQueryArgs = {
+  sql: Scalars["String"]["input"];
 };
 
 export type GQLRecordHotelProfileResponse = {
@@ -3066,12 +3091,12 @@ export type GQLValidateActionPlanResponse = {
   valid: Scalars["Boolean"]["output"];
 };
 
-export type GQLValidateRdFoxRuleInput = {
+export type GQLValidateRuleInput = {
   rule: Scalars["DatalogRule"]["input"];
 };
 
-export type GQLValidateRdFoxRuleResponse = {
-  __typename?: "ValidateRDFoxRuleResponse";
+export type GQLValidateRuleResponse = {
+  __typename?: "ValidateRuleResponse";
   error?: Maybe<Scalars["String"]["output"]>;
   ruleStructure?: Maybe<GQLRuleStructure>;
   valid: Scalars["Boolean"]["output"];
@@ -3086,6 +3111,28 @@ export type GQLValidateStagedRulesResponse = {
   syntaxErrors: Array<Scalars["String"]["output"]>;
 };
 
+export type GQLWarehouseColumn = {
+  __typename?: "WarehouseColumn";
+  name: Scalars["String"]["output"];
+  type: Scalars["String"]["output"];
+};
+
+export type GQLWarehouseQueryResponse = {
+  __typename?: "WarehouseQueryResponse";
+  columns?: Maybe<Array<GQLWarehouseColumn>>;
+  error?: Maybe<Scalars["String"]["output"]>;
+  rows?: Maybe<Array<Scalars["JSON"]["output"]>>;
+  statistics?: Maybe<GQLWarehouseStatistics>;
+  truncated: Scalars["Boolean"]["output"];
+};
+
+export type GQLWarehouseStatistics = {
+  __typename?: "WarehouseStatistics";
+  returnRows?: Maybe<Scalars["SafeInt"]["output"]>;
+  scanBytes?: Maybe<Scalars["SafeInt"]["output"]>;
+  scanRows?: Maybe<Scalars["SafeInt"]["output"]>;
+};
+
 export type GQLWoocommerceProviderInput = {
   consumerKey: Scalars["String"]["input"];
   consumerSecret: Scalars["String"]["input"];
@@ -3642,6 +3689,15 @@ export type GQLExecuteAgentActionUpdateMutation = {
   };
 };
 
+export type GQLForgetUserNameMutationVariables = Exact<{
+  userId: Scalars["String"]["input"];
+}>;
+
+export type GQLForgetUserNameMutation = {
+  __typename?: "Mutation";
+  forgetUserName: { __typename?: "ForgetUserNameResponse"; id: string; datastoreIds: Array<string> };
+};
+
 export type GQLGenerateStructuredDataMutationVariables = Exact<{
   input: GQLStructuredDataInput;
 }>;
@@ -4965,6 +5021,7 @@ export type GQLDataConnectionsQuery = {
     id: string;
     name: string;
     connectorType: string;
+    sourceName?: string | null;
     status: GQLDataConnectionStatus;
     error?: string | null;
     createdAt: string;
@@ -5051,6 +5108,10 @@ export type GQLDatastoreSubscriptionsQuery = {
     datastoreId: string;
     status: string;
     inGoodStanding: boolean;
+    amount?: number | null;
+    currency?: string | null;
+    interval?: string | null;
+    intervalCount?: number | null;
   }>;
 };
 
@@ -5101,7 +5162,7 @@ export type GQLDetectMemoryConflictsQuery = {
   };
 };
 
-export type GQLExplainRdFoxFactQueryVariables = Exact<{
+export type GQLExplainFactQueryVariables = Exact<{
   subject: Scalars["String"]["input"];
   predicate?: InputMaybe<Scalars["String"]["input"]>;
   object: Scalars["String"]["input"];
@@ -5110,10 +5171,10 @@ export type GQLExplainRdFoxFactQueryVariables = Exact<{
   namedGraph?: InputMaybe<Scalars["IRI"]["input"]>;
 }>;
 
-export type GQLExplainRdFoxFactQuery = {
+export type GQLExplainFactQuery = {
   __typename?: "Query";
-  explainRDFoxFact: {
-    __typename?: "RDFoxExplanationResponse";
+  explainFact: {
+    __typename?: "FactExplanationResponse";
     facts: unknown;
     formatted: string;
     summary?: string | null;
@@ -5818,6 +5879,27 @@ export type GQLValidateActionPlanQuery = {
   };
 };
 
+export type GQLWarehouseQueryQueryVariables = Exact<{
+  sql: Scalars["String"]["input"];
+}>;
+
+export type GQLWarehouseQueryQuery = {
+  __typename?: "Query";
+  warehouseQuery: {
+    __typename?: "WarehouseQueryResponse";
+    rows?: Array<unknown> | null;
+    truncated: boolean;
+    error?: string | null;
+    columns?: Array<{ __typename?: "WarehouseColumn"; name: string; type: string }> | null;
+    statistics?: {
+      __typename?: "WarehouseStatistics";
+      scanRows?: number | null;
+      scanBytes?: number | null;
+      returnRows?: number | null;
+    } | null;
+  };
+};
+
 export const CustomAppConnectionFieldsFragmentDoc = `
     fragment CustomAppConnectionFields on CustomAppConnection {
   issuer
@@ -7006,6 +7088,33 @@ export const useExecuteAgentActionUpdateMutation = <TError = unknown, TContext =
   });
 };
 
+export const ForgetUserNameDocument = `
+    mutation ForgetUserName($userId: String!) {
+  forgetUserName(userId: $userId) {
+    id
+    datastoreIds
+  }
+}
+    `;
+
+export const useForgetUserNameMutation = <TError = unknown, TContext = unknown>(
+  client: GraphQLClient,
+  options?: UseMutationOptions<GQLForgetUserNameMutation, TError, GQLForgetUserNameMutationVariables, TContext>,
+  headers?: RequestInit["headers"],
+) => {
+  return useMutation<GQLForgetUserNameMutation, TError, GQLForgetUserNameMutationVariables, TContext>({
+    mutationKey: ["ForgetUserName"],
+    mutationFn: (variables?: GQLForgetUserNameMutationVariables) =>
+      fetcher<GQLForgetUserNameMutation, GQLForgetUserNameMutationVariables>(
+        client,
+        ForgetUserNameDocument,
+        variables,
+        headers,
+      )(),
+    ...options,
+  });
+};
+
 export const GenerateStructuredDataDocument = `
     mutation GenerateStructuredData($input: StructuredDataInput!) {
   generateStructuredData(input: $input) {
@@ -9697,6 +9806,7 @@ export const DataConnectionsDocument = `
     id
     name
     connectorType
+    sourceName
     status
     error
     createdAt
@@ -9962,6 +10072,10 @@ export const DatastoreSubscriptionsDocument = `
     datastoreId
     status
     inGoodStanding
+    amount
+    currency
+    interval
+    intervalCount
   }
 }
     `;
@@ -10152,9 +10266,9 @@ useSuspenseDetectMemoryConflictsQuery.getKey = (variables: GQLDetectMemoryConfli
   variables,
 ];
 
-export const ExplainRdFoxFactDocument = `
-    query ExplainRDFoxFact($subject: String!, $predicate: String, $object: String!, $objectDatatype: String, $summarize: Boolean = false, $namedGraph: IRI) {
-  explainRDFoxFact(
+export const ExplainFactDocument = `
+    query ExplainFact($subject: String!, $predicate: String, $object: String!, $objectDatatype: String, $summarize: Boolean = false, $namedGraph: IRI) {
+  explainFact(
     subject: $subject
     predicate: $predicate
     object: $object
@@ -10170,19 +10284,19 @@ export const ExplainRdFoxFactDocument = `
 }
     `;
 
-export const useExplainRdFoxFactQuery = <TData = GQLExplainRdFoxFactQuery, TError = unknown>(
+export const useExplainFactQuery = <TData = GQLExplainFactQuery, TError = unknown>(
   client: GraphQLClient,
-  variables: GQLExplainRdFoxFactQueryVariables,
-  options?: Omit<UseQueryOptions<GQLExplainRdFoxFactQuery, TError, TData>, "queryKey"> & {
-    queryKey?: UseQueryOptions<GQLExplainRdFoxFactQuery, TError, TData>["queryKey"];
+  variables: GQLExplainFactQueryVariables,
+  options?: Omit<UseQueryOptions<GQLExplainFactQuery, TError, TData>, "queryKey"> & {
+    queryKey?: UseQueryOptions<GQLExplainFactQuery, TError, TData>["queryKey"];
   },
   headers?: RequestInit["headers"],
 ) => {
-  return useQuery<GQLExplainRdFoxFactQuery, TError, TData>({
-    queryKey: ["ExplainRDFoxFact", variables],
-    queryFn: fetcher<GQLExplainRdFoxFactQuery, GQLExplainRdFoxFactQueryVariables>(
+  return useQuery<GQLExplainFactQuery, TError, TData>({
+    queryKey: ["ExplainFact", variables],
+    queryFn: fetcher<GQLExplainFactQuery, GQLExplainFactQueryVariables>(
       client,
-      ExplainRdFoxFactDocument,
+      ExplainFactDocument,
       variables,
       headers,
     ),
@@ -10190,21 +10304,21 @@ export const useExplainRdFoxFactQuery = <TData = GQLExplainRdFoxFactQuery, TErro
   });
 };
 
-useExplainRdFoxFactQuery.getKey = (variables: GQLExplainRdFoxFactQueryVariables) => ["ExplainRDFoxFact", variables];
+useExplainFactQuery.getKey = (variables: GQLExplainFactQueryVariables) => ["ExplainFact", variables];
 
-export const useSuspenseExplainRdFoxFactQuery = <TData = GQLExplainRdFoxFactQuery, TError = unknown>(
+export const useSuspenseExplainFactQuery = <TData = GQLExplainFactQuery, TError = unknown>(
   client: GraphQLClient,
-  variables: GQLExplainRdFoxFactQueryVariables,
-  options?: Omit<UseSuspenseQueryOptions<GQLExplainRdFoxFactQuery, TError, TData>, "queryKey"> & {
-    queryKey?: UseSuspenseQueryOptions<GQLExplainRdFoxFactQuery, TError, TData>["queryKey"];
+  variables: GQLExplainFactQueryVariables,
+  options?: Omit<UseSuspenseQueryOptions<GQLExplainFactQuery, TError, TData>, "queryKey"> & {
+    queryKey?: UseSuspenseQueryOptions<GQLExplainFactQuery, TError, TData>["queryKey"];
   },
   headers?: RequestInit["headers"],
 ) => {
-  return useSuspenseQuery<GQLExplainRdFoxFactQuery, TError, TData>({
-    queryKey: ["ExplainRDFoxFactSuspense", variables],
-    queryFn: fetcher<GQLExplainRdFoxFactQuery, GQLExplainRdFoxFactQueryVariables>(
+  return useSuspenseQuery<GQLExplainFactQuery, TError, TData>({
+    queryKey: ["ExplainFactSuspense", variables],
+    queryFn: fetcher<GQLExplainFactQuery, GQLExplainFactQueryVariables>(
       client,
-      ExplainRdFoxFactDocument,
+      ExplainFactDocument,
       variables,
       headers,
     ),
@@ -10212,10 +10326,7 @@ export const useSuspenseExplainRdFoxFactQuery = <TData = GQLExplainRdFoxFactQuer
   });
 };
 
-useSuspenseExplainRdFoxFactQuery.getKey = (variables: GQLExplainRdFoxFactQueryVariables) => [
-  "ExplainRDFoxFactSuspense",
-  variables,
-];
+useSuspenseExplainFactQuery.getKey = (variables: GQLExplainFactQueryVariables) => ["ExplainFactSuspense", variables];
 
 export const GmailIntegrationsDocument = `
     query GmailIntegrations {
@@ -12156,3 +12267,69 @@ useSuspenseValidateActionPlanQuery.getKey = (variables: GQLValidateActionPlanQue
   "ValidateActionPlanSuspense",
   variables,
 ];
+
+export const WarehouseQueryDocument = `
+    query WarehouseQuery($sql: String!) {
+  warehouseQuery(sql: $sql) {
+    columns {
+      name
+      type
+    }
+    rows
+    truncated
+    statistics {
+      scanRows
+      scanBytes
+      returnRows
+    }
+    error
+  }
+}
+    `;
+
+export const useWarehouseQueryQuery = <TData = GQLWarehouseQueryQuery, TError = unknown>(
+  client: GraphQLClient,
+  variables: GQLWarehouseQueryQueryVariables,
+  options?: Omit<UseQueryOptions<GQLWarehouseQueryQuery, TError, TData>, "queryKey"> & {
+    queryKey?: UseQueryOptions<GQLWarehouseQueryQuery, TError, TData>["queryKey"];
+  },
+  headers?: RequestInit["headers"],
+) => {
+  return useQuery<GQLWarehouseQueryQuery, TError, TData>({
+    queryKey: ["WarehouseQuery", variables],
+    queryFn: fetcher<GQLWarehouseQueryQuery, GQLWarehouseQueryQueryVariables>(
+      client,
+      WarehouseQueryDocument,
+      variables,
+      headers,
+    ),
+    ...options,
+  });
+};
+
+useWarehouseQueryQuery.getKey = (variables: GQLWarehouseQueryQueryVariables) => ["WarehouseQuery", variables];
+
+export const useSuspenseWarehouseQueryQuery = <TData = GQLWarehouseQueryQuery, TError = unknown>(
+  client: GraphQLClient,
+  variables: GQLWarehouseQueryQueryVariables,
+  options?: Omit<UseSuspenseQueryOptions<GQLWarehouseQueryQuery, TError, TData>, "queryKey"> & {
+    queryKey?: UseSuspenseQueryOptions<GQLWarehouseQueryQuery, TError, TData>["queryKey"];
+  },
+  headers?: RequestInit["headers"],
+) => {
+  return useSuspenseQuery<GQLWarehouseQueryQuery, TError, TData>({
+    queryKey: ["WarehouseQuerySuspense", variables],
+    queryFn: fetcher<GQLWarehouseQueryQuery, GQLWarehouseQueryQueryVariables>(
+      client,
+      WarehouseQueryDocument,
+      variables,
+      headers,
+    ),
+    ...options,
+  });
+};
+
+useSuspenseWarehouseQueryQuery.getKey = (variables: GQLWarehouseQueryQueryVariables) => [
+  "WarehouseQuerySuspense",
+  variables,
+];

package/src/shared/datastore-id.ts

@@ -17,7 +17,7 @@ export function deriveDatastoreId(organizationName: string, segments: string[]):
   );
 }
 
-// Validated before reaching Infisical/KGC/RDFox since the Keycloak attribute is admin-settable.
+// Validated before reaching Infisical, KGC or the datastore, since the Keycloak attribute is admin-settable.
 const datastoreIdPattern = /^[a-z0-9]+(?:-[a-z0-9]+)*$/;
 
 export function isValidDatastoreId(id: string): boolean {

package/src/shared/memory-citation-sampling.ts

@@ -9,18 +9,18 @@ export function shouldRequestCitations(seed: string, rate: number = memoryCitati
 
 export type CitationRequestTurn = {
   hasPersonalPreload: boolean;
+  hasPersonalRecall: boolean;
   hasSharedSections: boolean;
   seed: string;
 };
 
 export type CitationDecision =
   | { requested: false; reason: "none" }
-  | { requested: true; reason: "personal" | "sampled" };
+  | { requested: true; reason: "personal" | "recalled" | "sampled" };
 
 export function citationDecisionForTurn(turn: CitationRequestTurn, rate?: number): CitationDecision {
   if (turn.hasPersonalPreload) return { requested: true, reason: "personal" };
-  if (turn.hasSharedSections && shouldRequestCitations(turn.seed, rate)) {
-    return { requested: true, reason: "sampled" };
-  }
-  return { requested: false, reason: "none" };
+  if (!turn.hasPersonalRecall && !turn.hasSharedSections) return { requested: false, reason: "none" };
+  if (!shouldRequestCitations(turn.seed, rate)) return { requested: false, reason: "none" };
+  return { requested: true, reason: turn.hasPersonalRecall ? "recalled" : "sampled" };
 }

package/src/shared/memory-diversity.ts

@@ -7,7 +7,7 @@ export type DiversityCandidate = {
   description?: string | null;
 };
 
-const tokenPattern = /[a-z0-9]+/g;
+const tokenPattern = /[\p{L}\p{N}]+/gu;
 
 const minimumTokenLength = 3;
 

package/src/shared/memory-keyword-recall.ts

@@ -5,6 +5,8 @@ export const rankFusionConstant = 60;
 
 export const maxKeywordTerms = 8;
 
+export const maxKeywordTermLength = 40;
+
 export type KeywordCandidate = DiversityCandidate & { iri: string };
 
 // A term feeds a SPARQL CONTAINS that admits rows, so a stray function word like "the" would widen the pool to the
@@ -103,7 +105,7 @@ const stopWords = new Set([
 
 export function keywordTerms(text: string, limit: number = maxKeywordTerms): string[] {
   return [...memoryTokens({ label: text })]
-    .filter((token) => !stopWords.has(token))
+    .filter((token) => !stopWords.has(token) && token.length <= maxKeywordTermLength)
     .sort((a, b) => b.length - a.length || a.localeCompare(b))
     .slice(0, limit);
 }

package/src/shared/rdf-namespaces.ts

@@ -3,12 +3,8 @@ import {
   expandPrefix as expandPrefixIri,
   iriLocalName as iriLocalNameValue,
   iriNamespace as iriNamespaceValue,
-  rdfoxDefaultBaseIri as rdfoxDefaultBaseIriValue,
-  shaclGraphIri as shaclGraphIriValue,
 } from "@manta-eu/rdf-lib/namespaces";
 
-export const rulesGraphIri = `${rdfoxDefaultBaseIriValue}rules`;
-
 // KGC uploads the realm ontology here (kgc/kgc/realms/lib/ontology.py: realm_base + "ontology").
 export const ontologyGraphIri = expandPrefixIri(":ontology");
 
@@ -19,8 +15,6 @@ export const writeLifecycle = {
   createOnly: expandPrefixIri("relay:CreateOnly"),
 };
 
-export const nonDataGraphIris = [ontologyGraphIri, shaclGraphIriValue, rulesGraphIri];
-
 export function mintInstanceIri(classIri: string, localId: string): string {
   return `${iriNamespaceValue(classIri)}${encodeIriLocalNameValue(`${iriLocalNameValue(classIri)}_${localId}`)}`;
 }
@@ -33,7 +27,7 @@ export {
   encodeIriUnsafeChars,
   expandPrefix,
   formatEntityName,
-  formatIriForRdfox,
+  formatIriForSparql,
   getNamespaceValues,
   isFullIri,
   isValidFullIri,
@@ -52,8 +46,7 @@ export {
   nfo,
   parseEntityUri,
   prependMissingSparqlPrefixes,
-  rdfoxDefaultBaseIri,
-  shaclGraphIri,
+  shaclGraph,
   sh,
   toTurtlePrefixes,
   validateFullIri,

package/src/shared/tools/coercers.ts

@@ -1,5 +1,5 @@
 import type {
-  RDFoxFactsOutput,
+  QueryKnowledgeGraphOutput,
   ShowEntitiesOutput,
   SparqlJsonResults,
   ToolErrorResult,
@@ -43,8 +43,11 @@ function coerceWith<T>(
   return { is_error: true, content: `Unexpected ${toolName} output format` };
 }
 
-function coerceRDFoxFactsOutput(raw: unknown): RDFoxFactsOutput {
-  return coerceWith<RDFoxFactsOutput>("getRDFoxFacts", raw, [isSparqlJsonResults, isJsonLdGraphDocument]);
+function coerceQueryKnowledgeGraphOutput(raw: unknown): QueryKnowledgeGraphOutput {
+  return coerceWith<QueryKnowledgeGraphOutput>("queryKnowledgeGraph", raw, [
+    isSparqlJsonResults,
+    isJsonLdGraphDocument,
+  ]);
 }
 
 export function isShowEntitiesResult(data: unknown): data is Exclude<ShowEntitiesOutput, ToolErrorResult> {
@@ -62,7 +65,7 @@ function coerceShowEntitiesOutput(raw: unknown): ShowEntitiesOutput {
 }
 
 const coercers: Partial<Record<ToolName, ToolOutputCoercer<ToolName>>> = {
-  getRDFoxFacts: coerceRDFoxFactsOutput as ToolOutputCoercer<"getRDFoxFacts">,
+  queryKnowledgeGraph: coerceQueryKnowledgeGraphOutput as ToolOutputCoercer<"queryKnowledgeGraph">,
   showEntities: coerceShowEntitiesOutput as ToolOutputCoercer<"showEntities">,
 };