Method _query_native

methods/mirix_agent.py:705–787

Query using MIRIX native send_message (recommended). This leverages MIRIX's built-in memory retrieval and conversation capabilities, providing a more integrated experience.

(
        self,
        question: str,
        system_message: Optional[str],
        start_time: float
    )

            return self._query_manual(question, system_message, start_time)

    def _query_native(
        self,
        question: str,
        system_message: Optional[str],
        start_time: float
    ) -> AgentResponse:
        """Query using MIRIX native send_message (recommended).

        This leverages MIRIX's built-in memory retrieval and conversation
        capabilities, providing a more integrated experience.
        """
        # Truncate question
        full_question = self._truncate_to_tokens(question, self.max_question_tokens)

        # Add timestamp for context
        query_with_time = f"{full_question}\n\nCurrent Time: {time.strftime('%Y-%m-%d %H:%M:%S')}"

        # If there's a system message, prepend it as context
        if system_message:
            query_with_context = f"[System Context: {system_message}]\n\n{query_with_time}"
        else:
            query_with_context = query_with_time

        response_content = ""
        retrieved_memories = []
        query_usage = {"input": 0, "output": 0}

        async def do_query():
            nonlocal response_content, retrieved_memories, query_usage

            # Use MIRIX native send_message - it will automatically retrieve
            # relevant memories and use them in the response
            # NOTE: We don't pass user_id - MIRIX uses LocalClient's default user
            response = await self._client.send_message(
                agent_id=self._meta_agent.id,
                role="user",
                message=query_with_context,
            )

            # Extract response content and usage
            if response:
                # Record usage to global tracker
                if hasattr(response, 'usage') and response.usage:
                    self._record_mirix_usage(response.usage, time.time() - start_time)
                    query_usage["input"] = getattr(response.usage, 'prompt_tokens', 0)
                    query_usage["output"] = getattr(response.usage, 'completion_tokens', 0)

                # Extract assistant response
                if hasattr(response, 'messages'):
                    for msg in response.messages:
                        msg_type = getattr(msg, 'message_type', '')
                        if msg_type == "assistant_message":
                            response_content = getattr(msg, 'message', '')
                            break

                    # Also extract any function calls that retrieved memories
                    for msg in response.messages:
                        msg_type = getattr(msg, 'message_type', '')