Bump version to 10.0.6 and refine AGE extension initialization logic in AgeConnectionManager.

Author: VladKovtun99

Directory.Build.props

@@ -26,8 +26,8 @@
         <RepositoryUrl>https://github.com/managedcode/graphrag</RepositoryUrl>
         <PackageProjectUrl>https://github.com/managedcode/graphrag</PackageProjectUrl>
         <Product>Managed Code GraphRag</Product>
-        <Version>10.0.5</Version>
-        <PackageVersion>10.0.5</PackageVersion>
+        <Version>10.0.6</Version>
+        <PackageVersion>10.0.6</PackageVersion>
 
     </PropertyGroup>
     <PropertyGroup Condition="'$(GITHUB_ACTIONS)' == 'true'">

src/ManagedCode.GraphRag.Postgres/ApacheAge/AgeConnectionManager.cs

@@ -137,10 +137,9 @@ private async Task LoadAgeAsync(NpgsqlConnection connection, CancellationToken c
             checkCommand.CommandTimeout = 0;
             var result = await checkCommand.ExecuteScalarAsync(cancellationToken).ConfigureAwait(false);
 
-            if (result is not null)
+            if (result is null)
             {
-                LogMessages.ExtensionLoaded(_logger, ConnectionString);
-                return;
+                throw new AgeException("AGE extension is not installed.");
             }
 
             await using var load = connection.CreateCommand();
@@ -151,6 +150,18 @@ private async Task LoadAgeAsync(NpgsqlConnection connection, CancellationToken c
         }
         catch (PostgresException ex) when (ex.SqlState == "42501")
         {
+            await using var initCommand = connection.CreateCommand();
+            initCommand.CommandText = "SELECT ag_catalog.create_graph('__age_init__'); SELECT ag_catalog.drop_graph('__age_init__', true);";
+            initCommand.CommandTimeout = 0;
+
+            try
+            {
+                await initCommand.ExecuteNonQueryAsync(cancellationToken).ConfigureAwait(false);
+            }
+            catch (PostgresException)
+            {
+            }
+
             LogMessages.ExtensionLoaded(_logger, ConnectionString);
         }
         catch (PostgresException ex)