feat: add file tools and grep search functionality

README.md

@@ -87,11 +87,7 @@ You: /help
 Available commands:
   /help              - Show this help message
   /clear             - Clear current conversation
-  /save              - Save current conversation
-  /load <id>         - Load a saved conversation
-  /list              - List all saved conversations
-  /delete <id>       - Delete a saved conversation
-  /models            - List and select available AI models
+  /model             - Show and select available AI models
   /render            - Render conversation as raw prompt text
   /quit              - Exit the program
 
@@ -132,7 +128,7 @@ code-editing-agent/
 
 ## Model
 
-Uses `LGAI-EXAONE/K-EXAONE-236B-A23B` via FriendliAI serverless endpoints by default. Use `/models` command to switch models.
+Uses `LGAI-EXAONE/K-EXAONE-236B-A23B` via FriendliAI serverless endpoints by default. Use `/model` command to switch models.
 
 ## License
 

package.json

@@ -29,6 +29,7 @@
     "node": ">=18"
   },
   "dependencies": {
+    "@ai-sdk/openai-compatible": "^2.0.4",
     "@friendliai/ai-provider": "^1.0.3",
     "@t3-oss/env-core": "^0.13.10",
     "ai": "^6.0.7",

src/agent.ts

@@ -1,187 +1,62 @@
-import type { TextStreamPart } from "ai";
-import {
-  type LanguageModel,
-  type ModelMessage,
-  stepCountIs,
-  streamText,
-} from "ai";
+import { createFriendli } from "@friendliai/ai-provider";
+import type { ModelMessage } from "ai";
+import { ToolLoopAgent, wrapLanguageModel } from "ai";
+import { SYSTEM_PROMPT } from "./context/system-prompt";
 import { env } from "./env";
-import { SYSTEM_PROMPT } from "./prompts/system";
-import type { tools } from "./tools/index";
-import { tools as agentTools } from "./tools/index";
-import {
-  printAIPrefix,
-  printChunk,
-  printNewline,
-  printReasoningChunk,
-  printReasoningEnd,
-  printReasoningPrefix,
-  printTool,
-} from "./utils/colors";
-import { withRetry } from "./utils/retry";
-
-type StreamChunk = TextStreamPart<typeof tools>;
-
-interface StreamState {
-  hasStartedText: boolean;
-  hasStartedReasoning: boolean;
-}
-
-function endReasoningIfNeeded(state: StreamState): void {
-  if (state.hasStartedReasoning) {
-    printReasoningEnd();
-    state.hasStartedReasoning = false;
-  }
-}
-
-function endTextIfNeeded(state: StreamState): void {
-  if (state.hasStartedText) {
-    printNewline();
-    state.hasStartedText = false;
-  }
-}
+import { trimLeadingNewlinesMiddleware } from "./middleware/trim-leading-newlines";
+import { tools } from "./tools";
+
+const DEFAULT_MODEL_ID = "Qwen/Qwen3-235B-A22B-Instruct-2507";
+const OUTPUT_TOKEN_MAX = 32_000;
+
+const friendli = createFriendli({
+  apiKey: env.FRIENDLI_TOKEN,
+  includeUsage: true,
+});
+
+const createAgent = (modelId: string) =>
+  new ToolLoopAgent({
+    model: wrapLanguageModel({
+      model: friendli(modelId),
+      middleware: trimLeadingNewlinesMiddleware,
+    }),
+    instructions: SYSTEM_PROMPT,
+    tools: {
+      ...tools,
+    },
+    maxOutputTokens: OUTPUT_TOKEN_MAX,
+    providerOptions: {
+      friendli: {
+        chat_template_kwargs: {
+          enable_thinking: true,
+        },
+      },
+    },
+  });
 
-function handleReasoningDelta(chunk: StreamChunk, state: StreamState): void {
-  if (chunk.type !== "reasoning-delta") {
-    return;
-  }
-  if (!state.hasStartedReasoning) {
-    printReasoningPrefix();
-    state.hasStartedReasoning = true;
-  }
-  printReasoningChunk(chunk.text);
-}
+class AgentManager {
+  private modelId: string = DEFAULT_MODEL_ID;
 
-function handleTextDelta(chunk: StreamChunk, state: StreamState): void {
-  if (chunk.type !== "text-delta") {
-    return;
-  }
-  endReasoningIfNeeded(state);
-  if (!state.hasStartedText) {
-    printAIPrefix();
-    state.hasStartedText = true;
+  getModelId(): string {
+    return this.modelId;
   }
-  printChunk(chunk.text);
-}
 
-function handleToolCall(chunk: StreamChunk, state: StreamState): void {
-  if (chunk.type !== "tool-call") {
-    return;
+  setModelId(modelId: string): void {
+    this.modelId = modelId;
   }
-  endReasoningIfNeeded(state);
-  endTextIfNeeded(state);
-  printTool(chunk.toolName, chunk.input);
-}
 
-function logDebugChunk(chunk: StreamChunk, chunkCount: number): void {
-  const skipTypes = ["text-delta", "reasoning-delta", "tool-result"];
-  if (!skipTypes.includes(chunk.type)) {
-    console.log(`[DEBUG] #${chunkCount} type: ${chunk.type}`);
+  getInstructions(): string {
+    return SYSTEM_PROMPT;
   }
-}
 
-function logDebugError(chunk: StreamChunk): void {
-  if (chunk.type === "error") {
-    console.log("[DEBUG] Error:", chunk.error);
+  getTools() {
+    return tools;
   }
-}
 
-function logDebugFinish(chunk: StreamChunk): void {
-  if (chunk.type === "finish") {
-    console.log(`[DEBUG] Finish reason: ${chunk.finishReason}`);
+  stream(messages: ModelMessage[]) {
+    const agent = createAgent(this.modelId);
+    return agent.stream({ messages });
   }
 }
 
-const DEFAULT_MAX_STEPS = 255;
-
-export class Agent {
-  private model: LanguageModel;
-  private conversation: ModelMessage[] = [];
-  private readonly maxSteps: number;
-
-  constructor(model: LanguageModel, maxSteps = DEFAULT_MAX_STEPS) {
-    this.model = model;
-    this.maxSteps = maxSteps;
-  }
-
-  getModel(): LanguageModel {
-    return this.model;
-  }
-
-  setModel(model: LanguageModel): void {
-    this.model = model;
-  }
-
-  getConversation(): ModelMessage[] {
-    return [...this.conversation];
-  }
-
-  loadConversation(messages: ModelMessage[]): void {
-    this.conversation = [...messages];
-  }
-
-  clearConversation(): void {
-    this.conversation = [];
-  }
-
-  async chat(userInput: string): Promise<void> {
-    this.conversation.push({
-      role: "user",
-      content: userInput,
-    });
-
-    await withRetry(async () => {
-      await this.executeStreamingChat();
-    });
-  }
-
-  private async executeStreamingChat(): Promise<void> {
-    const result = streamText({
-      model: this.model,
-      system: SYSTEM_PROMPT,
-      messages: this.conversation,
-      tools: agentTools,
-      stopWhen: stepCountIs(this.maxSteps),
-      providerOptions: {
-        friendliai: {
-          // enable_thinking for hybrid reasoning models
-          chat_template_kwargs: {
-            enable_thinking: true,
-          },
-        },
-      },
-    });
-
-    const state: StreamState = {
-      hasStartedText: false,
-      hasStartedReasoning: false,
-    };
-
-    let chunkCount = 0;
-    const debug = env.DEBUG_CHUNK_LOG;
-
-    for await (const chunk of result.fullStream) {
-      chunkCount++;
-
-      if (debug) {
-        logDebugChunk(chunk, chunkCount);
-        logDebugError(chunk);
-        logDebugFinish(chunk);
-      }
-
-      handleReasoningDelta(chunk, state);
-      handleTextDelta(chunk, state);
-      handleToolCall(chunk, state);
-    }
-
-    endReasoningIfNeeded(state);
-    endTextIfNeeded(state);
-
-    const response = await result.response;
-    if (debug) {
-      console.log(`[DEBUG] Total chunks: ${chunkCount}`);
-      console.log(`[DEBUG] Response messages: ${response.messages.length}`);
-    }
-    this.conversation.push(...response.messages);
-  }
-}
+export const agentManager = new AgentManager();

src/commands/clear.ts

@@ -0,0 +1,16 @@
+import type { MessageHistory } from "../context/message-history";
+import type { Command, CommandContext, CommandResult } from "./types";
+
+export const createClearCommand = (
+  messageHistory: MessageHistory
+): Command => ({
+  name: "clear",
+  description: "Clear current conversation history",
+  execute: (_context: CommandContext): CommandResult => {
+    messageHistory.clear();
+    return {
+      success: true,
+      message: "Conversation history cleared.",
+    };
+  },
+});

src/commands/help.ts

@@ -0,0 +1,18 @@
+import type { Command, CommandResult } from "./types";
+
+export const createHelpCommand = (
+  getCommands: () => Map<string, Command>
+): Command => ({
+  name: "help",
+  description: "Show available commands",
+  execute: (): CommandResult => {
+    const commandList = Array.from(getCommands().values())
+      .map((cmd) => `  /${cmd.name} - ${cmd.description}`)
+      .join("\n");
+
+    return {
+      success: true,
+      message: `Available commands:\n${commandList}`,
+    };
+  },
+});