Skip to content

Commit

Permalink
Optional 'user' input param for OpenAI API (#2270)
Browse files Browse the repository at this point in the history
* added user input param for OpenAI

* added to completions api

* added to chat_models/openai.ts
  • Loading branch information
ilyasubkhankulov authored Aug 15, 2023
1 parent ff230cf commit 8d60a12
Show file tree
Hide file tree
Showing 4 changed files with 15 additions and 0 deletions.
4 changes: 4 additions & 0 deletions langchain/src/chat_models/openai.ts
Original file line number Diff line number Diff line change
Expand Up @@ -219,6 +219,8 @@ export class ChatOpenAI

stop?: string[];

user?: string;

timeout?: number;

streaming = false;
Expand Down Expand Up @@ -291,6 +293,7 @@ export class ChatOpenAI
this.n = fields?.n ?? this.n;
this.logitBias = fields?.logitBias;
this.stop = fields?.stop;
this.user = fields?.user;

this.streaming = fields?.streaming ?? false;

Expand Down Expand Up @@ -329,6 +332,7 @@ export class ChatOpenAI
n: this.n,
logit_bias: this.logitBias,
stop: options?.stop ?? this.stop,
user: this.user,
stream: this.streaming,
functions:
options?.functions ??
Expand Down
4 changes: 4 additions & 0 deletions langchain/src/llms/openai-chat.ts
Original file line number Diff line number Diff line change
Expand Up @@ -106,6 +106,8 @@ export class OpenAIChat

stop?: string[];

user?: string;

streaming = false;

openAIApiKey?: string;
Expand Down Expand Up @@ -177,6 +179,7 @@ export class OpenAIChat
this.logitBias = fields?.logitBias;
this.maxTokens = fields?.maxTokens;
this.stop = fields?.stop;
this.user = fields?.user;

this.streaming = fields?.streaming ?? false;

Expand Down Expand Up @@ -221,6 +224,7 @@ export class OpenAIChat
logit_bias: this.logitBias,
max_tokens: this.maxTokens === -1 ? undefined : this.maxTokens,
stop: options?.stop ?? this.stop,
user: this.user,
stream: this.streaming,
...this.modelKwargs,
};
Expand Down
4 changes: 4 additions & 0 deletions langchain/src/llms/openai.ts
Original file line number Diff line number Diff line change
Expand Up @@ -105,6 +105,8 @@ export class OpenAI

stop?: string[];

user?: string;

streaming = false;

openAIApiKey?: string;
Expand Down Expand Up @@ -185,6 +187,7 @@ export class OpenAI
this.bestOf = fields?.bestOf ?? this.bestOf;
this.logitBias = fields?.logitBias;
this.stop = fields?.stop;
this.user = fields?.user;

this.streaming = fields?.streaming ?? false;

Expand Down Expand Up @@ -228,6 +231,7 @@ export class OpenAI
best_of: this.bestOf,
logit_bias: this.logitBias,
stop: options?.stop ?? this.stop,
user: this.user,
stream: this.streaming,
...this.modelKwargs,
};
Expand Down
3 changes: 3 additions & 0 deletions langchain/src/types/openai-types.ts
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,9 @@ export declare interface OpenAIBaseInput {
/** Dictionary used to adjust the probability of specific tokens being generated */
logitBias?: Record<string, number>;

/** Unique string identifier representing your end-user, which can help OpenAI to monitor and detect abuse. */
user?: string;

/** Whether to stream the results or not. Enabling disables tokenUsage reporting */
streaming: boolean;

Expand Down

1 comment on commit 8d60a12

@vercel
Copy link

@vercel vercel bot commented on 8d60a12 Aug 15, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please sign in to comment.