superagent-ai · homanp · Jul 22, 2025 · Jul 22, 2025
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@vibe-kit/grok-cli",
-  "version": "0.0.2",
+  "version": "0.0.3",
   "description": "An open-source AI agent that brings the power of Grok directly into your terminal.",
   "main": "dist/index.js",
   "bin": {

diff --git a/src/agent/grok-agent.ts b/src/agent/grok-agent.ts
@@ -7,7 +7,7 @@ import { createTokenCounter, TokenCounter } from "../utils/token-counter";
 import { loadCustomInstructions } from "../utils/custom-instructions";
 
 export interface ChatEntry {
-  type: "user" | "assistant" | "tool_result";
+  type: "user" | "assistant" | "tool_result" | "tool_call";
   content: string;
   timestamp: Date;
   toolCalls?: GrokToolCall[];
@@ -64,6 +64,9 @@ You have access to these tools:
 - create_todo_list: Create a visual todo list for planning and tracking tasks
 - update_todo_list: Update existing todos in your todo list
 
+REAL-TIME INFORMATION:
+You have access to real-time web search and X (Twitter) data. When users ask for current information, latest news, or recent events, you automatically have access to up-to-date information from the web and social media.
+
 IMPORTANT TOOL USAGE RULES:
 - NEVER use create_file on files that already exist - this will overwrite them completely
 - ALWAYS use str_replace_editor to modify existing files, even for small changes
@@ -127,7 +130,9 @@ Current working directory: ${process.cwd()}`,
     try {
       let currentResponse = await this.grokClient.chat(
         this.messages,
-        GROK_TOOLS
+        GROK_TOOLS,
+        undefined,
+        { search_parameters: { mode: "auto" } }
       );
 
       // Agent loop - continue until no more tool calls or max rounds reached
@@ -191,7 +196,9 @@ Current working directory: ${process.cwd()}`,
           // Get next response - this might contain more tool calls
           currentResponse = await this.grokClient.chat(
             this.messages,
-            GROK_TOOLS
+            GROK_TOOLS,
+            undefined,
+            { search_parameters: { mode: "auto" } }
           );
         } else {
           // No more tool calls, add final response
@@ -270,7 +277,7 @@ Current working directory: ${process.cwd()}`,
   ): AsyncGenerator<StreamingChunk, void, unknown> {
     // Create new abort controller for this request
     this.abortController = new AbortController();
-    
+
     // Add user message to conversation
     const userEntry: ChatEntry = {
       type: "user",
@@ -307,7 +314,12 @@ Current working directory: ${process.cwd()}`,
         }
 
         // Stream response and accumulate
-        const stream = this.grokClient.chatStream(this.messages, GROK_TOOLS);
+        const stream = this.grokClient.chatStream(
+          this.messages,
+          GROK_TOOLS,
+          undefined,
+          { search_parameters: { mode: "auto" } }
+        );
         let accumulatedMessage: any = {};
         let accumulatedContent = "";
         let toolCallsYielded = false;

diff --git a/src/grok/client.ts b/src/grok/client.ts
@@ -1,15 +1,15 @@
-import OpenAI from 'openai';
-import type { ChatCompletionMessageParam } from 'openai/resources/chat';
+import OpenAI from "openai";
+import type { ChatCompletionMessageParam } from "openai/resources/chat";
 
 export type GrokMessage = ChatCompletionMessageParam;
 
 export interface GrokTool {
-  type: 'function';
+  type: "function";
   function: {
     name: string;
     description: string;
     parameters: {
-      type: 'object';
+      type: "object";
       properties: Record<string, any>;
       required: string[];
     };
@@ -18,13 +18,22 @@ export interface GrokTool {
 
 export interface GrokToolCall {
   id: string;
-  type: 'function';
+  type: "function";
   function: {
     name: string;
     arguments: string;
   };
 }
 
+export interface SearchParameters {
+  mode?: "auto" | "on" | "off";
+  // sources removed - let API use default sources to avoid format issues
+}
+
+export interface SearchOptions {
+  search_parameters?: SearchParameters;
+}
+
 export interface GrokResponse {
   choices: Array<{
     message: {
@@ -38,12 +47,12 @@ export interface GrokResponse {
 
 export class GrokClient {
   private client: OpenAI;
-  private currentModel: string = 'grok-3-latest';
+  private currentModel: string = "grok-3-latest";
 
   constructor(apiKey: string, model?: string, baseURL?: string) {
     this.client = new OpenAI({
       apiKey,
-      baseURL: baseURL || process.env.GROK_BASE_URL || 'https://api.x.ai/v1',
+      baseURL: baseURL || process.env.GROK_BASE_URL || "https://api.x.ai/v1",
       timeout: 360000,
     });
     if (model) {
@@ -62,17 +71,27 @@ export class GrokClient {
   async chat(
     messages: GrokMessage[],
     tools?: GrokTool[],
-    model?: string
+    model?: string,
+    searchOptions?: SearchOptions
   ): Promise<GrokResponse> {
     try {
-      const response = await this.client.chat.completions.create({
+      const requestPayload: any = {
         model: model || this.currentModel,
         messages,
         tools: tools || [],
-        tool_choice: tools ? 'auto' : undefined,
+        tool_choice: tools && tools.length > 0 ? "auto" : undefined,
         temperature: 0.7,
         max_tokens: 4000,
-      });
+      };
+
+      // Add search parameters if specified
+      if (searchOptions?.search_parameters) {
+        requestPayload.search_parameters = searchOptions.search_parameters;
+      }
+
+      const response = await this.client.chat.completions.create(
+        requestPayload
+      );
 
       return response as GrokResponse;
     } catch (error: any) {
@@ -83,18 +102,28 @@ export class GrokClient {
   async *chatStream(
     messages: GrokMessage[],
     tools?: GrokTool[],
-    model?: string
+    model?: string,
+    searchOptions?: SearchOptions
   ): AsyncGenerator<any, void, unknown> {
     try {
-      const stream = await this.client.chat.completions.create({
+      const requestPayload: any = {
         model: model || this.currentModel,
         messages,
         tools: tools || [],
-        tool_choice: tools ? 'auto' : undefined,
+        tool_choice: tools && tools.length > 0 ? "auto" : undefined,
         temperature: 0.7,
         max_tokens: 4000,
         stream: true,
-      });
+      };
+
+      // Add search parameters if specified
+      if (searchOptions?.search_parameters) {
+        requestPayload.search_parameters = searchOptions.search_parameters;
+      }
+
+      const stream = (await this.client.chat.completions.create(
+        requestPayload
+      )) as any;
 
       for await (const chunk of stream) {
         yield chunk;
@@ -103,4 +132,20 @@ export class GrokClient {
       throw new Error(`Grok API error: ${error.message}`);
     }
   }
-}
+
+  async search(
+    query: string,
+    searchParameters?: SearchParameters
+  ): Promise<GrokResponse> {
+    const searchMessage: GrokMessage = {
+      role: "user",
+      content: query,
+    };
+
+    const searchOptions: SearchOptions = {
+      search_parameters: searchParameters || { mode: "on" },
+    };
+
+    return this.chat([searchMessage], [], undefined, searchOptions);
+  }
+}