Merge pull request #10 from Portkey-AI/noble-varghese/add-generations…

…-api

feat: Add support for the saved prompt generations

examples/promptGeneration.ts

@@ -0,0 +1,16 @@
+import { Portkey } from "../src";
+
+const portkey = new Portkey({
+    mode: "fallback"
+});
+
+async function main() {
+    const chatCompletion = await portkey.generations.create({
+        promptId: "your-prompt-id",
+        // variables: {hello: "world"} # Add variables if required
+    });
+
+    console.log(chatCompletion.data);
+};
+
+main();

src/apis/generations.ts

@@ -1 +1,27 @@
-export class Generations { }
+import { ModelParams } from "../_types/portkeyConstructs";
+import { ApiResource } from "../apiResource";
+import { APIPromise, RequestOptions } from "../baseClient";
+
+export class Generations extends ApiResource {
+    create(
+        _body: GenerationsBody,
+        opts?: RequestOptions
+    ): APIPromise<Generation> {
+        const config = this.client.config || {
+            mode: this.client.mode,
+            options: this.client.llms
+        }
+        const body = { "variables": _body.variables }
+        return this.post<Generation>(`/v1/prompts/${_body.promptId}/generate`, { body, ...opts })
+    }
+}
+
+export interface GenerationsBody extends ModelParams {
+    promptId: string;
+    variables?: Record<string, any>
+}
+
+export interface Generation {
+    success: boolean,
+    data: Record<string, any>
+}

src/index.ts

@@ -61,7 +61,7 @@ export class Portkey extends ApiClient {
 
     completions: API.Completions = new API.Completions(this);
     chatCompletions = new API.ChatCompletions(this);
-    // generations =  new API.Generations();
+    generations =  new API.Generations(this);
 }
 
 export import LLMOptions = Types.LLMOptions;