Grok MCP

Implementation Reference

• src/server.py:18-46 (handler)

  Main handler function for the 'chat' tool. Takes a prompt, optional session, model, and system_prompt as parameters. Loads conversation history, creates a Grok chat instance, appends messages to the conversation, gets a response, saves history if session is provided, and returns the response content.

  async def chat(
      prompt: str,
      session: Optional[str] = None,
      model: str = "grok-4",
      system_prompt: Optional[str] = None,
  ):
      history = load_history(session) if session else []
  
      client = Client(api_key=XAI_API_KEY)
      grok = client.chat.create(model=model)
      if system_prompt:
          grok.append(system(system_prompt))
  
      for message in history:
          if message["role"] == "user":
              grok.append(user(message["content"]))
          elif message["role"] == "assistant":
              grok.append(assistant(message["content"]))
  
      grok.append(user(prompt))
      response = grok.sample()
      client.close()
  
      if session:
          history.append({"role": "user", "content": prompt, "time": datetime.now().strftime("%d.%m.%Y %H:%M:%S")})
          history.append({"role": "assistant", "content": response.content, "time": datetime.now().strftime("%d.%m.%Y %H:%M:%S")})
          save_history(session, history)
  
      return response.content

• src/server.py:17-17 (registration)

  Registration of the 'chat' tool using the @mcp.tool() decorator from FastMCP, which registers this async function as an MCP tool endpoint.

  @mcp.tool()

• src/utils.py:39-43 (helper)

  Helper function load_history() used by the chat tool to load previous conversation history from a JSON file for the given session.

  def load_history(session: str):
      path = Path("chats") / f"{session}.json"
      if path.exists():
          return json.loads(path.read_text())
      return []

• src/utils.py:46-48 (helper)

  Helper function save_history() used by the chat tool to persist conversation history to a JSON file for the given session.

  def save_history(session: str, history: list):
      Path("chats").mkdir(exist_ok=True)
      (Path("chats") / f"{session}.json").write_text(json.dumps(history, indent=2, ensure_ascii=False))

• src/server.py:19-23 (schema)

  Input schema/parameters for the chat tool: prompt (required string), session (optional string), model (string with default 'grok-4'), and system_prompt (optional string). These define the tool's input validation and type definitions.

      prompt: str,
      session: Optional[str] = None,
      model: str = "grok-4",
      system_prompt: Optional[str] = None,
  ):