@stacksjs/ts-cloud 0.1.3 → 0.1.5

package/{src/aws/kendra.ts → dist/aws/kendra.d.ts}

@@ -1,16 +1,16 @@
+import { AWSClient } from './client';
 /**
- * AWS Kendra Client
- * Enterprise search service
- * No external SDK dependencies - implements AWS Signature V4 directly
+ * Quick search
  */
-
-import { AWSClient } from './client'
-
+export declare function search(indexId: string, query: string, options?: { region?: string; pageSize?: number }): Promise<Array<{ title: string; excerpt: string; uri: string; score: string }>>;
+/**
+ * Retrieve for RAG
+ */
+export declare function retrieveForRag(indexId: string, query: string, options?: { region?: string; pageSize?: number }): Promise<Array<{ content: string; uri: string; score: string }>>;
 // ============================================================================
 // Types
 // ============================================================================
-
-export interface CreateIndexCommandInput {
+export declare interface CreateIndexCommandInput {
   Name: string
   Edition?: 'DEVELOPER_EDITION' | 'ENTERPRISE_EDITION'
   RoleArn: string
@@ -40,16 +40,13 @@ export interface CreateIndexCommandInput {
     UserGroupResolutionMode: 'AWS_SSO' | 'NONE'
   }
 }
-
-export interface CreateIndexCommandOutput {
+export declare interface CreateIndexCommandOutput {
   Id?: string
 }
-
-export interface DescribeIndexCommandInput {
+export declare interface DescribeIndexCommandInput {
   Id: string
 }
-
-export interface Index {
+export declare interface Index {
   Name?: string
   Id?: string
   Edition?: 'DEVELOPER_EDITION' | 'ENTERPRISE_EDITION'
@@ -112,8 +109,7 @@ export interface Index {
     UserGroupResolutionMode?: 'AWS_SSO' | 'NONE'
   }
 }
-
-export interface DescribeIndexCommandOutput {
+export declare interface DescribeIndexCommandOutput {
   Name?: string
   Id?: string
   Edition?: 'DEVELOPER_EDITION' | 'ENTERPRISE_EDITION'
@@ -130,13 +126,11 @@ export interface DescribeIndexCommandOutput {
   ErrorMessage?: string
   CapacityUnits?: Index['CapacityUnits']
 }
-
-export interface ListIndicesCommandInput {
+export declare interface ListIndicesCommandInput {
   NextToken?: string
   MaxResults?: number
 }
-
-export interface IndexSummary {
+export declare interface IndexSummary {
   Name?: string
   Id?: string
   Edition?: 'DEVELOPER_EDITION' | 'ENTERPRISE_EDITION'
@@ -144,21 +138,17 @@ export interface IndexSummary {
   UpdatedAt?: string
   Status?: 'CREATING' | 'ACTIVE' | 'DELETING' | 'FAILED' | 'UPDATING' | 'SYSTEM_UPDATING'
 }
-
-export interface ListIndicesCommandOutput {
+export declare interface ListIndicesCommandOutput {
   IndexConfigurationSummaryItems?: IndexSummary[]
   NextToken?: string
 }
-
-export interface DeleteIndexCommandInput {
+export declare interface DeleteIndexCommandInput {
   Id: string
 }
+export declare interface DeleteIndexCommandOutput {
 
-export interface DeleteIndexCommandOutput {
-  // Empty
 }
-
-export interface CreateDataSourceCommandInput {
+export declare interface CreateDataSourceCommandInput {
   Name: string
   IndexId: string
   Type: 'S3' | 'SHAREPOINT' | 'DATABASE' | 'SALESFORCE' | 'ONEDRIVE' | 'SERVICENOW' | 'CUSTOM' | 'CONFLUENCE' | 'GOOGLEDRIVE' | 'WEBCRAWLER' | 'WORKDOCS' | 'FSX' | 'SLACK' | 'BOX' | 'QUIP' | 'JIRA' | 'GITHUB' | 'ALFRESCO' | 'TEMPLATE'
@@ -320,17 +310,14 @@ export interface CreateDataSourceCommandInput {
     RoleArn?: string
   }
 }
-
-export interface CreateDataSourceCommandOutput {
+export declare interface CreateDataSourceCommandOutput {
   Id?: string
 }
-
-export interface DescribeDataSourceCommandInput {
+export declare interface DescribeDataSourceCommandInput {
   Id: string
   IndexId: string
 }
-
-export interface DataSource {
+export declare interface DataSource {
   Id?: string
   IndexId?: string
   Name?: string
@@ -350,16 +337,15 @@ export interface DataSource {
   LanguageCode?: string
   CustomDocumentEnrichmentConfiguration?: Record<string, unknown>
 }
+export declare interface DescribeDataSourceCommandOutput extends DataSource {
 
-export interface DescribeDataSourceCommandOutput extends DataSource {}
-
-export interface ListDataSourcesCommandInput {
+}
+export declare interface ListDataSourcesCommandInput {
   IndexId: string
   NextToken?: string
   MaxResults?: number
 }
-
-export interface DataSourceSummary {
+export declare interface DataSourceSummary {
   Name?: string
   Id?: string
   Type?: string
@@ -368,31 +354,25 @@ export interface DataSourceSummary {
   Status?: 'CREATING' | 'DELETING' | 'FAILED' | 'UPDATING' | 'ACTIVE'
   LanguageCode?: string
 }
-
-export interface ListDataSourcesCommandOutput {
+export declare interface ListDataSourcesCommandOutput {
   SummaryItems?: DataSourceSummary[]
   NextToken?: string
 }
-
-export interface StartDataSourceSyncJobCommandInput {
+export declare interface StartDataSourceSyncJobCommandInput {
   Id: string
   IndexId: string
 }
-
-export interface StartDataSourceSyncJobCommandOutput {
+export declare interface StartDataSourceSyncJobCommandOutput {
   ExecutionId?: string
 }
-
-export interface StopDataSourceSyncJobCommandInput {
+export declare interface StopDataSourceSyncJobCommandInput {
   Id: string
   IndexId: string
 }
+export declare interface StopDataSourceSyncJobCommandOutput {
 
-export interface StopDataSourceSyncJobCommandOutput {
-  // Empty
 }
-
-export interface QueryCommandInput {
+export declare interface QueryCommandInput {
   IndexId: string
   QueryText?: string
   AttributeFilter?: {
@@ -517,8 +497,7 @@ export interface QueryCommandInput {
     }
   }
 }
-
-export interface QueryResultItem {
+export declare interface QueryResultItem {
   Id?: string
   Type?: 'DOCUMENT' | 'QUESTION_ANSWER' | 'ANSWER'
   Format?: 'TABLE' | 'TEXT'
@@ -594,8 +573,7 @@ export interface QueryResultItem {
     ExpandedResults?: QueryResultItem[]
   }
 }
-
-export interface FacetResult {
+export declare interface FacetResult {
   DocumentAttributeKey?: string
   DocumentAttributeValueType?: 'STRING_VALUE' | 'STRING_LIST_VALUE' | 'LONG_VALUE' | 'DATE_VALUE'
   DocumentAttributeValueCountPairs?: Array<{
@@ -609,8 +587,7 @@ export interface FacetResult {
     FacetResults?: FacetResult[]
   }>
 }
-
-export interface QueryCommandOutput {
+export declare interface QueryCommandOutput {
   QueryId?: string
   ResultItems?: QueryResultItem[]
   FacetResults?: FacetResult[]
@@ -663,8 +640,7 @@ export interface QueryCommandOutput {
     FeedbackToken?: string
   }>
 }
-
-export interface RetrieveCommandInput {
+export declare interface RetrieveCommandInput {
   IndexId: string
   QueryText: string
   AttributeFilter?: QueryCommandInput['AttributeFilter']
@@ -674,8 +650,7 @@ export interface RetrieveCommandInput {
   PageSize?: number
   UserContext?: QueryCommandInput['UserContext']
 }
-
-export interface RetrieveResultItem {
+export declare interface RetrieveResultItem {
   Id?: string
   DocumentId?: string
   DocumentTitle?: string
@@ -694,13 +669,11 @@ export interface RetrieveResultItem {
     ScoreConfidence?: 'VERY_HIGH' | 'HIGH' | 'MEDIUM' | 'LOW' | 'NOT_AVAILABLE'
   }
 }
-
-export interface RetrieveCommandOutput {
+export declare interface RetrieveCommandOutput {
   QueryId?: string
   ResultItems?: RetrieveResultItem[]
 }
-
-export interface BatchPutDocumentCommandInput {
+export declare interface BatchPutDocumentCommandInput {
   IndexId: string
   RoleArn?: string
   Documents: Array<{
@@ -739,16 +712,14 @@ export interface BatchPutDocumentCommandInput {
   }>
   CustomDocumentEnrichmentConfiguration?: CreateDataSourceCommandInput['CustomDocumentEnrichmentConfiguration']
 }
-
-export interface BatchPutDocumentCommandOutput {
+export declare interface BatchPutDocumentCommandOutput {
   FailedDocuments?: Array<{
     Id?: string
     ErrorCode?: 'InternalError' | 'InvalidRequest'
     ErrorMessage?: string
   }>
 }
-
-export interface BatchDeleteDocumentCommandInput {
+export declare interface BatchDeleteDocumentCommandInput {
   IndexId: string
   DocumentIdList: string[]
   DataSourceSyncJobMetricTarget?: {
@@ -756,342 +727,56 @@ export interface BatchDeleteDocumentCommandInput {
     DataSourceSyncJobId?: string
   }
 }
-
-export interface BatchDeleteDocumentCommandOutput {
+export declare interface BatchDeleteDocumentCommandOutput {
   FailedDocuments?: Array<{
     Id?: string
     ErrorCode?: 'InternalError' | 'InvalidRequest'
     ErrorMessage?: string
   }>
 }
-
 // ============================================================================
 // Kendra Client
 // ============================================================================
-
-export class KendraClient {
-  private client: AWSClient
-  private region: string
-
-  constructor(region: string = 'us-east-1') {
-    this.region = region
-    this.client = new AWSClient()
-  }
-
-  private async request<T>(action: string, params: Record<string, unknown>): Promise<T> {
-    return this.client.request({
-      service: 'kendra',
-      region: this.region,
-      method: 'POST',
-      path: '/',
-      headers: {
-        'Content-Type': 'application/x-amz-json-1.1',
-        'X-Amz-Target': `AWSKendraFrontendService.${action}`,
-      },
-      body: JSON.stringify(params),
-    })
-  }
-
-  // -------------------------------------------------------------------------
-  // Index Management
-  // -------------------------------------------------------------------------
-
-  async createIndex(params: CreateIndexCommandInput): Promise<CreateIndexCommandOutput> {
-    return this.request('CreateIndex', params as unknown as Record<string, unknown>)
-  }
-
-  async describeIndex(params: DescribeIndexCommandInput): Promise<DescribeIndexCommandOutput> {
-    return this.request('DescribeIndex', params as unknown as Record<string, unknown>)
-  }
-
-  async listIndices(params?: ListIndicesCommandInput): Promise<ListIndicesCommandOutput> {
-    return this.request('ListIndices', (params || {}) as unknown as Record<string, unknown>)
-  }
-
-  async deleteIndex(params: DeleteIndexCommandInput): Promise<DeleteIndexCommandOutput> {
-    return this.request('DeleteIndex', params as unknown as Record<string, unknown>)
-  }
-
-  // -------------------------------------------------------------------------
-  // Data Source Management
-  // -------------------------------------------------------------------------
-
-  async createDataSource(params: CreateDataSourceCommandInput): Promise<CreateDataSourceCommandOutput> {
-    return this.request('CreateDataSource', params as unknown as Record<string, unknown>)
-  }
-
-  async describeDataSource(params: DescribeDataSourceCommandInput): Promise<DescribeDataSourceCommandOutput> {
-    return this.request('DescribeDataSource', params as unknown as Record<string, unknown>)
-  }
-
-  async listDataSources(params: ListDataSourcesCommandInput): Promise<ListDataSourcesCommandOutput> {
-    return this.request('ListDataSources', params as unknown as Record<string, unknown>)
-  }
-
-  async startDataSourceSyncJob(params: StartDataSourceSyncJobCommandInput): Promise<StartDataSourceSyncJobCommandOutput> {
-    return this.request('StartDataSourceSyncJob', params as unknown as Record<string, unknown>)
-  }
-
-  async stopDataSourceSyncJob(params: StopDataSourceSyncJobCommandInput): Promise<StopDataSourceSyncJobCommandOutput> {
-    return this.request('StopDataSourceSyncJob', params as unknown as Record<string, unknown>)
-  }
-
-  // -------------------------------------------------------------------------
-  // Search
-  // -------------------------------------------------------------------------
-
-  async query(params: QueryCommandInput): Promise<QueryCommandOutput> {
-    return this.request('Query', params as unknown as Record<string, unknown>)
-  }
-
-  async retrieve(params: RetrieveCommandInput): Promise<RetrieveCommandOutput> {
-    return this.request('Retrieve', params as unknown as Record<string, unknown>)
-  }
-
-  // -------------------------------------------------------------------------
-  // Document Management
-  // -------------------------------------------------------------------------
-
-  async batchPutDocument(params: BatchPutDocumentCommandInput): Promise<BatchPutDocumentCommandOutput> {
-    return this.request('BatchPutDocument', params as unknown as Record<string, unknown>)
-  }
-
-  async batchDeleteDocument(params: BatchDeleteDocumentCommandInput): Promise<BatchDeleteDocumentCommandOutput> {
-    return this.request('BatchDeleteDocument', params as unknown as Record<string, unknown>)
-  }
-
-  // -------------------------------------------------------------------------
-  // Convenience Methods
-  // -------------------------------------------------------------------------
-
-  /**
-   * Simple search query
-   */
-  async search(indexId: string, queryText: string, options?: {
+export declare class KendraClient {
+  private client: AWSClient;
+  private region: string;
+  constructor(region?: string);
+  private request<T>(action: string, params: Record<string, unknown>): Promise<T>;
+  createIndex(params: CreateIndexCommandInput): Promise<CreateIndexCommandOutput>;
+  describeIndex(params: DescribeIndexCommandInput): Promise<DescribeIndexCommandOutput>;
+  listIndices(params?: ListIndicesCommandInput): Promise<ListIndicesCommandOutput>;
+  deleteIndex(params: DeleteIndexCommandInput): Promise<DeleteIndexCommandOutput>;
+  createDataSource(params: CreateDataSourceCommandInput): Promise<CreateDataSourceCommandOutput>;
+  describeDataSource(params: DescribeDataSourceCommandInput): Promise<DescribeDataSourceCommandOutput>;
+  listDataSources(params: ListDataSourcesCommandInput): Promise<ListDataSourcesCommandOutput>;
+  startDataSourceSyncJob(params: StartDataSourceSyncJobCommandInput): Promise<StartDataSourceSyncJobCommandOutput>;
+  stopDataSourceSyncJob(params: StopDataSourceSyncJobCommandInput): Promise<StopDataSourceSyncJobCommandOutput>;
+  query(params: QueryCommandInput): Promise<QueryCommandOutput>;
+  retrieve(params: RetrieveCommandInput): Promise<RetrieveCommandOutput>;
+  batchPutDocument(params: BatchPutDocumentCommandInput): Promise<BatchPutDocumentCommandOutput>;
+  batchDeleteDocument(params: BatchDeleteDocumentCommandInput): Promise<BatchDeleteDocumentCommandOutput>;
+  search(indexId: string, queryText: string, options?: {
     pageSize?: number
     pageNumber?: number
     attributeFilter?: QueryCommandInput['AttributeFilter']
-  }): Promise<QueryResultItem[]> {
-    const result = await this.query({
-      IndexId: indexId,
-      QueryText: queryText,
-      PageSize: options?.pageSize,
-      PageNumber: options?.pageNumber,
-      AttributeFilter: options?.attributeFilter,
-    })
-    return result.ResultItems || []
-  }
-
-  /**
-   * Retrieve documents (for RAG)
-   */
-  async retrieveDocuments(indexId: string, queryText: string, options?: {
+  }): Promise<QueryResultItem[]>;
+  retrieveDocuments(indexId: string, queryText: string, options?: {
     pageSize?: number
     pageNumber?: number
-  }): Promise<RetrieveResultItem[]> {
-    const result = await this.retrieve({
-      IndexId: indexId,
-      QueryText: queryText,
-      PageSize: options?.pageSize,
-      PageNumber: options?.pageNumber,
-    })
-    return result.ResultItems || []
-  }
-
-  /**
-   * Add a text document to the index
-   */
-  async addDocument(indexId: string, document: {
+  }): Promise<RetrieveResultItem[]>;
+  addDocument(indexId: string, document: {
     id: string
     title?: string
     content: string
     attributes?: Record<string, string | string[] | number>
-  }): Promise<void> {
-    const attributes = document.attributes
-      ? Object.entries(document.attributes).map(([key, value]) => ({
-          Key: key,
-          Value: typeof value === 'number'
-            ? { LongValue: value }
-            : Array.isArray(value)
-              ? { StringListValue: value }
-              : { StringValue: value },
-        }))
-      : undefined
-
-    const result = await this.batchPutDocument({
-      IndexId: indexId,
-      Documents: [
-        {
-          Id: document.id,
-          Title: document.title,
-          Blob: new TextEncoder().encode(document.content),
-          ContentType: 'PLAIN_TEXT',
-          Attributes: attributes,
-        },
-      ],
-    })
-
-    if (result.FailedDocuments?.length) {
-      throw new Error(`Failed to add document: ${result.FailedDocuments[0].ErrorMessage}`)
-    }
-  }
-
-  /**
-   * Add multiple documents
-   */
-  async addDocuments(indexId: string, documents: Array<{
+  }): Promise<void>;
+  addDocuments(indexId: string, documents: Array<{
     id: string
     title?: string
     content: string
     attributes?: Record<string, string | string[] | number>
-  }>): Promise<{ succeeded: number; failed: Array<{ id: string; error: string }> }> {
-    const docs = documents.map(doc => {
-      const attributes = doc.attributes
-        ? Object.entries(doc.attributes).map(([key, value]) => ({
-            Key: key,
-            Value: typeof value === 'number'
-              ? { LongValue: value }
-              : Array.isArray(value)
-                ? { StringListValue: value }
-                : { StringValue: value },
-          }))
-        : undefined
-
-      return {
-        Id: doc.id,
-        Title: doc.title,
-        Blob: new TextEncoder().encode(doc.content),
-        ContentType: 'PLAIN_TEXT' as const,
-        Attributes: attributes,
-      }
-    })
-
-    const result = await this.batchPutDocument({
-      IndexId: indexId,
-      Documents: docs,
-    })
-
-    const failed = result.FailedDocuments?.map(f => ({
-      id: f.Id || 'unknown',
-      error: f.ErrorMessage || 'Unknown error',
-    })) || []
-
-    return {
-      succeeded: documents.length - failed.length,
-      failed,
-    }
-  }
-
-  /**
-   * Delete documents
-   */
-  async deleteDocuments(indexId: string, documentIds: string[]): Promise<void> {
-    const result = await this.batchDeleteDocument({
-      IndexId: indexId,
-      DocumentIdList: documentIds,
-    })
-
-    if (result.FailedDocuments?.length) {
-      throw new Error(`Failed to delete documents: ${result.FailedDocuments.map(f => f.Id).join(', ')}`)
-    }
-  }
-
-  /**
-   * Wait for index to be active
-   */
-  async waitForIndex(
-    indexId: string,
-    options?: { maxWaitMs?: number; pollIntervalMs?: number },
-  ): Promise<Index> {
-    const maxWaitMs = options?.maxWaitMs ?? 3600000 // 1 hour
-    const pollIntervalMs = options?.pollIntervalMs ?? 30000
-    const startTime = Date.now()
-
-    while (Date.now() - startTime < maxWaitMs) {
-      const result = await this.describeIndex({ Id: indexId })
-
-      if (result.Status === 'ACTIVE') {
-        return result as Index
-      }
-      if (result.Status === 'FAILED') {
-        throw new Error(`Index creation failed: ${result.ErrorMessage}`)
-      }
-
-      await new Promise(resolve => setTimeout(resolve, pollIntervalMs))
-    }
-
-    throw new Error(`Timeout waiting for index ${indexId}`)
-  }
-
-  /**
-   * Wait for data source to be active
-   */
-  async waitForDataSource(
-    indexId: string,
-    dataSourceId: string,
-    options?: { maxWaitMs?: number; pollIntervalMs?: number },
-  ): Promise<DataSource> {
-    const maxWaitMs = options?.maxWaitMs ?? 1800000 // 30 minutes
-    const pollIntervalMs = options?.pollIntervalMs ?? 30000
-    const startTime = Date.now()
-
-    while (Date.now() - startTime < maxWaitMs) {
-      const result = await this.describeDataSource({
-        Id: dataSourceId,
-        IndexId: indexId,
-      })
-
-      if (result.Status === 'ACTIVE') {
-        return result
-      }
-      if (result.Status === 'FAILED') {
-        throw new Error(`Data source creation failed: ${result.ErrorMessage}`)
-      }
-
-      await new Promise(resolve => setTimeout(resolve, pollIntervalMs))
-    }
-
-    throw new Error(`Timeout waiting for data source ${dataSourceId}`)
-  }
-}
-
-// ============================================================================
-// Helper Functions
-// ============================================================================
-
-/**
- * Quick search
- */
-export async function search(
-  indexId: string,
-  query: string,
-  options?: { region?: string; pageSize?: number },
-): Promise<Array<{ title: string; excerpt: string; uri: string; score: string }>> {
-  const client = new KendraClient(options?.region || 'us-east-1')
-  const results = await client.search(indexId, query, { pageSize: options?.pageSize })
-
-  return results.map(r => ({
-    title: r.DocumentTitle?.Text || '',
-    excerpt: r.DocumentExcerpt?.Text || '',
-    uri: r.DocumentURI || '',
-    score: r.ScoreAttributes?.ScoreConfidence || 'NOT_AVAILABLE',
-  }))
-}
-
-/**
- * Retrieve for RAG
- */
-export async function retrieveForRag(
-  indexId: string,
-  query: string,
-  options?: { region?: string; pageSize?: number },
-): Promise<Array<{ content: string; uri: string; score: string }>> {
-  const client = new KendraClient(options?.region || 'us-east-1')
-  const results = await client.retrieveDocuments(indexId, query, { pageSize: options?.pageSize })
-
-  return results.map(r => ({
-    content: r.Content || '',
-    uri: r.DocumentURI || '',
-    score: r.ScoreAttributes?.ScoreConfidence || 'NOT_AVAILABLE',
-  }))
-}
+  }>): Promise<{ succeeded: number; failed: Array<{ id: string; error: string }> }>;
+  deleteDocuments(indexId: string, documentIds: string[]): Promise<void>;
+  waitForIndex(indexId: string, options?: { maxWaitMs?: number; pollIntervalMs?: number }): Promise<Index>;
+  waitForDataSource(indexId: string, dataSourceId: string, options?: { maxWaitMs?: number; pollIntervalMs?: number }): Promise<DataSource>;
+}