Skip to content

Commit

Permalink
fix: use Supabase namespace for non-std Supabase APIs
Browse files Browse the repository at this point in the history
  • Loading branch information
soedirgo committed Apr 15, 2024
1 parent 26a23e0 commit 5af445a
Showing 1 changed file with 35 additions and 33 deletions.
68 changes: 35 additions & 33 deletions src/edge-runtime.d.ts
Original file line number Diff line number Diff line change
@@ -1,45 +1,47 @@
interface ModelOptions {
/**
* Pool embeddings by taking their mean. Applies only for `gte-small` model
*/
mean_pool?: boolean
declare namespace Supabase {
export interface ModelOptions {
/**
* Pool embeddings by taking their mean. Applies only for `gte-small` model
*/
mean_pool?: boolean

/**
* Normalize the embeddings result. Applies only for `gte-small` model
*/
normalize?: boolean
/**
* Normalize the embeddings result. Applies only for `gte-small` model
*/
normalize?: boolean

/**
* Stream response from model. Applies only for LLMs like `mistral` (default: false)
*/
stream?: boolean
/**
* Stream response from model. Applies only for LLMs like `mistral` (default: false)
*/
stream?: boolean

/**
* Automatically abort the request to the model after specified time (in seconds). Applies only for LLMs like `mistral` (default: 60)
*/
timeout?: number
}
/**
* Automatically abort the request to the model after specified time (in seconds). Applies only for LLMs like `mistral` (default: 60)
*/
timeout?: number
}

interface Session {
/**
* Execute the given prompt in model session
*/
run(prompt: string, modelOptions?: ModelOptions): unknown
}
export class Session {
/**
* Create a new model session using given model
*/
constructor(model: string, sessionOptions?: unknown)

declare var Session: {
prototype: Session
/**
* Create a new model session using given model
*/
new (model: string, sessionOptions?: unknown): Session
}
/**
* Execute the given prompt in model session
*/
run(prompt: string, modelOptions?: ModelOptions): unknown
}

declare var Supabase: {
/**
* Provides AI related APIs
*/
readonly ai: {
export interface Ai {
readonly Session: typeof Session
}

/**
* Provides AI related APIs
*/
export const ai: Ai
}

0 comments on commit 5af445a

Please sign in to comment.