From 4cc768c3fd60e93dc0fb0df0f566743a506a40d4 Mon Sep 17 00:00:00 2001
From: Andrei Alistar <andrei.alistar@genisoft.eu>
Date: Thu, 10 Oct 2024 12:21:42 +0300
Subject: [PATCH] Fix Cosmos DB dependencies

---
 src/dotnet/ManagementAPI/Program.cs    | 2 ++
 src/dotnet/OrchestrationAPI/Program.cs | 4 +++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/dotnet/ManagementAPI/Program.cs b/src/dotnet/ManagementAPI/Program.cs
index 34394d606f..c451f16ce2 100644
--- a/src/dotnet/ManagementAPI/Program.cs
+++ b/src/dotnet/ManagementAPI/Program.cs
@@ -106,6 +106,8 @@ public static void Main(string[] args)
             builder.AddAttachmentResourceProvider();
             builder.AddAIModelResourceProvider();
 
+            builder.AddAzureCosmosDBService();
+
             // Add authentication configuration.
             var e2ETestEnvironmentValue = Environment.GetEnvironmentVariable(EnvironmentVariables.FoundationaLLM_Environment) ?? string.Empty;
             var isE2ETestEnvironment = e2ETestEnvironmentValue.Equals(EnvironmentTypes.E2ETest, StringComparison.CurrentCultureIgnoreCase);
diff --git a/src/dotnet/OrchestrationAPI/Program.cs b/src/dotnet/OrchestrationAPI/Program.cs
index cd71d1cf81..0e20198601 100644
--- a/src/dotnet/OrchestrationAPI/Program.cs
+++ b/src/dotnet/OrchestrationAPI/Program.cs
@@ -60,7 +60,7 @@ public static void Main(string[] args)
                 options.Select(AppConfigurationKeyFilters.FoundationaLLM_ResourceProviders_AIModel_Storage);
                 options.Select(AppConfigurationKeyFilters.FoundationaLLM_ResourceProviders_Prompt_Storage);
                 options.Select(AppConfigurationKeyFilters.FoundationaLLM_ResourceProviders_AzureOpenAI_Storage);
-
+                options.Select(AppConfigurationKeyFilters.FoundationaLLM_APIEndpoints_CoreAPI_Configuration_CosmosDB);
                 options.Select(AppConfigurationKeyFilters.FoundationaLLM_APIEndpoints_AzureEventGrid_Essentials);
                 options.Select(AppConfigurationKeyFilters.FoundationaLLM_APIEndpoints_AzureEventGrid_Configuration);
                 options.Select(AppConfigurationKeys.FoundationaLLM_Events_Profiles_OrchestrationAPI);
@@ -134,6 +134,8 @@ public static void Main(string[] args)
             builder.AddLLMOrchestrationServices();
             builder.AddOrchestrationService();
 
+            builder.AddAzureCosmosDBService();
+
             builder.Services
                 .AddApiVersioning(options =>
                 {