Find answers from the community

Updated 2 months ago

Hello all, I have created an ReAct agent

Hello all, I have created an ReAct agent and given some tools to it. Now I will like to include a parser such that the response is always just a python dictionary instead of a sentence. Does anyone know how can I do that?

ReActAgent.from_tools has a output_parser argument however I had no luck with it.

This is current set up and I will be adding on more tools to extract more data:
Plain Text
recursive_query_engine = recursive_index.as_query_engine(
    similarity_top_k=5,
    node_postprocessors=[reranker],
    verbose=True
)

class DocumentTypeResponse(BaseModel):
    """Data model for the document type"""
    document_type: str

document_type_identifier = QueryEngineTool(
    query_engine=recursive_query_engine,
    metadata=ToolMetadata(
        name='document_type',
        description=(
            "Only use this tool when required. "
            "Answer to question relating to document type. "
            "Identify document types as either purchase order or invoice. "
        ),
        fn_schema=DocumentTypeResponse
    )
)

context_document_agent = """\
You are an expert administraive assistant who specialized in answering questions about document. 
The questions mainly revolves around extracting key information from either an invoice or purchase order. 
Only use the necessary tool to answer the questions. Only use more tools when needed.
"""

document_agent = ReActAgent.from_tools(
    tools=document_type_identifier,
    verbose=True,
    context=context_document_agent,
)
L
1 comment
I think you'd just have to post-process the response, or add some instruction for it to always respond in JSON that you can parse
Add a reply
Sign up and join the conversation on Discord