Skip to content

Commit

Permalink
chore(openai): update azure user agent (#7543)
Browse files Browse the repository at this point in the history
Co-authored-by: jacoblee93 <jacoblee93@gmail.com>
  • Loading branch information
sinedied and jacoblee93 authored Feb 11, 2025
1 parent b81807a commit b33a19f
Showing 1 changed file with 10 additions and 4 deletions.
14 changes: 10 additions & 4 deletions libs/langchain-openai/src/azure/chat_models.ts
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ import {
LangSmithParams,
type BaseChatModelParams,
} from "@langchain/core/language_models/chat_models";
import { getEnvironmentVariable } from "@langchain/core/utils/env";
import { getEnv, getEnvironmentVariable } from "@langchain/core/utils/env";
import { BaseLanguageModelInput } from "@langchain/core/language_models/base";
import { BaseMessage } from "@langchain/core/messages";
import { Runnable } from "@langchain/core/runnables";
Expand Down Expand Up @@ -570,11 +570,17 @@ export class AzureChatOpenAI extends ChatOpenAI {
delete params.baseURL;
}

let env = getEnv();
if (env === "node" || env === "deno") {
env = `(${env}/${process.version}; ${process.platform}; ${process.arch})`;
}

const specifiedUserAgent = params.defaultHeaders?.["User-Agent"];
params.defaultHeaders = {
...params.defaultHeaders,
"User-Agent": params.defaultHeaders?.["User-Agent"]
? `${params.defaultHeaders["User-Agent"]}: langchainjs-azure-openai-v2`
: `langchainjs-azure-openai-v2`,
"User-Agent": `langchainjs-azure-openai/2.0.0 (${env})${
specifiedUserAgent ? ` ${specifiedUserAgent}` : ""
}`,
};

this.client = new AzureOpenAIClient({
Expand Down

0 comments on commit b33a19f

Please sign in to comment.