using tool decorator with result_as_answer

[amindadgar]

Hey Everybody, I've created my custom tool with @tool decorator but I have no idea how to use the result_as_answer flag for the tool function. Can anyone help me on this?

crewai version 0.102.0

My code for the tool

import os
from uuid import uuid1

from crewai.tools import tool
from dotenv import load_dotenv
from tc_temporal_backend.client import TemporalClient
from tc_temporal_backend.schema.hivemind import HivemindQueryPayload


class QueryDataSources:
    def __init__(self, community_id: str, enable_answer_skipping: bool):
        self.community_id = community_id
        self.enable_answer_skipping = enable_answer_skipping

    @tool
    async def query(self, query: str) -> str:
        """
        query data sources for the given community
        """
        client = await TemporalClient().get_client()

        payload = HivemindQueryPayload(
            community_id=self.community_id,
            query=query,
            enable_answer_skipping=self.enable_answer_skipping,
        )

        hivemind_queue = self.load_hivemind_queue()
        result = await client.execute_workflow(
            "HivemindWorkflow",
            payload,
            id=f"hivemind-query-{self.community_id}-{uuid1()}",
            task_queue=hivemind_queue,
        )

        return result

    def load_hivemind_queue(self) -> str:
        """
        load the hivemind queue name
        """
        load_dotenv()
        hivemind_queue = os.getenv("TEMPORAL_HIVEMIND_TASK_QUEUE")
        if not hivemind_queue:
            raise ValueError("env `TEMPORAL_HIVEMIND_TASK_QUEUE` is not loaded!")

        return hivemind_queue

How to apply the result_as_answer flag?

Note: I've already tried passing this as a kwarg for the query but no success there.

[sei-li-miidas]

I'm a newbie too.
As from the source, I guess @tool(result_as_answer=True) should work