chore: initial commit
This commit is contained in:
60
test_agent_sdk.py
Normal file
60
test_agent_sdk.py
Normal file
@@ -0,0 +1,60 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import os
|
||||
import asyncio
|
||||
from claude_agent_sdk import ClaudeSDKClient
|
||||
|
||||
# 设置 API key
|
||||
os.environ['ANTHROPIC_API_KEY'] = 'cr_9792f20a98f055e204248a41f280780ca2fb8f08f35e60c785e5245653937e06'
|
||||
|
||||
async def test_agent_sdk():
|
||||
print('🔧 测试 claude-agent-sdk 配置...')
|
||||
|
||||
try:
|
||||
# 初始化客户端
|
||||
client = ClaudeSDKClient()
|
||||
print('✅ Claude SDK 客户端初始化成功')
|
||||
|
||||
# 连接
|
||||
await client.connect()
|
||||
print('✅ 连接成功')
|
||||
|
||||
# 获取服务器信息
|
||||
info = await client.get_server_info()
|
||||
print(f'✅ 服务器信息: {info}')
|
||||
|
||||
# 测试对话
|
||||
await client.query('你好,请简短回复测试成功')
|
||||
print(f'✅ 查询已发送')
|
||||
|
||||
# 接收响应
|
||||
full_response = ''
|
||||
async for chunk in client.receive_response():
|
||||
if hasattr(chunk, 'text'):
|
||||
full_response += chunk.text
|
||||
elif isinstance(chunk, dict) and 'text' in chunk:
|
||||
full_response += chunk['text']
|
||||
else:
|
||||
full_response += str(chunk)
|
||||
|
||||
print(f'✅ 对话测试成功')
|
||||
print(f'📝 AI回复: {full_response}')
|
||||
|
||||
await client.disconnect()
|
||||
print('✅ 断开连接成功')
|
||||
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
print(f'❌ 测试失败: {e}')
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
return False
|
||||
|
||||
if __name__ == '__main__':
|
||||
result = asyncio.run(test_agent_sdk())
|
||||
if result:
|
||||
print('\n🎉 claude-agent-sdk 配置成功!')
|
||||
else:
|
||||
print('\n❌ 配置失败')
|
||||
Reference in New Issue
Block a user