fix: embeddings and prompts completion

Portkey-AI · Dec 6, 2023 · 56c853b · 56c853b
1 parent 824b753
commit 56c853b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/apis/embeddings.ts b/src/apis/embeddings.ts
@@ -14,10 +14,6 @@ export interface EmbeddingsBody extends ModelParams {
 export type EmbeddingsResponse = Record<string, any> & APIResponseType
 
 export class Embeddings extends ApiResource {
- completions: EmbeddingsCompletions = new EmbeddingsCompletions(this.client);
-}
-
-export class EmbeddingsCompletions extends ApiResource {
  create(
  _body: EmbeddingsBody,
  params?: ApiClientInterface,

diff --git a/src/apis/generations.ts b/src/apis/generations.ts
@@ -44,7 +44,13 @@ export interface PromptsCreateNonStreaming extends PromptBodyBase {
 export type PromptsCreateParams = PromptsCreateNonStreaming | PromptsCreateStreaming
 
 type PromptsResponse = Record<string, any> & APIResponseType;
+
 export class Prompt extends ApiResource {
+ completions: PromptCompletions = new PromptCompletions(this.client);
+}
+
+
+export class PromptCompletions extends ApiResource {
  create(
  _body: PromptsCreateNonStreaming,
  params?: ApiClientInterface,