import { Agent, run, setDefaultModelProvider, setTracingDisabled } from '@openai/agents';
import { OpenAIProvider } from '@openai/agents-openai';
import OpenAI from 'openai';
import { setTracingExportApiKey } from '@openai/agents';
import { setGlobalDispatcher, ProxyAgent } from 'undici';
setGlobalDispatcher(new ProxyAgent('http://127.0.0.1:7890'));
//process.env.OPENAI_AGENTS_DISABLE_TRACING = '1';

// 你的代码...
setTracingExportApiKey('sk-。。。');

//setTracingDisabled(true);

/* const deepseekClient = new OpenAI({
  apiKey: 'ollama',
  baseURL: 'http://192.168.2.116:11434/v1',
  defaultHeaders: {
    'Content-Type': 'application/json',
  },
}); 
const model_name = "llama3.2:3b";
*/

const deepseekClient = new OpenAI({
  apiKey: process.env.DEEPSEEK_API_KEY || 'sk。。。',
  baseURL: 'https://api.deepseek.com',
  defaultHeaders: {
    'Content-Type': 'application/json',
  },
});
const model_name = "deepseek-v4-flash";

const deepseekProvider = new OpenAIProvider({
  openAIClient: deepseekClient,
  useResponses: false,
});

setDefaultModelProvider(deepseekProvider);

const agent = new Agent({
  name: '数学老师',
  instructions: '解决数学问题',
  model: model_name,
  modelSettings: {
    temperature: 0.7,
    maxTokens: 1000
  },
});

async function main() {
  try {
    console.log('正在连接到DeepSeek API...');
    const result = await run(agent, '2+2等于多少？');
    console.log('回答:', result.finalOutput);
    // console.log('end');
  } catch (error) {
    console.error('错误详情:');
    console.error('状态码:', error.status);
    console.error('错误信息:', error.message);
  }
}

main();