from code_mode import CodeModeClient

# Initialize the client
client = CodeModeClient.create()

# Register tools (example with mock manual)
await client.register_manual({
    'name': 'github',
    'call_template_type': 'http',
    'config': {
        'base_url': 'https://api.github.com'
    }
})

# Execute Python code with tool access
result = await client.call_tool_chain("""
# Call tools and process data
pr = await github.get_pull_request(owner='microsoft', repo='vscode', pull_number=1234)
comments = await github.get_pull_request_comments(owner='microsoft', repo='vscode', pull_number=1234)

# Return processed result
return {
    'title': pr['title'],
    'comment_count': len(comments)
}
""")

print(result['result'])  # Access the execution result
print(result['logs'])     # Access console output