@kevisual/cnb 0.0.56 → 0.0.58

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kevisual/cnb",
-  "version": "0.0.56",
+  "version": "0.0.58",
   "description": "",
   "main": "index.js",
   "basename": "/root/cnb",

package/src/index.ts

@@ -8,6 +8,8 @@ import { Issue } from "./issue/index.ts";
 import { Mission } from "./mission/index.ts";
 import { AiBase } from "./ai/index.ts";
 import { RepoLabel, IssueLabel } from "./labels/index.ts";
+import { RegistryPackage, PackageManagement } from "./package/index.ts";
+import { Organization } from "./org/index.ts";
 
 type CNBOptions = CNBCoreOptions<{
 }>;
@@ -25,6 +27,11 @@ export class CNB extends CNBCore {
     repoLabel: RepoLabel;
     issueLabel: IssueLabel;
   };
+  packages!: {
+    registry: RegistryPackage;
+    package: PackageManagement;
+  };
+  org!: Organization;
   constructor(options: CNBOptions) {
     super({ ...options, token: options.token, cookie: options.cookie, cnb: options.cnb });
     this.init(options);
@@ -47,6 +54,11 @@ export class CNB extends CNBCore {
       repoLabel: new RepoLabel(options),
       issueLabel: new IssueLabel(options),
     };
+    this.packages = {
+      registry: new RegistryPackage(options),
+      package: new PackageManagement(options),
+    };
+    this.org = new Organization(options);
   }
   setToken(token: string) {
     this.token = token;
@@ -58,6 +70,9 @@ export class CNB extends CNBCore {
     this.mission.token = token;
     this.labels.repoLabel.token = token;
     this.labels.issueLabel.token = token;
+    this.packages.registry.token = token;
+    this.packages.package.token = token;
+    this.org.token = token;
   }
   setCookie(cookie: string) {
     this.cookie = cookie;
@@ -69,6 +84,9 @@ export class CNB extends CNBCore {
     this.mission.cookie = cookie;
     this.labels.repoLabel.cookie = cookie;
     this.labels.issueLabel.cookie = cookie;
+    this.packages.registry.cookie = cookie;
+    this.packages.package.cookie = cookie;
+    this.org.cookie = cookie;
   }
   getCNBVersion = getCNBVersion
 }
@@ -95,4 +113,6 @@ type VersionInfo = {
 }
 
 export * from './issue/npc/env.ts'
-export * from './labels/index.ts'
+export * from './labels/index.ts'
+export * from './package/index.ts'
+export * from './org/index.ts'

package/src/issue/index.ts

@@ -63,8 +63,11 @@ export type IssueItem = {
   number: string;
   priority: string;
   started_at: string;
-  state: string;
-  state_reason: string;
+  state: 'open' | 'closed';
+  /**
+   * state_reason must be one of completed, not_planned, reopened
+   */
+  state_reason: 'completed' | 'not_planned' | 'reopened';
   title: string;
   updated_at: string;
 };

package/src/knowledge/index.ts

@@ -1,5 +1,6 @@
 import { CNBCore, CNBCoreOptions, Result } from "../cnb-core.ts";
 
+// https://cnb.cool/cnb/plugins/cnbcool/knowledge-base/-/blob/main/src/api.py
 export class KnowledgeBase extends CNBCore {
   constructor(options: CNBCoreOptions) {
     super(options);
@@ -12,10 +13,7 @@ export class KnowledgeBase extends CNBCore {
     metadata_filtering_conditions?: MetadataFilteringConditions
   }): Promise<Result<QueryRag[]>> {
     const url = `/${repo}/-/knowledge/base/query`;
-    let postData = {
-      query: data.query,
-    };
-    return this.post({ url, data: postData });
+    return this.post({ url, data });
   }
   getEmbeddingModels(repo: string): Promise<Result<Array<{ name: string, dimensions: number }>>> {
     const url = `/${repo}/-/knowledge/embedding/models`;
@@ -31,6 +29,95 @@ export class KnowledgeBase extends CNBCore {
     const url = `/${repo}/-/knowledge/base`;
     return this.request({ url, method: 'DELETE' });
   }
+
+  /**
+   * 未暴露
+   * @param repo 
+   * @param data 
+   * @returns 
+   */
+  createKnowledgeBase(repo: string, data: {
+    embedding_model_name: string;
+    include: string;
+    exclude: string;
+    issue_sync_enabled?: boolean;
+    processing?: {
+      chunk_size: number;
+      chunk_overlap: number;
+      text_separator: string;
+    };
+    issue?: {
+      labels: string;
+      state: string;
+    };
+  }): Promise<Result<any>> {
+    const url = `/${repo}/-/knowledge/base`;
+    return this.post({ url, data });
+  }
+  /**
+   * 未暴露
+   * @param repo 
+   * @param data 
+   * @returns
+   */
+  updateKnowledgeBase(repo: string, data: {
+    last_commit_sha?: string;
+    issue_last_sync_time?: string;
+  }): Promise<Result<any>> {
+    const url = `/${repo}/-/knowledge/base`;
+    return this.put({ url, data });
+  }
+
+  /**
+   * 未暴露
+   * @param repo 
+   * @param text 
+   * @returns 
+   */
+  getEmbedding(repo: string, text: string): Promise<Result<{ embedding: number[] }>> {
+    const url = `/${repo}/-/knowledge/embedding`;
+    return this.post({ url, data: { text } });
+  }
+  /**
+   *  未暴露
+   * @param repo 
+   * @param chunksData 
+   * @returns 
+   */
+  addDocument(repo: string, chunksData: {
+    path: string;
+    chunks: Array<{
+      content: string;
+      hash: string;
+      offset: number;
+      size: number;
+    }>;
+  }): Promise<Result<any>> {
+    const url = `/${repo}/-/knowledge/documents/upsert-document-with-chunks`;
+    return this.post({ url, data: chunksData });
+  }
+  /**
+   * 未暴露
+   * @param repo 
+   * @param paths 
+   * @returns 
+   */
+
+  deleteDocument(repo: string, paths: string[]): Promise<Result<any>> {
+    const url = `/${repo}/-/knowledge/documents`;
+    return this.delete({ url, data: { paths } });
+  }
+  /**
+   * 未暴露
+   * @param repo 
+   * @param page 
+   * @param page_size 
+   * @returns 
+   */
+  listDocument(repo: string, page: number = 1, page_size: number = 50): Promise<Result<any[]>> {
+    const url = `/${repo}/-/knowledge/documents`;
+    return this.get({ url, params: { page, page_size } });
+  }
 }
 
 type MetadataFilteringConditions = {
@@ -42,7 +129,7 @@ type MetadataFilteringConditions = {
      */
     value?: string
   }>
-  logical_operator?: 'adn' | 'or'
+  logical_operator?: 'and' | 'or'
 }
 
 type QueryRag = {