how to infer partial ?

[dcsan]

I have a query that doesn't include all columns in a table.
how to I create a type that reflects this?

eg i have a func returning values that skips some columns:

    const chunks = await db.query.chunk.findMany({
      // where: eq(schema.chunk.sourceId, this.data.sourceId),
      orderBy: l2Distance(schema.chunk.embedding, embedding),
      limit,
      columns: {
        sourceId: true,
        text: true,
        // embedding: true,
        // l2Distance: true,
      },
    })
    return chunks

elsewhere I have the schema and then this to define the Chunk type

export type Chunk = InferSelectModel<typeof schema.chunk>

currently I have a little TS workaround but not sure if there's a better way to do it with Drizzle inference?

type StrictOmit<T, K extends keyof T> = Omit<T, K>

export type ChunkSlim = StrictOmit<
  Chunk,
  "embedding" | "id" | "idx" | "offset" | "duration"
>
export type ChunkInsert = InferInsertModel<typeof schema.chunk>

[danisal]

Hi @dcsan, did you find a solution for this? I'm looking for a solution to this type of problem

[dcsan]

no i just have to do all the typing manually myself. pita. hope there's a workaround.

[ReinisL]

What if you create a function and then create type from it like

async function getChunks(limit, embedding) {
  return db.query.chunk.findMany({
      orderBy: l2Distance(schema.chunk.embedding, embedding),
      limit,
      columns: {
        sourceId: true,
        text: true,
      },
    })
}

type Chunks = Awaited<ReturnType<typeof getChunks>>;

[dlford]

Really old thread, but I just ran into a similar situation, here's what I came up with in case anyone else finds themselves on this page looking for an answer:

const chunkPartialColumns = {
  sourceId: true,
  text: true,
} as const;

export type ChunkPartial = Pick<typeof chunk.$inferSelect, keyof typeof chunkPartialColumns>;

let chunks: ChunkPartial[];

chunks = await db.query.chunk.findMany({
  columns: chunkPartialColumns,
})

return chunks;