Merge pull request #136 from samuelzedec/develop

feat: atualizar relacionamentos de entidades e melhorar resiliência do banco de dados

Author: samuelzedec

CHANGELOG.md

@@ -1,5 +1,11 @@
 # Riber - Changelog
 
+## v4.3.1 - 09/12/2025
+**REFATOÇÃO**: Remove o método de busca por id da entidade nos IAiModelService
+Adiciona relationacimento entre entidade `ProductEmbeddings` e `Company` para uma melhor busca
+
+---
+
 ## v4.3.0 - 08/11/2025
 - **NOVO**: Novas entidades par uso de chat e agendes de IA
 - Adiciona entidade `Chat` para o usuário armazenar conversas com IA

Directory.Build.props

@@ -4,7 +4,7 @@
     <Companies>Samuel Zedec</Companies>
     <Copyright>Copyright © $([System.DateTime]::Now.Year)</Copyright>
     <Description>Sistema de gestão financeira para um lanchonete local</Description>
-    <Version>4.3.0</Version>
+    <Version>4.3.1</Version>
   </PropertyGroup>
 
   <!-- Repositório e Documentação -->

src/Riber.Application/Abstractions/Services/AI/IAiModelService.cs

@@ -1,6 +1,6 @@
 namespace Riber.Application.Abstractions.Services.AI;
 
-public interface IAiModelService<TInput, TOutput>
+public interface IAiModelService<in TInput, TOutput>
     where TInput : class
     where TOutput : class
 {
@@ -23,17 +23,10 @@ public interface IAiModelService<TInput, TOutput>
     /// <summary>
     /// Encontra itens semelhantes com base no vetor de consulta fornecido.
     /// </summary>
+    /// <param name="companyId">O identificador único da empresa a ser consultada.</param>
     /// <param name="query">Um array de números de ponto flutuante representando o vetor de consulta para a operação de similaridade.</param>
     /// <param name="cancellationToken">Um token para monitorar solicitações de cancelamento.</param>
     /// <typeparam name="TOutput">O tipo de item que será retornado como resultado da operação.</typeparam>
     /// <returns>Uma tarefa que representa a operação assíncrona, contendo uma memória somente leitura de itens semelhantes do tipo especificado.</returns>
-    Task<ReadOnlyMemory<TOutput>> FindSimilarAsync(float[] query, CancellationToken cancellationToken = default);
-
-    /// <summary>
-    /// Obtém um recurso ou dado com base no identificador da entidade especificado.
-    /// </summary>
-    /// <param name="entityId">O identificador único da entidade a ser recuperada.</param>
-    /// <param name="cancellationToken">Um token para monitorar solicitações de cancelamento.</param>
-    /// <returns>Uma tarefa que representa a operação assíncrona, contendo o recurso ou dado encontrado, ou null se não encontrado.</returns>
-    Task<TInput?> GetByEntityIdAsync(Guid entityId, CancellationToken cancellationToken = default);
+    Task<ReadOnlyMemory<TOutput>> FindSimilarAsync(Guid companyId, float[] query, CancellationToken cancellationToken = default);
 }

src/Riber.Domain/Enums/AssistantType.cs

@@ -4,5 +4,4 @@ public enum AssistantType
 {
     SalesAnalysis = 1,
     ProductAnalysis = 2,
-    MonthlyReport = 3
 }

src/Riber.Infrastructure/DependencyInjection.cs

@@ -105,7 +105,10 @@ private static void AddPersistence(this IServiceCollection services, string defa
                 b.MigrationsAssembly(typeof(AppDbContext).Assembly.FullName)
                     .MigrationsHistoryTable("__EFMigrationsHistory");
             })
-            .AddInterceptors(new CaseInsensitiveInterceptor(), new AuditInterceptor()));
+            .AddInterceptors(new CaseInsensitiveInterceptor(), new AuditInterceptor())
+            .EnableDetailedErrors()
+            .EnableServiceProviderCaching()
+        );
     }
 
     private static void AddRepositories(this IServiceCollection services)

src/Riber.Infrastructure/Extensions/AssistantTypeExtension.cs

@@ -7,7 +7,7 @@ public static class AssistantTypeExtension
     public static string GetModel(this AssistantType assistantType)
         => assistantType switch
         {
-            AssistantType.SalesAnalysis or AssistantType.MonthlyReport => "claude-sonnet-4-20250514",
+            AssistantType.SalesAnalysis => "claude-sonnet-4-20250514",
             AssistantType.ProductAnalysis => "claude-haiku-4-5-20251001",
             _ => throw new ArgumentOutOfRangeException(nameof(assistantType), assistantType, null)
         };

src/Riber.Infrastructure/Messaging/Consumers/GenerateProductEmbeddingsMessageConsumer.cs

@@ -40,8 +40,7 @@ public async Task Consume(ConsumeContext<GenerateProductEmbeddingsMessage> conte
             logger.LogInformation("Embeddings do produto {ProductId} gerados.", productId);
             var productEmbeddingsModel = new ProductEmbeddingsModel
             {
-                ProductId = productId, 
-                Embeddings = new Vector(embeddings)
+                ProductId = productId, CompanyId = product.CompanyId, Embeddings = new Vector(embeddings)
             };
 
             await modelService.CreateAsync(productEmbeddingsModel, context.CancellationToken);

src/Riber.Infrastructure/Persistence/Factories/AppDbContextFactory.cs

@@ -25,9 +25,9 @@ public AppDbContext CreateDbContext(string[] args)
                     b.MigrationsAssembly(typeof(AppDbContext).Assembly.FullName)
                         .MigrationsHistoryTable("__EFMigrationsHistory");
                 })
-            .AddInterceptors(
-                new CaseInsensitiveInterceptor(),
-                new AuditInterceptor());
+            .AddInterceptors(new CaseInsensitiveInterceptor(), new AuditInterceptor())
+            .EnableSensitiveDataLogging()
+            .EnableServiceProviderCaching();
 
         return new AppDbContext(optionsBuilder.Options);
     }

src/Riber.Infrastructure/Persistence/Mappings/Models/ProductEmbeddingsModelMap.cs

@@ -30,5 +30,18 @@ protected override void Mapping(EntityTypeBuilder<ProductEmbeddingsModel> builde
             .HasForeignKey(p => p.ProductId)
             .HasConstraintName("fk_product_embeddings_product_id")
             .OnDelete(DeleteBehavior.NoAction);
+
+        builder
+            .Property(p => p.CompanyId)
+            .HasColumnName("company_id")
+            .HasColumnType("uuid")
+            .IsRequired();
+
+        builder
+            .HasOne(p => p.Company)
+            .WithMany()
+            .HasForeignKey(p => p.CompanyId)
+            .HasConstraintName("fk_product_embeddings_company_id")
+            .OnDelete(DeleteBehavior.NoAction);
     }
 }