@tstdl/base 0.92.104 → 0.92.106

package/ai/ai.service.js

@@ -11,6 +11,7 @@ import { GoogleGenerativeAI } from '@google/generative-ai';
 import { NotSupportedError } from '../errors/not-supported.error.js';
 import { Singleton } from '../injector/decorators.js';
 import { inject, injectArgument } from '../injector/inject.js';
+import { Logger } from '../logger/logger.js';
 import { getShutdownSignal } from '../process-shutdown.js';
 import { DeferredPromise } from '../promise/deferred-promise.js';
 import { LazyPromise } from '../promise/lazy-promise.js';
@@ -28,7 +29,6 @@ import { resolveValueOrAsyncProvider } from '../utils/value-or-provider.js';
 import { AiFileService } from './ai-file.service.js';
 import { AiSession } from './ai-session.js';
 import { isSchemaFunctionDeclarationWithHandler } from './types.js';
-import { Logger } from '../logger/logger.js';
 export class AiServiceOptions {
     apiKey;
     keyFile;
@@ -252,7 +252,6 @@ Always output the content and tags in ${options?.targetLanguage ?? 'the same lan
             let lastUsageMetadata;
             let candidate;
             for await (const generationResponse of generation.stream) {
-                this.#logger.trace('Handling generation stream response');
                 candidate = generationResponse.candidates.at(0);
                 inputContent.push(candidate.content);
                 const { promptTokenCount = 0, candidatesTokenCount = 0 } = generationResponse.usageMetadata ?? {};

package/document-management/server/services/document-management.service.d.ts

@@ -71,6 +71,8 @@ export declare class DocumentManagementService extends DocumentManagementService
     loadDocument(id: string): Promise<Document>;
     loadDocumentFile(id: string): Promise<DocumentFile>;
     loadType(id: string): Promise<DocumentType>;
+    getFileContent(fileId: string): Promise<Uint8Array>;
+    getFileContentStream(fileId: string): ReadableStream<Uint8Array>;
     getFileContentUrl(fileId: string, title: string | null, download?: boolean): Promise<string>;
     createCategory(parameters: CreateDocumentCategoryParameters): Promise<DocumentCategory>;
     createType(parameters: CreateDocumentTypeParameters): Promise<DocumentType>;

package/document-management/server/services/document-management.service.js

@@ -250,6 +250,14 @@ let DocumentManagementService = DocumentManagementService_1 = class DocumentMana
     async loadType(id) {
         return this.documentTypeService.load(id);
     }
+    async getFileContent(fileId) {
+        const key = getDocumentFileKey(fileId);
+        return this.fileObjectStorage.getContent(key);
+    }
+    getFileContentStream(fileId) {
+        const key = getDocumentFileKey(fileId);
+        return this.fileObjectStorage.getContentStream(key);
+    }
     async getFileContentUrl(fileId, title, download = false) {
         const file = await this.documentFileService.load(fileId);
         return this.getDocumentFileContentObjectUrl(title ?? fileId, file, download);
@@ -554,6 +562,7 @@ let DocumentManagementService = DocumentManagementService_1 = class DocumentMana
     async enrichDocument(documentId) {
         const document = await this.documentService.load(documentId);
         const { properties, ...extractionResult } = await this.extractFileInformation(document.fileId, document.typeId);
+        this.logger.trace(`Applying extraction to document ${document.id}`);
         await this.documentService.transaction(async (documentService, transaction) => {
             await documentService.update(document.id, { ...extractionResult, validated: false });
             await this.setPropertyValues({ documentId, properties: properties }, transaction);
@@ -727,6 +736,7 @@ Ordne die Datei unter "file" der passenden Anforderungen unter "requests" zu. Gi
                 .where(isString(assumeTypeId) ? eq(documentType.id, assumeTypeId) : undefined);
             const typeLabelEntries = types.map((type) => ({ id: type.typeId, label: `${type.categoryLabel} | ${type.typeGroup} | ${type.typeLabel}` }));
             const typeLabels = typeLabelEntries.map(({ label }) => label);
+            this.logger.trace(`Classifying document file ${fileId}`);
             const documentTypeGeneration = await this.#aiService.generate({
                 ...defaultGenerationOptions,
                 generationSchema: object({
@@ -766,6 +776,7 @@ Ordne die Datei unter "file" der passenden Anforderungen unter "requests" zu. Gi
                     ? {}
                     : { documentProperties: object(fromEntries(propertiesSchemaEntries)) })
             });
+            this.logger.trace(`Extracting document file ${fileId}`);
             const { json: extraction } = await this.#aiService.generate({
                 model: 'gemini-2.0-flash',
                 generationOptions: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tstdl/base",
-  "version": "0.92.104",
+  "version": "0.92.106",
   "author": "Patrick Hein",
   "publishConfig": {
     "access": "public"