|
1 | | -#!/usr/bin/env node |
2 | | - |
| 1 | +import path from 'node:path'; |
| 2 | +import fs from 'node:fs/promises'; |
| 3 | +import os from 'node:os'; |
| 4 | +import { DefaultAzureCredential, getBearerTokenProvider } from '@azure/identity'; |
3 | 5 | import { AzureChatOpenAI } from '@langchain/openai'; |
4 | 6 | import { BaseChatModel } from '@langchain/core/language_models/chat_models'; |
5 | 7 | import { ChatPromptTemplate } from '@langchain/core/prompts'; |
6 | 8 | import { createToolCallingAgent } from 'langchain/agents'; |
7 | 9 | import { AgentExecutor } from 'langchain/agents'; |
8 | 10 | import { loadMcpTools } from '@langchain/mcp-adapters'; |
9 | 11 | import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js'; |
10 | | -import { getAzureOpenAiTokenProvider } from './src/auth.js'; |
11 | 12 | import { Client } from '@modelcontextprotocol/sdk/client/index.js'; |
12 | 13 | import { BaseMessage, HumanMessage, AIMessage } from '@langchain/core/messages'; |
13 | | -import path from 'node:path'; |
14 | | -import fs from 'node:fs/promises'; |
15 | | -import os from 'node:os'; |
16 | 14 | import dotenv from 'dotenv'; |
17 | 15 |
|
18 | 16 | dotenv.config({ path: path.join(process.cwd(), '../../.env'), quiet: true }); |
@@ -150,7 +148,7 @@ export async function run() { |
150 | 148 | } |
151 | 149 | } |
152 | 150 |
|
153 | | - const azureADTokenProvider = getAzureOpenAiTokenProvider(); |
| 151 | + const azureADTokenProvider = getBearerTokenProvider(new DefaultAzureCredential(), 'https://cognitiveservices.azure.com/.default');; |
154 | 152 |
|
155 | 153 | model = new AzureChatOpenAI({ |
156 | 154 | temperature: 0.3, |
@@ -228,7 +226,3 @@ export async function run() { |
228 | 226 | process.exitCode = 0; |
229 | 227 | } |
230 | 228 | } |
231 | | - |
232 | | -if (process.argv[1] && process.argv[1].endsWith('agent-cli.ts') || process.argv[1].endsWith('agent-cli.js')) { |
233 | | - run(); |
234 | | -} |
0 commit comments