"""Test script to run the research agent with simpler query."""

from agent import agent

query = "搜索前列腺癌相关的基因标志物"

print("=== Starting agent invocation ===")
print(f"Query: {query}")
print()

result = agent.invoke(
    {
        "messages": [
            {
                "role": "user",
                "content": query,
            }
        ],
    },
)

print("=== Result ===")
print("Result keys:", list(result.keys()))

if "files" in result:
    print("Files:", list(result["files"].keys()))
else:
    print("No 'files' in result")

print()
print("=== Messages ===")
if "messages" in result:
    for i, msg in enumerate(result["messages"]):
        role = msg.role if hasattr(msg, 'role') else "unknown"
        content = msg.content if hasattr(msg, 'content') else ""
        
        print(f"\n--- Message {i}: role={role} ---")
        if isinstance(content, str):
            if len(content) > 1000:
                print(f"Content (first 1000 chars): {content[:1000]}...")
            else:
                print(f"Content: {content}")
        else:
            print(f"Content type: {type(content)}")