BE-456: HashQL: Entity type model enrichment and graph store query extensions

libs/@local/graph/migrations/graph-migrations/v010__query/down.sql

@@ -0,0 +1 @@
+DROP TYPE CONTINUATION;

libs/@local/graph/migrations/graph-migrations/v010__query/mod.rs

@@ -0,0 +1,43 @@
+use error_stack::Report;
+use hash_graph_migrations::{Context, Migration};
+use tokio_postgres::Client;
+use tracing::Instrument as _;
+
+pub struct Query;
+
+impl Migration for Query {
+    type Context = Client;
+    type Error = tokio_postgres::Error;
+
+    async fn up(
+        self,
+        context: &mut <Self::Context as Context>::Transaction<'_>,
+    ) -> Result<(), Report<Self::Error>> {
+        context
+            .simple_query(include_str!("up.sql"))
+            .instrument(tracing::info_span!(
+                "BATCH",
+                otel.kind = "client",
+                db.system = "postgresql",
+                peer.service = "Postgres",
+            ))
+            .await?;
+        Ok(())
+    }
+
+    async fn down(
+        self,
+        context: &mut <Self::Context as Context>::Transaction<'_>,
+    ) -> Result<(), Report<Self::Error>> {
+        context
+            .simple_query(include_str!("down.sql"))
+            .instrument(tracing::info_span!(
+                "BATCH",
+                otel.kind = "client",
+                db.system = "postgresql",
+                peer.service = "Postgres",
+            ))
+            .await?;
+        Ok(())
+    }
+}

libs/@local/graph/migrations/graph-migrations/v010__query/up.sql

@@ -0,0 +1 @@
+CREATE TYPE continuation AS (filter boolean, block int, locals int [], values jsonb []);

libs/@local/graph/postgres-store/postgres_migrations/V49__query.sql

@@ -0,0 +1 @@
+CREATE TYPE continuation AS (filter boolean, block int, locals int[], values jsonb[]);

libs/@local/graph/postgres-store/src/store/mod.rs

@@ -3,7 +3,7 @@ pub mod error;
 mod config;
 mod validation;
 
-pub(crate) mod postgres;
+pub mod postgres;
 
 pub use self::{
     config::{DatabaseConnectionInfo, DatabasePoolConfig, DatabaseType},

libs/@local/graph/postgres-store/src/store/postgres/query/compile.rs

@@ -18,8 +18,9 @@ use type_system::knowledge::Entity;
 
 use super::expression::{JoinType, TableName, TableReference};
 use crate::store::postgres::query::{
-    Alias, Column, Distinctness, EqualityOperator, Expression, Function, PostgresQueryPath,
-    PostgresRecord, SelectExpression, SelectStatement, Table, Transpile as _, WindowStatement,
+    Alias, Column, Distinctness, EqualityOperator, Expression, Function, Identifier,
+    PostgresQueryPath, PostgresRecord, SelectExpression, SelectStatement, Table, Transpile as _,
+    WindowStatement,
     expression::{FromItem, GroupByExpression, PostgresType},
     table::{
         DataTypeEmbeddings, DatabaseColumn as _, EntityEditions, EntityEmbeddings,
@@ -440,7 +441,7 @@ impl<'p, 'q: 'p, R: PostgresRecord> SelectCompiler<'p, 'q, R> {
                     .map(|filter| self.compile_filter(filter))
                     .collect::<Result<_, _>>()?,
             ),
-            Filter::Not(filter) => Expression::not(self.compile_filter(filter)?),
+            Filter::Not(filter) => self.compile_filter(filter)?.not(),
             Filter::Equal(lhs, rhs) => Expression::equal(
                 self.compile_filter_expression(lhs).0,
                 self.compile_filter_expression(rhs).0,
@@ -616,7 +617,7 @@ impl<'p, 'q: 'p, R: PostgresRecord> SelectCompiler<'p, 'q, R> {
                                                     parameter_expression,
                                                 )),
                                             )),
-                                            alias: Some("distance"),
+                                            alias: Some(Identifier::from("distance")),
                                         }))
                                         .collect(),
                                 )
@@ -744,7 +745,7 @@ impl<'p, 'q: 'p, R: PostgresRecord> SelectCompiler<'p, 'q, R> {
                                 Column::OntologyIds(OntologyIds::BaseUrl).aliased(alias),
                             )),
                         ),
-                        alias: Some("latest_version"),
+                        alias: Some(Identifier::from("latest_version")),
                     },
                 ])
                 .from(