chore: initial commit
This commit is contained in:
36
test_with_token.py
Normal file
36
test_with_token.py
Normal file
@@ -0,0 +1,36 @@
|
||||
#!/usr/bin/env python3
|
||||
import os
|
||||
import asyncio
|
||||
from claude_agent_sdk import ClaudeSDKClient
|
||||
|
||||
os.environ['ANTHROPIC_AUTH_TOKEN'] = 'cr_9792f20a98f055e204248a41f280780ca2fb8f08f35e60c785e5245653937e06'
|
||||
|
||||
async def test():
|
||||
print('Testing claude-agent-sdk...')
|
||||
try:
|
||||
client = ClaudeSDKClient()
|
||||
await client.connect()
|
||||
info = await client.get_server_info()
|
||||
print('Account info:', info.get('account'))
|
||||
await client.query('Hello, reply with OK')
|
||||
response_text = ''
|
||||
async for chunk in client.receive_response():
|
||||
chunk_str = str(chunk)
|
||||
if 'AssistantMessage' in chunk_str:
|
||||
import re
|
||||
match = re.search(r"text='([^']*)'", chunk_str)
|
||||
if match:
|
||||
response_text += match.group(1)
|
||||
if response_text:
|
||||
print('SUCCESS! AI replied:', response_text)
|
||||
await client.disconnect()
|
||||
return True
|
||||
except Exception as e:
|
||||
print('FAILED:', e)
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
return False
|
||||
|
||||
if __name__ == '__main__':
|
||||
result = asyncio.run(test())
|
||||
print('\nResult:', 'SUCCESS' if result else 'FAILED')
|
||||
Reference in New Issue
Block a user