@rizom/brain 0.2.0-alpha.50 → 0.2.0-alpha.51

package/dist/entities.d.ts

@@ -232,18 +232,58 @@ interface EntityTypeConfig {
  * Core entity service interface for read-only operations
  * Used by core plugins that need entity access but shouldn't modify entities
  */
+interface GetEntityRequest {
+    entityType: string;
+    id: string;
+}
+type GetEntityRawRequest = GetEntityRequest;
+interface ListEntitiesRequest {
+    entityType: string;
+    options?: ListOptions | undefined;
+}
+interface CountEntitiesRequest {
+    entityType: string;
+    options?: Pick<ListOptions, "publishedOnly" | "filter"> | undefined;
+}
+interface CreateEntityRequest<T extends BaseEntity> {
+    entity: EntityInput<T>;
+    options?: CreateEntityOptions | undefined;
+}
+interface CreateEntityFromMarkdownRequest {
+    input: CreateEntityFromMarkdownInput;
+    options?: CreateEntityOptions | undefined;
+}
+interface UpdateEntityRequest<T extends BaseEntity> {
+    entity: T;
+    options?: EntityJobOptions | undefined;
+}
+interface DeleteEntityRequest {
+    entityType: string;
+    id: string;
+}
+interface UpsertEntityRequest<T extends BaseEntity> {
+    entity: T;
+    options?: EntityJobOptions | undefined;
+}
+interface EntitySearchRequest {
+    query: string;
+    options?: SearchOptions | undefined;
+}
+interface SearchWithDistancesRequest {
+    query: string;
+}
 interface ICoreEntityService {
-    getEntity<T extends BaseEntity>(entityType: string, id: string): Promise<T | null>;
+    getEntity<T extends BaseEntity>(request: GetEntityRequest): Promise<T | null>;
     /**
      * Get entity without content resolution (raw)
      * Used internally to avoid recursion when resolving image references
      */
-    getEntityRaw<T extends BaseEntity>(entityType: string, id: string): Promise<T | null>;
-    listEntities<T extends BaseEntity>(type: string, options?: ListOptions): Promise<T[]>;
-    search<T extends BaseEntity = BaseEntity>(query: string, options?: SearchOptions): Promise<SearchResult<T>[]>;
+    getEntityRaw<T extends BaseEntity>(request: GetEntityRawRequest): Promise<T | null>;
+    listEntities<T extends BaseEntity>(request: ListEntitiesRequest): Promise<T[]>;
+    search<T extends BaseEntity = BaseEntity>(request: EntitySearchRequest): Promise<SearchResult<T>[]>;
     getEntityTypes(): string[];
     hasEntityType(type: string): boolean;
-    countEntities(entityType: string, options?: Pick<ListOptions, "publishedOnly" | "filter">): Promise<number>;
+    countEntities(request: CountEntitiesRequest): Promise<number>;
     getEntityCounts(): Promise<Array<{
         entityType: string;
         count: number;
@@ -255,18 +295,18 @@ interface ICoreEntityService {
  * Entity service interface for managing brain entities
  */
 interface EntityService extends ICoreEntityService {
-    createEntity<T extends BaseEntity>(entity: EntityInput<T>, options?: CreateEntityOptions): Promise<EntityMutationResult>;
-    createEntityFromMarkdown(input: CreateEntityFromMarkdownInput, options?: CreateEntityOptions): Promise<EntityMutationResult>;
-    updateEntity<T extends BaseEntity>(entity: T, options?: EntityJobOptions): Promise<EntityMutationResult>;
-    deleteEntity(entityType: string, id: string): Promise<boolean>;
-    upsertEntity<T extends BaseEntity>(entity: T, options?: EntityJobOptions): Promise<EntityMutationResult & {
+    createEntity<T extends BaseEntity>(request: CreateEntityRequest<T>): Promise<EntityMutationResult>;
+    createEntityFromMarkdown(request: CreateEntityFromMarkdownRequest): Promise<EntityMutationResult>;
+    updateEntity<T extends BaseEntity>(request: UpdateEntityRequest<T>): Promise<EntityMutationResult>;
+    deleteEntity(request: DeleteEntityRequest): Promise<boolean>;
+    upsertEntity<T extends BaseEntity>(request: UpsertEntityRequest<T>): Promise<EntityMutationResult & {
         created: boolean;
     }>;
     storeEmbedding(data: StoreEmbeddingData): Promise<void>;
     serializeEntity(entity: BaseEntity): string;
     deserializeEntity(markdown: string, entityType: string): Partial<BaseEntity>;
     countEmbeddings(): Promise<number>;
-    searchWithDistances(query: string): Promise<Array<{
+    searchWithDistances(request: SearchWithDistancesRequest): Promise<Array<{
         entityId: string;
         entityType: string;
         distance: number;