04_agent_langchain.py

import json
import os
import subprocess

from langchain.agents import initialize_agent
from langchain.tools import Tool
from langchain_ollama import OllamaLLM

OLLAMA_URL = os.environ.get("OLLAMA_URL", "http://localhost:11434")
model = os.environ.get("MODEL", "gemma3:12b")
llm = OllamaLLM(model=model, base_url=OLLAMA_URL)


def list_transactions(email: str) -> list[dict]:
    """List user transaction by email"""
    try:
        # Executa o comando da CLI
        result = subprocess.run(
            ["dundie", "list", "--email", email, "--asjson"],
            capture_output=True,
            text=True,
            check=True,
        )
        return json.loads(result.stdout.strip())
    except subprocess.CalledProcessError as e:
        print(f"Error listing transactions: {e}")
        return []


tools = [
    Tool(
        name="list_transactions",
        func=list_transactions,
        description="Use this when you need to list user transactions.",
    )
]

agent = initialize_agent(
    llm=llm, tools=tools, verbose=True, agent_type="zero-shot-react-description"
)


def invoke_agent(prompt: str) -> dict:
    return agent.invoke({"input": prompt})


def main():
    while True:
        prompt = input("Enter your prompt: ")
        if not prompt:
            print("Prompt cannot be empty.")
            continue
        if prompt.strip() in ["exit", "quit", "q"]:
            print("Exiting...")
            break
        result = invoke_agent(prompt)
        print("-" * 50)
        print(result)


if __name__ == "__main__":
    main()