エージェントをトレースする

import { tracing } from '@coreweave/forge-sdk/agentlens';
import type { Message, Usage } from '@coreweave/forge-sdk/agentlens/tracing';

// プレースホルダー関数: 独自の実装に置き換えてください。
async function getWeatherApi(city: string): Promise<string> {
  return '24°C, sunny';  // 使用する天気 API の呼び出しに置き換えてください。
}

await tracing.init('[YOUR-TEAM]/[YOUR-PROJECT]');

await tracing.runIsolated(async () => {
  const conversation = tracing.startConversation({ agentName: 'weather-bot' });
  try {
    const turn = conversation.startTurn({ userMessage: 'What is the weather in Tokyo?' });
    try {
      // 1 回目の LLM Call: ツール呼び出しを返します。
      const llm = tracing.startLLM({ model: 'gpt-4o', providerName: 'openai' });
      try {
        llm.inputMessages = [{ role: 'user', content: 'What is the weather?' }];
        llm.think('User wants weather data, I should call get_weather.');
        llm.output('Let me check the weather for you.');
        llm.usage = { inputTokens: 100, outputTokens: 20 };

        // ツール呼び出し: 呼び出し元の LLM Call の子になります。
        const tool = tracing.startTool({ name: 'get_weather', args: '{"city":"Tokyo"}' });
        try {
          tool.result = await getWeatherApi('Tokyo');  // "24°C, sunny" を返します。
        } finally {
          tool.end();
        }
      } finally {
        llm.end();
      }

      // 2 回目の LLM Call: 最終的な回答を生成します。
      const llm2 = tracing.startLLM({ model: 'gpt-4o', providerName: 'openai' });
      try {
        llm2.inputMessages = [{ role: 'user', content: 'What is the weather?' }];
        llm2.output('It is 24°C and sunny in Tokyo today.');
        llm2.usage = { inputTokens: 150, outputTokens: 30 };
      } finally {
        llm2.end();
      }
    } finally {
      turn.end();
    }
  } finally {
    conversation.end();
  }
});

await tracing.shutdown();

await tracing.runIsolated(async () => {
  const conversation = tracing.startConversation({ agentName: 'weather-bot' });
  const turn = conversation.startTurn({ userMessage: 'What is the weather?' });

  const llm = tracing.startLLM({ model: 'gpt-4o', providerName: 'openai' });
  llm.inputMessages = [{ role: 'user', content: 'What is the weather?' }];
  llm.output('Let me check.');
  llm.usage = { inputTokens: 100, outputTokens: 20 };

  const tool = tracing.startTool({ name: 'get_weather', args: '{"city": "Tokyo"}' });
  tool.result = '24°C, sunny';
  tool.end();  // end() はべき等なので、複数回呼び出しても問題ありません。

  llm.end();

  const llm2 = tracing.startLLM({ model: 'gpt-4o', providerName: 'openai' });
  llm2.output('It is 24°C and sunny in Tokyo.');
  llm2.usage = { inputTokens: 150, outputTokens: 30 };
  llm2.end();

  turn.end();
  conversation.end();
});

import { tracing } from '@coreweave/forge-sdk/agentlens';

// プレースホルダー関数: 独自の実装に置き換えてください。
async function runToolInBrowser(name: string, args: Record<string, string>): Promise<Record<string, string>> {
  return { forecast: '24°C, sunny' };  // ブラウザーとの往復処理に置き換えてください。
}

await tracing.init('[TEAM-NAME]/[PROJECT-NAME]');

await tracing.runIsolated(async () => {
  const conversation = tracing.startConversation({ agentName: 'weather-bot' });
  const turn = conversation.startTurn({ userMessage: 'What is the weather in Tokyo?' });
  try {
    // ブラウザーがツールを実行する前後にサーバーのクロックを読み取ります。
    const sentAt = new Date();
    const result = await runToolInBrowser('get_weather', { city: 'Tokyo' });
    const receivedAt = new Date();

    // サーバーのタイムスタンプでツール呼び出しを記録します。
    const tool = turn.startTool({ name: 'get_weather', args: { city: 'Tokyo' }, startTime: sentAt });
    tool.end({ result, endTime: receivedAt });
  } finally {
    turn.end();
    conversation.end();
  }
});

await tracing.shutdown();