Skip to content

feat: integrate vikingdb-python-sdk for memory backend#349

Merged
yaozheng-fang merged 4 commits intovolcengine:mainfrom
ethfoo:feat/viking-memory-sdk
Dec 11, 2025
Merged

feat: integrate vikingdb-python-sdk for memory backend#349
yaozheng-fang merged 4 commits intovolcengine:mainfrom
ethfoo:feat/viking-memory-sdk

Conversation

@ethfoo
Copy link
Copy Markdown
Contributor

@ethfoo ethfoo commented Dec 5, 2025

This PR integrates the official vikingdb-python-sdk into the VeADK memory backend system, replacing the previous custom implementation with the official SDK client.

Changes

• Dependencies: Added vikingdb-python-sdk>=0.1.3 to pyproject.toml
• Memory Backend: Updated VikingDBLTMBackend to use the official SDK client (VikingMem) instead of custom implementation
• API Updates:
• Replaced custom add_messages method with SDK's add_session method
• Updated search_memory to use SDK's native implementation
• Added _get_sdk_client() method for proper SDK client initialization
• Code Cleanup: Removed deprecated memory operation methods from ve_viking_db_memory.py

Benefits

• Official Support: Uses the officially maintained VikingDB Python SDK
• Better Reliability: Leverages tested and maintained SDK methods
• Future Compatibility: Ensures compatibility with future VikingDB API updates
• Simplified Maintenance: Reduces custom code maintenance overhead

Testing

image image

The changes maintain the same interface for memory operations while using the underlying SDK. All existing functionality for saving and searching memories remains intact.

@ethfoo ethfoo force-pushed the feat/viking-memory-sdk branch from 9942746 to 8322e25 Compare December 11, 2025 05:55
@yaozheng-fang yaozheng-fang merged commit 2fb31b3 into volcengine:main Dec 11, 2025
5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants