autogen - 💡(How to fix) Fix Feature: Cryptographic action receipts (AAR) for multi-agent conversation provenance [3 comments, 4 participants]

from autogen import AssistantAgent
from botindex_aar import AARMiddleware
import hashlib
import ed25519

# Generate a signing key
signing_key = ed25519.SigningKey.generate()

# Create an AARMiddleware instance
agent = AARMiddleware(
    AssistantAgent("analyst", llm_config=llm_config),
    signing_key=signing_key
)

# Define a function to generate a receipt
def generate_receipt(agent, action, scope, inputs, outputs):
    # Calculate SHA-256 hashes of inputs and outputs
    input_hash = hashlib.sha256(str(inputs).encode()).hexdigest()
    output_hash = hashlib.sha256(str(outputs).encode()).hexdigest()
    
    # Create a receipt
    receipt = {
        "agent_identity": agent.identity,
        "action": action,
        "scope": scope,
        "input_hash": input_hash,
        "output_hash": output_hash
    }
    
    # Sign the receipt
    signed_receipt = agent.signing_key.sign(str(receipt).encode())
    
    return receipt, signed_receipt

# Generate a reply and a receipt
reply = agent.generate_reply("Hello, how are you?")
receipt, signed_receipt = generate_receipt(agent, "generate_reply", "conversation", "Hello, how are you?", reply)

# Print the reply and the receipt
print("Reply:", reply)
print("Receipt:", receipt)
print("Signed Receipt:", signed_receipt)