feat: implement agent cli

Author: sinedied

package-lock.json

@@ -15,3 +15,23 @@ You can use the `api.http` file to test the API using the [REST Client](https://
 ### `npm run build`
 
 To build the API for production to the `dist` folder.
+
+### `npm run cli`
+
+Run the burger agent CLI. Usage examples:
+
+```bash
+# Ask a simple question
+npm run cli "show me the burger menu"
+
+# Ask with a specific user ID
+npm run cli "place an order" -- --userId user123
+
+# Start a new session
+npm run cli "hello" -- --new
+
+# Combine options
+npm run cli "cancel my order" -- --userId user123 --new
+```
+
+The CLI maintains conversation history in `~/.burger-agent-cli/burger-agent-cli.json` for context across sessions.

packages/agent-api/README.md

@@ -1,3 +1,5 @@
+#!/usr/bin/env node
+
 import { AzureChatOpenAI } from '@langchain/openai';
 import { BaseChatModel } from '@langchain/core/language_models/chat_models';
 import { ChatPromptTemplate } from '@langchain/core/prompts';
@@ -7,30 +9,117 @@ import { loadMcpTools } from '@langchain/mcp-adapters';
 import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
 import { getAzureOpenAiTokenProvider } from './src/auth.js';
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { BaseMessage, HumanMessage, AIMessage } from '@langchain/core/messages';
 import path from 'node:path';
+import fs from 'node:fs/promises';
+import os from 'node:os';
 import dotenv from 'dotenv';
 
-dotenv.config({ path: path.join(process.cwd(), '../../.env') });
+dotenv.config({ path: path.join(process.cwd(), '../../.env'), quiet: true });
 
 const agentSystemPrompt = `
-# Role
+## Role
 You an expert assistant that helps users with managing burger orders. Use the provided tools to get the information you need and perform actions on behalf of the user.
 Only answer to requests that are related to burger orders and the menu. If the user asks for something else, politely inform them that you can only assist with burger orders.
 
-# Task
+## Task
 Help the user with their request, ask any clarifying questions if needed.
 
-# Instructions
+## Instructions
 - Always use the tools provided to get the information requested or perform any actions
 - If you get any errors when trying to use a tool that does not seem related to missing parameters, try again
 - If you cannot get the information needed to answer the user's question or perform the specified action, inform the user that you are unable to do so. Never make up information.
 - The get_burger tool can help you get informations about the burgers
 - Creating or cancelling an order requires a \`userId\`: if not provided, ask the user to provide it or to run the CLI with the \`--userId\` option.
+
+## Output
+Your response will be printed to a terminal. Do not use markdown formatting or any other special formatting. Just provide the plain text response.
 `;
 
-const question = 'show me the burger menu';
+interface CliArgs {
+  question: string;
+  userId?: string;
+  isNew: boolean;
+}
+
+interface SessionData {
+  history: Array<{ type: 'human' | 'ai'; content: string }>;
+  userId?: string;
+}
+
+function parseArgs(): CliArgs {
+  const args = process.argv.slice(2);
+
+  if (args.length === 0 || args.includes('--help') || args.includes('-h')) {
+    console.log('Usage: agent-cli <question> [--userId <userId>] [--new]');
+    console.log('  question: Your question about burger orders');
+    console.log('  --userId: Optional user ID (needed for some tasks)');
+    console.log('  --new: Start a new session');
+    process.exit(0);
+  }
+
+  const questionParts: string[] = [];
+  let userId: string | undefined;
+  let isNew = false;
+
+  for (let i = 0; i < args.length; i++) {
+    const arg = args[i];
+
+    if (arg === '--userId') {
+      userId = args[i + 1];
+      i++;
+    } else if (arg === '--new') {
+      isNew = true;
+    } else {
+      questionParts.push(arg);
+    }
+  }
+
+  const question = questionParts.join(' ');
+
+  if (!question) {
+    console.error('Error: Question is required');
+    process.exit(1);
+  }
+
+  return { question, userId, isNew };
+}
+
+async function getSessionPath(): Promise<string> {
+  const userDataDir = path.join(os.homedir(), '.burger-agent-cli');
+  await fs.mkdir(userDataDir, { recursive: true });
+  return path.join(userDataDir, 'burger-agent-cli.json');
+}
+
+async function loadSession(): Promise<SessionData> {
+  try {
+    const sessionPath = await getSessionPath();
+    const content = await fs.readFile(sessionPath, 'utf-8');
+    return JSON.parse(content);
+  } catch {
+    return { history: [] };
+  }
+}
+
+async function saveSession(session: SessionData): Promise<void> {
+  try {
+    const sessionPath = await getSessionPath();
+    await fs.writeFile(sessionPath, JSON.stringify(session, null, 2));
+  } catch (error) {
+    console.error('Failed to save session:', error);
+  }
+}
+
+function convertHistoryToMessages(history: SessionData['history']): BaseMessage[] {
+  return history.map(msg =>
+    msg.type === 'human'
+      ? new HumanMessage(msg.content)
+      : new AIMessage(msg.content)
+  );
+}
 
 export async function run() {
+  const { question, userId, isNew } = parseArgs();
   const azureOpenAiEndpoint = process.env.AZURE_OPENAI_API_ENDPOINT;
   const burgerMcpEndpoint = process.env.BURGER_MCP_URL ?? 'http://localhost:3000/mcp';
 
@@ -40,20 +129,25 @@ export async function run() {
     if (!azureOpenAiEndpoint || !burgerMcpEndpoint) {
       const errorMessage = 'Missing required environment variables: AZURE_OPENAI_API_ENDPOINT or BURGER_MCP_URL';
       console.error(errorMessage);
-      return {
-        status: 500,
-        jsonBody: {
-          error: errorMessage,
-        },
-      };
+      return;
+    }
+
+    let session: SessionData;
+    if (isNew) {
+      session = { history: [], userId };
+    } else {
+      session = await loadSession();
+      if (userId && session.userId !== userId) {
+        session.userId = userId;
+      }
     }
 
     const azureADTokenProvider = getAzureOpenAiTokenProvider();
 
     model = new AzureChatOpenAI({
-      // Controls randomness. 0 = deterministic, 1 = maximum randomness
       temperature: 0.3,
       azureADTokenProvider,
+      azureOpenAIApiVersion: process.env.AZURE_OPENAI_API_VERSION,
     });
 
     const client = new Client({
@@ -67,8 +161,10 @@ export async function run() {
     const tools = await loadMcpTools('burger', client);
     console.log(`Loaded ${tools.length} tools from Burger MCP server`);
 
+    console.log(`Thinking...`);
+
     const prompt = ChatPromptTemplate.fromMessages([
-      ['system', agentSystemPrompt],
+      ['system', agentSystemPrompt + (session.userId ? `\n\nUser ID: ${session.userId}` : '')],
       ['placeholder', '{chat_history}'],
       ['human', '{input}'],
       ['placeholder', '{agent_scratchpad}'],
@@ -82,17 +178,31 @@ export async function run() {
     const agentExecutor = new AgentExecutor({
       agent,
       tools,
-      returnIntermediateSteps: true,
+      returnIntermediateSteps: false,
+    });
+
+    const chatHistory = convertHistoryToMessages(session.history);
+
+    const response = await agentExecutor.invoke({
+      input: question,
+      chat_history: chatHistory
     });
 
-    const response = await agentExecutor.invoke({ input: question });
+    console.log('----------\n' + response.output);
+
+    session.history.push({ type: 'human', content: question });
+    session.history.push({ type: 'ai', content: response.output });
+
+    await saveSession(session);
+
+    console.log('----------\nDone.');
 
-    console.log('Intermediate steps:', JSON.stringify(response.intermediateSteps, null, 2));
-    console.log('Final answer:', response.output);
   } catch (_error: unknown) {
     const error = _error as Error;
-    console.error(`Error when processing chat-post request: ${error.message}`);
+    console.error(`Error when processing request: ${error.message}`);
   }
 }
 
-run();
+if (process.argv[1] && process.argv[1].endsWith('agent-cli.ts') || process.argv[1].endsWith('agent-cli.js')) {
+  run();
+}

packages/agent-api/agent-cli.ts

@@ -12,7 +12,8 @@
     "start:host": "func start --port 7072",
     "start": "concurrently npm:start:* npm:watch --raw --kill-others",
     "update:local-settings": "node ./scripts/update-local-settings.mjs",
-    "postinstall": "npm run update:local-settings"
+    "postinstall": "npm run update:local-settings",
+    "cli": "tsx agent-cli.ts --"
   },
   "author": "Microsoft",
   "license": "MIT",
@@ -24,14 +25,15 @@
     "@langchain/core": "^0.3.18",
     "@langchain/langgraph": "^0.3.6",
     "@langchain/mcp-adapters": "^0.5.2",
-    "@langchain/openai": "^0.6.2",
+    "@langchain/openai": "^0.5.18",
     "@microsoft/ai-chat-protocol": "^1.0.0-beta.20240814.1",
     "dotenv": "^17.0.1",
     "langchain": "^0.3.6"
   },
   "devDependencies": {
     "@types/node": "^20",
     "azure-functions-core-tools": "^4",
+    "tsx": "^4",
     "typescript": "^5"
   }
 }